From 301cd1a6cc2490a271a5037adf0bcdc7cfab2d56 Mon Sep 17 00:00:00 2001
From: huangge1199 <huangge1199@hotmail.com>
Date: Wed, 28 May 2025 11:18:28 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E7=94=A8=E5=B7=A5=E5=85=B7=EF=BC=9AAI?=
 =?UTF-8?q?=E5=86=99=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../huangge1199/aiagent/Service/ToolsService.java  |  2 ++
 .../aiagent/Service/impl/ToolsServiceImpl.java     |  9 +++++++++
 .../aiagent/controller/ToolController.java         | 14 +++++++++++---
 3 files changed, 22 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/huangge1199/aiagent/Service/ToolsService.java b/src/main/java/com/huangge1199/aiagent/Service/ToolsService.java
index 96510a3..d8c7bf4 100644
--- a/src/main/java/com/huangge1199/aiagent/Service/ToolsService.java
+++ b/src/main/java/com/huangge1199/aiagent/Service/ToolsService.java
@@ -12,4 +12,6 @@ public interface ToolsService {
     String writeFileTest(String context, String name);
 
     String readFileTest(String name);
+
+    String aiWriteFile(String question);
 }
diff --git a/src/main/java/com/huangge1199/aiagent/Service/impl/ToolsServiceImpl.java b/src/main/java/com/huangge1199/aiagent/Service/impl/ToolsServiceImpl.java
index 4c47673..a9116f3 100644
--- a/src/main/java/com/huangge1199/aiagent/Service/impl/ToolsServiceImpl.java
+++ b/src/main/java/com/huangge1199/aiagent/Service/impl/ToolsServiceImpl.java
@@ -41,4 +41,13 @@ public class ToolsServiceImpl implements ToolsService {
         FileTool fileTool = new FileTool();
         return fileTool.readFile(name);
     }
+
+    @Override
+    public String aiWriteFile(String question) {
+        return ChatClient.create(ollamaChatModel)
+                .prompt(question)
+                .advisors(new MyLoggerAdvisor())
+                .tools(new FileTool())
+                .call().content();
+    }
 }
diff --git a/src/main/java/com/huangge1199/aiagent/controller/ToolController.java b/src/main/java/com/huangge1199/aiagent/controller/ToolController.java
index 0b058a3..aee224f 100644
--- a/src/main/java/com/huangge1199/aiagent/controller/ToolController.java
+++ b/src/main/java/com/huangge1199/aiagent/controller/ToolController.java
@@ -30,9 +30,9 @@ public class ToolController {
     @Resource
     private ToolsService toolsService;
 
-    @PostMapping("/getWeather")
-    @Operation(summary = "获取天气")
-    public R<String> getWeather(@RequestBody String question) {
+    @PostMapping("/getAiWeather")
+    @Operation(summary = "AI获取天气")
+    public R<String> getAiWeather(@RequestBody String question) {
         String result = toolsService.getWeather(question);
         return R.ok(result);
     }
@@ -56,4 +56,12 @@ public class ToolController {
         String result = toolsService.readFileTest(name);
         return R.ok(result);
     }
+
+    @PostMapping("/aiWriteFile")
+    @Operation(summary = "AI写文件")
+    public R<String> aiWriteFile(@RequestBody String question) {
+        CheckUtils.checkEmpty(question, "问题");
+        String result = toolsService.aiWriteFile(question);
+        return R.ok(result);
+    }
 }