vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2026-05-05 07:10:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

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

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",