Model Context Protocol (MCP) Rust SDK

Model Context Protocol (MCP) Rust SDK

⚠️ 警告：この SDK は現在開発中であり、本番環境での使用には対応していません。

AI モデルとその実行環境の間でシームレスに通信できるように設計された、Model Context Protocol（MCP）の Rust 実装です。

機能

• 🚀 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<(), Error> {
    // 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<Message>, UnboundedReceiver<Message>) =
        tokio::sync::mpsc::unbounded_channel();
    let (response_tx, response_rx): (UnboundedSender<Message>, UnboundedReceiver<Message>) =
        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<(), Box<dyn std::error::Error>> {
    // 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 ライセンスのもとで公開されています。詳細は LICENSE ファイルをご覧ください。

当初は https://github.com/Derek-X-Wang/mcp-rust-sdk からフォークされました。