当前位置：首页 > wzjs >正文

网站建设价格专注制作网站设计广州seo服务外包

wzjs 2025/9/9 8:50:26

网站建设价格专注制作网站设计,广州seo服务外包,电商网站布局设计,随州seo推广前言以大模型向量存储的方式，提交本地文件到知识库。并在 AI 对话中增强检索知识库符合 AI 对话内容的资料，合并提交问题。技术组件 1:SpringAI : 提供向量模型处理框架，支持文件的解析、拆分和向量化操作。 2:TikaDocumentReader : 用于解…

前言
以大模型向量存储的方式，提交本地文件到知识库。并在 AI 对话中增强检索知识库符合 AI 对话内容的资料，合并提交问题。
技术组件
1:SpringAI : 提供向量模型处理框架，支持文件的解析、拆分和向量化操作。

2:TikaDocumentReader : 用于解析上传的文件，支持多种文件格式（如 MD、TXT、SQL 等）。

3:TokenTextSplitter : 用于将解析后的文本内容拆分为更小的片段，便于后续处理和存储。

4:PostgreSQL向量库 : 用于存储处理后的文本向量数据，支持高效的相似性搜索和检索。
方案流程
文件上传与解析
文件上传 : 用户通过前端界面或 API 上传文件，文件可以是多种格式（如 MD、TXT、SQL 等）。
文件解析 : 使用 TikaDocumentReader 对上传的文件进行解析，提取出文本内容。TikaDocumentReader 能够处理多种文件格式，并提取出结构化的文本数据。
文本拆分
文本拆分 : 使用 TokenTextSplitter 将解析后的文本内容拆分为更小的片段。拆分策略可以根据需求进行调整，例如按句子、段落或固定长度的 token 进行拆分。
拆分后的文本片段 : 每个文本片段将作为后续处理和存储的基本单元。
向量化与存储
向量化 : 使用 Spring AI 提供的向量模型将标记后的文本片段转换为向量表示。向量化过程将文本内容映射到高维向量空间，便于后续的相似性搜索和检索。
存储到PostgreSQL向量库 : 将向量化后的文本片段及其标记存储到 PostgreSQL 向量库中。PostgreSQL 提供了高效的向量索引和搜索功能，能够支持大规模的文本数据存储和检索。
功能实现
在这里插入图片描述
引入组件

<dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-tika-document-reader</artifactId>
</dependency>
<dependency><groupId>org.springframework.ai</groupId><artifactId>spring-ai-pgvector-store</artifactId>
</dependency>

spring-ai-tika-document-reader，解析文档
spring-ai-pgvector-store，向量存储库

文件配置

server:port: 8090spring:datasource:driver-class-name: org.postgresql.Driverusername: postgrespassword: postgresurl: jdbc:postgresql://ip:15432/ai-rag-knowledgetype: com.zaxxer.hikari.HikariDataSource# hikari连接池配置hikari:#连接池名pool-name: HikariCP#最小空闲连接数minimum-idle: 5# 空闲连接存活最大时间，默认10分钟idle-timeout: 600000# 连接池最大连接数，默认是10maximum-pool-size: 10# 此属性控制从池返回的连接的默认自动提交行为,默认值：trueauto-commit: true# 此属性控制池中连接的最长生命周期，值0表示无限生命周期，默认30分钟max-lifetime: 1800000# 数据库连接超时时间,默认30秒connection-timeout: 30000# 连接测试queryconnection-test-query: SELECT 1ai:ollama:base-url: http://ip:11434embedding:options:num-batch: 512model: nomic-embed-textopenai:base-url: https://pro-share-aws-api.zcyai.com/api-key: sk-esdsdfxptPgbfXd3Z164260740E0494161Bd8**embedding-model: text-embedding-ada-002rag:embed: nomic-embed-text #nomic-embed-text、text-embedding-ada-002# Redis
redis:sdk:config:host: ipport: 16379pool-size: 10min-idle-size: 5idle-timeout: 30000connect-timeout: 5000retry-attempts: 3retry-interval: 1000ping-interval: 60000keep-alive: truelogging:level:root: infoconfig: classpath:logback-spring.xml

