From f9330b7d4604dcb38fd1213f8e86465f68f18e49 Mon Sep 17 00:00:00 2001 From: louis-xiv Date: Thu, 24 Apr 2025 16:46:19 +0800 Subject: [PATCH] init --- .../mcp/dbconnect/config/AppConfiguration.java | 7 +++++-- .../mcp/dbconnect/tool/ToUpperCaseTool.java | 14 ++++++++++++++ src/main/resources/application.yml | 2 +- 3 files changed, 20 insertions(+), 3 deletions(-) create mode 100644 src/main/java/cn/louisxiv/mcp/dbconnect/tool/ToUpperCaseTool.java diff --git a/src/main/java/cn/louisxiv/mcp/dbconnect/config/AppConfiguration.java b/src/main/java/cn/louisxiv/mcp/dbconnect/config/AppConfiguration.java index 6e422d3..6d44076 100644 --- a/src/main/java/cn/louisxiv/mcp/dbconnect/config/AppConfiguration.java +++ b/src/main/java/cn/louisxiv/mcp/dbconnect/config/AppConfiguration.java @@ -1,6 +1,7 @@ package cn.louisxiv.mcp.dbconnect.config; import cn.louisxiv.mcp.dbconnect.tool.DbConnectTool; +import cn.louisxiv.mcp.dbconnect.tool.ToUpperCaseTool; import org.springframework.ai.tool.ToolCallbackProvider; import org.springframework.ai.tool.method.MethodToolCallbackProvider; import org.springframework.context.annotation.Bean; @@ -13,7 +14,9 @@ public class AppConfiguration { * 声明ToolCallbackProvider */ @Bean - public ToolCallbackProvider toolCallbackProvider(DbConnectTool dbConnectTool) { - return MethodToolCallbackProvider.builder().toolObjects(dbConnectTool).build(); + public ToolCallbackProvider toolCallbackProvider(DbConnectTool dbConnectTool + , ToUpperCaseTool toUpperCaseTool) { + return MethodToolCallbackProvider.builder() + .toolObjects(dbConnectTool, toUpperCaseTool).build(); } } diff --git a/src/main/java/cn/louisxiv/mcp/dbconnect/tool/ToUpperCaseTool.java b/src/main/java/cn/louisxiv/mcp/dbconnect/tool/ToUpperCaseTool.java new file mode 100644 index 0000000..62e382c --- /dev/null +++ b/src/main/java/cn/louisxiv/mcp/dbconnect/tool/ToUpperCaseTool.java @@ -0,0 +1,14 @@ +package cn.louisxiv.mcp.dbconnect.tool; + +import org.springframework.ai.tool.annotation.Tool; +import org.springframework.stereotype.Component; + +@Component +public class ToUpperCaseTool { + + @Tool(description = "把输入字符串转为大写") + public String toUpperCase(String input) { + return input.toUpperCase(); + } +} + diff --git a/src/main/resources/application.yml b/src/main/resources/application.yml index a77b689..683e0d4 100644 --- a/src/main/resources/application.yml +++ b/src/main/resources/application.yml @@ -6,7 +6,7 @@ spring: name: mcp-db-connect version: 0.0.1 type: ASYNC - sse-message-endpoint: /mcp/sse + sse-message-endpoint: /mcp/sse/message # 退化为STDIO模式需要关闭banner main: banner-mode: off