模型上下文协议(MCP)Rust SDK
⚠️ 警告:此SDK目前正在开发中,尚未准备好用于生产。
模型上下文协议(MCP)的Rust实现,旨在实现AI模型与其运行时环境之间的无缝通信。
特性
- 🚀 全面实施MCP协议规范
- 🔄 多个传输层(WebSocket、stdio)
- ⚡ 使用Tokio支持异步/等待
- 🛡️ 类型安全消息处理
- 🔍 全面的错误处理
- 📦 零拷贝序列化/反序列化
安装
将此添加到您的 Cargo.toml:
[dependencies]
mcp_sdk_rs = "0.1.0"快速开始
客户端示例
use mcp_sdk_rs::{Client, transport::WebSocketTransport};
#[tokio::main]
async fn main() -> Result {
// Create a transport
let transport = WebSocketTransport::new(self.url.as_str())
.await
.map_err(|_| Error::Internal)?;
//Create mpsc channels for communication between the client and session
let (request_tx, request_rx): (UnboundedSender, UnboundedReceiver) =
tokio::sync::mpsc::unbounded_channel();
let (response_tx, response_rx): (UnboundedSender, UnboundedReceiver) =
tokio::sync::mpsc::unbounded_channel();
// Create and start the session
// Optionally pass an implementation of ClientHandler for custom handling of requests and notifications
let session = Session::new(Arc::new(transport), response_tx, request_rx, None);
session.start().await.map_err(|_| Error::Internal)?;
// Create the client and make requests, receive notifications etc
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> {
// Create a stdio transport
let (transport, _) = StdioTransport::new();
// Create and start the server
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许可证获得许可-请参阅 许可证 文件以获取详细信息。
最初分叉自 .
