Struct StreamingToolExecutor 

pub struct StreamingToolExecutor { /* private fields */ }

Executes tools as they stream in with concurrency control.

Implementations

impl StreamingToolExecutor

pub fn new( executor: Arc<dyn Fn(String, Value, String) -> Pin<Box<dyn Future<Output = Result<ToolResult, AgentError>> + Send + Sync>> + Send + Sync>, tools_def: Vec<ToolDefinition>, ) -> (Self, UnboundedReceiver<ToolMessageUpdate>)

Create a new streaming tool executor.

pub fn add_tool(&self, name: String, id: String, args: Value)

Add a tool to the execution queue. Will start executing immediately if conditions allow.

pub async fn mark_complete(&self, tool_use_id: &str)

Mark a tool use as complete.

pub async fn get_is_concurrency_safe(&self, tool_use_id: &str) -> bool

Get a tool’s concurrency safety flag.

pub async fn has_unfinished_tools(&self) -> bool

Check if there are unfinished tools.

pub async fn has_executing_tools(&self) -> bool

Check if any tools are currently executing.

pub async fn discard(&self)

Discard all pending and in-progress tools.

pub async fn trigger_sibling_abort(&self)

Trigger sibling abort (called when Bash tool errors).

pub async fn set_tool_result( &self, tool_call_id: String, result: Result<ToolResult, AgentError>, )

Set tool result from external execution.

pub async fn get_completed_results(&self) -> Vec<ToolMessageUpdate>

Get completed results that haven’t been yielded yet. Yields progress messages immediately, then results in order. Stops yielding when encountering a non-concurrency-safe executing tool.

pub async fn get_remaining_results( &self, result_rx: &mut UnboundedReceiver<ToolMessageUpdate>, ) -> Vec<ToolMessageUpdate>

Wait for remaining tools and collect their results.

pub async fn discard_sync(&self)

Discard all pending and in-progress tools.