1use anyhow::{Result, anyhow};
2
3const DEFAULT_MODEL: &str = "copilot";
4const DEFAULT_BASE_URL: &str = "https://api.openai.com/v1";
5
6#[derive(Clone, Debug)]
8pub struct LlmConfig {
9 pub api_key: String,
11 pub model: String,
13 pub base_url: String,
15}
16
17pub 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}