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(