模型上下文协议 (MCP) Rust SDK

⚠️ 警告: 此 SDK 目前仍在开发中，尚未准备好用于生产环境。

一个模型上下文协议 (MCP) 的 Rust 实现，旨在实现 AI 模型与其运行时环境之间的无缝通信。

特性

• 🚀 MCP 协议规范的完整实现
• 🔄 多种传输层 (WebSocket, stdio)
• ⚡ 使用 Tokio 的 Async/await 支持
• 🛡️ 类型安全的消息处理
• 🔍 完善的错误处理
• 📦 零拷贝序列化/反序列化

安装

将以下内容添加到你的 Cargo.toml 文件中：

[dependencies]
mcp_sdk_rs = "0.1.0"

快速开始

客户端示例

use mcp_sdk_rs::{Client, transport::WebSocketTransport};

#[tokio::main]
async fn main() -> Result<(), Error> {
    // 创建一个传输层
    let transport = WebSocketTransport::new(self.url.as_str())
        .await
        .map_err(|_| Error::Internal)?;

    // 创建 mpsc 通道，用于客户端和会话之间的通信
    let (request_tx, request_rx): (UnboundedSender<Message>, UnboundedReceiver<Message>) =
        tokio::sync::mpsc::unbounded_channel();
    let (response_tx, response_rx): (UnboundedSender<Message>, UnboundedReceiver<Message>) =
        tokio::sync::mpsc::unbounded_channel();

    // 创建并启动会话
    // 可选地传递 ClientHandler 的实现，用于自定义处理请求和通知
    let session = Session::new(Arc::new(transport), response_tx, request_rx, None);
    session.start().await.map_err(|_| Error::Internal)?;

    // 创建客户端并发出请求，接收通知等
    let client = Client::new(request_tx, response_rx);
    let response = client.request(
        "tools/call",
        Some(json!({
            "name": "methondName",
            "arguments": json!({})
        })),
    )
    .await?
}

服务端示例

use mcp_sdk_rs::{Server, transport::StdioTransport};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // 创建一个 stdio 传输层
    let (transport, _) = StdioTransport::new();

    // 创建并启动服务器
    let server = Server::new(transport);
    server.start().await?;

    Ok(())
}

传输层

该 SDK 支持多种传输层：

WebSocket 传输层

• 适用于基于网络的通信
• 支持安全 (WSS) 和标准 (WS) 连接
• 内置重连处理

stdio 传输层

• 适用于本地进程通信
• 轻量且高效
• 非常适合命令行工具和本地开发

错误处理

该 SDK 通过 Error 类型提供完善的错误处理：

use mcp_sdk_rs::Error;

match result {
    Ok(value) => println!("Success: {:?}", value),
    Err(Error::Protocol(code, msg)) => println!("Protocol error {}: {}", code, msg),
    Err(Error::Transport(e)) => println!("Transport error: {}", e),
    Err(e) => println!("Other error: {}", e),
}

许可证

本项目基于 MIT 许可证 - 详情请参阅 LICENSE 文件。

最初 fork 自 https://github.com/Derek-X-Wang/mcp-rust-sdk。