pmcp 2.1.0

High-quality Rust SDK for Model Context Protocol (MCP) with full TypeScript SDK compatibility
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

//! Core traits for handling server-side MCP requests.

use crate::error::Result;
use crate::shared::cancellation::RequestHandlerExtra;
use crate::types::{
    CallToolResult, CreateMessageParams, CreateMessageResult, GetPromptResult,
    ListResourcesResult, ListToolsResult, ReadResourceResult,
};
use async_trait::async_trait;
use serde_json::Value;
use std::collections::HashMap;

/// A comprehensive handler for all MCP server operations.
///
/// This trait combines all the specific handler traits into a single interface.
/// It is the main entry point for implementing a custom MCP server.
#[async_trait]
pub trait RequestHandler:
    ToolHandler + PromptHandler + ResourceHandler + SamplingHandler + Send + Sync
{
}

/// Handler for tool execution.
#[async_trait]
pub trait ToolHandler: Send + Sync {
    /// List available tools.
    async fn list_tools(&self, _req: crate::types::ListToolsRequest) -> Result<ListToolsResult> {
        Ok(ListToolsResult {
            tools: vec![],
            next_cursor: None,
        })
    }

    /// Handle a tool call with the given arguments.
    async fn call_tool(&self, req: crate::types::CallToolRequest) -> Result<CallToolResult>;
}

/// Handler for prompt generation.
#[async_trait]
pub trait PromptHandler: Send + Sync {
    /// Generate a prompt with the given arguments.
    async fn handle(
        &self,
        args: HashMap<String, String>,
        extra: RequestHandlerExtra,
    ) -> Result<GetPromptResult>;
}

/// Handler for resource access.
#[async_trait]
pub trait ResourceHandler: Send + Sync {
    /// Read a resource at the given URI.
    async fn read(&self, uri: &str, extra: RequestHandlerExtra) -> Result<ReadResourceResult>;

    /// List available resources.
    async fn list(&self, cursor: Option<String>, extra: RequestHandlerExtra)
        -> Result<ListResourcesResult>;
}

/// Handler for message sampling (LLM operations).
#[async_trait]
pub trait SamplingHandler: Send + Sync {
    /// Create a message using the language model.
    async fn create_message(
        &self,
        params: CreateMessageParams,
        extra: RequestHandlerExtra,
    ) -> Result<CreateMessageResult>;
}