shu/gemini 2.5 pro support (#2109)

This commit is contained in:
Shuchang Zheng 2025-04-06 00:03:24 -04:00 committed by GitHub
parent a72fcadd9a
commit 7e5e684f53
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 37 additions and 2 deletions

View file

@ -279,7 +279,15 @@ def setup_llm_providers() -> None:
else:
update_or_add_env_var("GEMINI_API_KEY", gemini_api_key)
update_or_add_env_var("ENABLE_GEMINI", "true")
model_options.extend(["GEMINI_FLASH_2_0", "GEMINI_FLASH_2_0_LITE", "GEMINI_PRO"])
model_options.extend(
[
"GEMINI_FLASH_2_0",
"GEMINI_FLASH_2_0_LITE",
"GEMINI_PRO",
"GEMINI_2.5_PRO_PREVIEW_03_25",
"GEMINI_2.5_PRO_EXP_03_25",
]
)
else:
update_or_add_env_var("ENABLE_GEMINI", "false")