feat(bedrock): add support in agents for nova models

awslabs · Dec 3, 2024 · f5ddbac · f5ddbac
1 parent ac57fc7
commit f5ddbac
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/apidocs/namespaces/bedrock/classes/BedrockFoundationModel.md b/apidocs/namespaces/bedrock/classes/BedrockFoundationModel.md
@@ -89,6 +89,24 @@ The ARN of the Bedrock invokable abstraction.
 
 ***
 
+### AMAZON\_NOVA\_LITE\_V1
+
+> `readonly` `static` **AMAZON\_NOVA\_LITE\_V1**: [`BedrockFoundationModel`](BedrockFoundationModel.md)
+
+***
+
+### AMAZON\_NOVA\_MICRO\_V1
+
+> `readonly` `static` **AMAZON\_NOVA\_MICRO\_V1**: [`BedrockFoundationModel`](BedrockFoundationModel.md)
+
+***
+
+### AMAZON\_NOVA\_PRO\_V1
+
+> `readonly` `static` **AMAZON\_NOVA\_PRO\_V1**: [`BedrockFoundationModel`](BedrockFoundationModel.md)
+
+***
+
 ### AMAZON\_TITAN\_PREMIER\_V1\_0
 
 > `readonly` `static` **AMAZON\_TITAN\_PREMIER\_V1\_0**: [`BedrockFoundationModel`](BedrockFoundationModel.md)

diff --git a/src/cdk-lib/bedrock/models.ts b/src/cdk-lib/bedrock/models.ts
@@ -76,6 +76,18 @@ export class BedrockFoundationModel implements IInvokable {
     supportsAgents: true,
   });
 
+  public static readonly AMAZON_NOVA_MICRO_V1 = new BedrockFoundationModel('amazon.nova-micro-v1:0', {
+    supportsAgents: true,
+  });
+
+  public static readonly AMAZON_NOVA_LITE_V1 = new BedrockFoundationModel('amazon.nova-lite-v1:0', {
+    supportsAgents: true,
+  });
+
+  public static readonly AMAZON_NOVA_PRO_V1 = new BedrockFoundationModel('amazon.nova-pro-v1:0', {
+    supportsAgents: true,
+  });
+
   public static readonly TITAN_EMBED_TEXT_V1 = new BedrockFoundationModel('amazon.titan-embed-text-v1', {
     supportsKnowledgeBase: true,
     vectorDimensions: 1536,