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

feat(tool): add _meta to tool definitions (#534)

Browse source
This commit is contained in:
Dylan Anthony 2025-11-10 18:27:52 -07:00 • committed by GitHub
commit 03040f8b0b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 33 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -77,6 +77,8 @@ pub struct ToolAttribute {
pub annotations: Option<ToolAnnotationsAttribute>,
/// Optional icons for the tool
pub icons: Option<Expr>,
/// Optional metadata for the tool
pub meta: Option<Expr>,
}
pub struct ResolvedToolAttribute {
@ -87,6 +89,7 @@ pub struct ResolvedToolAttribute {
pub output_schema: Option<Expr>,
pub annotations: Expr,
pub icons: Option<Expr>,
pub meta: Option<Expr>,
}
impl ResolvedToolAttribute {
@ -99,6 +102,7 @@ impl ResolvedToolAttribute {
output_schema,
annotations,
icons,
meta,
} = self;
let description = if let Some(description) = description {
quote! { Some(#description.into()) }
@ -120,6 +124,11 @@ impl ResolvedToolAttribute {
} else {
quote! { None }
};
let meta = if let Some(meta) = meta {
quote! { Some(#meta) }
} else {
quote! { None }
};
let doc_comment = format!("Generated tool metadata function for {name}");
let doc_attr: syn::Attribute = parse_quote!(#[doc = #doc_comment]);
let tokens = quote! {
@ -133,6 +142,7 @@ impl ResolvedToolAttribute {
output_schema: #output_schema,
annotations: #annotations,
icons: #icons,
meta: #meta,
}
}
};
@ -264,6 +274,7 @@ pub fn tool(attr: TokenStream, input: TokenStream) -> syn::Result<TokenStream> {
annotations: annotations_expr,
title: attribute.title,
icons: attribute.icons,
meta: attribute.meta,
};
let tool_attr_fn = resolved_tool_attr.into_fn(tool_attr_fn_ident)?;
// modify the the input function