rust-mcp-sdk 0.9.0

An asynchronous SDK and framework for building MCP-Servers and MCP-Clients, leveraging the rust-mcp-schema for type safe MCP Schema Objects.
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

#[macro_export]
/// Generates an enum representing a toolbox with mcp tool variants and associated functionality.
///
/// **Note:** The macro assumes that tool types provided are annotated with the mcp_tool() macro.
///
/// This macro creates:
/// - An enum with the specified name containing variants for each mcp tool
/// - A `tools()` function returning a vector of supported tools
/// - A `TryFrom<CallToolRequestParams>` implementation for converting requests to tool instances
///
/// # Arguments
/// * `$enum_name` - The name to give the generated enum
/// * `[$($tool:ident),*]` - A comma-separated list of tool types to include in the enum
///
///
/// # Example
/// ```ignore
/// tool_box!(FileSystemTools, [ReadFileTool, EditFileTool, SearchFilesTool]);
/// // Creates:
/// // pub enum FileSystemTools {
/// //     ReadFileTool(ReadFileTool),
/// //     EditFileTool(EditFileTool),
/// //     SearchFilesTool(SearchFilesTool),
/// // }
/// // pub fn tools() -> Vec<Tool> {
/// //     vec![ReadFileTool::tool(), EditFileTool::tool(), SearchFilesTool::tool()]
/// // }
///
/// // impl TryFrom<CallToolRequestParams> for FileSystemTools {
/// //  //.......
/// // }
macro_rules! tool_box {
    ($enum_name:ident, [$($tool:ident),* $(,)?]) => {
        #[derive(Debug)]
        pub enum $enum_name {
            $(
                // Just create enum variants for each tool
                $tool($tool),
            )*
        }

        /// Returns the name of the tool as a String
        impl $enum_name {
            pub fn tool_name(&self) -> String {
                match self {
                    $(
                        $enum_name::$tool(_) => $tool::tool_name(),
                    )*
                }
            }

            /// Returns a vector containing instances of all supported tools
            pub fn tools() -> Vec<rust_mcp_sdk::schema::Tool> {
                vec![
                    $(
                        $tool::tool(),
                    )*
                ]
            }
        }




        impl TryFrom<rust_mcp_sdk::schema::CallToolRequestParams> for $enum_name {
            type Error = rust_mcp_sdk::schema::schema_utils::CallToolError;

            /// Attempts to convert a tool request into the appropriate tool variant
            fn try_from(value: rust_mcp_sdk::schema::CallToolRequestParams) -> Result<Self, Self::Error> {
                let arguments = value
                    .arguments
                    .ok_or(rust_mcp_sdk::schema::schema_utils::CallToolError::invalid_arguments(
                        &value.name,
                        Some("Missing 'arguments' field in the request".to_string())
                    ))?;

                    let v = serde_json::to_value(arguments).map_err(|err| {
                        rust_mcp_sdk::schema::schema_utils::CallToolError::invalid_arguments(
                            &value.name,
                            Some(format!("{err}")),
                        )
                    })?;

                    match value.name {
                        $(
                            name if name == $tool::tool_name().as_str() => {
                                Ok(Self::$tool(serde_json::from_value(v).map_err(rust_mcp_sdk::schema::schema_utils::CallToolError::new)?))
                            }
                        )*
                        _ => {
                               Err(
                                rust_mcp_sdk::schema::schema_utils::CallToolError::unknown_tool(value.name.to_string())
                              )
                        }
                    }

            }
        }
    }
}