systemprompt_models/services/agent_config/
mod.rs

1//! Agent configuration models — the on-disk YAML shape, the runtime
2//! shape, and the lightweight summary projection.
3//!
4//! Copyright (c) systemprompt.io — Business Source License 1.1.
5//! See <https://systemprompt.io> for licensing details.
6
7mod card;
8mod disk;
9mod summary;
10
11pub use card::{
12    AgentCardConfig, AgentMetadataConfig, AgentProviderInfo, CapabilitiesConfig, OAuthConfig,
13};
14pub use disk::DiskAgentConfig;
15pub use summary::AgentSummary;
16
17use crate::auth::Permission;
18use crate::errors::ConfigValidationError;
19use serde::{Deserialize, Serialize};
20
21pub const AGENT_CONFIG_FILENAME: &str = "config.yaml";
22pub const DEFAULT_AGENT_SYSTEM_PROMPT_FILE: &str = "system_prompt.md";
23
24#[derive(Debug, Clone, Serialize, Deserialize)]
25pub struct AgentConfig {
26    pub name: String,
27    pub port: u16,
28    pub endpoint: String,
29    pub enabled: bool,
30    #[serde(default)]
31    pub dev_only: bool,
32    #[serde(default)]
33    pub is_primary: bool,
34    #[serde(default)]
35    pub default: bool,
36    #[serde(default)]
37    pub tags: Vec<String>,
38    pub card: AgentCardConfig,
39    pub metadata: AgentMetadataConfig,
40    #[serde(default)]
41    pub oauth: OAuthConfig,
42}
43
44impl AgentConfig {
45    pub fn validate(&self, name: &str) -> Result<(), ConfigValidationError> {
46        if self.name != name {
47            return Err(ConfigValidationError::invalid_field(format!(
48                "Agent config key '{}' does not match name field '{}'",
49                name, self.name
50            )));
51        }
52
53        if !self
54            .name
55            .chars()
56            .all(|c| c.is_ascii_lowercase() || c.is_ascii_digit() || c == '_')
57        {
58            return Err(ConfigValidationError::invalid_field(format!(
59                "Agent name '{}' must be lowercase alphanumeric with underscores only",
60                self.name
61            )));
62        }
63
64        if self.name.len() < 3 || self.name.len() > 50 {
65            return Err(ConfigValidationError::invalid_field(format!(
66                "Agent name '{}' must be between 3 and 50 characters",
67                self.name
68            )));
69        }
70
71        if self.port == 0 {
72            return Err(ConfigValidationError::invalid_field(format!(
73                "Agent '{}' has invalid port {}",
74                self.name, self.port
75            )));
76        }
77
78        Ok(())
79    }
80
81    pub fn extract_oauth_scopes_from_card(&mut self) {
82        let Some(security_vec) = &self.card.security else {
83            return;
84        };
85        for security_obj in security_vec {
86            let Some(oauth2_scopes) = security_obj.get("oauth2") else {
87                continue;
88            };
89            let permissions: Vec<Permission> = oauth2_scopes
90                .iter()
91                .filter_map(|scope| match scope.as_str() {
92                    "admin" => Some(Permission::Admin),
93                    "user" => Some(Permission::User),
94                    "service" => Some(Permission::Service),
95                    "a2a" => Some(Permission::A2a),
96                    "mcp" => Some(Permission::Mcp),
97                    "anonymous" => Some(Permission::Anonymous),
98                    _ => None,
99                })
100                .collect();
101            if !permissions.is_empty() {
102                self.oauth.scopes = permissions;
103                self.oauth.required = true;
104            }
105        }
106    }
107
108    #[must_use]
109    pub fn construct_url(&self, base_url: &str) -> String {
110        format!(
111            "{}/api/v1/agents/{}",
112            base_url.trim_end_matches('/'),
113            self.name
114        )
115    }
116}
systemprompt_models/services/agent_config/mod.rs

systemprompt_models/services/agent_config/
mod.rs
