From fa55580c378ef82229dcf7a8d1be069c61d095b5 Mon Sep 17 00:00:00 2001 From: Tongzhou Jiang Date: Wed, 11 Feb 2026 12:00:33 -0800 Subject: [PATCH] fix: Refactor session retrieval fallback _streaming_agent_run_with_events. PiperOrigin-RevId: 868781249 --- vertexai/agent_engines/templates/adk.py | 2 ++ vertexai/preview/reasoning_engines/templates/adk.py | 4 ++++ 2 files changed, 6 insertions(+) diff --git a/vertexai/agent_engines/templates/adk.py b/vertexai/agent_engines/templates/adk.py index cb27b97a31..acd6d02d44 100644 --- a/vertexai/agent_engines/templates/adk.py +++ b/vertexai/agent_engines/templates/adk.py @@ -1190,6 +1190,8 @@ async def streaming_agent_run_with_events(self, request_json: str): session_id=request.session_id, ) except ClientError: + session = None + if not session: # Fall back to create session if the session is not found. # Specifying session_id on creation is not supported, # so session id will be regenerated. diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py index 6d7b4aa255..76ef1ae20e 100644 --- a/vertexai/preview/reasoning_engines/templates/adk.py +++ b/vertexai/preview/reasoning_engines/templates/adk.py @@ -1024,7 +1024,11 @@ async def _invoke_agent_async(): session_id=request.session_id, ) except ClientError: + session = None + if not session: # Fall back to create session if the session is not found. + # Specifying session_id on creation is not supported, + # so session id will be regenerated. session = await self._init_session( session_service=session_service, artifact_service=artifact_service,