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

amazon nova support (#1315)

Browse source
This commit is contained in:
Shuchang Zheng 2024-12-05 00:12:53 -08:00 • committed by GitHub
parent 247ee7d3dd
commit d8c6471c1b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 30 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -208,6 +208,24 @@ if settings.ENABLE_BEDROCK:
add_assistant_prefix=True,
),
)
LLMConfigRegistry.register_config(
"BEDROCK_AMAZON_NOVA_PRO",
LLMConfig(
"bedrock/us.amazon.nova-pro-v1:0",
["AWS_REGION"],
supports_vision=True,
add_assistant_prefix=True,
),
)
LLMConfigRegistry.register_config(
"BEDROCK_AMAZON_NOVA_LITE",
LLMConfig(
"bedrock/us.amazon.nova-lite-v1:0",
["AWS_REGION"],
supports_vision=True,
add_assistant_prefix=True,
),
)
if settings.ENABLE_AZURE: