From db4bc9809c7bb6b0d261973ca7cfd87b392694be Mon Sep 17 00:00:00 2001
From: Shangjie Chen <deanchen@google.com>
Date: Thu, 29 May 2025 15:15:14 -0700
Subject: [PATCH] fix: Filter memory by app_name and user_id.

Fixing https://github.com/google/adk-python/issues/1019.

PiperOrigin-RevId: 764903966
---
 src/google/adk/memory/vertex_ai_rag_memory_service.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/google/adk/memory/vertex_ai_rag_memory_service.py b/src/google/adk/memory/vertex_ai_rag_memory_service.py
index 1b163a9..611cdb3 100644
--- 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 @@ class VertexAiRagMemoryService(BaseMemoryService):
     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: