sgr-agent 0.5.1

SGR LLM client + agent framework — structured output, function calling, agent loop, 3 agent variants
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

//! User config: provider and model selection.
//!
//! Stored in `~/<agent_home>/config.toml` (e.g. `~/.rust-code/config.toml`).

use serde::{Deserialize, Serialize};
use std::path::PathBuf;

/// User config stored in `~/<agent_home>/config.toml`.
#[derive(Debug, Default, Serialize, Deserialize)]
pub struct UserConfig {
    /// Default provider: "gemini", "codex", "openai", "claude", "ollama"
    #[serde(default)]
    pub provider: Option<String>,
    /// Override model name (BAML client name like "OpenAI", "Gemini31Pro", etc.)
    #[serde(default)]
    pub model: Option<String>,
}

fn config_path(agent_home: &str) -> PathBuf {
    let home = std::env::var("HOME").unwrap_or_else(|_| "/tmp".into());
    PathBuf::from(home).join(agent_home).join("config.toml")
}

/// Load user config from `~/<agent_home>/config.toml`.
pub fn load_config(agent_home: &str) -> UserConfig {
    let path = config_path(agent_home);
    if !path.exists() {
        return UserConfig::default();
    }
    let content = match std::fs::read_to_string(&path) {
        Ok(c) => c,
        Err(_) => return UserConfig::default(),
    };
    toml::from_str(&content).unwrap_or_default()
}

/// Save user config to `~/<agent_home>/config.toml`.
pub fn save_config(agent_home: &str, config: &UserConfig) -> Result<(), std::io::Error> {
    let path = config_path(agent_home);
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)?;
    }
    let content = toml::to_string_pretty(config)
        .map_err(|e| std::io::Error::new(std::io::ErrorKind::InvalidData, e.to_string()))?;
    std::fs::write(&path, content)
}

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

    #[test]
    fn default_config_is_empty() {
        let cfg = UserConfig::default();
        assert!(cfg.provider.is_none());
        assert!(cfg.model.is_none());
    }

    #[test]
    fn roundtrip_toml() {
        let cfg = UserConfig {
            provider: Some("gemini".into()),
            model: Some("Gemini31Pro".into()),
        };
        let s = toml::to_string_pretty(&cfg).unwrap();
        let parsed: UserConfig = toml::from_str(&s).unwrap();
        assert_eq!(parsed.provider.as_deref(), Some("gemini"));
        assert_eq!(parsed.model.as_deref(), Some("Gemini31Pro"));
    }
}