From 6b7f81374e968ff771cd6b65a36ef5f762e0bc08 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?/c=C2=B2?= Date: Mon, 17 Jun 2024 22:29:56 +0200 Subject: [PATCH] Introduce Titan Text Embedding V2 (#495) * Introduce Titan Text Embedding V2 --- apidocs/classes/bedrock.BedrockFoundationModel.md | 7 +++++++ src/cdk-lib/bedrock/models.ts | 7 +++++-- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/apidocs/classes/bedrock.BedrockFoundationModel.md b/apidocs/classes/bedrock.BedrockFoundationModel.md index b63711db..eae46b2f 100644 --- a/apidocs/classes/bedrock.BedrockFoundationModel.md +++ b/apidocs/classes/bedrock.BedrockFoundationModel.md @@ -31,6 +31,7 @@ can instantiate a `BedrockFoundationModel` object, e.g: `new BedrockFoundationMo - [COHERE\_EMBED\_ENGLISH\_V3](bedrock.BedrockFoundationModel.md#cohere_embed_english_v3) - [COHERE\_EMBED\_MULTILINGUAL\_V3](bedrock.BedrockFoundationModel.md#cohere_embed_multilingual_v3) - [TITAN\_EMBED\_TEXT\_V1](bedrock.BedrockFoundationModel.md#titan_embed_text_v1) +- [TITAN\_EMBED\_TEXT\_V2\_1024](bedrock.BedrockFoundationModel.md#titan_embed_text_v2_1024) ### Methods @@ -138,6 +139,12 @@ ___ ▪ `Static` `Readonly` **TITAN\_EMBED\_TEXT\_V1**: [`BedrockFoundationModel`](bedrock.BedrockFoundationModel.md) +___ + +### TITAN\_EMBED\_TEXT\_V2\_1024 + +▪ `Static` `Readonly` **TITAN\_EMBED\_TEXT\_V2\_1024**: [`BedrockFoundationModel`](bedrock.BedrockFoundationModel.md) + ## Methods ### asArn diff --git a/src/cdk-lib/bedrock/models.ts b/src/cdk-lib/bedrock/models.ts index 23d04ed1..db7e2011 100644 --- a/src/cdk-lib/bedrock/models.ts +++ b/src/cdk-lib/bedrock/models.ts @@ -69,11 +69,14 @@ export class BedrockFoundationModel { 'amazon.titan-text-premier-v1:0', { supportsAgents: true }, ); - public static readonly TITAN_EMBED_TEXT_V1 = new BedrockFoundationModel( 'amazon.titan-embed-text-v1', { supportsKnowledgeBase: true, vectorDimensions: 1536 }, ); + public static readonly TITAN_EMBED_TEXT_V2_1024 = new BedrockFoundationModel( + 'amazon.titan-embed-text-v2:0', + { supportsKnowledgeBase: true, vectorDimensions: 1024 }, + ); public static readonly COHERE_EMBED_ENGLISH_V3 = new BedrockFoundationModel( 'cohere.embed-english-v3', { supportsKnowledgeBase: true, vectorDimensions: 1024 }, @@ -102,4 +105,4 @@ export class BedrockFoundationModel { const region = cdk.Stack.of(construct).region; return `arn:aws:bedrock:${region}::foundation-model/${this.modelId}`; } -} \ No newline at end of file +}