solon ai的chat改为agent

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

@@ -1,9 +1,7 @@
 package com.ruoyi.web.llm.controller;
 
-import java.io.IOException;
 import java.util.Date;
 import java.util.List;
-import java.util.concurrent.ExecutionException;
 
 import javax.servlet.http.HttpServletResponse;
 
@@ -88,7 +86,7 @@ public class CmcAgentController extends BaseController
      * 保存目录到Word文件
      */
     @PostMapping("/writeTitles")
-    public AjaxResult writeTitles(@RequestBody JSONObject data) throws IOException
+    public AjaxResult writeTitles(@RequestBody JSONObject data) throws Throwable
     {
         JSONObject result = cmcAgentService.writeTitles(data);
         if (result.getIntValue("code") == 200) {
@@ -103,7 +101,7 @@ public class CmcAgentController extends BaseController
      * @return
      */
     @PostMapping("/upload")
-    public AjaxResult upload(MultipartFile file, String agentName) throws IOException
+    public AjaxResult upload(MultipartFile file, String agentName) throws Throwable
     {
         return success(cmcAgentService.uploadDocument(file, agentName));
     }
@@ -113,7 +111,7 @@ public class CmcAgentController extends BaseController
      * @return
      */
     @PostMapping("/modifyFile")
-    public AjaxResult uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException, InterruptedException, ExecutionException
+    public AjaxResult uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode) throws Throwable
     {
         return success(cmcAgentService.uploadModifyFile(topicId, file, agentName, selectedNode));
     }
@@ -123,7 +121,7 @@ public class CmcAgentController extends BaseController
      */
     @Anonymous
     @PostMapping(value = "/streamGenerate", produces = "text/event-stream;charset=UTF-8")
-    public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException
+    public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode) throws Throwable
     {
         return cmcAgentService.streamGenerateChapters(topicId, file, agentName, selectedNode);
     }
@@ -133,7 +131,7 @@ public class CmcAgentController extends BaseController
      * @return
      */
     @PostMapping("/uploadList")
-    public AjaxResult uploadList(MultipartFile[] fileList, String agentName) throws IOException
+    public AjaxResult uploadList(MultipartFile[] fileList, String agentName) throws Throwable
     {
         return success(cmcAgentService.uploadDocumentList(fileList, agentName));
     }

oa-back/ruoyi-system/pom.xml

@@ -60,7 +60,13 @@
         <dependency>
             <groupId>org.noear</groupId>
             <artifactId>solon-ai</artifactId>
-            <version>3.9.5</version>
+            <version>4.0.2</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.noear</groupId>
+            <artifactId>solon-ai-agent</artifactId>
+            <version>4.0.2</version>
         </dependency>
 
         <!-- Spring Boot Web -->

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

@@ -6,7 +6,6 @@ import com.ruoyi.llm.domain.CmcAgent;
 import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.io.IOException;
 import java.util.List;
 
 /**
@@ -47,7 +46,7 @@ public interface ICmcAgentService
      * @param file 文件
      * @return 结果
      */
-    public JSONObject uploadDocument(MultipartFile file, String agentName) throws IOException;
+    public JSONObject uploadDocument(MultipartFile file, String agentName) throws Throwable;
 
     /**
      * 上传修改文件
@@ -58,7 +57,7 @@ public interface ICmcAgentService
      * @param selectedNode 选中的节点标题（可选，为空时生成全部章节）
      * @return 结果
      */
-    public JSONObject uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException, InterruptedException, java.util.concurrent.ExecutionException;
+    public JSONObject uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode) throws Throwable;
 
     /**
      * 上传多文件
@@ -66,7 +65,7 @@ public interface ICmcAgentService
      * @param fileList 文件
      * @return 结果
      */
-    public String uploadDocumentList(MultipartFile[] fileList, String agentName) throws IOException;
+    public String uploadDocumentList(MultipartFile[] fileList, String agentName) throws Throwable;
 
     /**
      * 获取进度
@@ -113,7 +112,7 @@ public interface ICmcAgentService
      * @param data 目录数据
      * @return 结果
      */
-    public JSONObject writeTitles(JSONObject data) throws IOException;
+    public JSONObject writeTitles(JSONObject data) throws Throwable;
 
     /**
      * 流式生成章节内容（SSE方式）
@@ -124,6 +123,6 @@ public interface ICmcAgentService
      * @param selectedNode 选中的节点标题（可选，为空时生成全部章节）
      * @return SseEmitter
      */
-    public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode) throws IOException;
+    public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode) throws Throwable;
 
 }

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

@@ -42,8 +42,8 @@ import org.apache.poi.hwpf.usermodel.Range;
 import org.apache.poi.xwpf.usermodel.*;
 import org.apache.xmlbeans.XmlCursor;
 import org.noear.solon.ai.AiUsage;
