add support for gpt-4o-2024-08-06 (#715)

2024-08-22 14:08:43 -07:00
parent 43cbfa7451
commit 1967e44acc
1 changed files with 4 additions and 0 deletions
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -87,6 +87,10 @@ if SettingsManager.get_settings().ENABLE_OPENAI:
            add_assistant_prefix=False,
        ),
    )
+    LLMConfigRegistry.register_config(
+        "OPENAI_GPT-4O-2024-08-06",
+        LLMConfig("gpt-4o-2024-08-06", ["OPENAI_API_KEY"], supports_vision=True, add_assistant_prefix=False),
+    )


 if SettingsManager.get_settings().ENABLE_ANTHROPIC: