根据招标文件要求编写技术方案到新文件

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

@@ -12,9 +12,11 @@ import org.noear.solon.annotation.Param;
 import org.springframework.stereotype.Service;
 
 import java.io.FileInputStream;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.util.Arrays;
 import java.util.Collection;
+import java.util.Collections;
 
 @Service
 @McpServerEndpoint(sseEndpoint = "/llm/mcp/sse")
@@ -25,10 +27,9 @@ public class McpServiceImpl implements IMcpService {
         return "晴，14度";
     }
 
-    @ToolMapping(description = "查看word文档")
-    public String openDocument() throws IOException {
-        String filePath = "C:\\Users\\lamphua\\Desktop\\智慧水利\\李家岩水库数字孪生招标文件.DOCX";
-        try (XWPFDocument doc = new XWPFDocument(new FileInputStream(filePath))) {
+    @ToolMapping(description = "查看招标文档")
+    public String openDocument(@Param(description = "文件路径") String document) throws IOException {
+        try (XWPFDocument doc = new XWPFDocument(new FileInputStream(document))) {
             StringBuilder content = new StringBuilder();
             for (XWPFParagraph paragraph : doc.getParagraphs()) {
                 content.append(paragraph.getText()).append("\n");
@@ -37,6 +38,31 @@ public class McpServiceImpl implements IMcpService {
         }
     }
 
+    @ToolMapping(description = "补全技术方案")
+    public String writeTechnicalPlan(@Param(description = "文件路径") String document, @Param(description = "技术方案内容") String resultContent) throws IOException {
+        // 创建临时输出文件路径
+        String outputPath = document.replace(".docx", "_modified.docx");
+
+        try (FileInputStream fis = new FileInputStream(document);
+             XWPFDocument doc = new XWPFDocument(fis);
+             FileOutputStream fos = new FileOutputStream(outputPath)) {
+
+            // 在最后添加分页符和新内容
+            XWPFParagraph pageBreak = doc.createParagraph();
+            pageBreak.setPageBreak(true);
+
+            XWPFParagraph newContent = doc.createParagraph();
+            newContent.createRun().setText(resultContent);
+
+            // 保存文档
+            doc.write(fos);
+
+            return "技术方案添加完成，已保存到: " + outputPath;
+        } catch (IOException e) {
+            throw new IOException("处理Word文档时出错: " + e.getMessage(), e);
+        }
+    }
+
     @ResourceMapping(uri = "config://app-version", description = "获取应用版本号")
     public String getAppVersion() {
         return "v3.2.0";
@@ -47,10 +73,10 @@ public class McpServiceImpl implements IMcpService {
         return user_id + "@example.com";
     }
 
-    @PromptMapping(description = "生成关于某个主题的提问")
-    public Collection<ChatMessage> askQuestion(@Param(description = "主题") String topic) {
-        return Arrays.asList(
-                ChatMessage.ofUser("请解释一下'" + topic + "'的概念？")
+    @PromptMapping(description = "提出技术问题")
+    public Collection<ChatMessage> askQuestion(@Param(description = "描述招标文件服务要求") String request) {
+        return Collections.singletonList(
+                ChatMessage.ofUser("请根据以下要求：'" + request + "\n编写技术方案")
         );
     }
 }

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

@@ -5,6 +5,8 @@ import com.ruoyi.web.llm.service.ILangChainMilvusService;
 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.tool.FunctionToolDesc;
 import org.noear.solon.ai.mcp.client.McpClientProvider;
 import org.reactivestreams.Publisher;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -13,6 +15,11 @@ import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 import reactor.core.publisher.Flux;
 
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
 
 /**
  * mcp模型上下文协议Controller
@@ -32,8 +39,7 @@ public class McpController extends BaseController
      * @return
      */
     @GetMapping("/answer")
-    public Flux<AssistantMessage> answer(String question)
-    {
+    public Flux<AssistantMessage> answer(String question) throws IOException {
         McpClientProvider clientProvider = McpClientProvider.builder()
                     .apiUrl("http://localhost:8080/llm/mcp/sse")
                     .build();
@@ -42,7 +48,7 @@ public class McpController extends BaseController
                 .provider("openai")
                 .model("DeepSeek-R1-Distill-Qwen-1.5B")
                 .apiKey("1")
-//                .defaultToolsAdd(clientProvider)
+                .defaultToolsAdd(clientProvider)
                 .build();
         Publisher<ChatResponse> publisher = chatModel.prompt(question).stream();
 
@@ -52,4 +58,35 @@ public class McpController extends BaseController
                 });
     }
 
+    /**
+     * 根据招标文件要求编写投标文件技术方案
+     * @return
+     */
+    @GetMapping("/tech")
+    public AssistantMessage tech(String file) 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("DeepSeek-R1-Distill-Qwen-1.5B")
+                .apiKey("1")
+                .defaultToolsAdd(clientProvider)
+                .build();
+
+        Map<String,Object> path = new HashMap<>();
+        path.put("document", file);
+        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", file);
+        write.put("resultContent", response.getChoices().get(0).getMessage().getResultContent());
+        clientProvider.callToolAsText("writeTechnicalPlan", write);
+        return response.getChoices().get(0).getMessage();
+    }
+
 }