Remove duplicate env vars

This commit is contained in:
f-trycua
2025-04-06 11:35:20 -07:00
parent 388f474068
commit 04a94711f1
2 changed files with 5 additions and 3 deletions

View File

@@ -14,6 +14,6 @@ DEFAULT_MODELS = {
ENV_VARS = {
LLMProvider.OPENAI: "OPENAI_API_KEY",
LLMProvider.ANTHROPIC: "ANTHROPIC_API_KEY",
LLMProvider.OLLAMA: "OLLAMA_API_KEY",
LLMProvider.OAICOMPAT: "OAICOMPAT_API_KEY", # OpenAI-compatible API typically doesn't require an API key
LLMProvider.OLLAMA: "none",
LLMProvider.OAICOMPAT: "none", # OpenAI-compatible API typically doesn't require an API key
}

View File

@@ -35,7 +35,9 @@ class LLM:
def __post_init__(self):
"""Set default model name if not provided."""
if self.name is None:
self.name = PROVIDER_TO_DEFAULT_MODEL.get(self.provider)
from .provider_config import DEFAULT_MODELS
self.name = DEFAULT_MODELS.get(self.provider)
# Set default provider URL if none provided
if self.provider_base_url is None and self.provider == LLMProvider.OAICOMPAT: