dirge-agent 0.7.6

Minimalistic coding agent written in Rust, optimized for memory footprint and performance
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

//! /mcp handler.

use crate::ui::slash::{SlashCtx, c_agent, c_error, c_result};

#[cfg(feature = "mcp")]
pub(crate) async fn cmd_mcp(ctx: &mut SlashCtx<'_>, parts: &[&str]) -> anyhow::Result<()> {
    let Some(mgr) = ctx.mcp_manager else {
        ctx.renderer
            .write_line("no MCP servers configured", c_agent())?;
        return Ok(());
    };
    let connections = mgr.connections_snapshot();
    if connections.is_empty() {
        ctx.renderer
            .write_line("no MCP servers connected", c_agent())?;
    } else if parts.len() == 1 {
        ctx.renderer.write_line("MCP servers:", c_agent())?;
        for (server_name, conn) in &connections {
            match crate::extras::mcp::client::list_tools(conn).await {
                Ok(tools) => {
                    ctx.renderer.write_line(
                        &format!("  {} ({} tools)", server_name, tools.len()),
                        c_result(),
                    )?;
                }
                Err(e) => {
                    ctx.renderer
                        .write_line(&format!("  {} (error: {})", server_name, e), c_error())?;
                }
            }
        }
    } else {
        let name = parts[1].trim();
        if let Some(conn) = connections.iter().find(|(n, _)| n == name).map(|(_, c)| c) {
            match crate::extras::mcp::client::list_tools(conn).await {
                Ok(tools) => {
                    if tools.is_empty() {
                        ctx.renderer
                            .write_line(&format!("server '{}' has no tools", name), c_agent())?;
                    } else {
                        ctx.renderer
                            .write_line(&format!("tools on '{}':", name), c_agent())?;
                        for tool in &tools {
                            let desc = tool.description.as_deref().unwrap_or("");
                            ctx.renderer
                                .write_line(&format!("  {}  {}", tool.name, desc), c_result())?;
                        }
                    }
                }
                Err(e) => {
                    ctx.renderer.write_line(
                        &format!("error listing tools on '{}': {}", name, e),
                        c_error(),
                    )?;
                }
            }
        } else {
            ctx.renderer
                .write_line(&format!("unknown MCP server: '{}'", name), c_error())?;
        }
    }
    Ok(())
}