Improved code by updating any interfaces
Change-Id: I9a2e630c37f3c427569f5bbd773978c2705dc08c
diff --git a/mcp/server.go b/mcp/server.go
index b6d7b10..b3ea43b 100644
--- a/mcp/server.go
+++ b/mcp/server.go
@@ -31,7 +31,7 @@
}
// AddTool registers a new tool with the server
-func (s *Server) AddTool(name, description string, inputSchema map[string]interface{}, handler func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error)) error {
+func (s *Server) AddTool(name, description string, inputSchema map[string]any, handler func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error)) error {
if name == "" {
return fmt.Errorf("tool name cannot be empty")
}
diff --git a/mcp/server_test.go b/mcp/server_test.go
index a0073c3..c00d315 100644
--- a/mcp/server_test.go
+++ b/mcp/server_test.go
@@ -21,10 +21,10 @@
server := NewServer("test-server", "1.0.0")
// Test adding a valid tool
- err := server.AddTool("test-tool", "A test tool", map[string]interface{}{
+ err := server.AddTool("test-tool", "A test tool", map[string]any{
"type": "object",
- "properties": map[string]interface{}{
- "input": map[string]interface{}{
+ "properties": map[string]any{
+ "input": map[string]any{
"type": "string",
},
},
@@ -35,12 +35,12 @@
assert.NoError(t, err)
// Test adding tool with empty name
- err = server.AddTool("", "description", map[string]interface{}{}, nil)
+ err = server.AddTool("", "description", map[string]any{}, nil)
assert.Error(t, err)
assert.Contains(t, err.Error(), "tool name cannot be empty")
// Test adding tool with empty description
- err = server.AddTool("name", "", map[string]interface{}{}, nil)
+ err = server.AddTool("name", "", map[string]any{}, nil)
assert.Error(t, err)
assert.Contains(t, err.Error(), "tool description cannot be empty")
}