fix impl doc

Author: 姚临倾

crates/rmcp-macros/src/tool.rs

@@ -190,22 +190,27 @@ pub enum ToolItem {
 
 impl Parse for ToolItem {
     fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
-        let lookahead = input.lookahead1();
-        if lookahead.peek(Token![impl]) {
-            let item = input.parse::<ItemImpl>()?;
-            Ok(ToolItem::Impl(item))
-        } else {
-            let item = input.parse::<ItemFn>()?;
-            Ok(ToolItem::Fn(item))
+        if let Ok(item)=input.parse::<ItemImpl>(){
+            return Ok(ToolItem::Impl(item));
+        }
+        if let Ok(item) = input.parse::<ItemFn>() {
+            return Ok(ToolItem::Fn(item));
         }
+        Err(syn::Error::new(
+            input.span(),
+            "expected function or impl block",
+        ))
     }
 }
 
 // dispatch impl function item and impl block item
 pub(crate) fn tool(attr: TokenStream, input: TokenStream) -> syn::Result<TokenStream> {
     let tool_item = syn::parse2::<ToolItem>(input)?;
     match tool_item {
-        ToolItem::Fn(item) => tool_fn_item(attr, item),
+        ToolItem::Fn(item) => {
+            println!("{:#?}",&item);
+            tool_fn_item(attr, item)
+        },
         ToolItem::Impl(item) => tool_impl_item(attr, item),
     }
 }
@@ -225,7 +230,6 @@ pub(crate) fn tool_impl_item(attr: TokenStream, mut input: ItemImpl) -> syn::Res
             .filter_map(extract_doc_line)
             .collect::<Vec<_>>()
             .join("\n");
-
         parse_quote! {
                 #doc_content.trim().to_string()
         }

examples/servers/src/common/calculator.rs

@@ -12,7 +12,8 @@ pub struct SumRequest {
 }
 #[derive(Debug, Clone)]
 pub struct Calculator;
-#[tool(tool_box,description = "A simple calculator")]
+
+#[tool(tool_box,description="A simple calculator")]
 impl Calculator {
     #[tool(description = "Calculate the sum of two numbers")]
     fn sum(&self, sum: SumRequest) -> String {