diff --git a/crates/tui/src/config.rs b/crates/tui/src/config.rs index 179f8a8e1..43edeb8be 100644 --- a/crates/tui/src/config.rs +++ b/crates/tui/src/config.rs @@ -5526,7 +5526,7 @@ pub fn active_provider_has_config_api_key(config: &Config) -> bool { return crate::oauth::auth_file_path().exists(); } if matches!(provider, ApiProvider::Huggingface) - && std::env::var("HF_TOKEN").is_ok_and(|k| !k.trim().is_empty()) + && provider_env_api_key(ApiProvider::Huggingface).is_some() { return true; } @@ -5736,6 +5736,17 @@ fn provider_config_table_name(provider: ApiProvider) -> Result { } fn provider_env_api_key(provider: ApiProvider) -> Option { + if matches!(provider, ApiProvider::Huggingface) { + return std::env::var("HUGGINGFACE_API_KEY") + .ok() + .filter(|value| !value.trim().is_empty()) + .or_else(|| { + std::env::var("HF_TOKEN") + .ok() + .filter(|value| !value.trim().is_empty()) + }); + } + provider.env_vars().iter().find_map(|var| { std::env::var(var) .ok()