modelcontextprotocol/rust-sdk
Watch
1
0
Fork
You've already forked rust-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: generic ServerHandler (#223)

Browse source
This commit is contained in:
Pete Hayman 2025-05-26 16:13:34 +10:00 • committed by GitHub
commit 4e8686ff47
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
crates/rmcp-macros/src/tool.rs
View file

@ -258,7 +258,7 @@ pub(crate) fn tool_impl_item(attr: TokenStream, mut input: ItemImpl) -> syn::Res
input.items.push(parse_quote! { input.items.push(parse_quote! {
async fn list_tools( async fn list_tools(
&self, &self,
request: rmcp::model::PaginatedRequestParam, request: Option<rmcp::model::PaginatedRequestParam>,
context: rmcp::service::RequestContext<rmcp::RoleServer>, context: rmcp::service::RequestContext<rmcp::RoleServer>,
) -> Result<rmcp::model::ListToolsResult, rmcp::Error> { ) -> Result<rmcp::model::ListToolsResult, rmcp::Error> {
self.list_tools_inner(request, context).await self.list_tools_inner(request, context).await
@ -316,7 +316,7 @@ pub(crate) fn tool_impl_item(attr: TokenStream, mut input: ItemImpl) -> syn::Res
input.items.push(parse_quote! { input.items.push(parse_quote! {
async fn list_tools_inner( async fn list_tools_inner(
&self, &self,
_: rmcp::model::PaginatedRequestParam, _: Option<rmcp::model::PaginatedRequestParam>,
_: rmcp::service::RequestContext<rmcp::RoleServer>, _: rmcp::service::RequestContext<rmcp::RoleServer>,
) -> Result<rmcp::model::ListToolsResult, rmcp::Error> { ) -> Result<rmcp::model::ListToolsResult, rmcp::Error> {
Ok(rmcp::model::ListToolsResult { Ok(rmcp::model::ListToolsResult {

2
crates/rmcp/tests/test_tool_macros.rs
View file

@ -83,6 +83,8 @@ impl<DS: DataService> GenericServer<DS> {
self.data_service.get_data() self.data_service.get_data()
} }
} }
#[tool(tool_box)]
impl<DS: DataService> ServerHandler for GenericServer<DS> {}
#[tokio::test] #[tokio::test]
async fn test_tool_macros() { async fn test_tool_macros() {