Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Inline tool schemas

Browse source
This commit is contained in:
Agus Zubiaga 2025-07-30 11:46:11 -03:00
parent f028ca4d1a
commit 738296345e
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
crates/context_server/src/listener.rs
View file

@ -83,14 +83,18 @@ impl McpServer {
} }
pub fn add_tool<T: McpServerTool + Clone + 'static>(&mut self, tool: T) { pub fn add_tool<T: McpServerTool + Clone + 'static>(&mut self, tool: T) {
let output_schema = schemars::schema_for!(T::Output); let mut settings = schemars::generate::SchemaSettings::draft07();
let unit_schema = schemars::schema_for!(()); settings.inline_subschemas = true;
let mut generator = settings.into_generator();
let output_schema = generator.root_schema_for::<T::Output>();
let unit_schema = generator.root_schema_for::<T::Output>();
let registered_tool = RegisteredTool { let registered_tool = RegisteredTool {
tool: Tool { tool: Tool {
name: T::NAME.into(), name: T::NAME.into(),
description: Some(tool.description().into()), description: Some(tool.description().into()),
input_schema: schemars::schema_for!(T::Input).into(), input_schema: generator.root_schema_for::<T::Input>().into(),
output_schema: if output_schema == unit_schema { output_schema: if output_schema == unit_schema {
None None
} else { } else {