Struct ToolMiddleware 

pub struct ToolMiddleware { /* private fields */ }

Intercepts execute() on a wrapped AgentTool.

All descriptor methods (name, label, description, parameters_schema, metadata, requires_approval, auth_config) delegate to the inner tool.

Implementations

impl ToolMiddleware

pub fn new<F>(inner: Arc<dyn AgentTool>, f: F) -> Self

where F: Fn(Arc<dyn AgentTool>, String, Value, CancellationToken, Option<Box<dyn Fn(AgentToolResult) + Send + Sync>>, Arc<RwLock<SessionState>>, Option<ResolvedCredential>) -> ToolFuture<'static> + Send + Sync + 'static,

Create a new middleware wrapping inner.

The closure receives (inner_tool, tool_call_id, params, cancel, on_update, state, credential) and can call through to the inner tool’s execute() at any point.

pub fn with_timeout(inner: Arc<dyn AgentTool>, timeout: Duration) -> Self

Create a middleware that enforces a timeout on tool execution.

If the inner tool does not complete within timeout, an error result is returned.

pub fn with_logging<F>(inner: Arc<dyn AgentTool>, callback: F) -> Self

where F: Fn(&str, &str, bool) + Send + Sync + 'static,

Create a middleware that calls a logging callback before and after tool execution.

The callback receives (tool_name, tool_call_id, is_start) where is_start is true before execution and false after.

Trait Implementations

impl AgentTool for ToolMiddleware

fn name(&self) -> &str

Unique routing key used to dispatch tool calls.

fn label(&self) -> &str

Human-readable display name for logging and UI.

fn description(&self) -> &str

Natural-language description included in the LLM prompt.

fn parameters_schema(&self) -> &Value

JSON Schema describing the tool’s input shape, used for validation.

fn metadata(&self) -> Option<ToolMetadata>

Optional organizational metadata (namespace, version).

fn requires_approval(&self) -> bool

Whether this tool requires user approval before execution. Default is false — tools execute immediately.

fn approval_context(&self, params: &Value) -> Option<Value>

Optional rich context for the approval UI.

fn auth_config(&self) -> Option<AuthConfig>

Optional authentication configuration for this tool.

fn execute( &self, tool_call_id: &str, params: Value, cancellation_token: CancellationToken, on_update: Option<Box<dyn Fn(AgentToolResult) + Send + Sync>>, state: Arc<RwLock<SessionState>>, credential: Option<ResolvedCredential>, ) -> ToolFuture<'_>

Execute the tool with validated parameters.

impl Debug for ToolMiddleware

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

Formats the value using the given formatter.