diff --git a/backend/services/llm.py b/backend/services/llm.py index 9102dcd8..0ec855a2 100644 --- a/backend/services/llm.py +++ b/backend/services/llm.py @@ -19,7 +19,6 @@ import litellm from litellm.files.main import ModelResponse from utils.logger import logger from utils.config import config -from utils.constants import PROVIDERS # litellm.set_verbose=True litellm.modify_params=True @@ -39,7 +38,8 @@ class LLMRetryError(LLMError): def setup_api_keys() -> None: """Set up API keys from environment variables.""" - for provider in PROVIDERS: + providers = ['OPENAI', 'ANTHROPIC', 'GROQ', 'OPENROUTER', 'XAI'] + for provider in providers: key = getattr(config, f'{provider}_API_KEY') if key: logger.debug(f"API key set for provider: {provider}") diff --git a/backend/utils/constants.py b/backend/utils/constants.py index c70538cf..7d9d4246 100644 --- a/backend/utils/constants.py +++ b/backend/utils/constants.py @@ -168,5 +168,3 @@ MODEL_ACCESS_TIERS = { "tier_125_800": PAID_TIER_MODELS, "tier_200_1000": PAID_TIER_MODELS, } - -PROVIDERS = ['OPENAI', 'ANTHROPIC', 'GROQ', 'OPENROUTER', 'XAI'] \ No newline at end of file