From f5ddbaca43cf370e34aa827550f1adbd85a62be0 Mon Sep 17 00:00:00 2001 From: krokoko Date: Tue, 3 Dec 2024 16:44:43 -0600 Subject: [PATCH] feat(bedrock): add support in agents for nova models --- .../bedrock/classes/BedrockFoundationModel.md | 18 ++++++++++++++++++ src/cdk-lib/bedrock/models.ts | 12 ++++++++++++ 2 files changed, 30 insertions(+) diff --git a/apidocs/namespaces/bedrock/classes/BedrockFoundationModel.md b/apidocs/namespaces/bedrock/classes/BedrockFoundationModel.md index ce861bc0..3f75ab84 100644 --- 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 index 09a3db2c..a0b8f169 100644 --- 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,