vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-16 02:09:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

add gpt4o-mini support (#625)

Browse source
This commit is contained in:
Shuchang Zheng 2024-07-18 22:33:19 -07:00 • committed by GitHub
parent 50ff8dc441
commit ec5a0a03c0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 36 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
skyvern/forge/sdk/api/llm/config_registry.py
View file

@ -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: