Go语言MCP协议库

项目简介

这是一个用于实现模型上下文协议（Model Context Protocol，简称MCP）的Go语言库。该协议定义了AI模型与外界交互的标准接口，支持文本、图像、音频等多种内容类型的操作。

为什么选择MCP？

• 统一标准：基于最新的2025-03-26 MCP规范开发
• 功能丰富：支持提示词、工具调用、资源操作等核心功能
• 扩展性强：内置了对多种媒体类型和资源格式的支持
• 高效可靠：采用异步非阻塞架构，性能优化明显

协议更新

新增特性

• 完整的JSON RPC支持
• 多模态内容传输能力
• 分页查询功能
• 资源模板系统
• 资源订阅通知
• 详细的服务器声明式能力
• 统一响应处理机制

安装

go get github.com/your-organization/mcp-go

快速上手

创建服务端

// 创建服务
server, err := mcp.NewServer(service, &types.ServerOptions{
    Address: ":8080",
    Capabilities: &types.ServerCapabilities{
        Prompts: &types.PromptCapabilities{
            ListChanged: true,
        },
        Resources: &types.ResourceCapabilities{
            ListChanged: true,
            Subscribe: true,
            Templates: true,
        },
    },
})

创建客户端

// 创建客户端
client, err := mcp.NewClient(&types.ClientOptions{
    ServerAddress: "localhost:8080",
    Type: "http",
    UseJSONRPC: true,
    SubscribeToNotifications: true,
})

接口定义

服务端接口

服务需要实现types.Server接口：

type Server interface {
    MCPService
    Initialize(ctx context.Context, options any) error
    Start() error
    Shutdown(ctx context.Context) error
}

核心功能由MCPService接口定义：

type MCPService interface {
    // 列出所有提示词，支持分页查询
    ListPrompts(ctx context.Context, cursor string) (*PromptListResult, error)
    
    // 获取特定提示词内容，可指定参数
    GetPrompt(ctx context.Context, name string, args map[string]any) (*PromptResult, error)

    // 列出所有工具，支持分页查询
    ListTools(ctx context.Context, cursor string) (*ToolListResult, error)

    // 调用特定工具，可指定参数
    CallTool(ctx context.Context, name string, args map[string]any) (*CallToolResult, error)

    // 列出所有资源，支持分页查询
    ListResources(ctx context.Context, cursor string) (*ResourceListResult, error)

    // 读取特定资源内容
    ReadResource(ctx context.Context, uri string) (*ResourceContent, error)

    // 获取资源模板列表
    ListResourceTemplates(ctx context.Context) ([]ResourceTemplate, error)

    // 订阅特定资源的变更通知
    SubscribeResource(ctx context.Context, resourceID string, callback func(*ResourceUpdateEvent)) error)
}

客户端接口

type MCPClient interface {
    // 发送提示词请求
    SendPrompt(prompt *Prompt) (*Response, error)

    // 调用工具方法
    CallTool(toolName string, args map[string]any) (*Response, error)

    // 订阅资源更新通知
    SubscribeResource(resourceID string, callback func(*ResourceUpdateEvent)) error)
}

错误处理

type MCPError struct {
    Code    int    `json:"code"`
    Message string `json:"message"`
}

func (e *MCPError) Error() string {
    return e.Message
}

使用示例

提示词操作

prompt := &Prompt{
    Content: "帮我写一首诗",
    Parameters: map[string]string{
        "temperature": "0.7",
    },
}

response, err := client.SendPrompt(prompt)
if err != nil {
    // 处理错误
}

工具调用

toolResult, err := client.CallTool("image_generator", map[string]string{
    "prompt": "一片孤帆远影碧空尽",
})
if err != nil {
    // 处理错误
}

贡献指南

• 代码规范：遵循Google Go语言风格指南
• 单元测试覆盖率要求达到100%
• 提交PR前必须通过CI检查
• 遵守GitHub的贡献协议

许可证

本项目使用MIT许可证，欢迎自由使用和修改。

更多详细文档请参考官方文档。