+import org.noear.solon.ai.agent.simple.SimpleAgent;
 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;
@@ -138,7 +138,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
     @Override
     public String getOpening(String agentName) {
         String content = "";
-        if (agentName.contains("技术"))
+        if (agentName.contains("技术标书"))
             content = "我是投标文件写作助手，我将助您完成技术文件部分撰写。请上传招标询价文件，分析后将依招标服务要求，运用技术方案知识库，为您提供参考。";
         else if (agentName.contains("检查"))
             content = "我是文档检查助手，我将助您完成错别字检查。请上传文件。";
@@ -152,7 +152,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      * @return 结果
      */
     @Override
-    public JSONObject uploadDocument(MultipartFile file, String agentName) throws IOException {
+    public JSONObject uploadDocument(MultipartFile file, String agentName) throws Throwable {
         processValue = "";
         String prefixPath = "/upload/agent/" + agentName;
         File profilePath = new File(RuoYiConfig.getProfile() + prefixPath);
@@ -176,7 +176,10 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         CmcChat cmcChat = new CmcChat();
         cmcChat.setChatId(jsonObject.getString("chatId"));
         cmcChat.setInputTime(new Date());
-        cmcChat.setInput("招标文件地址：" + prefixPath + "/" + file.getOriginalFilename());
+        if (agentName.contains("技术设计书"))
+            cmcChat.setInput("项目任务书地址：" + prefixPath + "/" + file.getOriginalFilename());
+        else
+            cmcChat.setInput("招标文件地址：" + prefixPath + "/" + file.getOriginalFilename());
         cmcChat.setUserId(SecurityUtils.getUserId());
         cmcChatMapper.insertCmcChat(cmcChat);
         if (agentName.contains("技术标书")) {
@@ -197,7 +200,9 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
             jsonObject.put("projectOverview", projectOverview);
 
             // 分析评分标准
-            String scoringRequirements = analyzeScoringRequirements(segments, agentName);
+            String scoringRequirements = "";
+            if (agentName.contains("技术标书"))
+                scoringRequirements = analyzeScoringRequirements(segments, agentName);
             jsonObject.put("scoringRequirements", scoringRequirements);
 
             // 提取输出文件名（不含路径）
@@ -209,10 +214,16 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
             jsonObject.put("detailedDirectory", detailedDirectoryTree);
             jsonObject.put("directoryText", detailedDirectory);
 
-            message = "好的，我已经收到您上传的招标文件。以下为根据招标文件生成的章节大纲：\n\n" + detailedDirectory + "\n\n" +
-                    "若您对章节标题有异议，请打开" + "【<a href='/profile" + outputFilename + "'> 技术文件 " + "</a>】"
-                    + "进行修改，后续将根据修改后的章节标题，帮您生成对应章节内容。\n\n" +
-                    "思考时间可能较长，请耐心等待！\n";
+            if (agentName.contains("技术标书"))
+                message = "好的，我已经收到您上传的招标文件。以下为根据招标文件生成的章节大纲：\n\n" + detailedDirectory + "\n\n" +
+                        "若您对章节标题有异议，请打开" + "【<a href='/profile" + outputFilename + "'> 技术文件 " + "</a>】"
+                        + "进行修改，后续将根据修改后的章节标题，帮您生成对应章节内容。\n\n" +
+                        "思考时间可能较长，请耐心等待！\n";
+            else
+                message = "好的，我已经收到您上传的项目任务书。以下为根据项目任务书生成的章节大纲：\n\n" + detailedDirectory + "\n\n" +
+                        "若您对章节标题有异议，请打开" + "【<a href='/profile" + outputFilename + "'> 项目任务书 " + "</a>】"
+                        + "进行修改，后续将根据修改后的章节标题，帮您生成对应章节内容。\n\n" +
+                        "思考时间可能较长，请耐心等待！\n";
 
             // 返回输出文件名，以便前端在保存目录时使用
             jsonObject.put("filename", file.getOriginalFilename()
@@ -244,7 +255,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
             StringBuilder output = new StringBuilder();
             output.append("【项目概况】\n").append(projectOverview).append("\n\n");
             // 当agentname包含"技术"且包含"询价"时，使用"应价人须知"，否则使用"评分标准"
-            String scoringTitle = agentName.contains("技术") && agentName.contains("询价") ? "【应价人须知】" : "【评分标准】";
+            String scoringTitle = agentName.contains("技术标书") && agentName.contains("询价") ? "【应价人须知】" : "【评分标准】";
             output.append(scoringTitle).append("\n").append(scoringRequirements).append("\n\n");
             output.append("【详细目录】\n").append(detailedDirectory);
 
@@ -269,7 +280,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      */
     @Override
     public JSONObject uploadModifyFile(String topicId, MultipartFile file, String agentName, String selectedNode)
-            throws IOException, InterruptedException, ExecutionException {
+            throws IOException, Throwable {
         processValue = "";
         String prefixPath = "/upload/agent/" + agentName;
         File profilePath = new File(RuoYiConfig.getProfile() + prefixPath);
@@ -332,7 +343,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         }
 
         String message = "";
-        if (agentName.contains("技术")) {
+        if (agentName.contains("技术标书")) {
             String docPath = profilePath + "/" + file.getOriginalFilename();
             // 优先从 Word 文档提取段落 text 格式的叶子标题
             List<String> allTitles = extractSubTitles(docPath, "技术文件");
@@ -463,7 +474,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      */
     private Map<String, String> generateChaptersContent(String inputPath, String templatePath,
             List<String> targetTitles, String projectOverview, String scoringRequirements, String agentName)
-            throws IOException, InterruptedException, ExecutionException {
+            throws Throwable {
         // 构建招标文件内容的嵌入向量存储（inputPath 是招标文件路径，通过 topicId 查询得到）
         File tenderFile = new File(inputPath);
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
@@ -527,7 +538,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
                         titleContentMap.put(title, chapterContent);
                         successTitles.add(title);
 
-                    } catch (Exception e) {
+                    } catch (Throwable e) {
                         failureTitles.add(title);
                         System.err.println("生成章节 " + title + " 内容时出错: " + e.getMessage());
                         titleContentMap.put(title, "该章节内容生成失败，请手动填写。");
@@ -633,7 +644,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
     /**
      * 调用LLM生成回答 - 检查文档内容
      */
-    public String generateAnswerWithDocumentContent(String documentPath) throws IOException {
+    public String generateAnswerWithDocumentContent(String documentPath) throws Throwable {
         File profilePath = new File(documentPath);
         List<TextSegment> segments = splitDocument(profilePath, 10000, 0);
         StringBuilder sb = new StringBuilder();
@@ -695,8 +706,9 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
 
         // 评分标准/应价人须知
         if (scoringRequirements != null && !scoringRequirements.isEmpty()) {
-            // 当agentname包含"技术"且包含"询价"时，使用"应价人须知"，否则使用"评分标准"
-            String scoringTitle = agentName != null && agentName.contains("技术") && agentName.contains("询价") ? "【应价人须知】"
+            // 当agentname包含"技术标书"且包含"询价"时，使用"应价人须知"，否则使用"评分标准"
+            String scoringTitle = agentName != null && agentName.contains("技术标书") && agentName.contains("询价")
+                    ? "【应价人须知】"
                     : "【评分标准】";
             prompt.append(scoringTitle).append(":\n");
             prompt.append(scoringRequirements).append("\n\n");
@@ -735,7 +747,12 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         prompt.append("【输出格式】:\n");
         prompt.append("请直接输出章节的正文内容（不要包含三级标题），正文内容为总-分结构，别写成总-分-总结构。");
         prompt.append("【标题编号规范】:\n");
-        prompt.append("如果需要在章节内添加子标题，必须严格遵循层级编号格式，最多到第六级标题（如6.1,6.1.1,6.1.1.1,6.1.1.1.1,6.1.1.1.1.1,6.1.1.1.1.1.1），且子标题后应添加换行符。\n");
+        if (agentName.contains("技术设计书"))
+            prompt.append(
+                    "如果需要在章节内添加子标题，必须严格遵循层级编号格式，最多到第六级标题（如1.1,1.1.1,1.1.1.1,1.1.1.1.1,1.1.1.1.1.1,1.1.1.1.1.1），且子标题后应添加换行符。\\n");
+        else
+            prompt.append(
+                    "如果需要在章节内添加子标题，必须严格遵循层级编号格式，最多到第六级标题（如6.1,6.1.1,6.1.1.1,6.1.1.1.1,6.1.1.1.1.1,6.1.1.1.1.1.1），且子标题后应添加换行符。\n");
         prompt.append("各级标题下，正文文字叙述部份的分点说明序号分为六级，\n");
         prompt.append("第一级按阿拉伯数字序号编排，如1,2,3\n");
         prompt.append("第二级按带右圆括号的阿拉伯数字序号编排，如1）,2）,3）\n");
@@ -864,33 +881,52 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      * 调用LLM生成目录结构
      */
     public JSONArray generateDetailedDirectory(String fileName, String agentName, List<TextSegment> segments)
-            throws IOException {
-        StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
-        InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
-        List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
-        embeddingStore.addAll(embeddings, segments);
-        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();
-        results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
-        for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
-            String requests = embeddingMatch.embedded().toString();
-            sb.append(requests).append("\n\n");
+            throws Throwable {
+        StringBuilder sb = new StringBuilder("");
+        if (agentName.contains("技术设计书")) {
+            sb.append("项目设计书内容：\n\n");
+            for (TextSegment segment : segments) {
+                sb.append(segment.toString()).append("\n\n");
+            }
+        } else if (agentName.contains("技术标书")) {
+            sb.append("招标文件内容：\n\n");
+
+            InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
+            List<Embedding> embeddings = embeddingModel.embedAll(segments).content();
+            embeddingStore.addAll(embeddings, segments);
+            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();
+            results.sort(Comparator.comparingDouble(EmbeddingMatch<TextSegment>::score).reversed());
+            for (EmbeddingMatch<TextSegment> embeddingMatch : results) {
+                String requests = embeddingMatch.embedded().toString();
+                sb.append(requests).append("\n\n");
+            }
         }
         // 根据招标文件工作大纲要求生成二级标题
-        sb.append("请基于上述招标文件中提到的")
-                .append("技术文件工作大纲要求")
-                .append("，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
-                .append("6.1 XX\n" +
-                        "6.2 XX\n" +
-                        "6.3 XX\n" +
-                        "......\n" +
-                        "6.n-1 XX\n" +
-                        "6.n XX\n");
+        if (agentName.contains("技术标书"))
+            sb.append("请基于上述招标文件中提到的")
+                    .append("技术文件工作大纲要求")
+                    .append("，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
+                    .append("6.1 XX\n" +
+                            "6.2 XX\n" +
+                            "6.3 XX\n" +
+                            "......\n" +
+                            "6.n-1 XX\n" +
+                            "6.n XX\n");
+        else if (agentName.contains("技术设计书"))
+            sb.append("请基于上述项目设计书内容")
+                    .append("，先列出二级章节标题，严格按以下格式，仅输出标题列表：\n")
+                    .append("1.1 XX\n" +
+                            "1.2 XX\n" +
+                            "1.3 XX\n" +
+                            "......\n" +
+                            "1.n-1 XX\n" +
+                            "1.n XX\n");
         String directoryContent = writeChapters(sb.toString(), fileName, agentName);
         return parseDirectoryToTree(directoryContent);
     }
@@ -907,7 +943,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         String[] lines = directoryContent.split("\n");
         for (String line : lines) {
             line = line.trim().replace("*", "").replace("#", "").replace("-", "");
-            if (!line.contains("6.") || line.isEmpty()) {
+            if ((!line.startsWith("6.") && !line.startsWith("1.")) || line.isEmpty()) {
                 continue;
             }
 
@@ -992,7 +1028,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      * @return 分析结果
      */
     private String analyzeDocumentContent(List<TextSegment> segments, String query, String analysisPrompt,
-            String progressStartMsg, String progressFoundMsg, String progressEndMsg) throws IOException {
+            String progressStartMsg, String progressFoundMsg, String progressEndMsg) throws Throwable {
         processValue = progressStartMsg;
         StringBuilder sb = new StringBuilder("招标文件内容：\n\n");
         InMemoryEmbeddingStore<TextSegment> embeddingStore = new InMemoryEmbeddingStore<>();
@@ -1026,7 +1062,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
     /**
      * 分析项目概况
      */
-    public String analyzeProjectOverview(String uploadFilePath, List<TextSegment> segments) throws IOException {
+    public String analyzeProjectOverview(String uploadFilePath, List<TextSegment> segments) throws Throwable {
         String query = uploadFilePath.split("/")[uploadFilePath.split("/").length - 1] + "项目概况";
         String analysisPrompt = "请基于上述招标文件内容，分析项目概况。\n" +
                 "请以清晰的结构输出分析结果。\n";
@@ -1040,13 +1076,13 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      * 分析评分标准/应价人须知
      */
     public String analyzeScoringRequirements(List<TextSegment> segments, String agentName)
-            throws IOException {
+            throws Throwable {
         String query = "评分标准、评标办法、打分规则";
         String analysisPrompt = "请基于上述招标文件内容，分析评分标准。\n" +
                 "请以清晰的结构输出分析结果，包括各项的具体分值和评分细则。\n";
 
-        // 当agentname包含"技术"且包含"询价"时，使用"应价人须知"相关查询
-        if (agentName.contains("技术") && agentName.contains("询价")) {
+        // 当agentname包含"技术标书"且包含"询价"时，使用"应价人须知"相关查询
+        if (agentName.contains("技术标书") && agentName.contains("询价")) {
             query = "应价人须知、报价要求、响应要求、投标报价";
             analysisPrompt = "请基于上述招标文件内容，分析应价人须知。\n" +
                     "请以清晰的结构输出分析结果，包括报价要求、响应要求等内容。\n";
@@ -1054,7 +1090,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
                     "分析应价人须知中: 0%",
                     "已查到应价人须知: 50%",
                     "分析应价人须知中: 100%");
-        } else if (agentName.contains("技术")) {
+        } else if (agentName.contains("技术标书")) {
             query = "技术部分评分标准、评标办法、打分规则";
         } else if (agentName.contains("商务")) {
             query = "商务部分评分标准、评标办法、打分规则";
@@ -1071,7 +1107,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      * 
      * @return
      */
-    public String writeChapters(String prompt, String fileName, String agentName) throws IOException {
+    public String writeChapters(String prompt, String fileName, String agentName) throws Throwable {
         String chapters2 = generateAnswer(prompt);
 
         // 根据技术文档知识库生成二级标题下三级标题
@@ -1079,7 +1115,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         List<String> chapter2List = new ArrayList<>();
         String[] contentLines = chapters2.split("\n");
         for (String line : contentLines) {
-            if (line.contains("6."))
+            if (line.startsWith("6.") || line.startsWith("1."))
                 chapter2List.add(line.replace("*", "").replace("#", "").replace("-", ""));
         }
         processValue = "已生成二级标题:100%";
@@ -1103,90 +1139,61 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         }
         String sb = "二级标题如下：\n" + chapters2 +
                 "\n 三级标题如下：" + chapter3 +
-                "请将上述二级与对应的三级标题进行合并，严格按以下格式输出完整大纲，仅输出标题：\n" +
-                "6 技术文件\n" +
-                "6.1 XX\n" +
-                "6.1.1 XX\n" +
-                "6.1.2 XX\n" +
-                "6.2 XX\n" +
-                "6.2.1 XX\n" +
-                "6.2.2 XX\n" +
-                "6.3 XX\n" +
-                "6.3.1 XX\n" +
-                "6.3.2 XX\n" +
-                "...... \n" +
-                "6.n XX\n" +
-                "6.n.1 XX\n" +
-                "6.n 2 XX\n";
+                "请将上述二级与对应的三级标题进行合并，严格按以下格式输出完整大纲，仅输出标题：\n";
+        if (agentName.contains("技术标书"))
+            sb += "6 技术文件\n" +
+                    "6.1 XX\n" +
+                    "6.1.1 XX\n" +
+                    "6.1.2 XX\n" +
+                    "6.2 XX\n" +
+                    "6.2.1 XX\n" +
+                    "6.2.2 XX\n" +
+                    "6.3 XX\n" +
+                    "6.3.1 XX\n" +
+                    "6.3.2 XX\n" +
+                    "...... \n" +
+                    "6.n XX\n" +
+                    "6.n.1 XX\n" +
+                    "6.n 2 XX\n";
+        if (agentName.contains("技术设计书"))
+            sb += "1 技术设计书\n" +
+                    "1.1 XX\n" +
+                    "1.1.1 XX\n" +
+                    "1.1.2 XX\n" +
+                    "1.2 XX\n" +
+                    "1.2.1 XX\n" +
+                    "1.2.2 XX\n" +
+                    "1.3 XX\n" +
+                    "1.3.1 XX\n" +
+                    "1.3.2 XX\n" +
+                    "...... \n" +
+                    "1.n XX\n" +
+                    "1.n.1 XX\n" +
+                    "1.n 2 XX\n";
         String content = generateAnswer(sb);
         writeTitles(content, fileName, agentName);
         return content;
     }
 
-    // /**
-    // * 调用LLM生成回答
-    // *
-    // * @return
-    // */
-    // public String generateAnswer(String prompt) throws IOException {
-    // ChatModel chatModel =
-    // ChatModel.of("https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions")
-    // .model("qwen3-vl-32b-instruct")
-    // .apiKey("sk-750a17cc723847f28b31fa1bc17c255c")
-    // .timeout(Duration.ofSeconds(240))
-    // .build();
-
-    // List<ChatMessage> messages = new ArrayList<>();
-    // messages.add(ChatMessage.ofUser(prompt));
-    // ChatSession chatSession =
-    // InMemoryChatSession.builder().messages(messages).build();
-
-    // Prompt prompt1 = Prompt.of(prompt).attrPut("session", chatSession);
-    // ChatResponse response = chatModel.prompt(prompt1).call();
-    // String content = response.lastChoice().getMessage().getContent();
-    // return content;
-    // }
-
     /**
      * 调用LLM生成回答
      *
      * @return
      */
-    public String generateAnswer(String prompt) throws IOException {
+    public String generateAnswer(String prompt) throws Throwable, IOException {
         ChatModel chatModel = ChatModel.of(llmServiceUrl)
                 .model("Qwen")
                 .timeout(Duration.ofSeconds(240))
                 .build();
-
-        List<ChatMessage> messages = new ArrayList<>();
-        messages.add(ChatMessage.ofUser(prompt));
-        ChatSession chatSession = InMemoryChatSession.builder().messages(messages).build();
-
-        Prompt prompt1 = Prompt.of(prompt).attrPut("session", chatSession);
-        ChatResponse response = chatModel.prompt(prompt1).call();
-        String content = response.lastChoice().getMessage().getContent();
-        return content;
-    }
-
-    /**
-     * 流式调用LLM生成回答 - 使用Flux
-     * 
-     * @param prompt 提示词
-     * @return Flux<String> 流式输出（原始内容）
-     */
-    public Flux<String> generateAnswerFlux(String prompt) {
-        ChatModel chatModel = ChatModel.of(llmServiceUrl)
-                .model("Qwen")
-                .timeout(Duration.ofSeconds(240))
+        SimpleAgent robot = SimpleAgent.of(chatModel)
                 .build();
         List<ChatMessage> messages = new ArrayList<>();
         messages.add(ChatMessage.ofUser(prompt));
         ChatSession chatSession = InMemoryChatSession.builder().messages(messages).build();
 
         Prompt prompt1 = Prompt.of(prompt).attrPut("session", chatSession);
-        return chatModel.prompt(prompt1).stream()
-                .map(resp -> resp.getContent())
-                .filter(content -> content != null && !content.isEmpty());
+        String content = robot.prompt(prompt1).call().getContent();
+        return content;
     }
 
     /**
@@ -1202,6 +1209,8 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
                 .model("Qwen")
                 .timeout(Duration.ofSeconds(240))
                 .build();
+        SimpleAgent robot = SimpleAgent.of(chatModel)
+                .build();
         List<ChatMessage> messages = new ArrayList<>();
         messages.add(ChatMessage.ofUser(prompt));
         ChatSession chatSession = InMemoryChatSession.builder().messages(messages).build();
@@ -1213,34 +1222,34 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
             tokenUsage[0].set((long) prompt.length());
         }
 
-        return chatModel.prompt(prompt1).stream()
+        return robot.prompt(prompt1).stream()
                 .doOnNext(resp -> {
                     if (tokenUsage != null && tokenUsage.length >= 3) {
                         try {
                             // 优先使用vLLM返回的usage信息（在最后一个chunk中）
-                            AiUsage usage = resp.getUsage();
-                            if (usage != null) {
-                                Long pt = usage.promptTokens();
-                                Long ct = usage.completionTokens();
-                                Long tt = usage.totalTokens();
-
-                                if (pt > 0)
-                                    tokenUsage[0].set(pt);
-                                if (ct > 0)
-                                    tokenUsage[1].set(ct);
-                                if (tt > 0)
-                                    tokenUsage[2].set(tt);
-                            } else {
-                                // 如果没有usage信息（中间chunk），估算completion tokens
-                                String content = resp.getContent();
-                                if (content != null && !content.isEmpty()) {
-                                    // 中文大约1个字符=1个token，实时累计
-                                    long currentCompletion = tokenUsage[1].get();
-                                    long increment = content.length();
-                                    tokenUsage[1].set(currentCompletion + increment);
-                                    tokenUsage[2].set(tokenUsage[0].get() + tokenUsage[1].get());
-                                }
+                            // AiUsage usage = resp.getUsage();
+                            // if (usage != null) {
+                            // Long pt = usage.promptTokens();
+                            // Long ct = usage.completionTokens();
+                            // Long tt = usage.totalTokens();
+
+                            // if (pt > 0)
+                            // tokenUsage[0].set(pt);
+                            // if (ct > 0)
+                            // tokenUsage[1].set(ct);
+                            // if (tt > 0)
+                            // tokenUsage[2].set(tt);
+                            // } else {
+                            // 如果没有usage信息（中间chunk），估算completion tokens
+                            String content = resp.getContent();
+                            if (content != null && !content.isEmpty()) {
+                                // 中文大约1个字符=1个token，实时累计
+                                long currentCompletion = tokenUsage[1].get();
+                                long increment = content.length();
+                                tokenUsage[1].set(currentCompletion + increment);
+                                tokenUsage[2].set(tokenUsage[0].get() + tokenUsage[1].get());
                             }
+                            // }
                         } catch (Exception e) {
                             // 如果获取usage失败，也尝试估算
                             try {
@@ -1333,7 +1342,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
         List<String> chapters = new ArrayList<>();
         String[] contentLines = content.split("\n");
         for (String line : contentLines) {
-            if (line.contains("6."))
+            if (line.startsWith("6.") || line.startsWith("1."))
                 chapters.add(line.replace("*", "").replace("#", "").replace("-", ""));
         }
         String prefixPath = "/upload/agent/" + agentName;
@@ -1628,7 +1637,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
 
             for (String line : lines) {
                 line = line.trim().replace("*", "").replace("#", "").replace("-", "");
-                if (!line.contains("6.") || line.isEmpty()) {
+                if (!line.startsWith("6.") && !line.startsWith("1.") || line.isEmpty()) {
                     continue;
                 }
 
@@ -1846,7 +1855,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      * 分割文档
      */
     private List<TextSegment> splitDocument(File transferFile, int maxSegmentSizeInChars, int maxOverlapSizeInChars)
-            throws IOException {
+            throws Throwable {
         String filename = transferFile.getName().toLowerCase();
         try (InputStream fileInputStream = new FileInputStream(transferFile)) {
             if (filename.endsWith(".docx")) {
@@ -2026,7 +2035,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
                     // 当前行按普通段落处理
                     blocks.add(new Block("para", line));
                 } else if (inTable) {
-                    tableLines.add(line);                
+                    tableLines.add(line);
                 } else {
                     // 匹配格式：1. **文本**（数字前缀+加粗）
                     Pattern numberedBoldPattern = Pattern.compile("^(\\d+\\.\\s+)\\*\\*([^*]+)\\*\\*(.*)$");
@@ -2282,7 +2291,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
      */
     @Override
     public SseEmitter streamGenerateChapters(String topicId, MultipartFile file, String agentName, String selectedNode)
-            throws IOException {
+            throws Throwable {
         // 设置SSE超时时间为10分钟
         SseEmitter emitter = new SseEmitter(600000L);
 
@@ -2494,7 +2503,7 @@ public class CmcAgentServiceImpl implements ICmcAgentService {
 
                 emitter.complete();
 
-            } catch (Exception e) {
+            } catch (Throwable e) {
                 try {
                     emitter.send(SseEmitter.event()
                             .name("error")

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

@@ -221,12 +221,11 @@
             <el-tab-pane label="文档上传" name="upload">
               <div class="task-content">
                 <div class="upload-section">
-                  <h4>上传招标文件</h4>
                   <el-upload class="upload-area" :action="uploadAction" :multiple="false" :auto-upload="false"
                     :file-list="taskFileList" :on-change="handleTaskFileChange" :before-upload="beforeUpload"
                     :show-file-list="true" :limit="1">
                     <el-button type="primary" size="large" icon="el-icon-upload">
-                      选择文件
+                      {{ selectDocument }}
                     </el-button>
                     <div class="upload-tip">支持 DOC、DOCX 格式</div>
                   </el-upload>
@@ -270,7 +269,7 @@
                       </div>
                     </div>
                   </el-tab-pane>
-                  <el-tab-pane :label="scoringTabLabel" name="scoring">
+                  <el-tab-pane v-if="!agentInfo.agentName.includes('设计')" :label="scoringTabLabel" name="scoring">
                     <div class="analysis-content">
                       <div v-if="analysisResults.scoring" class="analysis-result">
                         <div v-html="analysisResults.scoring"></div>
@@ -286,14 +285,14 @@
             </el-tab-pane>
 
             <!-- 标书内容 -->
-            <el-tab-pane label="标书内容" name="bid">
+            <el-tab-pane :label="bidTabLabel" name="bid">
               <div class="task-content">
                 <div class="bid-content">
                   <el-form :model="bidForm" label-width="70px">
-                    <el-form-item label="标书标题">
+                    <el-form-item :label="bidTitleLabel">
                       <div class="bid-title-row">
                         <div v-if="currentFilename" class="bid-title-display">
-                          <i class="el-icon-file-word"></i>
+                          <i clabidTabLabels="el-icon-file-word"></i>
                           <span>{{ currentFilename }}</span>
                         </div>
                         <el-upload v-else class="upload-area" :action="uploadAction" :multiple="false"
@@ -303,7 +302,7 @@
                         </el-upload>
                         <el-button type="primary" @click="isStreaming ? stopGeneration() : generateBid()"
                           :disabled="!currentFilename && !isStreaming">
-                          {{ isStreaming ? '停止生成' : '生成标书' }}
+                          {{ isStreaming ? '停止生成' : generateButtonText }}
                         </el-button>
                         <el-button type="primary" icon="el-icon-download" @click="exportWord"
                           :disabled="isStreaming">
@@ -380,7 +379,7 @@
                         <div v-if="generationCompleted" class="empty-bid-content generation-completed"
                           style="color: #67c23a;">
                           <i class="el-icon-success" style="font-size: 64px; margin-bottom: 20px; opacity: 0.8;"></i>
-                          <p style="font-size: 18px; font-weight: 500;">生成标书完成</p>
+                          <p style="font-size: 18px; font-weight: 500;">生成完成</p>
                         </div>
                         <!-- 初始空状态（未选中章节且未开始生成） -->
                         <div v-if="!isStreaming && !generationCompleted && !bidResult" class="empty-bid-content">
@@ -528,6 +527,15 @@ export default {
   computed: {
     uploadAction() {
       return '/llm/agent/upload'
+    },
+    bidTabLabel() {
+      return this.agentInfo?.agentName.includes('设计') ? '设计书内容' : '标书内容'
+    },
+    bidTitleLabel() {
+      return this.agentInfo?.agentName.includes('设计') ? '设计书标题' : '标书标题'
+    },
+    generateButtonText() {
+      return this.agentInfo?.agentName.includes('设计') ? '生成设计书' : '生成标书'
     }
   },
   watch: {
@@ -563,7 +571,7 @@ export default {
 
         // 设置评分标准/应价人须知标签
         if (this.agentInfo?.agentName) {
-          if (this.agentInfo.agentName.includes('技术') && this.agentInfo.agentName.includes('询价')) {
+          if (this.agentInfo.agentName.includes('技术标书') && this.agentInfo.agentName.includes('询价')) {
             this.scoringTabLabel = '应价人须知'
           } else {
             this.scoringTabLabel = '评分标准'
@@ -574,10 +582,14 @@ export default {
         if (this.agentInfo?.agentName) {
           const res = await opening(this.agentInfo.agentName)
           this.openingMessage = res.data.resultContent;
-          if (this.agentInfo.agentName.includes('技术')) {
+          if (this.agentInfo.agentName.includes('技术标书')) {
             this.selectDocument = '选择招标文件'
             this.selectDocumentTip = '请上传您需要分析的招标文件（单个文件）：'
           }
+          else if (this.agentInfo.agentName.includes('技术设计书')) {
+            this.selectDocument = '选择项目任务书'
+            this.selectDocumentTip = '请上传您需要分析的项目任务书（单个文件）：'
+          }
           else if (this.agentInfo.agentName.includes('商务')) {
           }
           else if (this.agentInfo.agentName.includes('文档检查')) {
@@ -1050,7 +1062,7 @@ export default {
               isHtml: true // 标记这是HTML内容
             }
             this.chatMessages.push(uploadMessage);
-            if (this.agentInfo.agentName.includes('技术'))
+            if (this.agentInfo.agentName.includes('技术标书'))
               this.techUploadDisplay = true;
             // this.$message.success('文件上传成功');
             const topicId = response.data.topicId;
@@ -1377,7 +1389,7 @@ export default {
        */
     async generateBid() {
       if (this.bidFileList.length === 0) {
-        this.$message.warning('请先上传标书文件')
+        this.$message.warning('请先上传文件')
         return
       }
 
@@ -1391,7 +1403,7 @@ export default {
             file = await this.downloadFileAsBlob(bidFileUrl, bidTitle)
             this.bidFileList[0].raw = file
           } catch (error) {
-            this.$message.error('下载标书文件失败')
+            this.$message.error('下载文件失败')
             return
           }
         }
@@ -1427,8 +1439,8 @@ export default {
         this.fetchStreamGeneration(formData, bidTitle)
 
       } catch (error) {
-        console.error('生成标书失败:', error)
-        this.$message.error('生成标书失败')
+        console.error('生成失败:', error)
+        this.$message.error('生成失败')
         this.isStreaming = false
       }
     },
@@ -1443,7 +1455,7 @@ export default {
 
       const onError = (error) => {
         console.error('流式请求失败:', error)
-        this.$message.error('生成标书失败: ' + error.message)
+        this.$message.error('生成失败: ' + error.message)
         this.isStreaming = false
         this.abortController = null
         this.$nextTick(() => {
@@ -1490,7 +1502,7 @@ export default {
           this.bidFilePath = process.env.VUE_APP_BASE_API + '/profile/upload/agent/' + this.agentInfo.agentName + '/' + this.currentFilename
         }
 
-        this.$message.success('标书生成完成')
+        this.$message.success('生成完成')
         return
       }
 
@@ -1673,7 +1685,7 @@ export default {
             this.selectedChapterContent = '暂无内容'
           }
         } else {
-          this.selectedChapterContent = '请先上传标书文件'
+          this.selectedChapterContent = '请先上传文件'
         }
       } catch (error) {
         console.error('获取章节内容失败:', error)
@@ -1814,7 +1826,7 @@ export default {
         this.bidResult = `<h3 style="color: #409EFF; margin: 10px 0;">【${node.title}】</h3>
                           <div class="empty-bid-content" style="padding: 40px; text-align: center; color: #999;">
                            <p>${isLeafNode ? '暂无内容' : '该节点为目录节点，请选择子章节'}</p>
-                           <p style="font-size: 12px; margin-top: 8px;">${isLeafNode ? '点击"生成标书"按钮生成该章节内容' : '请展开并选择具体章节'}</p>
+                           <p style="font-size: 12px; margin-top: 8px;">${isLeafNode ? '点击"生成"按钮生成该章节内容' : '请展开并选择具体章节'}</p>
                          </div>`
       }
     },