tau-agent-plugin 0.1.1

Plugin SDK for the tau agent — everything a plugin author needs in one crate
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

//! Tool execution abstraction.
//!
//! The `ToolExecutor` trait is the contract between "something that runs tools"
//! and "something that consumes tool results". Plugin authors don't implement it
//! directly (they speak the wire protocol), but consumers of plugin handles do
//! (`PluginExecutor` in server, `InProcessWorker` in worker).

use async_trait::async_trait;
use tau_agent_base::types::{ToolCall, ToolResultMessage};

/// Trait for tool execution (allows plugin-based or in-process).
#[async_trait]
pub trait ToolExecutor: Send {
    /// Execute a tool call.
    ///
    /// `cancel` is a shared cancellation flag. Long-running tools (bash) poll
    /// it and abort mid-execution when it becomes true; short tools (read,
    /// write, edit) may ignore it or check it once at the top.
    ///
    /// Implementations that cannot poll the flag directly (e.g. a plugin
    /// subprocess over an RPC channel) should spawn a background watcher
    /// that translates the flag into a [`PluginRequest::CancelToolCall`]
    /// sent to the plugin.
    async fn execute(
        &mut self,
        tool_call: &ToolCall,
        output_tx: &smol::channel::Sender<String>,
        cancel: &tau_agent_base::types::CancelToken,
    ) -> tau_agent_base::Result<ToolResultMessage>;
}