自动获取mcp工具撰写技术标书文档内容

llm-back/ruoyi-agent/pom.xml

@@ -54,7 +54,34 @@
             <groupId>org.apache.poi</groupId>
             <artifactId>poi-ooxml</artifactId>
         </dependency>
-
+        <dependency>
+            <groupId>com.alibaba.fastjson2</groupId>
+            <artifactId>fastjson2</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-embeddings-bge-small-zh-v15</artifactId>
+            <version>0.35.0</version>
+            <scope>compile</scope>
+        </dependency>
+        <dependency>
+            <groupId>io.milvus</groupId>
+            <artifactId>milvus-sdk-java</artifactId>
+            <version>2.3.3</version>
+            <scope>compile</scope>
+        </dependency>
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j</artifactId>
+            <version>0.35.0</version>
+            <scope>compile</scope>
+        </dependency>
+        <dependency>
+            <groupId>dev.langchain4j</groupId>
+            <artifactId>langchain4j-document-parser-apache-pdfbox</artifactId>
+            <version>0.35.0</version>
+            <scope>compile</scope>
+        </dependency>
     </dependencies>
 
     <dependencyManagement>

llm-back/ruoyi-agent/src/main/java/com/ruoyi/agent/service/McpServerConfig.java

@@ -24,7 +24,7 @@ import java.util.List;
 public class McpServerConfig {
     @PostConstruct
     public void start() {
-        Solon.start(McpServerConfig.class, new String[]{});
+        Solon.start(McpServerConfig.class, new String[]{"--cfg=application.yml"});
     }
 
     @PreDestroy

llm-back/ruoyi-agent/src/main/java/com/ruoyi/agent/service/impl/McpServiceImpl.java

@@ -1,26 +1,74 @@
 package com.ruoyi.agent.service.impl;
 
+import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.agent.service.IMcpService;
+import dev.langchain4j.data.document.Document;
+import dev.langchain4j.data.document.parser.apache.pdfbox.ApachePdfBoxDocumentParser;
+import dev.langchain4j.data.document.splitter.DocumentByParagraphSplitter;
+import dev.langchain4j.data.embedding.Embedding;
+import dev.langchain4j.data.segment.TextSegment;
+import dev.langchain4j.model.embedding.EmbeddingModel;
+import dev.langchain4j.model.embedding.onnx.bgesmallzhv15.BgeSmallZhV15EmbeddingModel;
+import dev.langchain4j.store.embedding.EmbeddingMatch;
+import dev.langchain4j.store.embedding.EmbeddingSearchRequest;
+import dev.langchain4j.store.embedding.inmemory.InMemoryEmbeddingStore;
+import io.milvus.client.MilvusClient;
+import io.milvus.client.MilvusServiceClient;
+import io.milvus.grpc.SearchResults;
+import io.milvus.param.ConnectParam;
+import io.milvus.param.MetricType;
+import io.milvus.param.R;
+import io.milvus.param.RpcStatus;
+import io.milvus.param.collection.LoadCollectionParam;
+import io.milvus.param.collection.ReleaseCollectionParam;
+import io.milvus.param.dml.SearchParam;
+import io.milvus.response.SearchResultsWrapper;
+import org.apache.poi.xwpf.extractor.XWPFWordExtractor;
 import org.apache.poi.xwpf.usermodel.XWPFDocument;
 import org.apache.poi.xwpf.usermodel.XWPFParagraph;
+import org.noear.solon.Solon;
 import org.noear.solon.ai.annotation.PromptMapping;
-import org.noear.solon.ai.annotation.ResourceMapping;
 import org.noear.solon.ai.annotation.ToolMapping;
+import org.noear.solon.ai.chat.ChatModel;
+import org.noear.solon.ai.chat.ChatResponse;
+import org.noear.solon.ai.chat.ChatSession;
+import org.noear.solon.ai.chat.ChatSessionDefault;
+import org.noear.solon.ai.chat.message.AssistantMessage;
 import org.noear.solon.ai.chat.message.ChatMessage;
 import org.noear.solon.ai.mcp.server.annotation.McpServerEndpoint;
 import org.noear.solon.annotation.Param;
+import org.reactivestreams.Publisher;
 import org.springframework.stereotype.Service;
+import reactor.core.publisher.Flux;
 
-import java.io.FileInputStream;
-import java.io.FileOutputStream;
-import java.io.IOException;
-import java.util.Collection;
-import java.util.Collections;
+import java.io.*;
+import java.util.*;
+import java.util.stream.Collectors;
 
 @Service
 @McpServerEndpoint(sseEndpoint = "/llm/mcp/sse")
 public class McpServiceImpl implements IMcpService {
 
+    private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
+
+    private static final MilvusServiceClient milvusClient = new MilvusServiceClient(
+            ConnectParam.newBuilder()
+                    .withHost("192.168.28.188")
+                    .withPort(19530)
+                    .build());
+    /**
+     * 调用LLM+RAG（外部文件+知识库）生成回答
+     */
+    @ToolMapping(description = "章节撰写")
+    public AssistantMessage writeParagraph(@Param(description = "智能体名称") String collectionName,
+                                                 @Param(description = "章节名称") String title,
+                                                 @Param(description = "招标文件地址") String templatePath) throws IOException
+    {
+        title = String.join(",", extractSubTitles( "/upload/agent/template/technical.docx", title));
+        List<JSONObject> requests = retrieveFromMilvus(milvusClient, embeddingModel, collectionName, title, 10);
+        return generateAnswerWithDocumentAndCollection(embeddingModel, templatePath, title, requests, "http://192.168.28.188:8000/v1/chat/completions");
+    }
+
     @ToolMapping(description = "梳理招标文件服务要求")
     public String getBidRequest(@Param(description = "文件路径") String document) throws IOException {
         try (XWPFDocument doc = new XWPFDocument(new FileInputStream(document))) {
@@ -66,4 +114,168 @@ public class McpServiceImpl implements IMcpService {
                         "回答以下问题：'" + part + "\n")
         );
     }
+
+    /**
+     * 从Milvus检索相关文档
+     * @return
+     */
+    public List<JSONObject> retrieveFromMilvus(MilvusClient milvusClient, EmbeddingModel embeddingModel, String collectionName, String query, int topK) {
+        SearchResultsWrapper wrapper = retrieve(milvusClient, embeddingModel, collectionName, query, topK);
+        return wrapper.getRowRecords(0).stream()
+                .map(record -> {
+                    JSONObject result = new JSONObject();
+                    result.put("file_name", record.get("file_name"));
+                    result.put("content", record.get("content"));
+                    return result;
+                })
+                .collect(Collectors.toList());
+    }
+
+    /**
+     * 调用LLM生成回答
+     */
+    public AssistantMessage generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String templatePath, String question, List<JSONObject> contexts, String llmServiceUrl) throws IOException {
+        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
+        File profilePath = new File(Solon.cfg().getProperty("cmc.profile") + templatePath);
+        List<TextSegment> segments = splitDocument(profilePath);
+        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
+        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+        embeddingStore.addAll(embeddings, segments);
+        Embedding queryEmbedding = embeddingModel.embed(question).content();
+        EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
+                .queryEmbedding(queryEmbedding)
+                .maxResults(3)
+                .build();
+        for (EmbeddingMatch embeddingMatch : embeddingStore.search(embeddingSearchRequest).matches()) {
+            String requests = embeddingMatch.embedded().toString();
+            sb.append(requests).append("\n\n");
+
+        }
+        sb.append("针对本项目招标文件内容，补全以下章节部分：\n\n").append(question);
+//        for (JSONObject context : contexts) {
+//            sb.append("文件").append(": ")
+//                    .append(context.getString("file_name")).append("\n\n")
+//                    .append("段落格式").append(": ")
+//                    .append(context.getString("content")).append("\n\n");
+//        }
+        return generateAnswer(sb.toString(), llmServiceUrl);
+    }
+
+    /**
+     * 调用LLM生成回答
+     * @return
+     */
+    public AssistantMessage generateAnswer(String prompt, String llmServiceUrl) throws IOException {
+        ChatSession chatSession = new ChatSessionDefault();
+        ChatModel chatModel = ChatModel.of(llmServiceUrl)
+                .provider("openai")
+                .model("Qwen2.5-1.5B-Instruct")
+                .build();
+
+        chatSession.addMessage(ChatMessage.ofUser(prompt));
+
+        ChatResponse response = chatModel.prompt(chatSession).call();
+        return response.lastChoice().getMessage();
+    }
+
+    /**
+     * 获取二级标题下三级标题列表
+     */
+    public List<String> extractSubTitles(String filename, String question) throws IOException {
+        List<String> subTitles = new ArrayList<>();
+        boolean inTargetSection = false;
+        filename = Solon.cfg().getProperty("cmc.profile") + filename;
+        InputStream fileInputStream = new FileInputStream(filename);
+        XWPFDocument document = new XWPFDocument(fileInputStream);
+        for (XWPFParagraph paragraph : document.getParagraphs()) {
+            String text = paragraph.getText().trim();
+            if (paragraph.getStyle() != null) {
+                // 判断主标题
+                if (paragraph.getStyle().equals("3") &&
+                        text.contains(question)) {
+                    inTargetSection = true;
+                    continue;
+                }
+
+                // 如果已经在目标节中，收集标题3级别的子标题
+                if (inTargetSection) {
+                    if (paragraph.getStyle().equals("4")) {
+                        subTitles.add(text);
+                    }
+                    // 遇到下一个Heading1则退出
+                    else if (paragraph.getStyle().equals("3")) {
+                        break;
+                    }
+                }
+            }
+        }
+        if (subTitles.size() == 0)
+            subTitles.add(question);
+        return subTitles;
+    }
+
+    /**
+     * 检索知识库
+     */
+    private SearchResultsWrapper retrieve(MilvusClient milvusClient, EmbeddingModel embeddingModel, String collectionName, String query, int topK) {
+        List<List<Float>> queryVector = Collections.singletonList(embeddingModel.embed(query).content().vectorAsList());
+
+        //  加载集合
+        LoadCollectionParam loadParam = LoadCollectionParam.newBuilder()
+                .withCollectionName(collectionName)
+                .build();
+
+        R<RpcStatus> loadResponse = milvusClient.loadCollection(loadParam);
+        if (loadResponse.getStatus() != R.Status.Success.getCode()) {
+            System.err.println("加载Collection失败: " + loadResponse.getMessage());
+            milvusClient.close();
+        }
+
+        // 构建SearchParam
+        SearchParam searchParam = SearchParam.newBuilder()
+                .withCollectionName(collectionName)
+                .withVectors(queryVector)
+                .withTopK(topK)
+                .withOutFields(Arrays.asList("file_name", "file_type", "content"))
+                .withVectorFieldName("embedding")
+                .withMetricType(MetricType.COSINE)
+                .withParams("{\"nprobe\": 8}")
+                .build();
+
+        R<SearchResults> response = milvusClient.search(searchParam);
+        SearchResultsWrapper wrapper = new SearchResultsWrapper(response.getData().getResults());
+
+        // 释放集合
+        ReleaseCollectionParam param = ReleaseCollectionParam.newBuilder()
+                .withCollectionName(collectionName)
+                .build();
+        milvusClient.releaseCollection(param);
+
+        return wrapper;
+    }
+
+    /**
+     * 检索知识库
+     */
+    private List<TextSegment> splitDocument(File transferFile) throws IOException {
+        // 加载文档
+        Document document;
+        InputStream fileInputStream = new FileInputStream(transferFile);
+        String filename = transferFile.getName().toLowerCase();
+        if (filename.endsWith(".docx")) {
+            XWPFDocument docx = new XWPFDocument(fileInputStream);
+            XWPFWordExtractor extractor = new XWPFWordExtractor(docx);
+            String text = extractor.getText();
+            document = Document.from(text);
+        }
+        else if (filename.endsWith(".pdf")) {
+            document = new ApachePdfBoxDocumentParser().parse(fileInputStream);
+        }
+        else {
+            throw new UnsupportedOperationException("不支持文件类型: " + filename);
+        }
+        DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(1000,200);
+        return splitter.split(document);
+    }
+
 }

