forked from kent/consciousness
92ca2bf2c8
Provenance now flows as a function parameter through the entire tool dispatch chain: thought::dispatch → memory::dispatch → store methods. Removed task_local (TASK_AGENT), thread_local (TASK_PHASE), and env var (POC_PROVENANCE) from the tool dispatch path. The env var remains only as a fallback for non-tool paths (CLI commands, digest). Phase names are passed from knowledge.rs → llm.rs → api.rs, and api.rs updates the provenance string between steps. No globals needed.
59 lines
1.9 KiB
Rust
59 lines
1.9 KiB
Rust
// tools/mod.rs — Agent-specific tool dispatch
|
|
//
|
|
// Shared tools (memory, files, bash, journal) live in thought/.
|
|
// This module handles agent-specific tools (control, vision,
|
|
// working_stack) and delegates everything else to thought::dispatch.
|
|
|
|
mod control;
|
|
mod vision;
|
|
pub mod working_stack;
|
|
|
|
// Re-export shared infrastructure from thought
|
|
pub use crate::thought::{ToolOutput, ProcessTracker, truncate_output};
|
|
pub use crate::thought::memory;
|
|
pub use crate::thought::journal;
|
|
|
|
use crate::agent::types::ToolDef;
|
|
|
|
/// Dispatch a tool call by name.
|
|
///
|
|
/// Tries agent-specific tools first (control, vision), then
|
|
/// delegates to thought::dispatch for shared tools.
|
|
///
|
|
/// Note: working_stack is handled in runner.rs before reaching this
|
|
/// function (it needs mutable context access).
|
|
pub async fn dispatch(
|
|
name: &str,
|
|
args: &serde_json::Value,
|
|
tracker: &ProcessTracker,
|
|
) -> ToolOutput {
|
|
// Agent-specific tools that return Result<ToolOutput> directly
|
|
let rich_result = match name {
|
|
"pause" => Some(control::pause(args)),
|
|
"switch_model" => Some(control::switch_model(args)),
|
|
"yield_to_user" => Some(control::yield_to_user(args)),
|
|
"view_image" => Some(vision::view_image(args)),
|
|
_ => None,
|
|
};
|
|
if let Some(result) = rich_result {
|
|
return result.unwrap_or_else(ToolOutput::error);
|
|
}
|
|
|
|
// Delegate to shared thought layer (poc-agent uses default provenance)
|
|
if let Some(output) = crate::thought::dispatch(name, args, tracker, None).await {
|
|
return output;
|
|
}
|
|
|
|
ToolOutput::error(format!("Unknown tool: {}", name))
|
|
}
|
|
|
|
/// Return all tool definitions (agent-specific + shared).
|
|
pub fn definitions() -> Vec<ToolDef> {
|
|
let mut defs = vec![
|
|
vision::definition(),
|
|
working_stack::definition(),
|
|
];
|
|
defs.extend(control::definitions());
|
|
defs.extend(crate::thought::all_definitions());
|
|
defs
|
|
}
|