feat(models): memory support

Signed-off-by: Rafael Mosca <[email protected]>
awslabs · Aug 19, 2024 · 84bcca7 · 84bcca7
1 parent 6b5bbeb
commit 84bcca7
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/cdk-lib/bedrock/models.ts b/src/cdk-lib/bedrock/models.ts
@@ -21,6 +21,12 @@ export interface BedrockFoundationModelProps {
    * @default - false
    */
   readonly supportsAgents?: boolean;
+  /**
+   * Whether Memory for Agents feature is supported for this model.
+   *
+   * @default - false
+   */
+  readonly supportsMemory?: boolean;
   /**
    * Bedrock Knowledge Base can use this model.
    *
@@ -59,11 +65,11 @@ export class BedrockFoundationModel {
   );
   public static readonly ANTHROPIC_CLAUDE_SONNET_V1_0 = new BedrockFoundationModel(
     'anthropic.claude-3-sonnet-20240229-v1:0',
-    { supportsAgents: true },
+    { supportsAgents: true, supportsMemory: true },
   );
   public static readonly ANTHROPIC_CLAUDE_HAIKU_V1_0 = new BedrockFoundationModel(
     'anthropic.claude-3-haiku-20240307-v1:0',
-    { supportsAgents: true },
+    { supportsAgents: true, supportsMemory: true },
   );
   public static readonly AMAZON_TITAN_PREMIER_V1_0 = new BedrockFoundationModel(
     'amazon.titan-text-premier-v1:0',
@@ -95,6 +101,7 @@ export class BedrockFoundationModel {
     this.supportsAgents = props.supportsAgents ?? false;
     this.vectorDimensions = props.vectorDimensions;
     this.supportsKnowledgeBase = props.supportsKnowledgeBase ?? false;
+    this.supportsMemory = props.supportsMemory ?? false;
   }
 
   toString(): string {