fix: investigation on session state

PiperOrigin-RevId: 831082835

Author: Tongzhou-Jiang

vertexai/agent_engines/templates/adk.py

@@ -1073,8 +1073,12 @@ async def streaming_agent_run_with_events(self, request_json: str):
         import json
         from google.genai import types
         from google.genai.errors import ClientError
+        from google.cloud.aiplatform import base
+
+        _LOGGER = base.Logger(__name__)
 
         request = _StreamRunRequest(**json.loads(request_json))
+        _LOGGER.info(f"request: {request.artifacts}")
         if not self._tmpl_attrs.get("in_memory_runner"):
             self.set_up()
         if not self._tmpl_attrs.get("runner"):
@@ -1140,6 +1144,7 @@ async def streaming_agent_run_with_events(self, request_json: str):
                 yield converted_event
         finally:
             if session and not request.session_id:
+                _LOGGER.info(f"Dumping session state: {session.state}")
                 app = self._tmpl_attrs.get("app")
                 await session_service.delete_session(
                     app_name=app.name if app else self._tmpl_attrs.get("app_name"),