oxios-kernel 1.0.0

Oxios kernel: supervisor, event bus, state store
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

//! MCP API — external tool server bridge.

use crate::mcp::{self, McpBridge, McpServer, McpToolCallResult};
use crate::tools::tool_types::ToolDef;
use std::sync::Arc;

/// MCP system calls.
pub struct McpApi {
    pub(crate) mcp_bridge: Arc<McpBridge>,
}

impl McpApi {
    /// Create a new McpApi.
    pub fn new(mcp_bridge: Arc<McpBridge>) -> Self {
        Self { mcp_bridge }
    }
    /// List registered MCP servers.
    pub fn list_servers(&self) -> Vec<String> {
        self.mcp_bridge.servers()
    }

    /// Get MCP server info.
    pub fn get_server(&self, name: &str) -> Option<McpServer> {
        self.mcp_bridge.get_server(name)
    }

    /// Register an MCP server.
    pub fn register_server(&self, server: McpServer) {
        self.mcp_bridge.register_server(server);
    }

    /// Initialize a specific MCP server.
    pub async fn init_server(&self, name: &str) -> anyhow::Result<()> {
        self.mcp_bridge.initialize_server(name).await
    }

    /// Get MCP client status.
    pub async fn client_status(&self, name: &str) -> Option<bool> {
        if let Some(client) = self.mcp_bridge.client(name).await {
            Some(client.is_initialized().await)
        } else {
            None
        }
    }

    /// List all MCP tools as ToolDefs.
    pub async fn list_tools(&self) -> anyhow::Result<Vec<ToolDef>> {
        mcp::list_tool_defs(&self.mcp_bridge).await
    }

    /// Get cached tools for a server as ToolDefs.
    pub async fn cached_tools(&self, server_name: &str) -> Option<Vec<ToolDef>> {
        mcp::cached_tool_defs(&self.mcp_bridge, server_name).await
    }

    /// Call an MCP tool.
    pub async fn call_tool(
        &self,
        server: &str,
        tool: &str,
        arguments: serde_json::Value,
    ) -> anyhow::Result<McpToolCallResult> {
        self.mcp_bridge.call_tool(server, tool, arguments).await
    }

    /// MCP bridge reference.
    pub fn bridge(&self) -> &Arc<McpBridge> {
        &self.mcp_bridge
    }

    /// Number of configured MCP servers.
    pub fn server_count(&self) -> usize {
        self.mcp_bridge.servers().len()
    }

    /// Remove (disconnect and delete) an MCP server.
    pub async fn remove_server(&self, name: &str) -> anyhow::Result<()> {
        self.mcp_bridge.remove_server(name).await
    }

    /// Toggle MCP server enabled/disabled. Returns the new enabled state.
    pub async fn toggle_server(&self, name: &str) -> anyhow::Result<bool> {
        self.mcp_bridge.toggle_server(name).await
    }

    /// Shutdown all MCP servers.
    pub async fn shutdown_all(&self) -> anyhow::Result<()> {
        self.mcp_bridge.shutdown_all().await
    }
}