Skip to main content

phi_agent/config/
llm.rs

1use anyhow::{Result, anyhow};
2
3const DEFAULT_MODEL: &str = "copilot";
4const DEFAULT_BASE_URL: &str = "https://api.openai.com/v1";
5
6/// Resolved LLM configuration.
7#[derive(Clone, Debug)]
8pub struct LlmConfig {
9    /// API key for the LLM provider.
10    pub api_key: String,
11    /// Model name (e.g. `"opus"`, `"gpt-4o"`).
12    pub model: String,
13    /// Base URL for the LLM API endpoint.
14    pub base_url: String,
15}
16
17/// Resolve LLM configuration (API key, model, base_url).
18///
19/// Priority: CLI arg > environment variable (.env) > default
20pub fn resolve_llm_config(model: Option<&str>, base_url: Option<&str>) -> Result<LlmConfig> {
21    let api_key = super::optional_env("LLM_API_KEY")
22        .or_else(|| super::optional_env("OPENAI_API_KEY"))
23        .ok_or_else(|| anyhow!("Missing environment variable LLM_API_KEY. Please configure it in .env."))?;
24
25    let resolved_model = model
26        .map(|s| s.to_string())
27        .or_else(|| super::optional_env("LLM_MODEL"))
28        .or_else(|| super::optional_env("OPENAI_MODEL"))
29        .unwrap_or_else(|| DEFAULT_MODEL.to_string());
30
31    let resolved_base_url = base_url
32        .map(|s| s.to_string())
33        .or_else(|| super::optional_env("LLM_BASE_URL"))
34        .or_else(|| super::optional_env("OPENAI_BASE_URL"))
35        .unwrap_or_else(|| DEFAULT_BASE_URL.to_string());
36
37    Ok(LlmConfig { api_key, model: resolved_model, base_url: resolved_base_url })
38}