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