systemprompt-mcp 0.0.13

Core MCP (Model Context Protocol) functionality for systemprompt.io OS
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

use anyhow::Result;
use systemprompt_identifiers::{AgentName, ContextId, SessionId, TaskId, TraceId, UserId};
use systemprompt_models::RequestContext;
use systemprompt_traits::{ToolContext, ToolProviderError};

use crate::services::deployment::DeploymentService;

pub fn create_request_context(ctx: &ToolContext) -> Result<RequestContext, ToolProviderError> {
    let session_id = ctx
        .session_id
        .as_ref()
        .map_or_else(SessionId::system, |s| SessionId::new(s.clone()));

    let trace_id = ctx
        .trace_id
        .as_ref()
        .map_or_else(TraceId::generate, |t| TraceId::new(t.clone()));

    let context_id = ctx
        .headers
        .get("x-context-id")
        .filter(|s| !s.is_empty())
        .map(|s| ContextId::new(s.clone()))
        .ok_or_else(|| {
            ToolProviderError::ConfigurationError(
                "Missing x-context-id header - context must be propagated from parent request"
                    .into(),
            )
        })?;

    let agent_name = ctx
        .headers
        .get("x-agent-name")
        .filter(|s| !s.is_empty())
        .map(|s| AgentName::new(s.clone()))
        .ok_or_else(|| {
            ToolProviderError::ConfigurationError(
                "Missing x-agent-name header - agent context must be propagated from parent \
                 request"
                    .into(),
            )
        })?;

    let mut request_ctx = RequestContext::new(session_id, trace_id, context_id, agent_name)
        .with_auth_token(ctx.auth_token.clone());

    if let Some(user_id) = ctx.headers.get("x-user-id").filter(|s| !s.is_empty()) {
        request_ctx = request_ctx.with_user_id(UserId::new(user_id.clone()));
    }

    if let Some(task_id) = ctx.headers.get("x-task-id").filter(|s| !s.is_empty()) {
        request_ctx = request_ctx.with_task_id(TaskId::new(task_id.clone()));
    }

    if let Some(ai_tool_call_id) = &ctx.ai_tool_call_id {
        request_ctx = request_ctx.with_ai_tool_call_id(ai_tool_call_id.clone().into());
    }

    Ok(request_ctx)
}

pub fn load_agent_servers(agent_name: &str) -> Result<Vec<String>> {
    let config = DeploymentService::load_config()?;
    let agent_name_type = AgentName::new(agent_name);

    let agent = config
        .agents
        .get(agent_name_type.as_str())
        .ok_or_else(|| anyhow::anyhow!("Agent '{agent_name}' not found in services.yaml"))?;

    Ok(agent.metadata.mcp_servers.clone())
}