feat(providers): add VOLCENGINE_API_KEY env var for VolcEngine/ByteDance gateway (#3725)

This commit is contained in:
Argenis 2026-03-16 15:29:36 -04:00 committed by Roman Tataurov
parent 301e8c70bf
commit 5271656e2a
No known key found for this signature in database
GPG Key ID: 70A51EF3185C334B

View File

@ -846,7 +846,9 @@ fn resolve_provider_credential(name: &str, credential_override: Option<&str>) ->
// not a single API key. Credential resolution happens inside BedrockProvider.
"bedrock" | "aws-bedrock" => return None,
name if is_qianfan_alias(name) => vec!["QIANFAN_API_KEY"],
name if is_doubao_alias(name) => vec!["ARK_API_KEY", "DOUBAO_API_KEY"],
name if is_doubao_alias(name) => {
vec!["ARK_API_KEY", "VOLCENGINE_API_KEY", "DOUBAO_API_KEY"]
}
name if is_qwen_alias(name) => vec!["DASHSCOPE_API_KEY"],
name if is_zai_alias(name) => vec!["ZAI_API_KEY"],
"nvidia" | "nvidia-nim" | "build.nvidia.com" => vec!["NVIDIA_API_KEY"],
@ -2607,6 +2609,14 @@ mod tests {
assert_eq!(resolved, Some("osaurus-test-key".to_string()));
}
#[test]
fn resolve_provider_credential_volcengine_env() {
let _env_lock = env_lock();
let _guard = EnvGuard::set("VOLCENGINE_API_KEY", Some("volc-test-key"));
let resolved = resolve_provider_credential("volcengine", None);
assert_eq!(resolved, Some("volc-test-key".to_string()));
}
// ── Extended ecosystem ───────────────────────────────────
#[test]