获取二级标题下三级标题列表

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

@@ -73,9 +73,9 @@ public class CmcDocumentController extends BaseController
      * @return
      */
     @PostMapping("/upload")
-    public AjaxResult upload(MultipartFile[] fileList) throws IOException
+    public AjaxResult upload(MultipartFile[] fileList, String agentName) throws IOException
     {
-        return success(cmcDocumentService.uploadDocument(fileList));
+        return success(cmcDocumentService.uploadDocument(fileList, agentName));
     }
 
     /**

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

@@ -1,6 +1,7 @@
 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.web.llm.service.ILangChainMilvusService;
 import dev.langchain4j.model.embedding.EmbeddingModel;
@@ -78,6 +79,7 @@ public class McpController extends BaseController
     @GetMapping("/answerWithDocument")
     public Flux<AssistantMessage> answerWithDocumentAndCollection(String collectionName, String topicId, String chatId, 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");
     }

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

@@ -55,4 +55,9 @@ public interface ILangChainMilvusService {
      */
     public Flux<AssistantMessage> generateAnswerWithDocumentAndCollection(EmbeddingModel embeddingModel, String topicId, String chatId, String question,  List<JSONObject> requests, String llmServiceUrl) throws IOException;
 
+    /**
+     * 获取二级标题下三级标题列表
+     * @return
+     */
+    public List<String> extractSubTitles(String filename, String question) throws IOException;
 }

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

@@ -13,6 +13,7 @@ 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.store.embedding.EmbeddingMatch;
 import dev.langchain4j.store.embedding.EmbeddingSearchRequest;
 import dev.langchain4j.store.embedding.inmemory.InMemoryEmbeddingStore;
 import io.milvus.client.MilvusClient;
@@ -28,6 +29,7 @@ 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.ai.chat.ChatModel;
 import org.noear.solon.ai.chat.ChatResponse;
 import org.noear.solon.ai.chat.ChatSession;
@@ -195,7 +197,7 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         List<CmcDocument> documentList = cmcDocumentService.selectCmcDocumentList(cmcDocument);
         StringBuilder sb = new StringBuilder("问题: " + question + "\n\n").append("根据以下上下文回答问题：\n\n");
         for (CmcDocument document : documentList) {
-            File profilePath = new File(RuoYiConfig.getProfile() + "/upload/rag/document/" + document.getPath());
+            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<>();
@@ -205,11 +207,13 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
                     .queryEmbedding(queryEmbedding)
                     .maxResults(1)
                     .build();
-            String contexts = embeddingStore.search(embeddingSearchRequest).matches().get(0).embedded().text();
-            sb.append("文件").append(": ")
-                    .append(document.getPath()).append("\n\n")
-                    .append("上下文").append(": ")
-                    .append(contexts).append("\n\n");
+            for (EmbeddingMatch embeddingMatch : embeddingStore.search(embeddingSearchRequest).matches()) {
+                String contexts = embeddingMatch.embedded().toString();
+                sb.append("文件").append(": ")
+                        .append(document.getPath()).append("\n\n")
+                        .append("上下文").append(": ")
+                        .append(contexts).append("\n\n");
+            }
         }
         return generateAnswer(topicId, sb.toString(), llmServiceUrl);
     }
@@ -218,18 +222,13 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
      * 调用LLM生成回答
      */
     @Override
-    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 chatId, 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);
-        StringBuilder sb = new StringBuilder("问题: " + question + "\n\n").append("根据以下要求：\n\n");
-        for (JSONObject request : requests) {
-            sb.append("要求").append(": ")
-                    .append(request.getString("content")).append("\n\n");
-        }
-        sb.append("参考以下文件上下文回答问题：\n\n");
         for (CmcDocument document : documentList) {
-            File profilePath = new File(RuoYiConfig.getProfile() + "/upload/rag/document/" + document.getPath());
+            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<>();
@@ -237,17 +236,60 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
             Embedding queryEmbedding = embeddingModel.embed(question).content();
             EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
                     .queryEmbedding(queryEmbedding)
-                    .maxResults(1)
+                    .maxResults(3)
                     .build();
-            String contexts = embeddingStore.search(embeddingSearchRequest).matches().get(0).embedded().text();
-            sb.append("文件").append(": ")
-                    .append(document.getPath()).append("\n\n")
-                    .append("上下文").append(": ")
-                    .append(contexts).append("\n\n");
+            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(topicId, sb.toString(), llmServiceUrl);
     }
 
