sage-cli 0.9.0

Command-line interface for Sage Agent
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

//! MCP registry building for the unified command

use sage_core::config::Config;
use sage_core::error::SageResult;
use sage_core::mcp::registry::McpRegistry;
use sage_core::mcp::transport::TransportConfig;

/// Build MCP registry from configuration
pub async fn build_mcp_registry_from_config(config: &Config) -> SageResult<McpRegistry> {
    let registry = McpRegistry::new();

    for (name, server_config) in &config.mcp.servers {
        if !server_config.enabled {
            continue;
        }

        let transport_config = match server_config.transport.as_str() {
            "stdio" => {
                let command = server_config.command.clone().unwrap_or_default();
                let args = server_config.args.clone();
                let env = server_config.env.clone();
                TransportConfig::Stdio { command, args, env }
            }
            "http" | "https" | "sse" => {
                let base_url = server_config.url.clone().unwrap_or_default();
                let headers = server_config.headers.clone();
                TransportConfig::Http { base_url, headers }
            }
            _ => {
                tracing::warn!("Unsupported MCP transport type: {}", server_config.transport);
                continue;
            }
        };

        match registry.register_server(name, transport_config).await {
            Ok(server_info) => {
                tracing::info!(
                    "Connected to MCP server '{}': {} v{}",
                    name,
                    server_info.name,
                    server_info.version
                );
            }
            Err(e) => {
                tracing::error!("Failed to connect to MCP server '{}': {}", name, e);
            }
        }
    }

    Ok(registry)
}