diff --git a/vertexai/agent_engines/templates/adk.py b/vertexai/agent_engines/templates/adk.py index cb27b97a31..a22e512ca2 100644 --- a/vertexai/agent_engines/templates/adk.py +++ b/vertexai/agent_engines/templates/adk.py @@ -1183,6 +1183,7 @@ async def streaming_agent_run_with_events(self, request_json: str): session_service = self._tmpl_attrs.get("session_service") artifact_service = self._tmpl_attrs.get("artifact_service") runner = self._tmpl_attrs.get("runner") + session = None try: session = await session_service.get_session( app_name=app.name if app else self._tmpl_attrs.get("app_name"), @@ -1190,6 +1191,8 @@ async def streaming_agent_run_with_events(self, request_json: str): session_id=request.session_id, ) except ClientError: + pass + 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..bf4cf42e92 100644 --- a/vertexai/preview/reasoning_engines/templates/adk.py +++ b/vertexai/preview/reasoning_engines/templates/adk.py @@ -1017,6 +1017,7 @@ async def _invoke_agent_async(): session_service = self._tmpl_attrs.get("session_service") artifact_service = self._tmpl_attrs.get("artifact_service") runner = self._tmpl_attrs.get("runner") + session = None try: session = await session_service.get_session( app_name=self._tmpl_attrs.get("app_name"), @@ -1024,7 +1025,11 @@ async def _invoke_agent_async(): session_id=request.session_id, ) except ClientError: + pass + 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,