技术标智能体开场白

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

@@ -21,11 +21,6 @@ import java.util.Collections;
 @McpServerEndpoint(sseEndpoint = "/llm/mcp/sse")
 public class McpServiceImpl implements IMcpService {
 
-    @ToolMapping(description = "查询天气预报")
-    public String getWeather(@Param(description = "城市位置") String location) {
-        return "晴，14度";
-    }
-
     @ToolMapping(description = "查看招标文档")
     public String openDocument(@Param(description = "文件路径") String document) throws IOException {
         try (XWPFDocument doc = new XWPFDocument(new FileInputStream(document))) {
@@ -37,25 +32,21 @@ public class McpServiceImpl implements IMcpService {
         }
     }
 
-    @ToolMapping(description = "编制技术方案")
-    public String openDocument(@Param(description = "招标文件") String inviteDocument, @Param(description = "投标文件") String submitDocument) throws IOException {
-        try (XWPFDocument doc = new XWPFDocument(new FileInputStream(inviteDocument))) {
-            StringBuilder content = new StringBuilder();
-            for (XWPFParagraph paragraph : doc.getParagraphs()) {
-                content.append(paragraph.getText()).append("\n");
-            }
-            return content.toString();
-        }
+    @ToolMapping(description = "开场白")
+    public String opening(@Param(description = "智能体名称") String agentName) {
+        String content = "";
+        if (agentName.contains("技术"))
+            content = "我是投标文件写作助手，我将助您完成技术文件部分撰写。请上传招标询价文件，分析后将依招标服务要求，运用技术方案知识库，为您提供参考。";
+        return content;
     }
 
     @ToolMapping(description = "补全技术方案")
     public String writeTechnicalPlan(@Param(description = "文件路径") String document, @Param(description = "技术方案内容") String resultContent) throws IOException {
         // 创建临时输出文件路径
-        String outputPath = document.replace(".docx", "_修改" + System.currentTimeMillis() + ".docx");
 
         try (FileInputStream fis = new FileInputStream(document);
              XWPFDocument doc = new XWPFDocument(fis);
-             FileOutputStream fos = new FileOutputStream(outputPath)) {
+             FileOutputStream fos = new FileOutputStream(document)) {
 
             // 在最后添加分页符和新内容
             XWPFParagraph pageBreak = doc.createParagraph();
@@ -67,26 +58,20 @@ public class McpServiceImpl implements IMcpService {
             // 保存文档
             doc.write(fos);
 
-            return "技术方案添加完成，已保存到: " + outputPath;
+            return "技术方案添加完成，已保存到: " + document;
         } catch (IOException e) {
             throw new IOException("处理Word文档时出错: " + e.getMessage(), e);
         }
     }
 
-    @ResourceMapping(uri = "config://app-version", description = "获取应用版本号")
-    public String getAppVersion() {
-        return "v3.2.0";
-    }
-
-    @ResourceMapping(uri = "db://users/{user_id}/email", description = "根据用户ID查询邮箱")
-    public String getEmail(@Param(description = "用户Id") String user_id) {
-        return user_id + "@example.com";
-    }
-
     @PromptMapping(description = "提出技术问题")
-    public Collection<ChatMessage> askQuestion(@Param(description = "描述招标文件服务要求") String request) {
+    public Collection<ChatMessage> askQuestion(@Param(description = "描述招标文件服务要求") String request,
+                                               @Param(description = "参考上下文") String contexts,
+                                               @Param(description = "补充文字部分") String part) {
         return Collections.singletonList(
-                ChatMessage.ofUser("请根据以下要求：'" + request + "\n编写技术方案")
+                ChatMessage.ofUser("请根据以下要求：'" + request + "\n" +
+                        "参考上下文：'" + contexts + "\n" +
+                        "回答以下问题：'" + part + "\n")
         );
     }
 }

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

@@ -1,6 +1,7 @@
 package com.ruoyi.web.llm.controller;
 
 import java.io.IOException;
+import java.util.Date;
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 
@@ -95,6 +96,8 @@ public class CmcAgentController extends BaseController
     @PostMapping
     public AjaxResult add(@RequestBody CmcAgent cmcAgent)
     {
+        cmcAgent.setCreateTime(new Date());
+        cmcAgent.setCreateBy(getLoginUser().getUserId().toString());
         return toAjax(cmcAgentService.insertCmcAgent(cmcAgent));
     }
 

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

@@ -1,15 +1,9 @@
 package com.ruoyi.web.llm.controller;
 
-import java.io.File;
 import java.io.IOException;
-import java.util.ArrayList;
 import java.util.List;
 import javax.servlet.http.HttpServletResponse;
 
-import com.alibaba.fastjson2.JSONObject;
-import com.ruoyi.common.config.RuoYiConfig;
-import com.ruoyi.common.utils.SnowFlake;
-import com.ruoyi.llm.domain.CmcChat;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;

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

@@ -11,6 +11,7 @@ import org.noear.solon.ai.chat.ChatModel;
 import org.noear.solon.ai.chat.ChatResponse;
 import org.noear.solon.ai.chat.message.AssistantMessage;
 import org.noear.solon.ai.chat.message.ChatMessage;
+import org.noear.solon.ai.chat.message.SystemMessage;
 import org.noear.solon.ai.chat.tool.FunctionTool;
 import org.noear.solon.ai.mcp.client.McpClientProvider;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -34,21 +35,20 @@ import java.util.*;
 public class McpController extends BaseController
 {
     /**
-     * 同步问答
+     * 获取开场白
      * @return
      */
-    @GetMapping("/answer")
-    public AssistantMessage answer(String question) throws IOException {
+    @GetMapping("/opening")
+    public AssistantMessage opening(String agentName) throws IOException {
         McpClientProvider clientProvider = McpClientProvider.builder()
                 .apiUrl("http://localhost:8080/llm/mcp/sse")
                 .build();
-
         ChatModel chatModel = ChatModel.of("http://192.168.28.188:8000/v1/chat/completions")
                 .provider("openai")
                 .model("Qwen2.5-1.5B-Instruct")
                 .defaultToolsAdd(clientProvider)
                 .build();
-        ChatResponse response = chatModel.prompt(question).call();
+        ChatResponse response = chatModel.prompt(agentName + "智能体开场白").call();
         String resultContent = response.lastChoice().getMessage().getResultContent();
         AssistantMessage assistantMessage = new AssistantMessage(resultContent);
         if (resultContent.startsWith("<tool_call>")) {
@@ -107,24 +107,16 @@ public class McpController extends BaseController
                 .model("DeepSeek-R1-Distill-Qwen-1.5B")
                 .defaultToolsAdd(clientProvider)
                 .build();
-        File profilePath = new File( RuoYiConfig.getProfile() + "/upload/knowledge");
-        if (!profilePath.exists())
-            profilePath.mkdirs();
-        File transferFile = new File( profilePath + File.separator + file.getOriginalFilename());
-        if (!transferFile.exists())
-            file.transferTo(transferFile);
         Map<String,Object> path = new HashMap<>();
-        path.put("document", transferFile.getPath());
         String content = clientProvider.callToolAsText("openDocument", path).getContent();
         Map<String,Object> request = new HashMap<>();
         request.put("request", content);
         List<ChatMessage> messages = clientProvider.getPromptAsMessages("askQuestion", request);
         ChatResponse response = chatModel.prompt(messages).call();
         System.out.println(response.getChoices().get(0).getMessage());
-//        Map<String,Object> write = new HashMap<>();
-//        write.put("document", transferFile.getPath());
-//        write.put("resultContent", response.getChoices().get(0).getMessage().getResultContent());
-//        String res = clientProvider.callToolAsText("writeTechnicalPlan", write).getContent();
+        Map<String,Object> write = new HashMap<>();
+        write.put("resultContent", response.getChoices().get(0).getMessage().getResultContent());
+        String res = clientProvider.callToolAsText("writeTechnicalPlan", write).getContent();
         return response.getChoices().get(0).getMessage();
     }
 

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

@@ -176,11 +176,11 @@ public class LangChainMilvusServiceImpl implements ILangChainMilvusService
         StringBuilder sb = new StringBuilder();
         sb.append("问题: ").append(question).append("\n\n");
         sb.append("根据以下上下文回答问题：\n\n");
-        for (int i = 0; i < contexts.size(); i++) {
+        for (JSONObject context : contexts) {
             sb.append("文件").append(": ")
-                    .append(contexts.get(i).getString("file_name")).append("\n\n")
+                    .append(context.getString("file_name")).append("\n\n")
                     .append("上下文").append(": ")
-                    .append(contexts.get(i).getString("content")).append("\n\n");
+                    .append(context.getString("content")).append("\n\n");
         }
         return generateAnswer(topicId, sb.toString(), llmServiceUrl);
     }

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

@@ -3,7 +3,6 @@ package com.ruoyi.llm.service;
 import java.io.IOException;
 import java.util.List;
 
-import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.llm.domain.CmcAgent;
 import org.springframework.web.multipart.MultipartFile;
 

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

@@ -1,11 +1,14 @@
 package com.ruoyi.llm.service.impl;
 
 import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.util.List;
 
 import com.ruoyi.common.config.RuoYiConfig;
 import com.ruoyi.common.utils.DateUtils;
+import org.apache.poi.xwpf.usermodel.XWPFDocument;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import com.ruoyi.llm.mapper.CmcAgentMapper;
@@ -64,7 +67,18 @@ public class CmcAgentServiceImpl implements ICmcAgentService
         if (!transferFile.exists())
             file.transferTo(transferFile);
         String[] filenameSplit = file.getOriginalFilename().split("\\.");
-        return "/upload/agent/" + agentName + File.separator + 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 (agentName.contains("技术")) {
+            XWPFDocument doc = new XWPFDocument(new FileInputStream(RuoYiConfig.getProfile() + "/upload/agent/technical/template/technical.docx"));
+            // 保存文档到本地文件系统
+            FileOutputStream out = new FileOutputStream(RuoYiConfig.getProfile() + outputFilename);
+            doc.write(out);
+            out.close();
+
+            // 关闭文档
+            doc.close();
+        }
+        return outputFilename;
     }
 
     /**