add support for gpt-4o-2024-08-06

Skyvern-AI · Aug 22, 2024 · d62efad · d62efad
1 parent 43cbfa7
commit d62efad
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/skyvern/forge/sdk/api/llm/config_registry.py b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -87,6 +87,10 @@ def get_config(cls, llm_key: str) -> LLMRouterConfig | LLMConfig:
             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: