From cdd88802772d1861dd038a76dab515e4c071a7c4 Mon Sep 17 00:00:00 2001 From: Tongzhou Jiang Date: Thu, 8 Jan 2026 15:37:14 -0800 Subject: [PATCH] chore: make sure memory service is setup in `streaming_agent_run_with_response` PiperOrigin-RevId: 853909960 --- vertexai/agent_engines/templates/adk.py | 4 ++++ vertexai/preview/reasoning_engines/templates/adk.py | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/vertexai/agent_engines/templates/adk.py b/vertexai/agent_engines/templates/adk.py index e717410fe6..87db2f29fe 100644 --- a/vertexai/agent_engines/templates/adk.py +++ b/vertexai/agent_engines/templates/adk.py @@ -1162,6 +1162,10 @@ async def streaming_agent_run_with_events(self, request_json: str): self.set_up() if not self._tmpl_attrs.get("session_service"): self.set_up() + if not self._tmpl_attrs.get("in_memory_memory_service"): + self.set_up() + if not self._tmpl_attrs.get("memory_service"): + self.set_up() app = self._tmpl_attrs.get("app") # Try to get the session, if it doesn't exist, create a new one. diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py index 872de508ef..f241a41067 100644 --- a/vertexai/preview/reasoning_engines/templates/adk.py +++ b/vertexai/preview/reasoning_engines/templates/adk.py @@ -1002,6 +1002,10 @@ async def _invoke_agent_async(): self.set_up() if not self._tmpl_attrs.get("session_service"): self.set_up() + if not self._tmpl_attrs.get("in_memory_memory_service"): + self.set_up() + if not self._tmpl_attrs.get("memory_service"): + self.set_up() # Try to get the session, if it doesn't exist, create a new one. if request.session_id: