feat: add MCP client capabilities and server request handling

Author: voocel

README.md

@@ -87,7 +87,7 @@ func WithSSETransport(url string) Option {
 func WithStreamableHTTPTransport(url string) Option {
 	return func(c *MCPClient) error {
 		t := streamable.New(url,
-			streamable.WithProtocolVersion("2025-03-26"))
+			streamable.WithProtocolVersion(protocol.MCPVersion))
 		c.transport = t
 		return nil
 	}
@@ -181,6 +181,11 @@ func (c *MCPClient) receiveLoop(ctx context.Context) {
 				}
 			}
 
+			// 处理服务器发起的请求
+			if message.Method != "" && !message.IsNotification() {
+				c.handleServerRequest(ctx, &message)
+			}
+
 			// 处理通知消息
 			if message.Method != "" && message.IsNotification() {
 				c.handleNotification(&message)
@@ -189,6 +194,48 @@ func (c *MCPClient) receiveLoop(ctx context.Context) {
 	}
 }
 
+// handleServerRequest 处理服务器发起的请求
+func (c *MCPClient) handleServerRequest(ctx context.Context, message *protocol.JSONRPCMessage) {
+	var response *protocol.JSONRPCMessage
+	var err error
+
+	switch message.Method {
+	case "sampling/createMessage":
+		// 处理 LLM 采样请求
+		response, err = c.handleSamplingRequest(ctx, message)
+	case "roots/list":
+		// 处理根目录列表请求
+		response, err = c.handleRootsListRequest(ctx, message)
+	default:
+		// 未知方法，返回方法未找到错误
+		response = &protocol.JSONRPCMessage{
+			JSONRPC: protocol.JSONRPCVersion,
+			ID:      message.ID,
+			Error: &protocol.JSONRPCError{
+				Code:    protocol.MethodNotFound,
+				Message: fmt.Sprintf("method not found: %s", message.Method),
+			},
+		}
+	}
+
+	if err != nil {
+		response = &protocol.JSONRPCMessage{
+			JSONRPC: protocol.JSONRPCVersion,
+			ID:      message.ID,
+			Error: &protocol.JSONRPCError{
+				Code:    protocol.InternalError,
+				Message: err.Error(),
+			},
+		}
+	}
+
+	// 发送响应
+	if response != nil {
+		responseBytes, _ := json.Marshal(response)
+		c.transport.Send(ctx, responseBytes)
+	}
+}
+
 // handleNotification 处理服务端通知
 func (c *MCPClient) handleNotification(message *protocol.JSONRPCMessage) {
 	switch message.Method {
@@ -211,6 +258,40 @@ func (c *MCPClient) handleNotification(message *protocol.JSONRPCMessage) {
 	}
 }
 
+// handleRootsListRequest 处理根目录列表请求
+func (c *MCPClient) handleRootsListRequest(ctx context.Context, message *protocol.JSONRPCMessage) (*protocol.JSONRPCMessage, error) {
+	// 默认返回空的根目录列表
+	// 实际应用中，客户端应该返回其暴露的文件系统根目录
+	result := map[string]interface{}{
+		"roots": []interface{}{},
+	}
+
+	resultBytes, err := json.Marshal(result)
+	if err != nil {
+		return nil, err
+	}
+
+	return &protocol.JSONRPCMessage{
+		JSONRPC: protocol.JSONRPCVersion,
+		ID:      message.ID,
+		Result:  json.RawMessage(resultBytes),
+	}, nil
+}
+
+// handleSamplingRequest 处理 LLM 采样请求
+func (c *MCPClient) handleSamplingRequest(ctx context.Context, message *protocol.JSONRPCMessage) (*protocol.JSONRPCMessage, error) {
+	// 默认返回不支持采样的错误
+	// 实际应用中，客户端应该将请求转发给 LLM 并返回结果
+	return &protocol.JSONRPCMessage{
+		JSONRPC: protocol.JSONRPCVersion,
+		ID:      message.ID,
+		Error: &protocol.JSONRPCError{
+			Code:    protocol.MethodNotFound,
+			Message: "sampling not implemented",
+		},
+	}, nil
+}
+
 // sendRequest 发送请求并等待响应
 func (c *MCPClient) sendRequest(ctx context.Context, method string, params interface{}) (*protocol.JSONRPCMessage, error) {
 	id := uuid.New().String()
@@ -277,7 +358,11 @@ func (c *MCPClient) Initialize(ctx context.Context, clientInfo protocol.ClientIn
 	initRequest := protocol.InitializeRequest{
 		ProtocolVersion: c.protocolVersion,
 		Capabilities: protocol.ClientCapabilities{
-			// 客户端能力可以根据需要扩展
+			// 根据 MCP 规范声明客户端能力
+			Roots: &protocol.RootsCapability{
+				ListChanged: true, // 支持根目录变更通知
+			},
+			Sampling:     &protocol.SamplingCapability{}, // 支持 LLM 采样
 			Experimental: make(map[string]interface{}),
 		},
 		ClientInfo: c.clientInfo,

client/client.go

@@ -25,7 +25,7 @@ func main() {
 	defer mcpClient.Close()
 
 	// 初始化握手
-	fmt.Println("🔄 连接到计算器服务...")
+	fmt.Println("连接到计算器服务...")
 	initResult, err := mcpClient.Initialize(ctx, protocol.ClientInfo{
 		Name:    "计算器客户端",
 		Version: "1.0.0",
@@ -34,7 +34,7 @@ func main() {
 		log.Fatalf("初始化失败: %v", err)
 	}
 
-	fmt.Printf("✅ 连接成功！服务器: %s v%s\n",
+	fmt.Printf("连接成功！服务器: %s v%s\n",
 		initResult.ServerInfo.Name, initResult.ServerInfo.Version)
 
 	// 发送初始化完成通知
@@ -48,15 +48,15 @@ func main() {
 		log.Fatalf("获取工具列表失败: %v", err)
 	}
 
-	fmt.Println("\n🔧 可用工具:")
+	fmt.Println("\n可用工具:")
 	for _, tool := range toolsResult.Tools {
 		fmt.Printf("  - %s: %s\n", tool.Name, tool.Description)
 	}
 	fmt.Println()
 
 	// 测试加法
-	fmt.Println("🧮 测试计算功能:")
-	result, err := mcpClient.CallTool(ctx, "add", map[string]interface{}{
+	fmt.Println("测试计算功能:")
+	result, err := mcpClient.CallTool(ctx, "add", map[string]any{
 		"a": 5.0,
 		"b": 3.0,
 	})
@@ -71,7 +71,7 @@ func main() {
 	}
 
 	// 测试减法
-	result, err = mcpClient.CallTool(ctx, "subtract", map[string]interface{}{
+	result, err = mcpClient.CallTool(ctx, "subtract", map[string]any{
 		"a": 10.0,
 		"b": 4.0,
 	})
@@ -86,7 +86,7 @@ func main() {
 	}
 
 	// 测试乘法
-	result, err = mcpClient.CallTool(ctx, "multiply", map[string]interface{}{
+	result, err = mcpClient.CallTool(ctx, "multiply", map[string]any{
 		"a": 6.0,
 		"b": 7.0,
 	})
@@ -101,7 +101,7 @@ func main() {
 	}
 
 	// 测试除法
-	result, err = mcpClient.CallTool(ctx, "divide", map[string]interface{}{
+	result, err = mcpClient.CallTool(ctx, "divide", map[string]any{
 		"a": 20.0,
 		"b": 5.0,
 	})
@@ -116,7 +116,7 @@ func main() {
 	}
 
 	// 测试除零错误
-	result, err = mcpClient.CallTool(ctx, "divide", map[string]interface{}{
+	result, err = mcpClient.CallTool(ctx, "divide", map[string]any{
 		"a": 20.0,
 		"b": 0.0,
 	})
@@ -129,7 +129,7 @@ func main() {
 	}
 
 	// 获取帮助提示模板
-	fmt.Println("\n💡 获取帮助信息:")
+	fmt.Println("\n获取帮助信息:")
 	promptResult, err := mcpClient.GetPrompt(ctx, "calculator_help", nil)
 	if err != nil {
 		log.Fatalf("获取提示模板失败: %v", err)
@@ -143,5 +143,5 @@ func main() {
 		}
 	}
 
-	fmt.Println("\n✨ 计算器演示完成！")
+	fmt.Println("\n end!")
 }

examples/calculator/client/main.go

@@ -30,7 +30,7 @@ func main() {
 	mcp.Tool("add", "两个数字相加").
 		WithNumberParam("a", "第一个数字", true).
 		WithNumberParam("b", "第二个数字", true).
-		Handle(func(ctx context.Context, args map[string]interface{}) (*protocol.CallToolResult, error) {
+		Handle(func(ctx context.Context, args map[string]any) (*protocol.CallToolResult, error) {
 			a, ok := args["a"].(float64)
 			if !ok {
 				return protocol.NewToolResultError("参数 'a' 必须是数字"), nil
@@ -47,7 +47,7 @@ func main() {
 	mcp.Tool("subtract", "一个数字减去另一个数字").
 		WithNumberParam("a", "被减数", true).
 		WithNumberParam("b", "减数", true).
-		Handle(func(ctx context.Context, args map[string]interface{}) (*protocol.CallToolResult, error) {
+		Handle(func(ctx context.Context, args map[string]any) (*protocol.CallToolResult, error) {
 			a, ok := args["a"].(float64)
 			if !ok {
 				return protocol.NewToolResultError("参数 'a' 必须是数字"), nil
@@ -64,7 +64,7 @@ func main() {
 	mcp.Tool("multiply", "两个数字相乘").
 		WithNumberParam("a", "第一个数字", true).
 		WithNumberParam("b", "第二个数字", true).
-		Handle(func(ctx context.Context, args map[string]interface{}) (*protocol.CallToolResult, error) {
+		Handle(func(ctx context.Context, args map[string]any) (*protocol.CallToolResult, error) {
 			a, ok := args["a"].(float64)
 			if !ok {
 				return protocol.NewToolResultError("参数 'a' 必须是数字"), nil
@@ -81,7 +81,7 @@ func main() {
 	mcp.Tool("divide", "一个数字除以另一个数字").
 		WithNumberParam("a", "被除数", true).
 		WithNumberParam("b", "除数", true).
-		Handle(func(ctx context.Context, args map[string]interface{}) (*protocol.CallToolResult, error) {
+		Handle(func(ctx context.Context, args map[string]any) (*protocol.CallToolResult, error) {
 			a, ok := args["a"].(float64)
 			if !ok {
 				return protocol.NewToolResultError("参数 'a' 必须是数字"), nil

examples/calculator/server/main.go

@@ -26,7 +26,7 @@ func main() {
 	defer mcpClient.Close()
 
 	// 执行 MCP 初始化握手
-	fmt.Println("🔄 连接到聊天机器人服务...")
+	fmt.Println("连接到聊天机器人服务...")
 	initResult, err := mcpClient.Initialize(ctx, protocol.ClientInfo{
 		Name:    "聊天机器人客户端",
 		Version: "1.0.0",
@@ -35,7 +35,7 @@ func main() {
 		log.Fatalf("初始化失败: %v", err)
 	}
 
-	fmt.Printf("✅ 连接成功！服务器: %s v%s\n",
+	fmt.Printf("连接成功！服务器: %s v%s\n",
 		initResult.ServerInfo.Name, initResult.ServerInfo.Version)
 
 	// 发送初始化完成通知
@@ -52,7 +52,7 @@ func main() {
 	}
 
 	// 获取问候语
-	result, err := mcpClient.CallTool(ctx, "greeting", map[string]interface{}{
+	result, err := mcpClient.CallTool(ctx, "greeting", map[string]any{
 		"name": username,
 	})
 	if err != nil {
@@ -88,7 +88,7 @@ func main() {
 	fmt.Println()
 
 	for {
-		fmt.Print("💬 > ")
+		fmt.Print("> ")
 		scanner.Scan()
 		input := strings.TrimSpace(scanner.Text())
 
@@ -111,75 +111,75 @@ func main() {
 
 			city = strings.TrimSpace(city)
 			if city == "" {
-				fmt.Println("❌ 请指定城市名称")
+				fmt.Println("请指定城市名称")
 				continue
 			}
 
-			result, err := mcpClient.CallTool(ctx, "weather", map[string]interface{}{
+			result, err := mcpClient.CallTool(ctx, "weather", map[string]any{
 				"city": city,
 			})
 			if err != nil {
-				fmt.Printf("❌ 错误: %v\n", err)
+				fmt.Printf("错误: %v\n", err)
 				continue
 			}
 
 			if len(result.Content) > 0 {
 				if textContent, ok := result.Content[0].(protocol.TextContent); ok {
-					fmt.Printf("🌤️  %s\n", textContent.Text)
+					fmt.Printf(" %s\n", textContent.Text)
 				}
 			}
 		} else if strings.Contains(input, " to ") {
 			// 处理翻译请求
 			parts := strings.Split(input, " to ")
 			if len(parts) != 2 || !strings.HasPrefix(parts[0], "translate ") {
-				fmt.Println("❌ 格式错误。请使用: translate [文本] to [zh/en]")
+				fmt.Println("格式错误。请使用: translate [文本] to [zh/en]")
 				continue
 			}
 
 			text := strings.TrimSpace(strings.TrimPrefix(parts[0], "translate "))
 			targetLang := strings.TrimSpace(parts[1])
 
 			if text == "" {
-				fmt.Println("❌ 请提供要翻译的文本")
+				fmt.Println("请提供要翻译的文本")
 				continue
 			}
 
 			if targetLang != "zh" && targetLang != "en" {
-				fmt.Println("❌ 目标语言必须是 'zh' 或 'en'")
+				fmt.Println("目标语言必须是 'zh' 或 'en'")
 				continue
 			}
 
-			result, err := mcpClient.CallTool(ctx, "translate", map[string]interface{}{
+			result, err := mcpClient.CallTool(ctx, "translate", map[string]any{
 				"text":        text,
 				"target_lang": targetLang,
 			})
 			if err != nil {
-				fmt.Printf("❌ 错误: %v\n", err)
+				fmt.Printf("错误: %v\n", err)
 				continue
 			}
 
 			if result.IsError && len(result.Content) > 0 {
 				if textContent, ok := result.Content[0].(protocol.TextContent); ok {
-					fmt.Printf("❌ %s\n", textContent.Text)
+					fmt.Printf("%s\n", textContent.Text)
 				}
 			} else if len(result.Content) > 0 {
 				if textContent, ok := result.Content[0].(protocol.TextContent); ok {
-					fmt.Printf("🔤 翻译结果: %s\n", textContent.Text)
+					fmt.Printf("翻译结果: %s\n", textContent.Text)
 				}
 			}
 		} else if input == "help" || input == "帮助" {
 			// 显示帮助信息
-			fmt.Println("📖 可用命令:")
+			fmt.Println("可用命令:")
 			fmt.Println("  - weather [城市] 或 天气 [城市] - 查看指定城市的天气")
 			fmt.Println("  - translate [文本] to [zh/en] - 翻译中英文")
 			fmt.Println("  - help 或 帮助 - 显示此帮助信息")
 			fmt.Println("  - exit 或 退出 - 退出程序")
 		} else {
 			// 未识别的命令
-			fmt.Printf("❓ 我不理解这个命令: '%s'\n", input)
-			fmt.Println("💡 请尝试 'weather [城市]'、'translate [文本] to [zh/en]'、'help' 或 'exit'")
+			fmt.Printf("未知命令: '%s'\n", input)
+			fmt.Println("请尝试 'weather [城市]'、'translate [文本] to [zh/en]'、'help' 或 'exit'")
 		}
 	}
 
-	fmt.Println("\n👋 再见！感谢使用聊天机器人！")
+	fmt.Println("\n end!")
 }