feat: Add subset_topics to Memory Bank GenerateMemories

PiperOrigin-RevId: 897366335

Author: vertex-sdk-bot

src/genai/converters/_memories_converters.ts

@@ -194,6 +194,19 @@ export function generateAgentEngineMemoriesConfigToVertex(
     );
   }
 
+  const fromAllowedTopics = common.getValueByPath(fromObject, [
+    'allowedTopics',
+  ]);
+  if (parentObject !== undefined && fromAllowedTopics != null) {
+    let transformedList = fromAllowedTopics;
+    if (Array.isArray(transformedList)) {
+      transformedList = transformedList.map((item) => {
+        return item;
+      });
+    }
+    common.setValueByPath(parentObject, ['allowedTopics'], transformedList);
+  }
+
   return toObject;
 }
 

src/genai/types/common.ts

@@ -1208,6 +1208,8 @@ export declare interface GenerateAgentEngineMemoriesConfig {
   metadata?: Record<string, MemoryMetadataValue>;
   /** Optional. The strategy to use when applying metadata to existing memories. */
   metadataMergeStrategy?: MemoryMetadataMergeStrategy;
+  /** Optional. Restricts memory generation to a subset of memory topics. */
+  allowedTopics?: MemoryTopicId[];
 }
 
 /** Parameters for generating agent engine memories. */