Struct WrapperToolset 

pub struct WrapperToolset<T, Deps = ()> { /* private fields */ }

Wrapper that allows custom pre/post processing.

This is useful for adding logging, metrics, or other cross-cutting concerns to tool calls.

Example

use serdes_ai_toolsets::{WrapperToolset, FunctionToolset};

let toolset = FunctionToolset::new().tool(my_tool);

let wrapped = WrapperToolset::new(toolset)
    .before(|name, args, ctx| {
        println!("Calling tool: {} with args: {:?}", name, args);
    })
    .after(|name, result, ctx| {
        match result {
            Ok(_) => println!("Tool {} succeeded", name),
            Err(e) => println!("Tool {} failed: {}", name, e),
        }
    });

Implementations

impl<T, Deps> WrapperToolset<T, Deps>

where T: AbstractToolset<Deps>,

pub fn new(inner: T) -> WrapperToolset<T, Deps>

Create a new wrapper toolset.

pub fn before<F>(self, f: F) -> WrapperToolset<T, Deps>

where F: Fn(&str, &Value, &RunContext<Deps>) + Send + Sync + 'static,

Add a before-call hook.

pub fn after<F>(self, f: F) -> WrapperToolset<T, Deps>

where F: Fn(&str, &Result<ToolReturn, ToolError>, &RunContext<Deps>) + Send + Sync + 'static,

Add an after-call hook.

pub fn inner(&self) -> &T

Get the inner toolset.

Trait Implementations

impl<T, Deps> AbstractToolset<Deps> for WrapperToolset<T, Deps>

where T: AbstractToolset<Deps>, Deps: Send + Sync,

fn id(&self) -> Option<&str>

Unique identifier for this toolset.

fn type_name(&self) -> &'static str

Type name for debugging.

fn label(&self) -> String

Human-readable label for error messages.

fn get_tools<'life0, 'life1, 'async_trait>( &'life0 self, ctx: &'life1 RunContext<Deps>, ) -> Pin<Box<dyn Future<Output = Result<HashMap<String, ToolsetTool>, ToolError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, WrapperToolset<T, Deps>: 'async_trait,

Get all available tools.

fn call_tool<'life0, 'life1, 'life2, 'life3, 'async_trait>( &'life0 self, name: &'life1 str, args: Value, ctx: &'life2 RunContext<Deps>, tool: &'life3 ToolsetTool, ) -> Pin<Box<dyn Future<Output = Result<ToolReturn, ToolError>> + Send + 'async_trait>>

where 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait, 'life3: 'async_trait, WrapperToolset<T, Deps>: 'async_trait,

Call a tool by name.

fn enter<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<(), ToolError>> + Send + 'async_trait>>

where 'life0: 'async_trait, WrapperToolset<T, Deps>: 'async_trait,

Enter context (for resource setup).

fn exit<'life0, 'async_trait>( &'life0 self, ) -> Pin<Box<dyn Future<Output = Result<(), ToolError>> + Send + 'async_trait>>

where 'life0: 'async_trait, WrapperToolset<T, Deps>: 'async_trait,

Exit context (for cleanup).

fn tool_name_conflict_hint(&self) -> String

Hint for resolving name conflicts.

impl<T, Deps> Debug for WrapperToolset<T, Deps>

where T: Debug,

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.