rmcp 2.0.0

Rust SDK for Model Context Protocol
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

#![cfg(all(feature = "server", feature = "macros", not(feature = "local")))]

use rmcp::{
    Json,
    handler::server::wrapper::Parameters,
    model::{ListToolsResult, NumberOrString, ServerJsonRpcMessage, ServerResult},
};

/// Parameters for adding two numbers.
#[derive(Debug, serde::Deserialize, schemars::JsonSchema)]
struct AddRequest {
    /// The left-hand number.
    a: f64,
    /// The right-hand number.
    b: f64,
}

/// Result of adding two numbers.
#[derive(Debug, serde::Serialize, schemars::JsonSchema)]
struct AddResult {
    /// The sum of the two numbers.
    sum: f64,
}

/// Add two numbers.
#[rmcp::tool]
fn add(Parameters(AddRequest { a, b }): Parameters<AddRequest>) -> Json<AddResult> {
    Json(AddResult { sum: a + b })
}

#[test]
fn list_tools_result_matches_expected_json() {
    let expected_json = std::fs::read("tests/test_list_tools_result/list_tools_result.json")
        .expect("missing expected list tools result JSON fixture");
    let expected: serde_json::Value =
        serde_json::from_slice(&expected_json).expect("invalid expected JSON fixture");

    assert_eq!(add(Parameters(AddRequest { a: 1.0, b: 2.0 })).0.sum, 3.0);

    let result = ListToolsResult::with_all_items(vec![add_tool_attr()]);
    let response = ServerJsonRpcMessage::response(
        ServerResult::ListToolsResult(result),
        NumberOrString::Number(2),
    );

    let actual = serde_json::to_value(response).expect("failed to serialize list tools response");
    assert_eq!(actual, expected);
}