From 6cc096f32362b9b27c7bd0fd5e133ae08c11a174 Mon Sep 17 00:00:00 2001 From: syeddhasnainn Date: Fri, 17 May 2024 02:50:01 +0100 Subject: [PATCH] gpt-4o integration --- setup.sh | 2 +- skyvern/forge/sdk/api/llm/config_registry.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/setup.sh b/setup.sh index 6284b2941..1543926c5 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 d246aa2e0..fe33eda07 100644 --- a/skyvern/forge/sdk/api/llm/config_registry.py +++ b/skyvern/forge/sdk/api/llm/config_registry.py @@ -63,6 +63,10 @@ def get_config(cls, llm_key: str) -> LLMRouterConfig | LLMConfig: LLMConfigRegistry.register_config( "OPENAI_GPT4V", LLMConfig("gpt-4-turbo", ["OPENAI_API_KEY"], supports_vision=True, 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(