diff --git a/crates/rmcp-macros/README.md b/crates/rmcp-macros/README.md index 7c2ff074..f009329b 100644 --- a/crates/rmcp-macros/README.md +++ b/crates/rmcp-macros/README.md @@ -72,7 +72,7 @@ Or specify the visibility and router name, which would be helpful when you want ```rust mod a { - #[tool_router(router = tool_router_a, vis = pub)] + #[tool_router(router = tool_router_a, vis = "pub")] impl MyToolHandler { #[tool] fn my_tool_a() { @@ -82,7 +82,7 @@ mod a { } mod b { - #[tool_router(router = tool_router_b, vis = pub)] + #[tool_router(router = tool_router_b, vis = "pub")] impl MyToolHandler { #[tool] fn my_tool_b() { diff --git a/crates/rmcp-macros/src/lib.rs b/crates/rmcp-macros/src/lib.rs index 3a5deb4d..6bb06827 100644 --- a/crates/rmcp-macros/src/lib.rs +++ b/crates/rmcp-macros/src/lib.rs @@ -75,7 +75,7 @@ pub fn tool(attr: TokenStream, input: TokenStream) -> TokenStream { /// /// ```rust,ignore /// mod a { -/// #[tool_router(router = tool_router_a, vis = pub)] +/// #[tool_router(router = tool_router_a, vis = "pub")] /// impl MyToolHandler { /// #[tool] /// fn my_tool_a() { @@ -85,7 +85,7 @@ pub fn tool(attr: TokenStream, input: TokenStream) -> TokenStream { /// } /// /// mod b { -/// #[tool_router(router = tool_router_b, vis = pub)] +/// #[tool_router(router = tool_router_b, vis = "pub")] /// impl MyToolHandler { /// #[tool] /// fn my_tool_b() { diff --git a/crates/rmcp-macros/src/tool_router.rs b/crates/rmcp-macros/src/tool_router.rs index f6bc0edd..91fa8d9d 100644 --- a/crates/rmcp-macros/src/tool_router.rs +++ b/crates/rmcp-macros/src/tool_router.rs @@ -76,6 +76,7 @@ mod test { fn test_router_attr() -> Result<(), Box> { let attr = quote! { router = test_router, + vis = "pub(crate)" }; let attr_args = NestedMeta::parse_meta_list(attr)?; let ToolRouterAttribute { router, vis } = ToolRouterAttribute::from_list(&attr_args)?;