fix: Filter memory by app_name and user_id.

DeanChensj · copybara-github · commit db4bc9809c7b · 2025-05-29T15:15:59.000-07:00
Fixing #1019. PiperOrigin-RevId: 764903966
diff --git a/src/google/adk/memory/vertex_ai_rag_memory_service.py b/src/google/adk/memory/vertex_ai_rag_memory_service.py
@@ -124,8 +124,8 @@ async def search_memory(
     for context in response.contexts.contexts:
       # filter out context that is not related
       # TODO: Add server side filtering by app_name and user_id.
-      # if not context.source_display_name.startswith(f"{app_name}.{user_id}."):
-      #   continue
+      if not context.source_display_name.startswith(f"{app_name}.{user_id}."):
+        continue
       session_id = context.source_display_name.split(".")[-1]
       events = []
       if context.text:
