amazon nova support (#1315)

2024-12-05 00:12:53 -08:00
parent 247ee7d3dd
commit d8c6471c1b
4 changed files with 30 additions and 11 deletions
--- a/skyvern/forge/sdk/api/llm/config_registry.py
+++ b/skyvern/forge/sdk/api/llm/config_registry.py
@@ -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: