Update default model to gemma
Browse files- routers/gemini.py +1 -1
routers/gemini.py
CHANGED
|
@@ -19,7 +19,7 @@ client = genai.Client(api_key=gemini_key)
|
|
| 19 |
ALLOWED_CHAT_MODELS = ("gemini-flash-latest", "gemma-4-26b-a4b-it")
|
| 20 |
|
| 21 |
# Default text-chat model. Override with GEMINI_CHAT_MODEL env var (must be one of ALLOWED_CHAT_MODELS).
|
| 22 |
-
GEMINI_DEFAULT_CHAT_MODEL = os.environ.get("GEMINI_CHAT_MODEL", "
|
| 23 |
if GEMINI_DEFAULT_CHAT_MODEL not in ALLOWED_CHAT_MODELS:
|
| 24 |
raise RuntimeError(
|
| 25 |
f"GEMINI_CHAT_MODEL must be one of {list(ALLOWED_CHAT_MODELS)}, got '{GEMINI_DEFAULT_CHAT_MODEL}'."
|
|
|
|
| 19 |
ALLOWED_CHAT_MODELS = ("gemini-flash-latest", "gemma-4-26b-a4b-it")
|
| 20 |
|
| 21 |
# Default text-chat model. Override with GEMINI_CHAT_MODEL env var (must be one of ALLOWED_CHAT_MODELS).
|
| 22 |
+
GEMINI_DEFAULT_CHAT_MODEL = os.environ.get("GEMINI_CHAT_MODEL", "gemma-4-26b-a4b-it")
|
| 23 |
if GEMINI_DEFAULT_CHAT_MODEL not in ALLOWED_CHAT_MODELS:
|
| 24 |
raise RuntimeError(
|
| 25 |
f"GEMINI_CHAT_MODEL must be one of {list(ALLOWED_CHAT_MODELS)}, got '{GEMINI_DEFAULT_CHAT_MODEL}'."
|