kevinkal commited on
Commit
e5d9213
·
verified ·
1 Parent(s): 17a1d3a

Update default model to gemma

Browse files
Files changed (1) hide show
  1. 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", "gemini-flash-latest")
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}'."