systemprompt-runtime 0.64.0

Application runtime for systemprompt.io AI governance infrastructure. AppContext, lifecycle builder, extension registry, and module wiring for the MCP governance pipeline.
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

//! Aggregated AI trace assembly across request, tool, and log rows.
//!
//! Copyright (c) systemprompt.io — Business Source License 1.1.
//! See <https://systemprompt.io> for licensing details.

use systemprompt_identifiers::{AiRequestId, ContextId, McpExecutionId, TaskId};

use super::TraceError;
use super::models::{
    AiRequestInfo, ConversationMessage, ExecutionStep, McpToolExecution, TaskArtifact, TaskInfo,
    ToolLogEntry,
};
use super::repository::TraceRepository;

pub(super) type Result<T> = std::result::Result<T, TraceError>;

#[derive(Debug, Clone)]
pub struct AiTraceService {
    repository: TraceRepository,
}

impl AiTraceService {
    pub const fn new(repository: TraceRepository) -> Self {
        Self { repository }
    }

    pub async fn resolve_task_id(&self, partial_id: &str) -> Result<TaskId> {
        self.repository
            .resolve_task_id(partial_id)
            .await?
            .map(TaskId::new)
            .ok_or_else(|| TraceError::TaskNotFound {
                partial_id: partial_id.to_owned(),
            })
    }

    pub async fn get_task_info(&self, task_id: &TaskId) -> Result<TaskInfo> {
        self.repository.fetch_task_info(task_id).await
    }

    pub async fn get_user_input(&self, task_id: &TaskId) -> Result<Option<String>> {
        self.repository.fetch_user_input(task_id).await
    }

    pub async fn get_agent_response(&self, task_id: &TaskId) -> Result<Option<String>> {
        self.repository.fetch_agent_response(task_id).await
    }

    pub async fn get_execution_steps(&self, task_id: &TaskId) -> Result<Vec<ExecutionStep>> {
        self.repository.fetch_execution_steps(task_id).await
    }

    pub async fn get_ai_requests(&self, task_id: &TaskId) -> Result<Vec<AiRequestInfo>> {
        self.repository.fetch_ai_requests(task_id).await
    }

    pub async fn get_system_prompt(&self, request_id: &AiRequestId) -> Result<Option<String>> {
        self.repository.fetch_system_prompt(request_id).await
    }

    pub async fn get_conversation_messages(
        &self,
        request_id: &AiRequestId,
    ) -> Result<Vec<ConversationMessage>> {
        self.repository
            .fetch_conversation_messages(request_id)
            .await
    }

    pub async fn get_mcp_executions(
        &self,
        task_id: &TaskId,
        context_id: &ContextId,
    ) -> Result<Vec<McpToolExecution>> {
        self.repository
            .fetch_mcp_executions(task_id, context_id)
            .await
    }

    pub async fn get_mcp_linked_ai_requests(
        &self,
        mcp_execution_id: &McpExecutionId,
    ) -> Result<Vec<AiRequestInfo>> {
        self.repository
            .fetch_mcp_linked_ai_requests(mcp_execution_id)
            .await
    }

    pub async fn get_ai_request_message_previews(
        &self,
        request_id: &AiRequestId,
    ) -> Result<Vec<ConversationMessage>> {
        self.repository
            .fetch_ai_request_message_previews(request_id)
            .await
    }

    pub async fn get_tool_logs(
        &self,
        task_id: &TaskId,
        context_id: &ContextId,
    ) -> Result<Vec<ToolLogEntry>> {
        self.repository.fetch_tool_logs(task_id, context_id).await
    }

    pub async fn get_task_artifacts(
        &self,
        task_id: &TaskId,
        context_id: &ContextId,
    ) -> Result<Vec<TaskArtifact>> {
        self.repository
            .fetch_task_artifacts(task_id, context_id)
            .await
    }
}