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 getWeather(@RequestBody String question) { + @PostMapping("/getAiWeather") + @Operation(summary = "AI获取天气") + public R 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 aiWriteFile(@RequestBody String question) { + CheckUtils.checkEmpty(question, "问题"); + String result = toolsService.aiWriteFile(question); + return R.ok(result); + } }