aimdb-mcp 0.7.0

Model Context Protocol (MCP) server for AimDB - enables LLM-powered introspection
Documentation 
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 tools implementation
//!
//! Tools for discovering and interacting with AimDB instances.

use crate::connection::ConnectionPool;
use once_cell::sync::OnceCell;

pub mod architecture;
pub mod graph;
pub mod instance;
pub mod record;
pub mod schema;

// Global connection pool (initialized once)
static CONNECTION_POOL: OnceCell<ConnectionPool> = OnceCell::new();

/// Initialize the connection pool for tools
pub fn init_connection_pool(pool: ConnectionPool) {
    CONNECTION_POOL.set(pool).ok();
}

/// Get the connection pool
pub(crate) fn connection_pool() -> Option<&'static ConnectionPool> {
    CONNECTION_POOL.get()
}

/// Resolve the socket path from an explicit argument or the `AIMDB_SOCKET` env var.
///
/// Returns an error if neither is set.
pub(crate) fn resolve_socket_path(explicit: Option<String>) -> crate::error::McpResult<String> {
    explicit
        .or_else(|| std::env::var("AIMDB_SOCKET").ok())
        .filter(|s| !s.is_empty())
        .ok_or_else(|| {
            crate::error::McpError::InvalidParams(
                "Missing socket_path (pass it explicitly or set AIMDB_SOCKET env var)".into(),
            )
        })
}

// Re-export tool functions
pub use architecture::{
    get_architecture, get_buffer_metrics, propose_add_binary, propose_add_connector,
    propose_add_record, propose_add_task, propose_modify_buffer, propose_modify_fields,
    propose_modify_key_variants, remove_binary, remove_record, remove_task, rename_record,
    reset_session, resolve_proposal, save_memory, validate_against_instance,
};
pub use graph::{graph_edges, graph_nodes, graph_topo_order};
pub use instance::{discover_instances, get_instance_info};
pub use record::{drain_record, get_record, list_records, set_record};
pub use schema::query_schema;