llm-back/ruoyi-agent/src/main/resources/application.yml

@@ -0,0 +1,5 @@
+# 项目相关配置
+cmc:
+  # 文件路径 示例（ Windows配置D:/ruoyi/uploadPath，Linux配置 /home/ruoyi/uploadPath）
+  #  profile: /home/cmc/projects/cmc-llm
+  profile: E:/home/cmc/projects/cmc-llm

llm-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/controller/CmcTopicController.java

@@ -4,6 +4,10 @@ import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 
 import com.ruoyi.common.utils.SnowFlake;
+import com.ruoyi.llm.domain.CmcChat;
+import com.ruoyi.llm.domain.CmcDocument;
+import com.ruoyi.llm.service.ICmcChatService;
+import com.ruoyi.llm.service.ICmcDocumentService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -35,6 +39,12 @@ public class CmcTopicController extends BaseController
     @Autowired
     private ICmcTopicService cmcTopicService;
 
+    @Autowired
+    private ICmcChatService cmcChatService;
+
+    @Autowired
+    private ICmcDocumentService cmcDocumentService;
+
     /**
      * 查询cmc聊天主题列表
      */
@@ -96,6 +106,24 @@ public class CmcTopicController extends BaseController
 	@DeleteMapping("/{topicIds}")
     public AjaxResult remove(@PathVariable String[] topicIds)
     {
+        for (String topicId : topicIds) {
+            CmcChat cmcChat = new CmcChat();
+            cmcChat.setTopicId(topicId);
+            List<CmcChat> cmcChatList = cmcChatService.selectCmcChatList(cmcChat);
+            String[] chatIds = new String[cmcChatList.size()];
+            for (int i = 0; i < cmcChatList.size(); i++)
+                chatIds[i] = cmcChatList.get(i).getChatId();
+            cmcChatService.deleteCmcChatByChatIds(chatIds);
+            for (String chatId : chatIds) {
+                CmcDocument cmcDocument = new CmcDocument();
+                cmcDocument.setChatId(chatId);
+                List<CmcDocument> cmcDocumentList = cmcDocumentService.selectCmcDocumentList(cmcDocument);
+                String[] documentIds = new String[cmcDocumentList.size()];
+                for (int i = 0; i < cmcDocumentList.size(); i++)
+                    documentIds[i] = cmcDocumentList.get(i).getDocumentId();
+                cmcChatService.deleteCmcChatByChatIds(documentIds);
+            }
+        }
         return success(cmcTopicService.deleteCmcTopicByTopicIds(topicIds));
     }
 }

llm-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/controller/McpController.java

@@ -3,6 +3,8 @@ package com.ruoyi.web.llm.controller;
 import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.core.controller.BaseController;
+import com.ruoyi.llm.domain.CmcChat;
+import com.ruoyi.llm.service.ICmcChatService;
 import com.ruoyi.web.llm.service.ILangChainMilvusService;
 import dev.langchain4j.model.embedding.EmbeddingModel;
 import dev.langchain4j.model.embedding.onnx.bgesmallzhv15.BgeSmallZhV15EmbeddingModel;
@@ -10,9 +12,12 @@ import io.milvus.client.MilvusServiceClient;
 import io.milvus.param.ConnectParam;
 import org.noear.solon.ai.chat.ChatModel;
 import org.noear.solon.ai.chat.ChatResponse;
+import org.noear.solon.ai.chat.ChatSession;
+import org.noear.solon.ai.chat.ChatSessionDefault;
 import org.noear.solon.ai.chat.message.AssistantMessage;
 import org.noear.solon.ai.chat.message.ChatMessage;
 import org.noear.solon.ai.mcp.client.McpClientProvider;
+import org.reactivestreams.Publisher;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
@@ -37,6 +42,9 @@ public class McpController extends BaseController
     @Autowired
     private ILangChainMilvusService langChainMilvusService;
 