可以指定向量库模型，这个是 ollama 安装的模型。如果不在这里指定，那么它会找一个默认的向量库。

datasource 配置 postgresql 地址。注意，这里为了避免端口冲突，设置为 15432，和 docker compose 设置的对外端口一致。

向量存储
Spring AI 提供了简单 SimpleVectorStore 实现类，把文档信息缓存到内存中。以及 PgVectorStore 实现类，把文档存储到向量库。

这块的需要现在 app/config 下，实例化 VectorStore 存储实现类。

@Configuration
public class RAGEmbeddingConfig {@Beanpublic TokenTextSplitter tokenTextSplitter() {return new TokenTextSplitter();}@Beanpublic OllamaApi ollamaApi(@Value("${spring.ai.ollama.base-url}") String baseUrl) {return new OllamaApi(baseUrl);}@Beanpublic SimpleVectorStore vectorStore(OllamaApi ollamaApi) {OllamaEmbeddingClient embeddingClient = new OllamaEmbeddingClient(ollamaApi);embeddingClient.withDefaultOptions(OllamaOptions.create().withModel("nomic-embed-text"));return new SimpleVectorStore(embeddingClient);}@Beanpublic PgVectorStore pgVectorStore(OllamaApi ollamaApi, JdbcTemplate jdbcTemplate) {OllamaEmbeddingClient embeddingClient = new OllamaEmbeddingClient(ollamaApi);embeddingClient.withDefaultOptions(OllamaOptions.create().withModel("nomic-embed-text"));return new PgVectorStore(jdbcTemplate, embeddingClient);}}

上传知识

package dev.tech.test;import com.alibaba.fastjson.JSON;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.ai.chat.ChatResponse;
import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.UserMessage;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.chat.prompt.SystemPromptTemplate;
import org.springframework.ai.document.Document;
import org.springframework.ai.ollama.OllamaChatClient;
import org.springframework.ai.ollama.api.OllamaOptions;
import org.springframework.ai.reader.tika.TikaDocumentReader;
import org.springframework.ai.transformer.splitter.TokenTextSplitter;
import org.springframework.ai.vectorstore.PgVectorStore;
import org.springframework.ai.vectorstore.SearchRequest;
import org.springframework.ai.vectorstore.SimpleVectorStore;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;@Slf4j
@RunWith(SpringRunner.class)
@SpringBootTest
public class RAGTest {@Resourceprivate OllamaChatClient ollamaChatClient;@Resourceprivate TokenTextSplitter tokenTextSplitter;@Resourceprivate SimpleVectorStore simpleVectorStore;@Resourceprivate PgVectorStore pgVectorStore;@Testpublic void upload() {TikaDocumentReader reader = new TikaDocumentReader("./data/file.text");List<Document> documents = reader.get();List<Document> documentSplitterList = tokenTextSplitter.apply(documents);documents.forEach(doc -> doc.getMetadata().put("knowledge", "知识库名称"));documentSplitterList.forEach(doc -> doc.getMetadata().put("knowledge", "知识库名称"));pgVectorStore.accept(documentSplitterList);log.info("上传完成");}@Testpublic void chat() {String message = "小明哥，哪年出生";String SYSTEM_PROMPT = """Use the information from the DOCUMENTS section to provide accurate answers but act as if you knew this information innately.If unsure, simply state that you don't know.Another thing you need to note is that your reply must be in Chinese!DOCUMENTS:{documents}""";SearchRequest request = SearchRequest.query(message).withTopK(5).withFilterExpression("knowledge == '知识库名称'");List<Document> documents = pgVectorStore.similaritySearch(request);String documentsCollectors = documents.stream().map(Document::getContent).collect(Collectors.joining());Message ragMessage = new SystemPromptTemplate(SYSTEM_PROMPT).createMessage(Map.of("documents", documentsCollectors));ArrayList<Message> messages = new ArrayList<>();messages.add(new UserMessage(message));messages.add(ragMessage);ChatResponse chatResponse = ollamaChatClient.call(new Prompt(messages, OllamaOptions.create().withModel("deepseek-r1:1.5b")));log.info("测试结果:{}", JSON.toJSONString(chatResponse));}}