diff --git a/setup.sh b/setup.sh index 409a4830..be98c53a 100755 --- 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" diff --git a/skyvern/cli/commands.py b/skyvern/cli/commands.py index 09d41b6a..d882e398 100644 --- a/skyvern/cli/commands.py +++ b/skyvern/cli/commands.py @@ -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") diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py index 0e025855..b03f1a3a 100644 --- a/skyvern/forge/sdk/api/llm/config_registry.py +++ b/skyvern/forge/sdk/api/llm/config_registry.py @@ -348,6 +348,27 @@ if settings.ENABLE_GEMINI: max_completion_tokens=8192, ), ) + LLMConfigRegistry.register_config( + "GEMINI_2.5_PRO_PREVIEW_03_25", + LLMConfig( + "gemini/gemini-2.5-pro-preview-03-25", + ["GEMINI_API_KEY"], + supports_vision=True, + add_assistant_prefix=False, + max_completion_tokens=1048576, + ), + ) + LLMConfigRegistry.register_config( + "GEMINI_2.5_PRO_EXP_03_25", + LLMConfig( + "gemini/gemini-2.5-pro-exp-03-25", + ["GEMINI_API_KEY"], + supports_vision=True, + add_assistant_prefix=False, + max_completion_tokens=1048576, + ), + ) + if settings.ENABLE_NOVITA: LLMConfigRegistry.register_config(