ares-server 0.9.1

ARES agent server with multi-provider LLM support, tool calling, RAG, and MCP integration
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

//! Adapter from [`Execute`](ares_agent::execution::Execute)
//! to [`ares_mcp::AgentRunner`].

use std::sync::Arc;

use cordis::Context;

/// Runs MCP `ares_run_agent` through Cordis `Execute`.
pub struct ExecutionAgentRunner {
    pub ctx: Arc<Context>,
}

impl ExecutionAgentRunner {
    /// Context already holding Execute.
    pub fn new(ctx: Arc<Context>) -> Self {
        Self { ctx }
    }

    /// Provide Execute::new() if missing, then wrap.
    pub fn attach(ctx: Arc<Context>) -> Self {
        if ctx.get::<ares_agent::execution::Execute>().is_none() {
            ctx.provide_arc(Arc::new(ares_agent::execution::Execute::new()));
        }
        Self { ctx }
    }

    /// Build a root context, attach tenant DB, provide the execution service.
    #[cfg(feature = "postgres")]
    pub fn with_tenant_db(tenant_db: Arc<ares_store::TenantDb>) -> Self {
        let ctx = Context::new_root();
        ctx.provide_arc(tenant_db);
        ctx.provide_arc(Arc::new(ares_agent::execution::Execute::new()));
        Self { ctx }
    }

    /// Context used by this runner (for tests and MCP stdio wiring).
    pub fn context(&self) -> &Arc<Context> {
        &self.ctx
    }
}

#[async_trait::async_trait]
impl ares_mcp::AgentRunner for ExecutionAgentRunner {
    async fn run_agent(
        &self,
        input: &ares_mcp::tools::RunAgentInput,
    ) -> Result<ares_mcp::tools::RunAgentOutput, String> {
        let exec = self
            .ctx
            .get::<ares_agent::execution::Execute>()
            .ok_or_else(|| "Execute not provided".to_string())?;
        let req = ares_agent::execution::AgentRequest {
            agent_name: input.agent_name.clone(),
            message: input.message.clone(),
            history: vec![],
            ctx_provider: None,
        };
        let exec_result = exec.run(&req, &self.ctx).await.map_err(|e| e.to_string())?;
        Ok(ares_mcp::tools::RunAgentOutput {
            response: exec_result.response.content,
            agent: input.agent_name.clone(),
            context_id: input.context_id.clone().unwrap_or_default(),
            sources: None,
        })
    }
}

#[cfg(test)]
mod tests {
    use super::ExecutionAgentRunner;
    use std::sync::Arc;

    use ares_mcp::AgentRunner;

    #[tokio::test]
    async fn execution_agent_runner_errors_without_service() {
        let ctx = cordis::Context::new_root();
        let runner = ExecutionAgentRunner { ctx };
        let err = runner
            .run_agent(&ares_mcp::tools::RunAgentInput {
                agent_name: "router".into(),
                message: "hi".into(),
                context_id: None,
            })
            .await
            .unwrap_err();
        assert!(err.contains("Execute"));
    }

    #[test]
    fn attach_provides_execution_service() {
        let runner = ExecutionAgentRunner::attach(cordis::Context::new_root());
        assert!(runner
            .context()
            .get::<ares_agent::execution::Execute>()
            .is_some());
    }

    #[test]
    fn attach_keeps_existing_service() {
        let ctx = cordis::Context::new_root();
        let existing = Arc::new(ares_agent::execution::Execute::new());
        ctx.provide_arc(existing.clone());
        let runner = ExecutionAgentRunner::attach(ctx);
        let got = runner
            .context()
            .get::<ares_agent::execution::Execute>()
            .expect("existing service kept");
        assert!(Arc::ptr_eq(&existing, &got));
    }
}