echo_agent 0.2.0

Production-grade AI Agent framework for Rust — ReAct engine, multi-agent, memory, streaming, MCP, IM channels, workflows
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! Tool execution subsystem
//!
//! Centralized management of tool registration/execution, Skill, Hook, MCP,
//! SubAgent, Sandbox, and other components directly related to tool invocation.

#[cfg(feature = "subagent")]
use crate::agent::subagent::SubagentRegistry;
#[cfg(feature = "mcp")]
use crate::mcp::McpManager;
use crate::sandbox::SandboxManager;
use crate::skills::SkillRegistry;
use crate::skills::hooks::HookRegistry;
use crate::skills::registry::SharedRegistry;
#[cfg(feature = "tasks")]
use crate::tasks::TaskManager;
use crate::tools::ToolManager;
use std::sync::Arc;

/// Tool execution subsystem
///
/// Aggregates all components directly related to tool invocation: tool registry,
/// Skill/Hook system, MCP connection management, SubAgent scheduling, and sandbox
/// environment.
pub(crate) struct ToolExecutionSubsystem {
    /// Tool registry (Arc for sharing with StreamRunner).
    pub(crate) tool_manager: Arc<ToolManager>,
    #[cfg(feature = "subagent")]
    pub(crate) subagent_registry: Arc<SubagentRegistry>,
    #[cfg(feature = "tasks")]
    pub(crate) task_manager: Arc<TaskManager>,
    pub(crate) skill_registry: SkillRegistry,
    pub(crate) progressive_skill_registry: Option<SharedRegistry>,
    pub(crate) hook_registry: Arc<tokio::sync::RwLock<HookRegistry>>,
    #[cfg(feature = "mcp")]
    pub(crate) mcp_manager: McpManager,
    pub(crate) sandbox_manager: Option<Arc<SandboxManager>>,
}

impl ToolExecutionSubsystem {
    /// Return a clone of the tool manager Arc (for StreamRunner construction).
    #[allow(dead_code)]
    pub(crate) fn tool_manager_arc(&self) -> Arc<ToolManager> {
        Arc::clone(&self.tool_manager)
    }

    #[cfg(feature = "mcp")]
    #[allow(dead_code)]
    pub(crate) fn mcp_manager_arc(&self) -> Option<Arc<McpManager>> {
        None // McpManager is not Arc-wrapped; use shared registry instead
    }

    #[cfg(feature = "subagent")]
    #[allow(dead_code)]
    pub(crate) fn subagent_registry(&self) -> Option<Arc<SubagentRegistry>> {
        Some(Arc::clone(&self.subagent_registry))
    }

    #[cfg(feature = "tasks")]
    pub(crate) fn task_manager(&self) -> Option<Arc<TaskManager>> {
        Some(Arc::clone(&self.task_manager))
    }

    #[allow(dead_code)]
    pub(crate) fn progressive_skill_registry(&self) -> Option<SharedRegistry> {
        self.progressive_skill_registry.clone()
    }
}