智能体页面修改

oa-back/ruoyi-admin/src/main/java/com/ruoyi/web/controller/oa/CmcDeviceApprovalController.java

@@ -145,6 +145,9 @@ public class CmcDeviceApprovalController extends BaseController
             cmcDeviceApproval.setDispatchTime(new Date());
         }
         if (formDataJson.getJSONArray("modifyDevices").size() > 0 && formDataJson.getString("managerComment") == null) {
+            cmcDeviceApproval.setDispatcher(getLoginUser().getUserId());
+            cmcDeviceApproval.setDispatchComment(formDataJson.getString("dispatchComment"));
+            cmcDeviceApproval.setDispatchTime(new Date());
             String deviceString = formDataJson.getString("modifyDevices").substring(1, formDataJson.getString("modifyDevices").length() - 1);
             cmcDeviceApproval.setModifyDevices(deviceString);
             if (!formDataJson.getString("projectId").equals("")) {

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

@@ -15,6 +15,8 @@ import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestBody;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
+
+import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.annotation.Log;
 import com.ruoyi.common.core.controller.BaseController;
 import com.ruoyi.common.core.domain.AjaxResult;
@@ -27,7 +29,7 @@ import org.springframework.web.multipart.MultipartFile;
 
 /**
  * 智能体Controller
- * 
+ *
  * @author ruoyi
  * @date 2025-07-17
  */
@@ -79,6 +81,20 @@ public class CmcAgentController extends BaseController
         return success(ChatMessage.ofAssistant(cmcAgentService.getOpening(agentName)));
     }
 
+    /**
+     * 保存目录到Word文件
+     */
+    @PostMapping("/writeTitles")
+    public AjaxResult writeTitles(@RequestBody JSONObject data) throws IOException
+    {
+        JSONObject result = cmcAgentService.writeTitles(data);
+        if (result.getIntValue("code") == 200) {
+            return AjaxResult.success(result.getString("message"));
+        } else {
+            return AjaxResult.error(result.getString("message"));
+        }
+    }
+
     /**
      * 上传单文件
      * @return
@@ -143,7 +159,7 @@ public class CmcAgentController extends BaseController
      * 删除智能体
      */
     @Log(title = "智能体", businessType = BusinessType.DELETE)
-	@DeleteMapping("/{agentIds}")
+    @DeleteMapping("/{agentIds}")
     public AjaxResult remove(@PathVariable Integer[] agentIds)
     {
         return success(cmcAgentService.deleteCmcAgentByAgentIds(agentIds));

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

@@ -9,15 +9,15 @@ import java.util.List;
 
 /**
  * 智能体Service接口
- * 
+ *
  * @author ruoyi
  * @date 2025-07-17
  */
-public interface ICmcAgentService 
+public interface ICmcAgentService
 {
     /**
      * 查询智能体
-     * 
+     *
      * @param agentId 智能体主键
      * @return 智能体
      */
@@ -25,7 +25,7 @@ public interface ICmcAgentService
 
     /**
      * 查询智能体列表
-     * 
+     *
      * @param cmcAgent 智能体
      * @return 智能体集合
      */
@@ -72,7 +72,7 @@ public interface ICmcAgentService
 
     /**
      * 新增智能体
-     * 
+     *
      * @param cmcAgent 智能体
      * @return 结果
      */
@@ -80,7 +80,7 @@ public interface ICmcAgentService
 
     /**
      * 修改智能体
-     * 
+     *
      * @param cmcAgent 智能体
      * @return 结果
      */
@@ -88,7 +88,7 @@ public interface ICmcAgentService
 
     /**
      * 批量删除智能体
-     * 
+     *
      * @param agentIds 需要删除的智能体主键集合
      * @return 结果
      */
@@ -96,9 +96,17 @@ public interface ICmcAgentService
 
     /**
      * 删除智能体信息
-     * 
+     *
      * @param agentId 智能体主键
      * @return 结果
      */
     public int deleteCmcAgentByAgentId(Integer agentId);
+
+    /**
+     * 保存目录到Word文件
+     *
+     * @param data 目录数据
+     * @return 结果
+     */
+    public JSONObject writeTitles(JSONObject data) throws IOException;
 }

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

@@ -1,17 +1,20 @@
 package com.ruoyi.llm.service.impl;
 
 import com.alibaba.fastjson2.JSONObject;
+import com.alibaba.fastjson2.JSONArray;
 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.CmcAgent;
 import com.ruoyi.llm.domain.CmcChat;
+import com.ruoyi.llm.domain.CmcTopic;
 import com.ruoyi.llm.domain.CmcDocument;
 import com.ruoyi.llm.mapper.CmcAgentMapper;
 import com.ruoyi.llm.mapper.CmcChatMapper;
 import com.ruoyi.llm.mapper.CmcDocumentMapper;
 import com.ruoyi.llm.service.ICmcAgentService;
+import com.ruoyi.llm.service.ICmcTopicService;
 import dev.langchain4j.data.document.Document;
 import dev.langchain4j.data.document.parser.TextDocumentParser;
 import dev.langchain4j.data.document.parser.apache.pdfbox.ApachePdfBoxDocumentParser;
@@ -41,10 +44,8 @@ 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.message.ChatMessage;
+import org.noear.solon.ai.chat.prompt.Prompt;
 import org.noear.solon.ai.chat.session.InMemoryChatSession;
-import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTDecimalNumber;
-import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTP;
-import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPr;
 import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTPPrGeneral;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
@@ -78,6 +79,9 @@ public class CmcAgentServiceImpl implements ICmcAgentService
     @Autowired
     private CmcChatMapper cmcChatMapper;
 
+    @Autowired
+    private ICmcTopicService cmcTopicService;
+
     private String processValue = "";
 
     private static final EmbeddingModel embeddingModel = new BgeSmallZhV15EmbeddingModel();
@@ -95,10 +99,10 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         if (milvusServiceUrl == null || milvusServiceUrl.isEmpty()) {
             throw new IllegalArgumentException("milvusServiceUrl 配置不能为空");
         }
-//        milvusClient = new MilvusClientV2(
-//                ConnectConfig.builder()
-//                        .uri(milvusServiceUrl)
-//                        .build());
+        milvusClient = new MilvusClientV2(
+                ConnectConfig.builder()
+                        .uri(milvusServiceUrl)
+                        .build());
     }
 
     @PreDestroy
@@ -189,7 +193,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         cmcChat.setInput("招标文件地址：" + prefixPath + "/" + file.getOriginalFilename());
         cmcChat.setUserId(SecurityUtils.getUserId());
         cmcChatMapper.insertCmcChat(cmcChat);
-        if (agentName.contains("技术")) {
+        if (agentName.contains("技术标书")) {
             String[] filenameSplit = file.getOriginalFilename().split("\\.");
             String outputFilename = prefixPath + "/" + file.getOriginalFilename()
                     .replace(filenameSplit[filenameSplit.length - 2], filenameSplit[filenameSplit.length - 2] + "_" + agentName);
@@ -197,36 +201,65 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                 outputFilename = outputFilename.replace(".doc", ".docx");
             if (file.getOriginalFilename().endsWith(".pdf"))
                 outputFilename = outputFilename.replace(".pdf", ".docx");
-            Path outputFilePath = Paths.get(RuoYiConfig.getProfile() + outputFilename);
-            Files.deleteIfExists(outputFilePath);
-            InputStream fileInputStream = new FileInputStream(RuoYiConfig.getProfile() + "/upload/agent/template/technical.docx");
-            try (XWPFDocument doc = new XWPFDocument(fileInputStream)) {
-                // 保存文档到本地文件系统
-                try (FileOutputStream out = new FileOutputStream(RuoYiConfig.getProfile() + outputFilename)) {
-                    doc.write(out);
-                }
-            }
-            String question = "工作大纲/工作范围/招标范围/服务范围";
-            String chapters = generateAnswerWithDocument(profilePath + "/" + file.getOriginalFilename(),
-                    RuoYiConfig.getProfile() + outputFilename, question);
-
-            // 生成详细目录结构（包含二三级标题）
-            String detailedDirectory = generateDetailedDirectory(profilePath + "/" + file.getOriginalFilename());
-
+            
+            // 分割文档
+            List<TextSegment> segments = splitDocument(transferFile, 300, 50);
+            
             // 分析项目概况
-            String projectOverview = analyzeProjectOverview(profilePath + "/" + file.getOriginalFilename());
+            String projectOverview = analyzeProjectOverview(profilePath + "/" + file.getOriginalFilename(), segments);
+            jsonObject.put("projectOverview", projectOverview);
 
             // 分析评分要求
-            String scoringRequirements = analyzeScoringRequirements(profilePath + "/" + file.getOriginalFilename());
+            String scoringRequirements = analyzeScoringRequirements(profilePath + "/" + file.getOriginalFilename(), segments, agentName);
+            jsonObject.put("scoringRequirements", scoringRequirements);
+
+            // 生成详细目录结构（包含二三级标题）并写入文件
+            JSONArray detailedDirectoryTree = generateDetailedDirectory(file.getOriginalFilename(), agentName);
+            String detailedDirectory = buildDirectoryText(detailedDirectoryTree);
+            jsonObject.put("detailedDirectory", detailedDirectoryTree);
+            jsonObject.put("directoryText", detailedDirectory);
 
-            message = "好的，我已经收到您上传的招标文件。以下为根据招标文件生成的章节大纲：\n\n"+ chapters + "\n\n" +
+            message = "好的，我已经收到您上传的招标文件。以下为根据招标文件生成的章节大纲：\n\n"+ detailedDirectory + "\n\n" +
                     "若您对章节标题有异议，请打开" + "【<a href='/profile" + outputFilename + "'> 技术文件 " + "</a>】" + "进行修改，后续将根据修改后的章节标题，帮您生成对应章节内容。\n\n" +
                     "思考时间可能较长，请耐心等待！\n";
 
-            // 将分析结果添加到返回对象中
-            jsonObject.put("detailedDirectory", detailedDirectory);
-            jsonObject.put("projectOverview", projectOverview);
-            jsonObject.put("scoringRequirements", scoringRequirements);
+            // 返回输出文件名，以便前端在保存目录时使用
+            jsonObject.put("filename", file.getOriginalFilename()
+                    .replace(filenameSplit[filenameSplit.length - 2], filenameSplit[filenameSplit.length - 2] + "_" + agentName));
+            
+            // 根据 agentName 查询 agentId
+            CmcAgent queryAgent = new CmcAgent();
+            queryAgent.setAgentName(agentName);
+            List<CmcAgent> agentList = cmcAgentMapper.selectCmcAgentList(queryAgent);
+            Integer agentId = null;
+            if (!agentList.isEmpty()) {
+                agentId = agentList.get(0).getAgentId();
+            }
+            
+            // 去除文件名后缀作为 topic 名称
+            String originalFilename = file.getOriginalFilename();
+            String topic = originalFilename.substring(0, originalFilename.lastIndexOf('.'));
+            
+            // 创建 topic
+            CmcTopic cmcTopic = new CmcTopic();
+            cmcTopic.setTopicId(new SnowFlake().generateId());
+            cmcTopic.setAgentId(agentId);
+            cmcTopic.setTopic(topic);
+            cmcTopicService.insertCmcTopic(cmcTopic);
+            String topicId = cmcTopic.getTopicId();
+            
+            // 合并 projectOverview、scoringRequirements、detailedDirectory 作为 output
+            StringBuilder output = new StringBuilder();
+            output.append("【项目概况】\n").append(projectOverview).append("\n\n");
+            output.append("【评分要求】\n").append(scoringRequirements).append("\n\n");
+            output.append("【详细目录】\n").append(detailedDirectory);
+            
+            // 更新 cmc_chat 表
+            cmcChat.setTopicId(topicId);
+            cmcChat.setOutput(output.toString());
+            cmcChat.setOutputTime(new Date());
+            cmcChatMapper.updateCmcChat(cmcChat);
+            jsonObject.put("topicId", topicId);
         }
         else if (agentName.contains("检查")) {
             message = generateAnswerWithDocumentContent(profilePath + "/" + file.getOriginalFilename());
@@ -472,7 +505,6 @@ public class CmcAgentServiceImpl implements ICmcAgentService
             EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
                     .queryEmbedding(queryEmbedding)
                     .minScore(0.7) // 降低阈值以获取更多相关内容
-                    .maxResults(5) // 限制结果数量
                     .build();
 
             List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
@@ -585,18 +617,20 @@ public class CmcAgentServiceImpl implements ICmcAgentService
     }
 
     /**
-     * 调用LLM生成回答
+     * 调用LLM生成目录结构
      */
-    public String generateAnswerWithDocument(String uploadFilePath, String templatePath, String question) throws IOException {
+    public JSONArray generateDetailedDirectory(String fileName, String agentName) throws IOException {
+        String prefixPath = "/upload/agent/" + agentName;
         StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
-        File profilePath = new File(uploadFilePath);
+        File profilePath = new File(new File( RuoYiConfig.getProfile() + prefixPath) + "/" + fileName);
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
         List<TextSegment> segments = splitDocument(profilePath, 300, 50);
         List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
         embeddingStore.addAll(embeddings, segments);
-        Embedding queryEmbedding = embeddingModel.embed(question).content();
+        Embedding queryEmbedding = embeddingModel.embed("工作大纲/工作范围/招标范围/服务范围").content();
         EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
                 .queryEmbedding(queryEmbedding)
+                .maxResults(5)
                 .minScore(0.7)
                 .build();
         List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
@@ -607,7 +641,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         }
         //根据招标文件工作大纲要求生成二级标题
         sb.append("请基于上述招标文件中提到的")
-                .append(question)
+                .append("工作大纲/工作范围/招标范围/服务范围")
                 .append("，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
                 .append("6.1 XX\n" +
                         "6.2 XX\n" +
@@ -615,117 +649,118 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         "......\n" +
                         "6.n-1 XX\n" +
                         "6.n XX\n");
-        return writeChapters(sb.toString(), templatePath);
+        String directoryContent = writeChapters(sb.toString(), fileName, agentName);
+        return parseDirectoryToTree(directoryContent);
     }
 
     /**
-     * 生成详细目录结构（包含二三级标题）
+     * 将目录文本解析为树结构
      */
-    public String generateDetailedDirectory(String uploadFilePath) throws IOException {
-        processValue = "生成详细目录结构中: 0%";
-        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
-        File profilePath = new File(uploadFilePath);
-        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
-        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
-        embeddingStore.addAll(embeddings, segments);
+    private JSONArray parseDirectoryToTree(String directoryContent) {
+        JSONArray tree = new JSONArray();
+        Map<String, JSONObject> level1Map = new LinkedHashMap<>();
+        Map<String, JSONObject> level2Map = new LinkedHashMap<>();
+        Map<String, JSONObject> level3Map = new LinkedHashMap<>();
+
+        String[] lines = directoryContent.split("\n");
+        for (String line : lines) {
+            line = line.trim().replace("*", "").replace("#", "").replace("-", "");
+            if (!line.contains("6.") || line.isEmpty()) {
+                continue;
+            }
 
-        // 搜索与目录相关的内容
-        String directoryQuery = "目录结构、章节划分、文件结构";
-        Embedding queryEmbedding = embeddingModel.embed(directoryQuery).content();
-        EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
-                .queryEmbedding(queryEmbedding)
-                .minScore(0.7)
-                .build();
-        List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
-        results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
+            String[] parts = line.split(" ", 2);
+            if (parts.length < 2) {
+                continue;
+            }
 
-        for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
-            String content = embeddingMatch.embedded().toString();
-            sb.append(content).append("\n\n");
+            String number = parts[0].trim();
+            String title = parts[1].trim();
+            int dotCount = number.split("\\.").length - 1;
+
+            if (dotCount == 1) {
+                JSONObject level1 = new JSONObject();
+                level1.put("title", number + " " + title);
+                level1.put("children", new JSONArray());
+                level1Map.put(number, level1);
+            } else if (dotCount == 2) {
+                JSONObject level2 = new JSONObject();
+                level2.put("title", number + " " + title);
+                level2.put("children", new JSONArray());
+                level2Map.put(number, level2);
+
+                String parentNumber = number.substring(0, number.lastIndexOf("."));
+                if (level1Map.containsKey(parentNumber)) {
+                    level1Map.get(parentNumber).getJSONArray("children").add(level2);
+                }
+            } else if (dotCount == 3) {
+                JSONObject level3 = new JSONObject();
+                level3.put("title", number + " " + title);
+                level3Map.put(number, level3);
+
+                String parentNumber = number.substring(0, number.lastIndexOf("."));
+                if (level2Map.containsKey(parentNumber)) {
+                    level2Map.get(parentNumber).getJSONArray("children").add(level3);
+                }
+            }
+        }
+
+        for (Map.Entry<String, JSONObject> entry : level1Map.entrySet()) {
+            tree.add(entry.getValue());
         }
 
-        // 生成详细目录结构
-        sb.append("请基于上述招标文件内容，生成详细的目录结构，包含一级、二级和三级标题，严格按以下格式输出：\n")
-                .append("1. 一级标题\n")
-                .append("  1.1 二级标题\n")
-                .append("    1.1.1 三级标题\n")
-                .append("    1.1.2 三级标题\n")
-                .append("  1.2 二级标题\n")
-                .append("2. 一级标题\n")
-                .append("  2.1 二级标题\n")
-                .append("    2.1.1 三级标题\n")
-                .append("......\n");
-
-        processValue = "生成详细目录结构中: 50%";
-        String directory = generateAnswer(sb.toString());
-        processValue = "生成详细目录结构中: 100%";
-        return directory;
+        return tree;
     }
 
     /**
-     * 分析项目概况
+     * 将目录树结构转换为文本格式
      */
-    public String analyzeProjectOverview(String uploadFilePath) throws IOException {
-        processValue = "分析项目概况中: 0%";
-        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
-        File profilePath = new File(uploadFilePath);
-        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
-        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
-        embeddingStore.addAll(embeddings, segments);
-
-        // 搜索与项目概况相关的内容
-        String overviewQuery = "项目概况、项目基本信息、项目背景、项目目标、项目范围";
-        Embedding queryEmbedding = embeddingModel.embed(overviewQuery).content();
-        EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
-                .queryEmbedding(queryEmbedding)
-                .minScore(0.7)
-                .build();
-        List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
-        results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
-
-        for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
-            String content = embeddingMatch.embedded().toString();
-            sb.append(content).append("\n\n");
+    private String buildDirectoryText(JSONArray tree) {
+        StringBuilder sb = new StringBuilder();
+        for (int i = 0; i < tree.size(); i++) {
+            JSONObject level1 = tree.getJSONObject(i);
+            sb.append(level1.getString("title")).append("\n");
+            if (level1.containsKey("children")) {
+                JSONArray children = level1.getJSONArray("children");
+                for (int j = 0; j < children.size(); j++) {
+                    JSONObject level2 = children.getJSONObject(j);
+                    sb.append(level2.getString("title")).append("\n");
+                    if (level2.containsKey("children")) {
+                        JSONArray grandchildren = level2.getJSONArray("children");
+                        for (int k = 0; k < grandchildren.size(); k++) {
+                            JSONObject level3 = grandchildren.getJSONObject(k);
+                            sb.append(level3.getString("title")).append("\n");
+                        }
+                    }
+                }
+            }
         }
-
-        // 分析项目概况
-        sb.append("请基于上述招标文件内容，分析项目概况，包括但不限于以下内容：\n")
-                .append("1. 项目名称\n")
-                .append("2. 项目类型\n")
-                .append("3. 项目预算\n")
-                .append("4. 项目周期\n")
-                .append("5. 项目地点\n")
-                .append("6. 招标人\n")
-                .append("7. 项目背景\n")
-                .append("8. 项目目标\n")
-                .append("9. 项目范围\n")
-                .append("请以清晰的结构输出分析结果。\n");
-
-        processValue = "分析项目概况中: 50%";
-        String overview = generateAnswer(sb.toString());
-        processValue = "分析项目概况中: 100%";
-        return overview;
+        return sb.toString();
     }
 
     /**
-     * 分析评分要求
+     * 分析招标文件内容（公共方法）
+     * @param uploadFilePath 文件路径
+     * @param query 查询关键词
+     * @param analysisPrompt 分析提示语
+     * @param progressStartMsg 进度开始消息
+     * @param progressFoundMsg 进度找到消息
+     * @param progressEndMsg 进度结束消息
+     * @return 分析结果
      */
-    public String analyzeScoringRequirements(String uploadFilePath) throws IOException {
-        processValue = "分析评分要求中: 0%";
+    private String analyzeDocumentContent(List<TextSegment> segments, String query, String analysisPrompt,
+                                         String progressStartMsg, String progressFoundMsg, String progressEndMsg) throws IOException {
+        processValue = progressStartMsg;
         StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
-        File profilePath = new File(uploadFilePath);
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-        List<TextSegment> segments = splitDocument(profilePath, 300, 50);
         List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
         embeddingStore.addAll(embeddings, segments);
 
-        // 搜索与评分要求相关的内容
-        String scoringQuery = "评分要求、评分标准、评标办法、打分规则";
-        Embedding queryEmbedding = embeddingModel.embed(scoringQuery).content();
+        // 搜索相关内容
+        Embedding queryEmbedding = embeddingModel.embed(query).content();
         EmbeddingSearchRequest embeddingSearchRequest = EmbeddingSearchRequest.builder()
                 .queryEmbedding(queryEmbedding)
+                .maxResults(5)
                 .minScore(0.7)
                 .build();
         List<EmbeddingMatch<TextSegment>> results = embeddingStore.search(embeddingSearchRequest).matches();
@@ -736,26 +771,50 @@ public class CmcAgentServiceImpl implements ICmcAgentService
             sb.append(content).append("\n\n");
         }
 
-        // 分析评分要求
-        sb.append("请基于上述招标文件内容，分析评分要求，包括但不限于以下内容：\n")
-                .append("1. 评分项及权重\n")
-                .append("2. 技术方案评分标准\n")
-                .append("3. 商务方案评分标准\n")
-                .append("4. 服务方案评分标准\n")
-                .append("5. 其他评分项\n")
-                .append("请以清晰的结构输出分析结果，包括各项的具体分值和评分细则。\n");
-
-        processValue = "分析评分要求中: 50%";
-        String scoring = generateAnswer(sb.toString());
-        processValue = "分析评分要求中: 100%";
-        return scoring;
+        // 添加分析提示
+        sb.append(analysisPrompt);
+
+        processValue = progressFoundMsg;
+        String result = generateAnswer(sb.toString());
+        processValue = progressEndMsg;
+        return result;
+    }
+
+    /**
+     * 分析项目概况
+     */
+    public String analyzeProjectOverview(String uploadFilePath, List<TextSegment> segments) throws IOException {
+        String query = uploadFilePath.split("/")[uploadFilePath.split("/").length - 1] + "项目概况";
+        String analysisPrompt = "请基于上述招标文件内容，分析项目概况。\n" +
+                "请以清晰的结构输出分析结果。\n";
+        return analyzeDocumentContent(segments, query, analysisPrompt,
+                "分析项目概况中: 0%",
+                "已查到项目概况: 50%",
+                "分析项目概况中: 100%");
+    }
+
+    /**
+     * 分析评分要求
+     */
+    public String analyzeScoringRequirements(String uploadFilePath, List<TextSegment> segments, String agentName) throws IOException {
+        String query = "评分要求、评分标准、评标办法、打分规则";
+        if (agentName.contains("技术"))
+            query = "技术部分评分要求、评分标准、评标办法、打分规则";
+        else if (agentName.contains("商务"))
+            query = "商务部分评分要求、评分标准、评标办法、打分规则";
+        String analysisPrompt = "请基于上述招标文件内容，分析评分要求。\n" +
+                "请以清晰的结构输出分析结果，包括各项的具体分值和评分细则。\n";
+        return analyzeDocumentContent(segments, query, analysisPrompt,
+                "分析评分要求中: 0%",
+                "已查到评分要求: 50%",
+                "分析评分要求中: 100%");
     }
 
     /**
      * 编排章节
      * @return
      */
-    public String writeChapters(String prompt, String templatePath) throws IOException {
+    public String writeChapters(String prompt, String fileName, String agentName) throws IOException {
         String chapters2 = generateAnswer(prompt);
 
         //根据技术文档知识库生成二级标题下三级标题
@@ -801,7 +860,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                 "6.n.1 XX\n" +
                 "6.n 2 XX\n";
         String content = generateAnswer(sb);
-        writeTitles(content, templatePath);
+        writeTitles(content, fileName, agentName);
         return content;
     }
 
@@ -812,14 +871,17 @@ public class CmcAgentServiceImpl implements ICmcAgentService
     public String generateAnswer(String prompt) throws IOException {
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
                 .model("Qwen")
+                .timeout(java.time.Duration.ofSeconds(120))
                 .build();
 
         List<ChatMessage> messages = new ArrayList<>();
         messages.add(ChatMessage.ofUser(prompt));
         ChatSession chatSession =  InMemoryChatSession.builder().messages(messages).build();
-        ChatResponse response = chatModel.prompt(chatSession).call();
 
-        return response.lastChoice().getMessage().getContent();
+        Prompt prompt1 = Prompt.of(prompt).attrPut("session", chatSession);
+        ChatResponse response = chatModel.prompt(prompt1).call();
+        String content = response.lastChoice().getMessage().getContent();
+        return content;
     }
 
     /**
@@ -893,13 +955,27 @@ public class CmcAgentServiceImpl implements ICmcAgentService
      * 写入章节大纲
      * @return
      */
-    public void writeTitles(String content, String templatePath) throws IOException {
+    public void writeTitles(String content, String fileName, String agentName) throws IOException {
         List<String> chapters = new ArrayList<>();
         String[] contentLines = content.split("\n");
         for (String line : contentLines) {
             if (line.contains("6."))
                 chapters.add(line.replace("*", "").replace("#", "").replace("-", ""));
         }
+        String prefixPath = "/upload/agent/" + agentName;
+        String[] filenameSplit = fileName.split("\\.");
+        String outputFilename = prefixPath + "/" + fileName
+                .replace(filenameSplit[filenameSplit.length - 2], filenameSplit[filenameSplit.length - 2] + "_" + agentName);
+        String templatePath = RuoYiConfig.getProfile() + outputFilename;
+        Path outputFilePath = Paths.get(templatePath);
+        Files.deleteIfExists(outputFilePath);
+        InputStream inputStream = new FileInputStream(RuoYiConfig.getProfile() + "/upload/agent/template/technical.docx");
+        try (XWPFDocument doc = new XWPFDocument(inputStream)) {
+            // 保存文档到本地文件系统
+            try (FileOutputStream out = new FileOutputStream(templatePath)) {
+                doc.write(out);
+            }
+        }
         File file = new File(templatePath);
         FileInputStream fileInputStream = new FileInputStream(file);
         try (XWPFDocument document = new XWPFDocument(fileInputStream)) {
@@ -1010,6 +1086,50 @@ public class CmcAgentServiceImpl implements ICmcAgentService
     /**
      * 获取最低级别子标题列表
      */
+    /**
+     * 保存目录到Word文件
+     */
+    @Override
+    public JSONObject writeTitles(JSONObject data) throws IOException {
+        JSONObject result = new JSONObject();
+        try {
+            String filename = data.getString("filename");
+            String agentName = data.getString("agentName");
+            JSONArray titlesArray = data.getJSONArray("titles");
+
+            // 构建目录字符串
+            StringBuilder titlesContent = new StringBuilder();
+            buildTitlesContent(titlesArray, titlesContent);
+
+            // 调用现有的writeTitles方法写入Word文件
+            writeTitles(titlesContent.toString(), filename, agentName);
+
+            result.put("code", 200);
+            result.put("message", "目录保存成功");
+        } catch (Exception e) {
+            e.printStackTrace();
+            result.put("code", 500);
+            result.put("message", "目录保存失败: " + e.getMessage());
+        }
+        return result;
+    }
+
+    /**
+     * 构建目录内容字符串
+     */
+    private void buildTitlesContent(JSONArray titlesArray, StringBuilder content) {
+        for (int i = 0; i < titlesArray.size(); i++) {
+            JSONObject titleObj = titlesArray.getJSONObject(i);
+            String title = titleObj.getString("title");
+            content.append(title).append("\n");
+
+            if (titleObj.containsKey("children")) {
+                JSONArray children = titleObj.getJSONArray("children");
+                buildTitlesContent(children, content);
+            }
+        }
+    }
+
     public List<String> extractSubTitles(String filename, String question) throws IOException {
         List<String> subTitles = new ArrayList<>();
         InputStream fileInputStream = new FileInputStream(filename);
@@ -1152,6 +1272,96 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         }
     }
 
+    /**
+     * 从 DOCX 文档中提取表格内容
+     */
+    private List<TextSegment> extractTablesFromDocx(XWPFDocument document) {
+        List<TextSegment> tableSegments = new ArrayList<>();
+        List<XWPFTable> tables = document.getTables();
+
+        for (int i = 0; i < tables.size(); i++) {
+            XWPFTable table = tables.get(i);
+            StringBuilder tableContent = new StringBuilder();
+
+            // 添加表格标识
+            tableContent.append("[表格 ").append(i + 1).append("]\n");
+
+            // 提取表格内容
+            for (XWPFTableRow row : table.getRows()) {
+                StringBuilder rowContent = new StringBuilder();
+                for (XWPFTableCell cell : row.getTableCells()) {
+                    String cellText = cell.getText().trim();
+                    if (!cellText.isEmpty()) {
+                        rowContent.append(cellText).append(" | ");
+                    }
+                }
+                if (rowContent.length() > 0) {
+                    // 移除末尾的分隔符
+                    if (rowContent.length() >= 3) {
+                        rowContent.setLength(rowContent.length() - 3);
+                    }
+                    tableContent.append(rowContent).append("\n");
+                }
+            }
+
+            // 只有当表格内容有实际内容时才添加
+            if (tableContent.length() > 10) {
+                TextSegment segment = TextSegment.from(tableContent.toString());
+                tableSegments.add(segment);
+            }
+        }
+
+        return tableSegments;
+    }
+
+    /**
+     * 从 DOC 文档中提取表格内容
+     */
+    private List<TextSegment> extractTablesFromDoc(HWPFDocument document) {
+        List<TextSegment> tableSegments = new ArrayList<>();
+        Range range = document.getRange();
+
+        int tableCount = 0;
+        StringBuilder tableContent = new StringBuilder();
+        boolean inTable = false;
+
+        for (int i = 0; i < range.numParagraphs(); i++) {
+            Paragraph paragraph = range.getParagraph(i);
+            String text = paragraph.text().trim();
+
+            // 检测表格段落
+            if (paragraph.isInTable()) {
+                if (!inTable) {
+                    // 开始新表格
+                    tableCount++;
+                    tableContent = new StringBuilder();
+                    tableContent.append("[表格 ").append(tableCount).append("]\n");
+                    inTable = true;
+                }
+
+                // 添加表格行内容
+                if (!text.isEmpty()) {
+                    tableContent.append(text).append("\n");
+                }
+            } else if (inTable) {
+                // 表格结束
+                inTable = false;
+                if (tableContent.length() > 10) {
+                    TextSegment segment = TextSegment.from(tableContent.toString());
+                    tableSegments.add(segment);
+                }
+            }
+        }
+
+        // 处理最后一个表格
+        if (inTable && tableContent.length() > 10) {
+            TextSegment segment = TextSegment.from(tableContent.toString());
+            tableSegments.add(segment);
+        }
+
+        return tableSegments;
+    }
+
     /**
      * 按三级标题分割DOC内容
      */
@@ -1243,6 +1453,10 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         splitDocxByLevel3(currentLevel2Content.toString(), xwpfDocument, segments);
                     }
                 }
+
+                // 提取文档中的所有表格
+                List<TextSegment> tableSegments = extractTablesFromDocx(xwpfDocument);
+                segments.addAll(tableSegments);
             }
             return segments;
         }
@@ -1289,6 +1503,10 @@ public class CmcAgentServiceImpl implements ICmcAgentService
                         splitDocByLevel3(currentLevel2Content.toString(), hwpfDocument, segments);
                     }
                 }
+
+                // 提取文档中的所有表格
+                List<TextSegment> tableSegments = extractTablesFromDoc(hwpfDocument);
+                segments.addAll(tableSegments);
             }
             return segments;
         }

