feat: Add subset_topics to Memory Bank GenerateMemories

PiperOrigin-RevId: 894126692

Author: vertex-sdk-bot

vertexai/_genai/memories.py

@@ -174,6 +174,13 @@ def _GenerateAgentEngineMemoriesConfig_to_vertex(
             getv(from_object, ["metadata_merge_strategy"]),
         )
 
+    if getv(from_object, ["allowed_topics"]) is not None:
+        setv(
+            parent_object,
+            ["allowedTopics"],
+            [item for item in getv(from_object, ["allowed_topics"])],
+        )
+
     return to_object
 
 

vertexai/_genai/types/common.py

@@ -9076,6 +9076,10 @@ class GenerateAgentEngineMemoriesConfig(_common.BaseModel):
         default=None,
         description="""Optional. The strategy to use when applying metadata to existing memories.""",
     )
+    allowed_topics: Optional[list[MemoryTopicId]] = Field(
+        default=None,
+        description="""Optional. Restricts memory generation to a subset of memory topics.""",
+    )
 
 
 class GenerateAgentEngineMemoriesConfigDict(TypedDict, total=False):
@@ -9113,6 +9117,9 @@ class GenerateAgentEngineMemoriesConfigDict(TypedDict, total=False):
     metadata_merge_strategy: Optional[MemoryMetadataMergeStrategy]
     """Optional. The strategy to use when applying metadata to existing memories."""
 
+    allowed_topics: Optional[list[MemoryTopicIdDict]]
+    """Optional. Restricts memory generation to a subset of memory topics."""
+
 
 GenerateAgentEngineMemoriesConfigOrDict = Union[
     GenerateAgentEngineMemoriesConfig, GenerateAgentEngineMemoriesConfigDict