ext/memories/src/schema.rs
42 lines
use schemars::JsonSchema;use schemars::r#gen::SchemaSettings;use serde_json::Map;use serde_json::Value;pub(crate) fn input_schema_for<T: JsonSchema>() -> Value { schema_for::<T>(/*option_add_null_type*/ false)}pub(crate) fn output_schema_for<T: JsonSchema>() -> Value { schema_for::<T>(/*option_add_null_type*/ true)}fn schema_for<T: JsonSchema>(option_add_null_type: bool) -> Value { let schema = SchemaSettings::draft2019_09() .with(|settings| { settings.inline_subschemas = true; settings.option_add_null_type = option_add_null_type; }) .into_generator() .into_root_schema_for::<T>(); let schema_value = serde_json::to_value(schema) .unwrap_or_else(|err| panic!("generated tool schema should serialize: {err}")); let Value::Object(mut schema_object) = schema_value else { unreachable!("root tool schema must be an object"); }; let mut tool_schema = Map::new(); for key in [ "properties", "required", "type", "additionalProperties", "$defs", "definitions", ] { if let Some(value) = schema_object.remove(key) { tool_schema.insert(key.to_string(), value); } } Value::Object(tool_schema)}