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

add gemini 2.0 flash (#1745)

Browse source
This commit is contained in:
Shuchang Zheng 2025-02-07 14:39:31 +08:00 • committed by GitHub
parent f9296ba1d6
commit 22cefee5f9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -305,6 +305,26 @@ if settings.ENABLE_AZURE_O3_MINI:
) )
if settings.ENABLE_GEMINI: if settings.ENABLE_GEMINI:
LLMConfigRegistry.register_config(
"GEMINI_FLASH_2_0",
LLMConfig(
"gemini/gemini-2.0-flash-001",
["GEMINI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=8192,
),
)
LLMConfigRegistry.register_config(
"GEMINI_FLASH_2_0_LITE",
LLMConfig(
"gemini/gemini-2.0-flash-lite-preview-02-05",
["GEMINI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=8192,
),
)
LLMConfigRegistry.register_config( LLMConfigRegistry.register_config(
"GEMINI_PRO", "GEMINI_PRO",
LLMConfig( LLMConfig(