systemprompt-agent 0.2.0

Core Agent protocol module for systemprompt.io
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

use super::{AgentInfo, AgentOrchestrator};
use crate::services::agent_orchestration::{
    AgentStatus, OrchestrationResult, ValidationReport, monitor,
};
use systemprompt_identifiers::AgentId;

impl AgentOrchestrator {
    pub async fn get_detailed_status(&self) -> OrchestrationResult<Vec<AgentInfo>> {
        self.get_comprehensive_agent_info().await
    }

    pub(super) async fn get_comprehensive_agent_info(&self) -> OrchestrationResult<Vec<AgentInfo>> {
        let agents = self.db_service.list_all_agents().await?;
        let mut agent_info = Vec::new();

        for (agent_id, status) in agents {
            if let Ok(config) = self.db_service.get_agent_config(&agent_id).await {
                agent_info.push(AgentInfo {
                    id: AgentId::new(agent_id),
                    name: config.name,
                    status,
                    port: config.port,
                });
            } else {
                let port = match status {
                    AgentStatus::Running { port, .. } => port,
                    AgentStatus::Failed { .. } => 8000,
                };
                agent_info.push(AgentInfo {
                    id: AgentId::new(agent_id.clone()),
                    name: "Unknown".to_string(),
                    status,
                    port,
                });
            }
        }

        Ok(agent_info)
    }

    pub async fn list_all(&self) -> OrchestrationResult<Vec<(String, AgentStatus)>> {
        self.db_service.list_all_agents().await
    }

    pub async fn validate_agent(&self, agent_name: &str) -> OrchestrationResult<ValidationReport> {
        let mut report = ValidationReport::new();

        let exists = self.db_service.agent_exists(agent_name).await?;
        if !exists {
            report.add_issue("Agent not found in database".to_string());
            return Ok(report);
        }

        match self.db_service.get_agent_config(agent_name).await {
            Ok(_) => {},
            Err(e) => {
                report.add_issue(format!("Configuration error: {e}"));
                return Ok(report);
            },
        }

        let status = self.db_service.get_status(agent_name).await?;
        match status {
            AgentStatus::Running { .. } => match self.health_check(agent_name).await {
                Ok(health) => {
                    if !health.healthy {
                        report.add_issue(format!("Health check failed: {}", health.message));
                    }
                },
                Err(e) => {
                    report.add_issue(format!("Health check error: {e}"));
                },
            },
            AgentStatus::Failed { reason, .. } => {
                report.add_issue(format!("Agent is in failed state: {reason}"));
            },
        }

        Ok(report)
    }

    pub async fn health_check_all(&self) -> OrchestrationResult<Vec<monitor::HealthCheckResult>> {
        let running_agents = self.db_service.list_running_agents().await?;
        let mut results = Vec::new();

        for agent_id in running_agents {
            match self.health_check(&agent_id).await {
                Ok(result) => results.push(result),
                Err(e) => {
                    tracing::warn!(agent_id = %agent_id, error = %e, "Health check failed");
                },
            }
        }

        Ok(results)
    }
}