vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-04-25 08:59:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions3

shu/gemini 2.5 pro support ()

Browse source
This commit is contained in:
Shuchang Zheng 2025-04-06 00:03:24 -04:00 • committed by GitHub
parent a72fcadd9a
commit 7e5e684f53
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 37 additions and 2 deletions
setup.sh
skyvern
cli
commands.py
forge/sdk/api/llm
config_registry.py

8
setup.sh
View file

@ -109,7 +109,13 @@ setup_llm_providers() {
else
update_or_add_env_var "GEMINI_API_KEY" "$gemini_api_key"
update_or_add_env_var "ENABLE_GEMINI" "true"
model_options+=("GEMINI_PRO")
model_options+=(
"GEMINI_FLASH_2_0"
"GEMINI_FLASH_2_0_LITE"
"GEMINI_PRO"
"GEMINI_2.5_PRO_PREVIEW_03_25"
"GEMINI_2.5_PRO_EXP_03_25"
)
fi
else
update_or_add_env_var "ENABLE_GEMINI" "false"

10
skyvern/cli/commands.py
View file

@ -279,7 +279,15 @@ def setup_llm_providers() -> None:
else:
update_or_add_env_var("GEMINI_API_KEY", gemini_api_key)
update_or_add_env_var("ENABLE_GEMINI", "true")
model_options.extend(["GEMINI_FLASH_2_0", "GEMINI_FLASH_2_0_LITE", "GEMINI_PRO"])
model_options.extend(
[
"GEMINI_FLASH_2_0",
"GEMINI_FLASH_2_0_LITE",
"GEMINI_PRO",
"GEMINI_2.5_PRO_PREVIEW_03_25",
"GEMINI_2.5_PRO_EXP_03_25",
]
)
else:
update_or_add_env_var("ENABLE_GEMINI", "false")

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

@ -348,6 +348,27 @@ if settings.ENABLE_GEMINI:
max_completion_tokens=8192,
),
)
LLMConfigRegistry.register_config(
"GEMINI_2.5_PRO_PREVIEW_03_25",
LLMConfig(
"gemini/gemini-2.5-pro-preview-03-25",
["GEMINI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=1048576,
),
)
LLMConfigRegistry.register_config(
"GEMINI_2.5_PRO_EXP_03_25",
LLMConfig(
"gemini/gemini-2.5-pro-exp-03-25",
["GEMINI_API_KEY"],
supports_vision=True,
add_assistant_prefix=False,
max_completion_tokens=1048576,
),
)
if settings.ENABLE_NOVITA:
LLMConfigRegistry.register_config(