写入章节内容

llm-back/ruoyi-agent/pom.xml

             <groupId>org.apache.poi</groupId>
             <artifactId>poi-ooxml</artifactId>
         </dependency>
+        <dependency>
+            <groupId>org.apache.poi</groupId>
+            <artifactId>poi-scratchpad</artifactId>
+            <version>5.2.3</version>
+        </dependency>
         <dependency>
             <groupId>com.alibaba.fastjson2</groupId>
             <artifactId>fastjson2</artifactId>
             <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>
 

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

 import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.agent.service.IMcpService;
 import dev.langchain4j.data.document.Document;
+import dev.langchain4j.data.document.parser.TextDocumentParser;
 import dev.langchain4j.data.document.parser.apache.pdfbox.ApachePdfBoxDocumentParser;
 import dev.langchain4j.data.document.splitter.DocumentByParagraphSplitter;
 import dev.langchain4j.data.embedding.Embedding;
 import io.milvus.param.collection.ReleaseCollectionParam;
 import io.milvus.param.dml.SearchParam;
 import io.milvus.response.SearchResultsWrapper;
+import org.apache.poi.hwpf.HWPFDocument;
 import org.apache.poi.xwpf.extractor.XWPFWordExtractor;
 import org.apache.poi.xwpf.usermodel.XWPFDocument;
 import org.apache.poi.xwpf.usermodel.XWPFParagraph;
+import org.apache.poi.xwpf.usermodel.XWPFRun;
 import org.noear.solon.Solon;
-import org.noear.solon.ai.annotation.PromptMapping;
 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.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.*;
 import java.util.*;
      * 调用LLM+RAG（外部文件+知识库）生成回答
      */
     @ToolMapping(description = "章节撰写")
-    public AssistantMessage writeParagraph(@Param(description = "智能体名称") String collectionName,
-                                                 @Param(description = "章节名称") String title,
-                                                 @Param(description = "招标文件地址") String templatePath) throws IOException
+    public AssistantMessage writeParagraph(@Param(description = "知识库名称") String collectionName,
+                                           @Param(description = "智能体名称") String agentName,
+                                           @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");
+        List<JSONObject> contexts = retrieveFromMilvus(milvusClient, embeddingModel, collectionName, title, 10);
+        return generateAnswerWithDocumentAndCollection(embeddingModel, agentName, templatePath, title, contexts, "http://192.168.28.188:8000/v1/chat/completions");
     }
 
     /**
     /**
      * 调用LLM生成回答
      */
-    public AssistantMessage generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String templatePath, String question, List<JSONObject> contexts, String llmServiceUrl) throws IOException {
+    public AssistantMessage generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String agentName, 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);
+        File profilePath = new File(templatePath.replace("/dev-api/profile", Solon.cfg().getProperty("cmc.profile")).replace("_" + agentName, ""));
         List<TextSegment> segments = splitDocument(profilePath);
         List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
 //                    .append("段落格式").append(": ")
 //                    .append(context.getString("content")).append("\n\n");
 //        }
-        return generateAnswer(sb.toString(), llmServiceUrl);
+        return generateAnswer(sb.toString(), question, templatePath, llmServiceUrl);
     }
 
     /**
      * 调用LLM生成回答
      * @return
      */
-    public AssistantMessage generateAnswer(String prompt, String llmServiceUrl) throws IOException {
+    public AssistantMessage generateAnswer(String prompt, String question, String templatePath, String llmServiceUrl) throws IOException {
         ChatSession chatSession = new ChatSessionDefault();
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
                 .provider("openai")
         chatSession.addMessage(ChatMessage.ofUser(prompt));
 
         ChatResponse response = chatModel.prompt(chatSession).call();
-        return response.lastChoice().getMessage();
+        String content = response.lastChoice().getMessage().getContent() + "\n\n" +
+                "招标文件分析完成，章节内容已写入【<a href='" + templatePath.replace("/dev-api", "") + "'> 技术文件" + "</a>】，请查阅";
+        String absolutePath = templatePath.replace("/dev-api/profile", Solon.cfg().getProperty("cmc.profile"));
+        writeContent(response.lastChoice().getMessage().getContent(), question, absolutePath);
+        return ChatMessage.ofAssistant(content);
+    }
+
+    /**
+     * 写入章节内容
+     * @return
+     */
+    public void writeContent(String content, String question, String absolutePath) throws IOException {
+        File file = new File(absolutePath);
+        FileInputStream fileInputStream = new FileInputStream(file);
+        XWPFDocument document = new XWPFDocument(fileInputStream);
+        String[] contentLines = content.split("\n");
+        Map<String, String> map = new HashMap<>();
+        String[] titles = question.split(",");
+        for (int i = 0; i < titles.length; i ++) {
+            int startIndex = Arrays.asList(contentLines).indexOf(titles[i]);
+            StringBuilder text = new StringBuilder("");
+            if (i < titles.length - 1) {
+                int endIndex = Arrays.asList(contentLines).indexOf(titles[i + 1]);
+                for (int c = startIndex + 1; c < endIndex; c++)
+                    text.append(contentLines[c]);
+            }
+            else {
+                if (startIndex + 1 < contentLines.length)
+                    for (int c = startIndex + 1; c < contentLines.length; c++)
+                        text.append(contentLines[c]);
+            }
+            map.put(titles[i], text.toString());
+        }
+        int count = 0;
+        for (int i = 0; i < document.getParagraphs().size(); i++) {
+            XWPFParagraph paragraph = document.getParagraphs().get(i);
+            for (String title : titles) {
+                if (paragraph.getText().equals(title)) {
+                    int pos = document.getBodyElements().indexOf(paragraph) + 1;
+                    XWPFParagraph contentParagraph = document.createParagraph();
+                    contentParagraph.setStyle("1");
+                    XWPFRun run = contentParagraph.createRun();
+                    run.setText(map.get(title));
+                    document.setParagraph(contentParagraph, pos);
+                    count++;
+                }
+            }
+            if (count == titles.length)
+                break;
+        }
+        FileOutputStream out = new FileOutputStream(absolutePath);
+        document.write(out);
+        // 关闭文档
+        out.close();
+        document.close();
     }
 
     /**
         Document document;
         InputStream fileInputStream = new FileInputStream(transferFile);
         String filename = transferFile.getName().toLowerCase();
-        if (filename.endsWith(".docx")) {
+        if (filename.endsWith(".doc")) {
+            HWPFDocument doc = new HWPFDocument(fileInputStream);
+            document = Document.from(doc.getDocumentText());
+        }
+        else if (filename.endsWith(".docx")) {
             XWPFDocument docx = new XWPFDocument(fileInputStream);
             XWPFWordExtractor extractor = new XWPFWordExtractor(docx);
             String text = extractor.getText();
         else if (filename.endsWith(".pdf")) {
             document = new ApachePdfBoxDocumentParser().parse(fileInputStream);
         }
+        else if (filename.endsWith(".txt")) {
+            document = new TextDocumentParser().parse(fileInputStream);
+        }
         else {
             throw new UnsupportedOperationException("不支持文件类型: " + filename);
         }

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

                 String[] documentIds = new String[cmcDocumentList.size()];
                 for (int i = 0; i < cmcDocumentList.size(); i++)
                     documentIds[i] = cmcDocumentList.get(i).getDocumentId();
-                cmcChatService.deleteCmcChatByChatIds(documentIds);
+                cmcDocumentService.deleteCmcDocumentByDocumentIds(documentIds);
             }
         }
         return success(cmcTopicService.deleteCmcTopicByTopicIds(topicIds));

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

 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.llm.domain.CmcChat;
+import com.ruoyi.llm.service.ICmcAgentService;
 import com.ruoyi.llm.service.ICmcChatService;
+import com.ruoyi.llm.service.ICmcTopicService;
 import com.ruoyi.web.llm.service.ILangChainMilvusService;
 import dev.langchain4j.model.embedding.EmbeddingModel;
 import dev.langchain4j.model.embedding.onnx.bgesmallzhv15.BgeSmallZhV15EmbeddingModel;
 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;
     @Autowired
     private ICmcChatService cmcChatService;
 
+    @Autowired
+    private ICmcTopicService cmcTopicService;
+
+    @Autowired
+    private ICmcAgentService cmcAgentService;
+
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
 
     private static final MilvusServiceClient milvusClient = new MilvusServiceClient(
             JSONObject jsonObject = JSONObject.parseObject(content);
             String name = jsonObject.getString("name");
             JSONObject arguments = jsonObject.getJSONObject("arguments");
-            if (arguments.getString("templatePath").contains("招标") || arguments.getString("templatePath").contains("询价"))
+            if (arguments.getString("templatePath").contains("招标") || arguments.getString("templatePath").contains("询价")) {
                 arguments.put("collectionName", "technical");
+                String agentName = cmcAgentService.selectCmcAgentByAgentId(cmcTopicService.selectCmcTopicByTopicId(topicId).getAgentId()).getAgentName();
+                arguments.put("agentName", agentName);
+            }
             resultContent = clientProvider.callToolAsText(name, arguments).getContent();
             assistantMessage = new AssistantMessage(resultContent);
         }

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

 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;
 import dev.langchain4j.data.document.Document;
+import dev.langchain4j.data.document.parser.TextDocumentParser;
 import dev.langchain4j.data.document.parser.apache.pdfbox.ApachePdfBoxDocumentParser;
 import dev.langchain4j.data.document.splitter.DocumentByParagraphSplitter;
 import dev.langchain4j.data.embedding.Embedding;
 import io.milvus.param.dml.InsertParam;
 import io.milvus.param.dml.SearchParam;
 import io.milvus.response.SearchResultsWrapper;
+import org.apache.poi.hwpf.HWPFDocument;
 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;
         Document document;
         InputStream fileInputStream = new FileInputStream(transferFile);
         filename = filename.toLowerCase();
-        if (filename.endsWith(".docx")) {
+        if (filename.endsWith(".doc")) {
+            HWPFDocument doc = new HWPFDocument(fileInputStream);
+            document = Document.from(doc.getDocumentText());
+        }
+        else if (filename.endsWith(".docx")) {
             XWPFDocument docx = new XWPFDocument(fileInputStream);
             XWPFWordExtractor extractor = new XWPFWordExtractor(docx);
             String text = extractor.getText();
         else if (filename.endsWith(".pdf")) {
             document = new ApachePdfBoxDocumentParser().parse(fileInputStream);
         }
+        else if (filename.endsWith(".txt")) {
+            document = new TextDocumentParser().parse(fileInputStream);
+        }
         else {
             throw new UnsupportedOperationException("不支持文件类型: " + filename);
         }

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

 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;
         JSONObject jsonObject = new JSONObject();
         jsonObject.put("chatId", chatId);
         String[] filenameSplit = file.getOriginalFilename().split("\\.");
-        String outputFilename = "/upload/agent/" + agentName + "/" + file.getOriginalFilename().replace(filenameSplit[filenameSplit.length - 2], filenameSplit[filenameSplit.length - 2] + "_" + agentName);
+        String outputFilename = "/upload/agent/" + agentName + "/" + file.getOriginalFilename()
+                .replace(filenameSplit[filenameSplit.length - 2], filenameSplit[filenameSplit.length - 2] + "_" + agentName);
+        if (file.getOriginalFilename().endsWith(".doc"))
+            outputFilename = outputFilename.replace(".doc", "docx");
         CmcDocument cmcDocument = new CmcDocument();
         cmcDocument.setDocumentId(new SnowFlake().generateId());
         cmcDocument.setChatId(chatId);
             CmcChat cmcChat = new CmcChat();
             cmcChat.setChatId(jsonObject.getString("chatId"));
             cmcChat.setInputTime(new Date());
-            cmcChat.setInput("招标文件地址：" + outputFilename);
+            cmcChat.setInput("招标文件地址：" + "/upload/agent/" + agentName + "/" + file.getOriginalFilename());
             cmcChat.setUserId(SecurityUtils.getUserId());
             cmcChatMapper.insertCmcChat(cmcChat);
             XWPFDocument doc = new XWPFDocument(new FileInputStream(RuoYiConfig.getProfile() + "/upload/agent/template/technical.docx"));