+    @Autowired
+    private ICmcChatService cmcChatService;
+
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
 
     private static final MilvusServiceClient milvusClient = new MilvusServiceClient(
@@ -44,47 +52,89 @@ public class McpController extends BaseController
                     .withHost("192.168.28.188")
                     .withPort(19530)
                     .build());
+//
+//    /**
+//     * 同步问答
+//     * @return
+//     */
+//    @GetMapping("/answer")
+//    public AssistantMessage answer(String question) throws IOException {
+//        McpClientProvider clientProvider = McpClientProvider.builder()
+//                .apiUrl("http://localhost:8080/llm/mcp/sse")
+//                .build();
+//        ChatModel chatModel = ChatModel.of("http://192.168.28.188:8000/v1/chat/completions")
+//                .provider("openai")
+//                .model("Qwen2.5-1.5B-Instruct")
+//                .defaultToolsAdd(clientProvider)
+//                .build();
+//        ChatResponse response = chatModel.prompt(question).call();
+//        String resultContent = response.lastChoice().getMessage().getResultContent();
+//        AssistantMessage assistantMessage = new AssistantMessage(resultContent);
+//        if (resultContent.startsWith("<tool_call>")) {
+//            String content = resultContent.replace("<tool_call>\n", "").replace("\n</tool_call>", "");
+//            JSONObject jsonObject = JSONObject.parseObject(content);
+//            String name = jsonObject.getString("name");
+//            JSONObject arguments = jsonObject.getJSONObject("arguments");
+//            resultContent = clientProvider.callToolAsText(name, arguments).getContent();
+//            assistantMessage = new AssistantMessage(resultContent);
+//        }
+//        return assistantMessage;
+//    }
 
     /**
-     * 同步问答
+     * 自动调用mcp工具问答
      * @return
      */
     @GetMapping("/answer")
