gpt-4o integration (#333)

2024-05-17 16:22:08 +01:00
parent 6358252012
commit 45a2840538
2 changed files with 5 additions and 1 deletions
--- 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(