From 45a28405381211f6bb3dfc320277fa47aa1e990d Mon Sep 17 00:00:00 2001
From: Hasnain <66178110+syeddhasnainn@users.noreply.github.com>
Date: Fri, 17 May 2024 16:22:08 +0100
Subject: [PATCH] gpt-4o integration (#333)

---
 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 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(