1f7f4d3055* refactor: refactor tool macros and router implementation - Updated the `#[tool(tool_box)]` macro to `#[tool_router]` across various modules for consistency. - Enhanced the `Calculator`, `Counter`, and `GenericService` structs to utilize `ToolRouter` for handling tool calls. - Introduced `Parameters` struct for better parameter handling in tool functions. - Added new methods for listing tools and calling tools in server handlers. - Improved test cases to reflect changes in tool routing and parameter handling. - Updated documentation and examples to align with the new router structure. * fix: fix fmt and build error * fix: fix test failure * docs: documents for macros, fix ci * fix: fix ci * fix: fix wrongly replaced documents * fix: remove useless file * fix: change the parameter format for tool_router * fix: update extract_doc_line to handle existing documentation and clean up unused code in server handler * doc: update document for macro and examples * doc: update readme and add contribute guide * fix: fix type
62 lines
1.6 KiB
Rust
62 lines
1.6 KiB
Rust
#![allow(dead_code)]
|
|
use rmcp::{
|
|
ServerHandler,
|
|
handler::server::{router::tool::ToolRouter, tool::Parameters},
|
|
model::{ServerCapabilities, ServerInfo},
|
|
schemars, tool, tool_router,
|
|
};
|
|
#[derive(Debug, serde::Deserialize, schemars::JsonSchema)]
|
|
pub struct SumRequest {
|
|
#[schemars(description = "the left hand side number")]
|
|
pub a: i32,
|
|
pub b: i32,
|
|
}
|
|
|
|
#[derive(Debug, serde::Deserialize, schemars::JsonSchema)]
|
|
pub struct SubRequest {
|
|
#[schemars(description = "the left hand side number")]
|
|
pub a: i32,
|
|
#[schemars(description = "the right hand side number")]
|
|
pub b: i32,
|
|
}
|
|
#[derive(Debug, Clone)]
|
|
pub struct Calculator {
|
|
tool_router: ToolRouter<Self>,
|
|
}
|
|
|
|
impl Calculator {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
tool_router: Self::tool_router(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Default for Calculator {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
#[tool_router]
|
|
impl Calculator {
|
|
#[tool(description = "Calculate the sum of two numbers")]
|
|
fn sum(&self, Parameters(SumRequest { a, b }): Parameters<SumRequest>) -> String {
|
|
(a + b).to_string()
|
|
}
|
|
|
|
#[tool(description = "Calculate the sub of two numbers")]
|
|
fn sub(&self, Parameters(SubRequest { a, b }): Parameters<SubRequest>) -> String {
|
|
(a - b).to_string()
|
|
}
|
|
}
|
|
|
|
impl ServerHandler for Calculator {
|
|
fn get_info(&self) -> ServerInfo {
|
|
ServerInfo {
|
|
instructions: Some("A simple calculator".into()),
|
|
capabilities: ServerCapabilities::builder().enable_tools().build(),
|
|
..Default::default()
|
|
}
|
|
}
|
|
}
|