Vapora/crates/vapora-agents/src/config.rs

// vapora-agents: Agent configuration module
// Load and parse agent definitions from TOML

use std::path::Path;

use serde::{Deserialize, Serialize};
use thiserror::Error;

#[derive(Debug, Error)]
pub enum ConfigError {
    #[error("Failed to read config file: {0}")]
    ReadError(#[from] std::io::Error),

    #[error("Failed to parse TOML: {0}")]
    ParseError(#[from] toml::de::Error),

    #[error("Invalid configuration: {0}")]
    ValidationError(String),
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentConfig {
    pub registry: RegistryConfig,
    pub agents: Vec<AgentDefinition>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RegistryConfig {
    #[serde(default = "default_max_agents")]
    pub max_agents_per_role: u32,
    #[serde(default = "default_health_check_interval")]
    pub health_check_interval: u64,
    #[serde(default = "default_agent_timeout")]
    pub agent_timeout: u64,
}

fn default_max_agents() -> u32 {
    5
}

fn default_health_check_interval() -> u64 {
    30
}

fn default_agent_timeout() -> u64 {
    300
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AgentDefinition {
    pub role: String,
    pub description: String,
    pub llm_provider: String,
    pub llm_model: String,
    #[serde(default)]
    pub parallelizable: bool,
    #[serde(default = "default_priority")]
    pub priority: u32,
    #[serde(default)]
    pub capabilities: Vec<String>,
}

fn default_priority() -> u32 {
    50
}

impl AgentConfig {
    /// Load configuration from TOML file
    pub fn load<P: AsRef<Path>>(path: P) -> Result<Self, ConfigError> {
        let content = std::fs::read_to_string(path)?;
        let config: Self = toml::from_str(&content)?;
        config.validate()?;
        Ok(config)
    }

    /// Load configuration from environment or default file
    pub fn from_env() -> Result<Self, ConfigError> {
        let config_path = std::env::var("VAPORA_AGENT_CONFIG")
            .unwrap_or_else(|_| "/etc/vapora/agents.toml".to_string());

        if Path::new(&config_path).exists() {
            Self::load(&config_path)
        } else {
            // Return default config if file doesn't exist
            Ok(Self::default())
        }
    }

    /// Validate configuration
    fn validate(&self) -> Result<(), ConfigError> {
        // Check that all agent roles are unique
        let mut roles = std::collections::HashSet::new();
        for agent in &self.agents {
            if !roles.insert(&agent.role) {
                return Err(ConfigError::ValidationError(format!(
                    "Duplicate agent role: {}",
                    agent.role
                )));
            }
        }

        // Check that we have at least one agent
        if self.agents.is_empty() {
            return Err(ConfigError::ValidationError(
                "No agents defined in configuration".to_string(),
            ));
        }

        Ok(())
    }

    /// Get agent definition by role
    pub fn get_by_role(&self, role: &str) -> Option<&AgentDefinition> {
        self.agents.iter().find(|a| a.role == role)
    }

    /// List all agent roles
    pub fn list_roles(&self) -> Vec<String> {
        self.agents.iter().map(|a| a.role.clone()).collect()
    }
}

impl Default for AgentConfig {
    fn default() -> Self {
        Self {
            registry: RegistryConfig {
                max_agents_per_role: default_max_agents(),
                health_check_interval: default_health_check_interval(),
                agent_timeout: default_agent_timeout(),
            },
            agents: vec![AgentDefinition {
                role: "developer".to_string(),
                description: "Code developer".to_string(),
                llm_provider: "claude".to_string(),
                llm_model: "claude-sonnet-4".to_string(),
                parallelizable: true,
                priority: 80,
                capabilities: vec!["coding".to_string()],
            }],
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_default_values() {
        let config = AgentConfig {
            registry: RegistryConfig {
                max_agents_per_role: 5,
                health_check_interval: 30,
                agent_timeout: 300,
            },
            agents: vec![AgentDefinition {
                role: "developer".to_string(),
                description: "Code developer".to_string(),
                llm_provider: "claude".to_string(),
                llm_model: "claude-sonnet-4".to_string(),
                parallelizable: true,
                priority: 80,
                capabilities: vec!["coding".to_string()],
            }],
        };

        assert!(config.validate().is_ok());
    }

    #[test]
    fn test_duplicate_roles() {
        let config = AgentConfig {
            registry: RegistryConfig {
                max_agents_per_role: 5,
                health_check_interval: 30,
                agent_timeout: 300,
            },
            agents: vec![
                AgentDefinition {
                    role: "developer".to_string(),
                    description: "Code developer 1".to_string(),
                    llm_provider: "claude".to_string(),
                    llm_model: "claude-sonnet-4".to_string(),
                    parallelizable: true,
                    priority: 80,
                    capabilities: vec![],
                },
                AgentDefinition {
                    role: "developer".to_string(),
                    description: "Code developer 2".to_string(),
                    llm_provider: "claude".to_string(),
                    llm_model: "claude-sonnet-4".to_string(),
                    parallelizable: true,
                    priority: 80,
                    capabilities: vec![],
                },
            ],
        };

        assert!(config.validate().is_err());
    }

    #[test]
    fn test_get_by_role() {
        let config = AgentConfig {
            registry: RegistryConfig {
                max_agents_per_role: 5,
                health_check_interval: 30,
                agent_timeout: 300,
            },
            agents: vec![AgentDefinition {
                role: "architect".to_string(),
                description: "System architect".to_string(),
                llm_provider: "claude".to_string(),
                llm_model: "claude-opus-4".to_string(),
                parallelizable: false,
                priority: 100,
                capabilities: vec!["architecture".to_string()],
            }],
        };

        let agent = config.get_by_role("architect");
        assert!(agent.is_some());
        assert_eq!(agent.unwrap().description, "System architect");

        assert!(config.get_by_role("nonexistent").is_none());
    }
}