-    public AssistantMessage answer(String question) throws IOException {
+    public AssistantMessage answer(String topicId, String question) throws IOException {
         McpClientProvider clientProvider = McpClientProvider.builder()
                 .apiUrl("http://localhost:8080/llm/mcp/sse")
                 .build();
+        ChatSession chatSession = new ChatSessionDefault(topicId);
         ChatModel chatModel = ChatModel.of("http://192.168.28.188:8000/v1/chat/completions")
                 .provider("openai")
                 .model("Qwen2.5-1.5B-Instruct")
                 .defaultToolsAdd(clientProvider)
                 .build();
-        ChatResponse response = chatModel.prompt(question).call();
+
+        CmcChat cmcChat = new CmcChat();
+        cmcChat.setTopicId(topicId);
+        List<CmcChat> cmcChatList = cmcChatService.selectCmcChatList(cmcChat);
+        for (CmcChat chat : cmcChatList) {
+            chatSession.addMessage(ChatMessage.ofUser(chat.getInput()));
+            chatSession.addMessage(ChatMessage.ofAssistant(chat.getOutput()));
+        }
+
+        chatSession.addMessage(ChatMessage.ofUser(question));
+        ChatResponse response = chatModel.prompt(chatSession).call();
         String resultContent = response.lastChoice().getMessage().getResultContent();
-        AssistantMessage assistantMessage = new AssistantMessage(resultContent);
+        AssistantMessage assistantMessage;
         if (resultContent.startsWith("<tool_call>")) {
             String content = resultContent.replace("<tool_call>\n", "").replace("\n</tool_call>", "");
             JSONObject jsonObject = JSONObject.parseObject(content);
             String name = jsonObject.getString("name");
             JSONObject arguments = jsonObject.getJSONObject("arguments");
+            if (arguments.getString("templatePath").contains("技术"))
+                arguments.put("collectionName", "technical");
             resultContent = clientProvider.callToolAsText(name, arguments).getContent();
             assistantMessage = new AssistantMessage(resultContent);
         }
+        else
+            throw new IOException("模型上下文工具未准备就绪，请重试");
         return assistantMessage;
     }
 
     /**
-     * 调用LLM+RAG（外部文件）生成回答
+     * 调用LLM+RAG（外部文件+知识库）生成回答
      */
     @GetMapping("/answerWithDocument")
-    public Flux<AssistantMessage> answerWithDocumentAndCollection(String collectionName, String topicId, String chatId, String question) throws IOException
+    public Flux<AssistantMessage> answerWithDocumentAndCollection(String collectionName, String topicId, String question) throws IOException
     {
         question = String.join(",", langChainMilvusService.extractSubTitles(RuoYiConfig.getProfile() + "/upload/agent/template/technical.docx", question));
         List<JSONObject> requests = langChainMilvusService.retrieveFromMilvus(milvusClient, embeddingModel, collectionName, question, 10);
-        return langChainMilvusService.generateAnswerWithDocumentAndCollection(embeddingModel, topicId, chatId, question, requests, "http://192.168.28.188:8000/v1/chat/completions");
+        return langChainMilvusService.generateAnswerWithDocumentAndCollection(embeddingModel, topicId, question, requests, "http://192.168.28.188:8000/v1/chat/completions");
     }
 
-
     /**
      * 根据招标文件要求编写投标文件技术方案
      * @return

llm-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/service/ILangChainMilvusService.java

@@ -53,7 +53,7 @@ public interface ILangChainMilvusService {
      * 调用LLM+RAG（外部文件+知识库）生成回答
      * @return
      */
-    public Flux<AssistantMessage> generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String topicId, String chatId, String question,  List<JSONObject> requests, String llmServiceUrl) throws IOException;
+    public Flux<AssistantMessage> generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String topicId, String question,  List<JSONObject> requests, String llmServiceUrl) throws IOException;
 
     /**
      * 获取二级标题下三级标题列表

llm-back/ruoyi-llm/src/main/java/com/ruoyi/web/llm/service/impl/LangChainMilvusServiceImpl.java

@@ -4,6 +4,7 @@ import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.llm.domain.CmcChat;
 import com.ruoyi.llm.domain.CmcDocument;
+import com.ruoyi.llm.domain.CmcTopic;
 import com.ruoyi.llm.service.ICmcChatService;
 import com.ruoyi.llm.service.ICmcDocumentService;
 import com.ruoyi.web.llm.service.ILangChainMilvusService;
@@ -30,6 +31,7 @@ import io.milvus.response.SearchResultsWrapper;
 import org.apache.poi.xwpf.extractor.XWPFWordExtractor;
 import org.apache.poi.xwpf.usermodel.XWPFDocument;
 import org.apache.poi.xwpf.usermodel.XWPFParagraph;
+import org.checkerframework.checker.units.qual.C;
 import org.noear.solon.ai.chat.ChatModel;
 import org.noear.solon.ai.chat.ChatResponse;
 import org.noear.solon.ai.chat.ChatSession;
@@ -222,25 +224,32 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
      * 调用LLM生成回答
      */
     @Override
