模型上下文协议（MCP）在 Go 语言中的实现

此仓库包含一个基于 JSON-RPC 的协议的 Go 实现，用于大语言模型 (LLM) 应用中客户端和服务端之间的交互。该协议名为 Model Context Protocol (MCP)，旨在增强 LLM 功能。

仓库内容

• mcp/schema.go：从 TypeScript 方案转换而来的 MCP 模式的 Go 结构体定义
• mcp/client.go：与 MCP 服务器交互的客户端实现
• examples/mcp_client_example.go：演示如何使用客户端的简单示例

MCP 概览

模型上下文协议 (MCP) 使得模型“宿主”（例如 LLM 客户端或与 LLM 接口的系统）和服务端（提供上下文信息、工具、提示语或增强 LLM 功能的资源的系统）之间的通信成为可能。

MCP 的关键功能包括：

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

入门指南

前提条件

• Go 1.18 或更高版本

运行示例

# 设置 MCP 服务器 URL（可选，默认为 http://localhost:3000/jsonrpc）
export MCP_SERVER_URL=http://your-mcp-server/jsonrpc

# 运行示例
go run examples/mcp_client_example.go

使用 MCP 客户端

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

func main() {
    // 创建一个新的客户端
    client := mcp.NewClient("http://your-mcp-server/jsonrpc")
    
    // 初始化客户端
    clientInfo := mcp.Implementation{
        Name:    "My MCP Client",
        Version: "1.0.0",
    }
    
    capabilities := mcp.ClientCapabilities{
        // 指定你客户端支持的功能
    }
    
    err := client.Initialize(clientInfo, capabilities)
    if err != nil {
        // 处理错误
    }
    
    // 现在你可以使用各种客户端方法：
    
    // 列举资源
    resources, err := client.ListResources("")
    
    // 读取一个资源
    resource, err := client.ReadResource("resource-uri")
    
    // 调用工具
    result, err := client.CallTool("tool-name", map[string]interface{}{
        "argument1": "value1",
        "argument2": 123,
    })
    
    // 获取提示语
    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 操作

贡献

欢迎贡献！你可以通过以下方式参与：

• 实现一个服务端
• 提交问题或功能请求
• 提交代码修复或新功能

请查看 CONTRIBUTING.md 了解详细信息。

许可证

此项目受 MIT License 保护。