From f7e68141ebea2e13f4662b49a9fbb0c4acfff7c4 Mon Sep 17 00:00:00 2001
From: pedrohsdb <pedro@skyvern.com>
Date: Tue, 18 Nov 2025 16:13:51 -0800
Subject: [PATCH] add vertex gemini 3 pro config (#4025)

---
 skyvern/config.py                            |  2 ++
 skyvern/forge/sdk/api/llm/config_registry.py | 19 +++++++++++++++++++
 2 files changed, 21 insertions(+)

diff --git a/skyvern/config.py b/skyvern/config.py
index 90fc2938..76c59545 100644
--- a/skyvern/config.py
+++ b/skyvern/config.py
@@ -401,6 +401,7 @@ class Settings(BaseSettings):
                     "llm_key": "VERTEX_GEMINI_2.5_FLASH",
                     "label": "Gemini 2.5 Flash",
                 },
+                "gemini-3-pro-preview": {"llm_key": "VERTEX_GEMINI_3.0_PRO", "label": "Gemini 3 Pro"},
                 "gemini-2.5-flash-lite": {
                     "llm_key": "VERTEX_GEMINI_2.5_FLASH_LITE",
                     "label": "Gemini 2.5 Flash Lite",
@@ -437,6 +438,7 @@ class Settings(BaseSettings):
                     "llm_key": "VERTEX_GEMINI_2.5_FLASH",
                     "label": "Gemini 2.5 Flash",
                 },
+                "gemini-3-pro-preview": {"llm_key": "VERTEX_GEMINI_3.0_PRO", "label": "Gemini 3 Pro"},
                 "gemini-2.5-flash-lite": {
                     "llm_key": "VERTEX_GEMINI_2.5_FLASH_LITE",
                     "label": "Gemini 2.5 Flash Lite",
diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py
index 5dedad72..b820105d 100644
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -1207,6 +1207,25 @@ if settings.ENABLE_VERTEX_AI:
             ),
         ),
     )
+    LLMConfigRegistry.register_config(
+        "VERTEX_GEMINI_3.0_PRO",
+        LLMConfig(
+            "vertex_ai/gemini-3-pro-preview",
+            [],
+            supports_vision=True,
+            add_assistant_prefix=False,
+            max_completion_tokens=65536,
+            litellm_params=LiteLLMParams(
+                api_base=f"{api_base}/gemini-3-pro-preview" if api_base else None,
+                vertex_location=settings.VERTEX_LOCATION,
+                thinking={
+                    "budget_tokens": settings.GEMINI_THINKING_BUDGET,
+                    "type": "enabled" if settings.GEMINI_INCLUDE_THOUGHT else None,
+                },
+                vertex_credentials=settings.VERTEX_CREDENTIALS,
+            ),
+        ),
+    )
     LLMConfigRegistry.register_config(
         "VERTEX_GEMINI_2.5_FLASH_LITE",
         LLMConfig(