vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2026-04-28 03:30:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Consolidate Gemini 3 Pro into version-agnostic VERTEX_GEMINI_3_PRO (#4926)

Browse source
This commit is contained in:
pedrohsdb 2026-02-27 19:13:23 -08:00 • committed by GitHub
parent 2e663beec0
commit 5bf555077f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 18 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

3
skyvern/config.py
View file

@ -471,9 +471,8 @@ 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-3-pro-preview": {"llm_key": "VERTEX_GEMINI_3_PRO", "label": "Gemini 3 Pro (Latest)"},
"gemini-3.0-flash": {"llm_key": "VERTEX_GEMINI_3.0_FLASH", "label": "Gemini 3 Flash"},
"gemini-3.1-pro-preview": {"llm_key": "VERTEX_GEMINI_3.1_PRO", "label": "Gemini 3.1 Pro"},
"mercury-2": {"llm_key": "INCEPTION_MERCURY_2", "label": "Inception Mercury 2"},
"gemini-2.5-flash-lite": {
"llm_key": "VERTEX_GEMINI_2.5_FLASH_LITE",

33
skyvern/forge/sdk/api/llm/config_registry.py
View file

@ -1034,7 +1034,7 @@ if settings.ENABLE_GEMINI:
),
)
LLMConfigRegistry.register_config(
"GEMINI_3.1_PRO",
"GEMINI_3_PRO",
LLMConfig(
"gemini/gemini-3.1-pro-preview",
["GEMINI_API_KEY"],
@ -1046,6 +1046,11 @@ if settings.ENABLE_GEMINI:
),
),
)
# Backward compat alias for non-Vertex Gemini 3 Pro
LLMConfigRegistry.register_config(
"GEMINI_3.1_PRO",
LLMConfigRegistry.get_config("GEMINI_3_PRO"),
)
if settings.ENABLE_NOVITA:
@ -1381,15 +1386,15 @@ if settings.ENABLE_VERTEX_AI:
),
)
LLMConfigRegistry.register_config(
"VERTEX_GEMINI_3.0_PRO",
"VERTEX_GEMINI_3_PRO",
LLMConfig(
"vertex_ai/gemini-3-pro-preview",
"vertex_ai/gemini-3.1-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,
api_base=f"{api_base}/gemini-3.1-pro-preview" if api_base else None,
vertex_location=settings.VERTEX_LOCATION,
thinking_level="medium" if settings.GEMINI_INCLUDE_THOUGHT else "minimal",
vertex_credentials=settings.VERTEX_CREDENTIALS,
@ -1412,21 +1417,15 @@ if settings.ENABLE_VERTEX_AI:
),
),
)
# Backward compat aliases — both resolve to the canonical VERTEX_GEMINI_3_PRO.
# Bump VERTEX_GEMINI_3_PRO above when Google ships a newer version.
LLMConfigRegistry.register_config(
"VERTEX_GEMINI_3.0_PRO",
LLMConfigRegistry.get_config("VERTEX_GEMINI_3_PRO"),
)
LLMConfigRegistry.register_config(
"VERTEX_GEMINI_3.1_PRO",
LLMConfig(
"vertex_ai/gemini-3.1-pro-preview",
[],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=65536,
litellm_params=LiteLLMParams(
api_base=f"{api_base}/gemini-3.1-pro-preview" if api_base else None,
vertex_location=settings.VERTEX_LOCATION,
thinking_level="medium" if settings.GEMINI_INCLUDE_THOUGHT else "minimal",
vertex_credentials=settings.VERTEX_CREDENTIALS,
),
),
LLMConfigRegistry.get_config("VERTEX_GEMINI_3_PRO"),
)
LLMConfigRegistry.register_config(
"VERTEX_GEMINI_2.5_FLASH_LITE",

2
tests/unit/test_text_prompt_block.py
View file

@ -18,7 +18,7 @@ block_module = sys.modules["skyvern.forge.sdk.workflow.models.block"]
("model_name", "expected_llm_key"),
[
("gemini-2.5-flash", "VERTEX_GEMINI_2.5_FLASH"),
("gemini-3-pro-preview", "VERTEX_GEMINI_3.0_PRO"),
("gemini-3-pro-preview", "VERTEX_GEMINI_3_PRO"),
],
)
async def test_text_prompt_block_uses_selected_model(monkeypatch, model_name, expected_llm_key):