diff --git a/app/core/llm.py b/app/core/llm.py index 80e14a5..3d49157 100644 --- a/app/core/llm.py +++ b/app/core/llm.py @@ -34,6 +34,8 @@ def _api_key_for_model(model: str) -> str | None: return settings.ANTHROPIC_API_KEY or None if model.startswith("gemini/") or model.startswith("google/"): return settings.GOOGLE_API_KEY or None + if model.startswith("cerebras/"): + return settings.CEREBRAS_API_KEY or None if model.startswith("github_copilot/"): # GitHub Copilot uses OAuth device-flow tokens managed by LiteLLM. # No API key is required; returning None lets LiteLLM handle auth.