+    /**
+     * 获取二级标题下三级标题列表
+     */
+    @Override
+    public List<String> extractSubTitles(String filename, String question) throws IOException {
+        List<String> subTitles = new ArrayList<>();
+        boolean inTargetSection = false;
+
+        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;
+    }
+
     /**
      * 检索知识库
      */
@@ -288,14 +330,13 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         return wrapper;
     }
 
-
     /**
      * 检索知识库
      */
-    private List<TextSegment> splitDocument(String filename, File profilePath) throws IOException {
+    private List<TextSegment> splitDocument(String filename, File transferFile) throws IOException {
         // 加载文档
         Document document;
-        InputStream fileInputStream = new FileInputStream(profilePath);
+        InputStream fileInputStream = new FileInputStream(transferFile);
         filename = filename.toLowerCase();
         if (filename.endsWith(".docx")) {
             XWPFDocument docx = new XWPFDocument(fileInputStream);
@@ -312,4 +353,5 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         DocumentByParagraphSplitter splitter = new DocumentByParagraphSplitter(1000,200);
         return splitter.split(document);
     }
+
 }

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

@@ -35,9 +35,10 @@ public interface ICmcDocumentService
      * 上传外部文件
      *
      * @param fileList 文件
+     * @param agentName 智能体名称
      * @return 结果
      */
-    public JSONObject uploadDocument(MultipartFile[] fileList) throws IOException;
+    public JSONObject uploadDocument(MultipartFile[] fileList, String agentName) throws IOException;
 
     /**
      * 新增cmc聊天附件

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

@@ -66,24 +66,30 @@ public class CmcDocumentServiceImpl implements ICmcDocumentService
      * 上传外部文件
      *
      * @param fileList 文件
+     * @param agentName 智能体名称
      * @return 结果
      */
     @Override
-    public JSONObject uploadDocument(MultipartFile[] fileList) throws IOException {
-        File profilePath = new File( RuoYiConfig.getProfile() + "/upload/rag/document" );
+    public JSONObject uploadDocument(MultipartFile[] fileList, String agentName) throws IOException {
+        String prefixPath = "";
+        if (!agentName.equals(""))
+            prefixPath = "/upload/agent/" + agentName;
+        else
+            prefixPath = "/upload/rag/document";
+        File profilePath = new File( RuoYiConfig.getProfile() + prefixPath);
         if (!profilePath.exists())
             profilePath.mkdirs();
         String chatId = new SnowFlake().generateId();
         JSONObject jsonObject = new JSONObject();
         jsonObject.put("chatId", chatId);
         for (MultipartFile file : fileList) {
-            File transferFile = new File(profilePath + File.separator + file.getOriginalFilename());
+            File transferFile = new File(profilePath + "/" + file.getOriginalFilename());
             if (!transferFile.exists())
                 file.transferTo(transferFile);
             CmcDocument cmcDocument = new CmcDocument();
             cmcDocument.setDocumentId(new SnowFlake().generateId());
             cmcDocument.setChatId(chatId);
-            cmcDocument.setPath(file.getOriginalFilename());
+            cmcDocument.setPath(prefixPath + "/" + file.getOriginalFilename());
             cmcDocumentMapper.insertCmcDocument(cmcDocument);
         }
         return jsonObject;

llm-ui/src/api/llm/document.js

@@ -51,11 +51,12 @@ export function delDocument(documentId) {
 
 
 // 上传聊天外部文件
-export function uploadDocument(fileList) {
+export function uploadDocument(fileList, agentName) {
   const formData = new FormData()
   for (let f of fileList) {
     formData.append('fileList', f)
   }
+  formData.append('agentName', agentName)
   return request({
     url: '/llm/document/upload',
     method: 'post',

llm-ui/src/views/llm/chat/index.vue

@@ -1,8 +1,8 @@
 <!--
  * @Author: ysh
  * @Date: 2025-04-07 14:14:05
- * @LastEditors: Please set LastEditors
- * @LastEditTime: 2025-07-29 15:25:17
+ * @LastEditors: wrh
+ * @LastEditTime: 2025-07-29 17:12:19
 -->
 <template>
   <div class="app-container">
@@ -590,7 +590,7 @@ const handleFileSelect = async (event) => {
     // 立即上传文件
     try {
       isUploading.value = true;
-      let res = await uploadDocument(files);
+      let res = await uploadDocument(files, '');
       documentChatId.value = res.data.chatId;
       // 上传成功后，将文件添加到已上传列表
       selectedFiles.value = [...selectedFiles.value, ...files];