oa-ui/src/api/llm/agent.js

@@ -2,7 +2,7 @@
  * @Author: wrh
  * @Date: 2025-07-17 18:06:24
  * @LastEditors: wrh
- * @LastEditTime: 2025-09-02 16:43:25
+ * @LastEditTime: 2026-04-24 13:52:03
  */
 import request from '@/utils/request'
 
@@ -65,11 +65,14 @@ export function uploadFile(file, agentName) {
 }
 
 // 上传单文件
-export function uploadModifyFile(topicId, file, agentName) {
+export function uploadModifyFile(topicId, file, agentName, chapterNumber) {
   const formData = new FormData()
   formData.append('topicId', topicId)
   formData.append('file', file)
   formData.append('agentName', agentName)
+  if (chapterNumber) {
+    formData.append('chapterNumber', chapterNumber)
+  }
   return request({
     url: '/llm/agent/modifyFile',
     method: 'post',
@@ -113,3 +116,12 @@ export function opening(agentName) {
     params: { agentName }
   })
 }
+
+// 保存目录到Word文件
+export function writeTitles(data) {
+  return request({
+    url: '/llm/agent/writeTitles',
+    method: 'post',
+    data: data
+  })
+}

oa-ui/src/views/llm/agent/index.vue

@@ -310,7 +310,9 @@ export default {
 }
 
 .left-panel {
-  width: 400px;
+  width: 25%;
+  min-width: 260px;
+  max-width: 320px;
   background: white;
   border-radius: 8px;
   box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);

oa-ui/src/views/llm/knowledge/index.vue

@@ -1277,7 +1277,7 @@ export default {
 }
 
 .left-panel {
-  width: 400px;
+  width: 250px;
   background: white;
   border-radius: 8px;
   box-shadow: 0 2px 8px rgba(0, 0, 0, 0.1);