From d2f97e4feeac4a388b56b1c53ccaa7e6ab596216 Mon Sep 17 00:00:00 2001 From: Tongzhou Jiang Date: Mon, 3 Nov 2025 15:25:34 -0800 Subject: [PATCH] fix: use _init_session for in_memory session case PiperOrigin-RevId: 827672820 --- vertexai/agent_engines/templates/adk.py | 8 ++++---- vertexai/preview/reasoning_engines/templates/adk.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/vertexai/agent_engines/templates/adk.py b/vertexai/agent_engines/templates/adk.py index f3c186e903..6e21a61be9 100644 --- a/vertexai/agent_engines/templates/adk.py +++ b/vertexai/agent_engines/templates/adk.py @@ -1095,10 +1095,10 @@ async def streaming_agent_run_with_events(self, request_json: str): session_service = self._tmpl_attrs.get("in_memory_session_service") artifact_service = self._tmpl_attrs.get("in_memory_artifact_service") runner = self._tmpl_attrs.get("in_memory_runner") - session = await session_service.create_session( - app_name=self._tmpl_attrs.get("app_name"), - user_id=request.user_id, - session_id=request.session_id, + session = await self._init_session( + session_service=session_service, + artifact_service=artifact_service, + request=request, ) if not session: raise RuntimeError("Session initialization failed.") diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py index 8b49b992c6..8385cd65c9 100644 --- a/vertexai/preview/reasoning_engines/templates/adk.py +++ b/vertexai/preview/reasoning_engines/templates/adk.py @@ -982,10 +982,10 @@ async def _invoke_agent_async(): session_service = self._tmpl_attrs.get("in_memory_session_service") artifact_service = self._tmpl_attrs.get("in_memory_artifact_service") runner = self._tmpl_attrs.get("in_memory_runner") - session = await session_service.create_session( - app_name=self._tmpl_attrs.get("app_name"), - user_id=request.user_id, - session_id=request.session_id, + session = await self._init_session( + session_service=session_service, + artifact_service=artifact_service, + request=request, ) if not session: raise RuntimeError("Session initialization failed.")