diff --git a/setup.sh b/setup.sh index 6284b294..1543926c 100755 --- a/setup.sh +++ b/setup.sh @@ -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" diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py index 1a926b7c..3096d3ae 100644 --- a/skyvern/forge/sdk/api/llm/config_registry.py +++ b/skyvern/forge/sdk/api/llm/config_registry.py @@ -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(