aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2025-12-17 06:04:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat(api): load provider config from CLI

Browse Source
This commit is contained in:
Aljaz
2025-05-29 11:25:39 +02:00
parent 7f2b19c51d
commit 23b480326e
1 changed files with 36 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

45
crates/goose-api/src/config.rs
View File

@@ -18,13 +18,31 @@ pub fn load_configuration() -> std::result::Result<config::Config, config::Confi
pub async fn initialize_provider_config() -> Result<(), anyhow::Error> {
let api_config = load_configuration()?;
let provider_name = std::env::var("GOOSE_API_PROVIDER")
.or_else(|_| api_config.get_string("provider"))
.unwrap_or_else(|_| "openai".to_string());
let global_config = Config::global();
let model_name = std::env::var("GOOSE_API_MODEL")
.or_else(|_| api_config.get_string("model"))
.unwrap_or_else(|_| "gpt-4o".to_string());
let provider_name = if let Ok(val) = std::env::var("GOOSE_API_PROVIDER") {
val
} else if let Ok(val) = api_config.get_string("provider") {
val
} else if global_config.exists() {
global_config
.get_param::<String>("GOOSE_PROVIDER")
.unwrap_or_else(|_| "openai".to_string())
} else {
"openai".to_string()
};
let model_name = if let Ok(val) = std::env::var("GOOSE_API_MODEL") {
val
} else if let Ok(val) = api_config.get_string("model") {
val
} else if global_config.exists() {
global_config
.get_param::<String>("GOOSE_MODEL")
.unwrap_or_else(|_| "gpt-4o".to_string())
} else {
"gpt-4o".to_string()
};
info!("Initializing with provider: {}, model: {}", provider_name, model_name);
@@ -44,12 +62,21 @@ pub async fn initialize_provider_config() -> Result<(), anyhow::Error> {
config.set_param(&key.name, Value::String(value))?;
info!("Set parameter: {}", key.name);
}
} else {
warn!("Environment variable not set for key: {}", key.name);
if key.required {
} else if global_config.exists() {
// If not provided via environment, try existing CLI config
let result: Result<String, _> = if key.secret {
global_config.get_secret(&key.name)
} else {
global_config.get_param(&key.name)
};
if result.is_err() && key.required {
error!("Required key {} not provided", key.name);
return Err(anyhow::anyhow!("Required key {} not provided", key.name));
}
} else if key.required {
error!("Required key {} not provided", key.name);
return Err(anyhow::anyhow!("Required key {} not provided", key.name));
}
}
}