Introduce Titan Text Embedding V2 (#495)

* Introduce Titan Text Embedding V2
awslabs · Jun 17, 2024 · 6b7f813 · 6b7f813
1 parent 879aa53
commit 6b7f813
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git 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
@@ -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}`;
   }
-}
+}