aidaemon 0.9.32

A personal AI agent that runs as a background daemon, accessible via Telegram, Slack, or Discord, with tool use, MCP integration, and persistent memory
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

use super::*;
use crate::testing::MockTool;
use crate::traits::ToolRole;

/// Mock tool that returns a specific ToolRole.
struct MockRoleTool {
    tool_name: String,
    role: ToolRole,
}

impl MockRoleTool {
    fn new(name: &str, role: ToolRole) -> Self {
        Self {
            tool_name: name.to_string(),
            role,
        }
    }
}

#[async_trait::async_trait]
impl Tool for MockRoleTool {
    fn name(&self) -> &str {
        &self.tool_name
    }
    fn description(&self) -> &str {
        "mock"
    }
    fn schema(&self) -> Value {
        json!({
            "name": self.tool_name,
            "description": "mock",
            "parameters": { "type": "object", "properties": {} }
        })
    }
    fn tool_role(&self) -> ToolRole {
        self.role
    }
    async fn call(&self, _args: &str) -> anyhow::Result<String> {
        Ok("ok".to_string())
    }
}

#[test]
fn test_tool_scoping_task_lead() {
    // Simulate tool filtering for task lead role
    let tools: Vec<Arc<dyn Tool>> = vec![
        Arc::new(MockRoleTool::new("terminal", ToolRole::Action)),
        Arc::new(MockRoleTool::new("web_search", ToolRole::Action)),
        Arc::new(MockRoleTool::new("system_info", ToolRole::Universal)),
        Arc::new(MockRoleTool::new("remember_fact", ToolRole::Universal)),
        Arc::new(MockRoleTool::new("plan_manager", ToolRole::Management)),
    ];

    // Task lead filter: Management + Universal only
    let tl_tools: Vec<String> = tools
        .iter()
        .filter(|t| t.name() != "spawn_agent")
        .filter(|t| matches!(t.tool_role(), ToolRole::Management | ToolRole::Universal))
        .map(|t| t.name().to_string())
        .collect();

    assert!(tl_tools.contains(&"system_info".to_string()));
    assert!(tl_tools.contains(&"remember_fact".to_string()));
    assert!(tl_tools.contains(&"plan_manager".to_string()));
    assert!(!tl_tools.contains(&"terminal".to_string()));
    assert!(!tl_tools.contains(&"web_search".to_string()));
    assert_eq!(tl_tools.len(), 3);
}

#[test]
fn test_tool_scoping_executor() {
    let tools: Vec<Arc<dyn Tool>> = vec![
        Arc::new(MockRoleTool::new("terminal", ToolRole::Action)),
        Arc::new(MockRoleTool::new("web_search", ToolRole::Action)),
        Arc::new(MockRoleTool::new("system_info", ToolRole::Universal)),
        Arc::new(MockRoleTool::new("remember_fact", ToolRole::Universal)),
        Arc::new(MockRoleTool::new("plan_manager", ToolRole::Management)),
    ];

    // Executor filter: Action + Universal only
    let exec_tools: Vec<String> = tools
        .iter()
        .filter(|t| t.name() != "spawn_agent")
        .filter(|t| matches!(t.tool_role(), ToolRole::Action | ToolRole::Universal))
        .map(|t| t.name().to_string())
        .collect();

    assert!(exec_tools.contains(&"terminal".to_string()));
    assert!(exec_tools.contains(&"web_search".to_string()));
    assert!(exec_tools.contains(&"system_info".to_string()));
    assert!(exec_tools.contains(&"remember_fact".to_string()));
    assert!(!exec_tools.contains(&"plan_manager".to_string()));
    assert_eq!(exec_tools.len(), 4);
}

#[test]
fn test_tool_scoping_legacy_no_filter() {
    let tools: Vec<Arc<dyn Tool>> = vec![
        Arc::new(MockRoleTool::new("terminal", ToolRole::Action)),
        Arc::new(MockRoleTool::new("system_info", ToolRole::Universal)),
        Arc::new(MockRoleTool::new("plan_manager", ToolRole::Management)),
        Arc::new(MockRoleTool::new("spawn_agent", ToolRole::Action)),
    ];

    // Legacy: filter out spawn_agent only, keep everything else
    let legacy_tools: Vec<String> = tools
        .iter()
        .filter(|t| t.name() != "spawn_agent")
        .map(|t| t.name().to_string())
        .collect();

    assert_eq!(legacy_tools.len(), 3);
    assert!(legacy_tools.contains(&"terminal".to_string()));
    assert!(legacy_tools.contains(&"system_info".to_string()));
    assert!(legacy_tools.contains(&"plan_manager".to_string()));
}

#[test]
fn test_agent_role_default() {
    assert_eq!(AgentRole::Orchestrator, AgentRole::Orchestrator);
    assert_ne!(AgentRole::TaskLead, AgentRole::Executor);
}

#[test]
fn test_tool_role_default() {
    // Verify that MockTool (from testing.rs) defaults to Action
    let mock = MockTool::new("test", "desc", "result");
    assert_eq!(mock.tool_role(), ToolRole::Action);
}

#[test]
fn test_system_info_tool_is_universal() {
    let tool = crate::tools::SystemInfoTool;
    assert_eq!(tool.tool_role(), ToolRole::Universal);
}