shu/gemini 2.5 pro support (#2109)

2025-04-06 00:03:24 -04:00
parent a72fcadd9a
commit 7e5e684f53
3 changed files with 37 additions and 2 deletions
--- a/setup.sh
+++ b/setup.sh
@@ -109,7 +109,13 @@ setup_llm_providers() {
        else
            update_or_add_env_var "GEMINI_API_KEY" "$gemini_api_key"
            update_or_add_env_var "ENABLE_GEMINI" "true"
-            model_options+=("GEMINI_PRO")
+            model_options+=(
+                "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"
+            )
        fi
    else
        update_or_add_env_var "ENABLE_GEMINI" "false"