systemprompt-models 0.1.21

Shared data models and types for systemprompt.io OS
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
52

use anyhow::Result;
use async_trait::async_trait;
use std::collections::HashMap;
use std::sync::Arc;

use crate::ai::tools::McpTool;
use crate::execution::context::RequestContext;

#[derive(Debug, Clone)]
pub struct McpServerState {
    pub name: String,
    pub host: String,
    pub port: u16,
    pub status: String,
}

#[async_trait]
pub trait McpRegistry: Send + Sync {
    async fn list_servers(&self) -> Result<Vec<String>>;

    async fn find_server(&self, name: &str) -> Result<Option<McpServerState>>;

    async fn server_exists(&self, name: &str) -> Result<bool>;
}

#[async_trait]
pub trait McpToolProvider: Send + Sync {
    async fn list_tools(&self, server_name: &str, context: &RequestContext)
    -> Result<Vec<McpTool>>;

    async fn load_tools_for_servers(
        &self,
        server_names: &[String],
        context: &RequestContext,
    ) -> Result<HashMap<String, Vec<McpTool>>>;
}

#[async_trait]
pub trait McpDeploymentProvider: Send + Sync {
    async fn load_config(&self) -> Result<crate::ServicesConfig>;

    fn protocol_version(&self) -> &str;
}

#[async_trait]
pub trait McpProvider: McpRegistry + McpToolProvider + McpDeploymentProvider {}

pub type DynMcpRegistry = Arc<dyn McpRegistry>;

pub type DynMcpToolProvider = Arc<dyn McpToolProvider>;

pub type DynMcpDeploymentProvider = Arc<dyn McpDeploymentProvider>;