Model Context Protocol (MCP) Implementation in Go

使用 Go 实现 Model Context Protocol（MCP）

此代码库包含 Model Context Protocol（MCP） 的 Go 实现。MCP 是一种基于 JSON-RPC 的协议，用于在由大模型驱动的应用中实现客户端与服务端之间的交互。

目录

• mcp/schema.go：MCP 架构的 Go 结构体定义，由 TypeScript 架构转换而来
• mcp/client.go：用于与 MCP 服务交互的客户端实现
• examples/mcp_client_example.go：展示如何使用客户端的简单示例

MCP 概述

Model Context Protocol（MCP）支持模型“宿主”（例如，大模型客户端或与大模型交互的系统）与“服务端”（提供上下文信息、工具、提示词或资源，以增强大模型能力的系统）之间的通信。

MCP 的主要特性包括：

• 资源：宿主可以读取的文件、文档和其他数据
• 工具：宿主可以代表模型调用的函数
• 提示词：宿主可以使用的模板或预定义消息序列
• 根目录：服务端可以操作的文件系统或数据根目录

开始使用

先决条件

• Go 1.18 或更高版本

运行示例

# Set the MCP server URL (optional, defaults to http://localhost:3000/jsonrpc)
export MCP_SERVER_URL=http://your-mcp-server/jsonrpc

# Run the example
go run examples/mcp_client_example.go

使用 MCP 客户端

import (
    "fmt"
    
    "path/to/mcp"
)

func main() {
    // Create a new client
    client := mcp.NewClient("http://your-mcp-server/jsonrpc")
    
    // Initialize the client
    clientInfo := mcp.Implementation{
        Name:    "My MCP Client",
        Version: "1.0.0",
    }
    
    capabilities := mcp.ClientCapabilities{
        // Specify the capabilities your client supports
    }
    
    err := client.Initialize(clientInfo, capabilities)
    if err != nil {
        // Handle error
    }
    
    // Now you can use various client methods:
    
    // List resources
    resources, err := client.ListResources("")
    
    // Read a resource
    resource, err := client.ReadResource("resource-uri")
    
    // Call a tool
    result, err := client.CallTool("tool-name", map[string]interface{}{
        "argument1": "value1",
        "argument2": 123,
    })
    
    // Get a prompt
    prompt, err := client.GetPrompt("prompt-name", map[string]string{
        "argName": "argValue",
    })
}

实现细节

架构

mcp/schema.go 中的 Go 架构定义为 MCP 消息提供了类型安全的结构，其中包括：

• JSON-RPC 协议结构
• 资源、提示词和工具定义
• 文本、图像和嵌入式资源的内容结构
• 用于通过 json.RawMessage 处理多态类型的辅助函数

客户端

mcp/client.go 中的客户端实现提供了以下方法：

• 初始化与 MCP 服务的连接
• 发送请求并处理响应
• 将 Go 结构体转换为 JSON-RPC 消息
• 用于常见 MCP 操作的辅助方法

贡献

欢迎贡献！以下是一些贡献方式：

• 实现服务端
• 为客户端添加 WebSocket 支持
• 改进错误处理
• 添加测试
• 记录更多使用场景

许可证

本项目采用 MIT 许可证授权，详情请参阅 LICENSE 文件。

参考资料

• Model Context Protocol 规范
• MCP GitHub 仓库