-    public Flux<AssistantMessage> generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String topicId, String chatId, String question, List<JSONObject> contexts, String llmServiceUrl) throws IOException {
+    public Flux<AssistantMessage> generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String topicId, String question, List<JSONObject> contexts, String llmServiceUrl) throws IOException {
         StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
-        CmcDocument cmcDocument = new CmcDocument();
-        cmcDocument.setChatId(chatId);
-        List<CmcDocument> documentList = cmcDocumentService.selectCmcDocumentList(cmcDocument);
-        for (CmcDocument document : documentList) {
-            File profilePath = new File(RuoYiConfig.getProfile() + document.getPath());
-            List<TextSegment> segments = splitDocument(document.getPath(), profilePath);
-            List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
-            InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-            embeddingStore.addAll(embeddings, segments);
-            Embedding queryEmbedding = embeddingModel.embed(question).content();
-            EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
-                    .queryEmbedding(queryEmbedding)
-                    .maxResults(3)
-                    .build();
-            for (EmbeddingMatch embeddingMatch : embeddingStore.search(embeddingSearchRequest).matches()) {
-                String requests = embeddingMatch.embedded().toString();
-                sb.append(requests).append("\n\n");
+        CmcChat cmcChat = new CmcChat();
+        cmcChat.setTopicId(topicId);
+        for (CmcChat chat : cmcChatService.selectCmcChatList(cmcChat)) {
+            CmcDocument cmcDocument = new CmcDocument();
+            cmcDocument.setChatId(chat.getChatId());
+            List<CmcDocument> documentList = cmcDocumentService.selectCmcDocumentList(cmcDocument);
+            //技术标书撰写智能体一个话题只会上传一个招标文件
+            if (documentList.size() == 1) {
+                for (CmcDocument document : documentList) {
+                    File profilePath = new File(RuoYiConfig.getProfile() + document.getPath());
+                    List<TextSegment> segments = splitDocument(document.getPath(), profilePath);
+                    List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
+                    InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+                    embeddingStore.addAll(embeddings, segments);
+                    Embedding queryEmbedding = embeddingModel.embed(question).content();
+                    EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
+                            .queryEmbedding(queryEmbedding)
+                            .maxResults(3)
+                            .build();
+                    for (EmbeddingMatch embeddingMatch : embeddingStore.search(embeddingSearchRequest).matches()) {
+                        String requests = embeddingMatch.embedded().toString();
+                        sb.append(requests).append("\n\n");
+                    }
+                }
             }
         }
         sb.append("针对本项目招标文件内容，补全以下章节部分：\n\n").append(question);

llm-back/ruoyi-system/src/main/java/com/ruoyi/llm/service/impl/CmcAgentServiceImpl.java

@@ -4,14 +4,19 @@ import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.util.Date;
 import java.util.List;
 
 import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.utils.DateUtils;
+import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.common.utils.SnowFlake;
+import com.ruoyi.llm.domain.CmcChat;
 import com.ruoyi.llm.domain.CmcDocument;
+import com.ruoyi.llm.mapper.CmcChatMapper;
 import com.ruoyi.llm.mapper.CmcDocumentMapper;
+import com.ruoyi.llm.service.ICmcChatService;
 import org.apache.poi.xwpf.usermodel.XWPFDocument;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
@@ -35,6 +40,9 @@ public class CmcAgentServiceImpl implements ICmcAgentService
     @Autowired
     private CmcDocumentMapper cmcDocumentMapper;
 
+    @Autowired
+    private CmcChatMapper cmcChatMapper;
+
     /**
      * 查询智能体
      * 
@@ -99,6 +107,12 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         cmcDocumentMapper.insertCmcDocument(cmcDocument);
         String message = "";
         if (agentName.contains("技术")) {
+            CmcChat cmcChat = new CmcChat();
+            cmcChat.setChatId(jsonObject.getString("chatId"));
+            cmcChat.setInputTime(new Date());
+            cmcChat.setInput("招标文件地址：" + outputFilename);
+            cmcChat.setUserId(SecurityUtils.getUserId());
+            cmcChatMapper.insertCmcChat(cmcChat);
             XWPFDocument doc = new XWPFDocument(new FileInputStream(RuoYiConfig.getProfile() + "/upload/agent/template/technical.docx"));
             // 保存文档到本地文件系统
             FileOutputStream out = new FileOutputStream(RuoYiConfig.getProfile() + outputFilename);