add cerebras models

2026-03-08 00:53:25 +01:00
parent 3b3b3baf25
commit ac71d99f9a
1 changed files with 2 additions and 0 deletions
--- 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.