fix bug (#2249)

2025-04-30 00:06:59 +08:00
parent f4ea546ae4
commit 53f4c23ef9
2 changed files with 4 additions and 4 deletions
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -739,9 +739,9 @@ if settings.ENABLE_NOVITA:
 # Set the value of my_vertex_credentials as the environment variable VERTEX_CREDENTIALS
 if settings.ENABLE_VERTEX_AI:
    LLMConfigRegistry.register_config(
-        "VERTEX_GEMINI_2.5_PRO_EXP_03_25",
+        "VERTEX_GEMINI_2.5_FLASH_PREVIEW_04_17",
        LLMConfig(
-            "vertex_ai/gemini-2.5-pro-preview-03-25",
+            "vertex_ai/gemini-2.5-flash-preview-04-17",
            ["VERTEX_CREDENTIALS"],
            supports_vision=True,
            add_assistant_prefix=False,