fix: correct API key environment variable for vLLM mode

This commit is contained in:
popcell 2026-02-03 12:14:14 +08:00
parent 30d6e4b4b6
commit 8cde0b3072

View File

@ -42,7 +42,7 @@ class LiteLLMProvider(LLMProvider):
os.environ["OPENROUTER_API_KEY"] = api_key
elif self.is_vllm:
# vLLM/custom endpoint - uses OpenAI-compatible API
os.environ["OPENAI_API_KEY"] = api_key
os.environ["HOSTED_VLLM_API_KEY"] = api_key
elif "anthropic" in default_model:
os.environ.setdefault("ANTHROPIC_API_KEY", api_key)
elif "openai" in default_model or "gpt" in default_model: