add gpt4o-mini support (#625)

2024-07-18 22:33:19 -07:00
parent 50ff8dc441
commit ec5a0a03c0
3 changed files with 36 additions and 3 deletions
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -77,6 +77,15 @@ if SettingsManager.get_settings().ENABLE_OPENAI:
    LLMConfigRegistry.register_config(
        "OPENAI_GPT4O", LLMConfig("gpt-4o", ["OPENAI_API_KEY"], supports_vision=True, add_assistant_prefix=False)
    )
+    LLMConfigRegistry.register_config(
+        "OPENAI_GPT4O_MINI",
+        LLMConfig(
+            "gpt-4o-mini",
+            ["OPENAI_API_KEY"],
+            supports_vision=True,
+            add_assistant_prefix=False,
+        ),
+    )


 if SettingsManager.get_settings().ENABLE_ANTHROPIC: