vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

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
2 changed files with 5 additions and 1 deletions
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 else
update_or_add_env_var "OPENAI_API_KEY" "$openai_api_key" update_or_add_env_var "OPENAI_API_KEY" "$openai_api_key"
update_or_add_env_var "ENABLE_OPENAI" "true" 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 fi
else else
update_or_add_env_var "ENABLE_OPENAI" "false" 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, 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: if SettingsManager.get_settings().ENABLE_ANTHROPIC:
LLMConfigRegistry.register_config( LLMConfigRegistry.register_config(