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

gpt-4o integration (#333)

Browse source
This commit is contained in:
Hasnain 2024-05-17 16:22:08 +01:00 • committed by GitHub
parent 6358252012
commit 45a2840538
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
setup.sh
View file

@ -52,7 +52,7 @@ setup_llm_providers() {
else
update_or_add_env_var "OPENAI_API_KEY" "$openai_api_key"
update_or_add_env_var "ENABLE_OPENAI" "true"
model_options+=("OPENAI_GPT4_TURBO" "OPENAI_GPT4V")
model_options+=("OPENAI_GPT4_TURBO" "OPENAI_GPT4V" "OPENAI_GPT4O")
fi
else
update_or_add_env_var "ENABLE_OPENAI" "false"

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

@ -74,6 +74,10 @@ if SettingsManager.get_settings().ENABLE_OPENAI:
add_assistant_prefix=False,
),
)
LLMConfigRegistry.register_config(
"OPENAI_GPT4O", LLMConfig("gpt-4o", ["OPENAI_API_KEY"], supports_vision=True, add_assistant_prefix=False)
)
if SettingsManager.get_settings().ENABLE_ANTHROPIC:
LLMConfigRegistry.register_config(