Struct ToolDispatcher 

pub struct ToolDispatcher { /* private fields */ }

Routes function calls to the right tool implementation.

Implementations

impl ToolDispatcher

pub fn new() -> Self

Create a new empty tool dispatcher with the default 30-second timeout.

Examples

use gemini_adk_rs::tool::{ToolDispatcher, SimpleTool};
use serde_json::json;

let mut dispatcher = ToolDispatcher::new();
dispatcher.register(SimpleTool::new(
    "echo", "Echo input", None,
    |args| async move { Ok(args) },
));

pub fn with_timeout(self, timeout: Duration) -> Self

Set the default timeout for tool calls.

pub fn default_timeout(&self) -> Duration

Returns the configured default timeout.

pub fn register(&mut self, tool: impl ToolFunction)

Register a tool that implements ToolFunction.

pub fn register_function(&mut self, tool: Arc<dyn ToolFunction>)

Register a regular function tool (pre-wrapped in Arc).

pub fn register_streaming(&mut self, tool: Arc<dyn StreamingTool>)

Register a streaming tool.

pub fn register_input_streaming(&mut self, tool: Arc<dyn InputStreamingTool>)

Register an input-streaming tool.

pub fn get_tool(&self, name: &str) -> Option<&ToolKind>

Get a tool by name (for introspection/streaming tool spawning).

pub fn classify(&self, name: &str) -> Option<ToolClass>

Classify a tool by name.

pub async fn call_function( &self, name: &str, args: Value, ) -> Result<Value, ToolError>

Call a regular function tool by name, using the default timeout.

pub async fn call_function_with_timeout( &self, name: &str, args: Value, timeout: Duration, ) -> Result<Value, ToolError>

Call a regular function tool by name with an explicit timeout.

If the tool does not complete within the given duration, its future is dropped (cancelling it) and ToolError::Timeout is returned.

pub async fn call_function_with_cancel( &self, name: &str, args: Value, cancel: CancellationToken, ) -> Result<Value, ToolError>

Call a regular function tool by name, racing against a cancellation token.

If the token is cancelled before the tool completes, its future is dropped and ToolError::Cancelled is returned.

pub fn build_response( call: &FunctionCall, result: Result<Value, ToolError>, ) -> FunctionResponse

Build a FunctionResponse from a FunctionCall result.

pub async fn cancel_streaming(&self, name: &str)

Cancel a streaming tool by name.

pub async fn cancel_by_ids(&self, ids: &[String])

Cancel streaming tools by IDs.

pub fn to_tool_declarations(&self) -> Vec<Tool>

Generate Tool declarations for the setup message.

Results are cached after first computation. The cache is invalidated when tools are registered via register*() methods.

pub fn len(&self) -> usize

Number of registered tools.

pub fn is_empty(&self) -> bool

Whether no tools are registered.

Trait Implementations

impl Default for ToolDispatcher

fn default() -> Self

Returns the “default value” for a type.

impl ToolProvider for ToolDispatcher

fn declarations(&self) -> Vec<Tool>

Return tool declarations for the setup message.