systemprompt-agent 0.5.0

Agent-to-Agent (A2A) protocol for systemprompt.io AI governance: streaming, JSON-RPC models, task lifecycle, .well-known discovery, and governed agent orchestration.
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

mod operations;
mod verification;

use std::sync::Arc;
use systemprompt_database::DbPool;
use systemprompt_models::AppPaths;
use systemprompt_traits::StartupEventSender;

use crate::services::agent_orchestration::database::AgentDatabaseService;
use crate::services::agent_orchestration::event_bus::AgentEventBus;
use crate::services::agent_orchestration::events::AgentEvent;
use crate::services::agent_orchestration::{OrchestrationError, OrchestrationResult};

#[derive(Debug)]
pub struct AgentLifecycle {
    pub(crate) db_service: AgentDatabaseService,
    pub(crate) event_bus: Option<Arc<AgentEventBus>>,
    pub(crate) app_paths: Arc<AppPaths>,
}

impl AgentLifecycle {
    pub fn new(db_pool: &DbPool, app_paths: Arc<AppPaths>) -> crate::error::AgentResult<Self> {
        use crate::repository::agent_service::AgentServiceRepository;

        let agent_service_repo = AgentServiceRepository::new(db_pool)
            .map_err(|e| crate::error::AgentError::Other(anyhow::anyhow!(e.to_string())))?;
        let db_service = AgentDatabaseService::new(agent_service_repo)
            .map_err(|e| crate::error::AgentError::Other(anyhow::anyhow!(e.to_string())))?;

        Ok(Self {
            db_service,
            event_bus: None,
            app_paths,
        })
    }

    pub fn with_event_bus(mut self, event_bus: Arc<AgentEventBus>) -> Self {
        self.event_bus = Some(event_bus);
        self
    }

    pub(crate) fn publish_event(&self, event: AgentEvent) {
        if let Some(ref bus) = self.event_bus {
            bus.publish(event);
        }
    }
}

pub async fn start_agent(
    pool: &DbPool,
    app_paths: Arc<AppPaths>,
    agent_name: &str,
    events: Option<&StartupEventSender>,
) -> OrchestrationResult<String> {
    let lifecycle = AgentLifecycle::new(pool, app_paths)
        .map_err(|e| OrchestrationError::Generic(anyhow::anyhow!(e.to_string())))?;
    lifecycle.start_agent(agent_name, events).await
}

pub async fn enable_agent(
    pool: &DbPool,
    app_paths: Arc<AppPaths>,
    agent_name: &str,
    events: Option<&StartupEventSender>,
) -> OrchestrationResult<String> {
    let lifecycle = AgentLifecycle::new(pool, app_paths)
        .map_err(|e| OrchestrationError::Generic(anyhow::anyhow!(e.to_string())))?;
    lifecycle.enable_agent(agent_name, events).await
}

pub async fn disable_agent(
    pool: &DbPool,
    app_paths: Arc<AppPaths>,
    agent_name: &str,
) -> OrchestrationResult<()> {
    let lifecycle = AgentLifecycle::new(pool, app_paths)
        .map_err(|e| OrchestrationError::Generic(anyhow::anyhow!(e.to_string())))?;
    lifecycle.disable_agent(agent_name).await
}

pub async fn restart_agent(
    pool: &DbPool,
    app_paths: Arc<AppPaths>,
    agent_name: &str,
    events: Option<&StartupEventSender>,
) -> OrchestrationResult<String> {
    let lifecycle = AgentLifecycle::new(pool, app_paths)
        .map_err(|e| OrchestrationError::Generic(anyhow::anyhow!(e.to_string())))?;
    lifecycle.restart_agent(agent_name, events).await
}