systemprompt-api 0.63.0

Axum-based HTTP server and API gateway for systemprompt.io AI governance infrastructure. Exposes governed agents, MCP, A2A, and admin endpoints with rate limiting and RBAC.
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
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

//! Axum handler adapters for the MCP and agent proxy routes, each delegating
//! to [`ProxyEngine::proxy_request`]. The service-name path segment is parsed
//! once here; a malformed one answers 400 `invalid_identifier`.
//!
//! Copyright (c) systemprompt.io — Business Source License 1.1.
//! See <https://systemprompt.io> for licensing details.

use axum::body::Body;
use axum::extract::{Path, Request, State};
use axum::response::{IntoResponse, Response};
use systemprompt_identifiers::ServiceName;
use systemprompt_runtime::AppContext;

use super::{ProxyEngine, ProxyKind, ProxyTarget};
use crate::services::proxy::backend::ProxyError;

impl ProxyEngine {
    pub async fn handle_mcp_request_with_path(
        &self,
        path_params: Path<(String, String)>,
        State(ctx): State<AppContext>,
        request: Request<Body>,
    ) -> Response<Body> {
        let Path((raw_service_name, path)) = path_params;
        let service_name = match ServiceName::try_new(raw_service_name) {
            Ok(name) => name,
            Err(error) => return ProxyError::from(error).into_response(),
        };
        let target = ProxyTarget {
            service_name: &service_name,
            path: &path,
            kind: ProxyKind::Mcp,
        };
        match self.proxy_request(target, request, ctx).await {
            Ok(response) => response,
            Err(e) => e.into_response(),
        }
    }

    pub async fn handle_agent_request(
        &self,
        path_params: Path<(String,)>,
        State(ctx): State<AppContext>,
        request: Request<Body>,
    ) -> Response<Body> {
        let Path((raw_service_name,)) = path_params;
        self.agent_request(&raw_service_name, "", ctx, request)
            .await
    }

    pub async fn handle_agent_request_with_path(
        &self,
        path_params: Path<(String, String)>,
        State(ctx): State<AppContext>,
        request: Request<Body>,
    ) -> Response<Body> {
        let Path((raw_service_name, path)) = path_params;
        self.agent_request(&raw_service_name, &path, ctx, request)
            .await
    }

    async fn agent_request(
        &self,
        raw_service_name: &str,
        path: &str,
        ctx: AppContext,
        request: Request<Body>,
    ) -> Response<Body> {
        let service_name = match ServiceName::try_new(raw_service_name) {
            Ok(name) => name,
            Err(error) => return ProxyError::from(error).into_response(),
        };
        let target = ProxyTarget {
            service_name: &service_name,
            path,
            kind: ProxyKind::Agent,
        };
        match self.proxy_request(target, request, ctx).await {
            Ok(response) => response,
            Err(e) => e.to_status_code().into_response(),
        }
    }
}