vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-15 17:59:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

add gemini 2.5 flash preview 05-20 (#2434)

Browse source
This commit is contained in:
Shuchang Zheng 2025-05-22 14:12:54 -07:00 • committed by GitHub
parent 9dd6b4e915
commit f49b5c4100
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -780,6 +780,21 @@ if settings.ENABLE_VERTEX_AI and settings.VERTEX_CREDENTIALS:
), ),
), ),
) )
LLMConfigRegistry.register_config(
"VERTEX_GEMINI_2.5_FLASH_PREVIEW_05_20",
LLMConfig(
"vertex_ai/gemini-2.5-flash-preview-05-20",
["VERTEX_CREDENTIALS"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=65535,
litellm_params=LiteLLMParams(
vertex_credentials=settings.VERTEX_CREDENTIALS,
api_base=f"{api_base}/gemini-2.5-flash-preview-05-20" if api_base else None,
vertex_location=settings.VERTEX_LOCATION,
),
),
)
LLMConfigRegistry.register_config( LLMConfigRegistry.register_config(
"VERTEX_GEMINI_FLASH_2_0", "VERTEX_GEMINI_FLASH_2_0",
LLMConfig( LLMConfig(