From 037d25505b2fa1beaeb27b768bd23d208a93c646 Mon Sep 17 00:00:00 2001
From: Shawn Yang <xiaolongyang@google.com>
Date: Fri, 24 Oct 2025 11:55:59 -0700
Subject: [PATCH] chore: check adk async_create_session output

PiperOrigin-RevId: 823614346
---
 vertexai/preview/reasoning_engines/templates/adk.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py
index 2d39ca6cdd..1b6d634f02 100644
--- a/vertexai/preview/reasoning_engines/templates/adk.py
+++ b/vertexai/preview/reasoning_engines/templates/adk.py
@@ -1073,6 +1073,13 @@ async def async_create_session(
             state=state,
             **kwargs,
         )
+        if asyncio.iscoroutine(session):
+            session = await session
+
+        from google.cloud.aiplatform import base
+        _LOGGER = base.Logger(__name__)
+        _LOGGER.debug("session content: %s", session)
+        _LOGGER.debug("session type: %s", type(session))
         return session
 
     def create_session(