From 8c876ef069d0fe6942790ede41e203196cd4a390 Mon Sep 17 00:00:00 2001 From: Yeesian Ng Date: Wed, 7 Jan 2026 15:44:04 -0800 Subject: [PATCH] fix: Replace asyncio.run with create_task in ADK async thread mains. PiperOrigin-RevId: 853441839 --- .../preview/reasoning_engines/templates/adk.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/vertexai/preview/reasoning_engines/templates/adk.py b/vertexai/preview/reasoning_engines/templates/adk.py index 872de508ef..e38120a5e0 100644 --- a/vertexai/preview/reasoning_engines/templates/adk.py +++ b/vertexai/preview/reasoning_engines/templates/adk.py @@ -1063,7 +1063,7 @@ async def _invoke_agent_async(): def _asyncio_thread_main(): try: - asyncio.run(_invoke_agent_async()) + asyncio.get_running_loop().create_task(_invoke_agent_async()) except RuntimeError as e: event_queue.put(e) finally: @@ -1225,7 +1225,9 @@ async def _invoke_async_get_session(): def _asyncio_thread_main(): try: - result = asyncio.run(_invoke_async_get_session()) + result = asyncio.get_running_loop().create_task( + _invoke_async_get_session() + ) event_queue.put(result) except RuntimeError as e: event_queue.put(e) @@ -1279,7 +1281,7 @@ async def _invoke_async_list_sessions(): def _asyncio_thread_main(): try: - asyncio.run(_invoke_async_list_sessions()) + asyncio.get_running_loop().create_task(_invoke_async_list_sessions()) finally: event_queue.put(None) @@ -1349,7 +1351,9 @@ async def _invoke_async_create_session(): def _asyncio_thread_main(): try: - result = asyncio.run(_invoke_async_create_session()) + result = asyncio.get_running_loop().create_task( + _invoke_async_create_session() + ) event_queue.put(result) except RuntimeError as e: event_queue.put(e) @@ -1411,7 +1415,7 @@ async def _invoke_async_delete_session(): def _asyncio_thread_main(): try: - asyncio.run(_invoke_async_delete_session()) + asyncio.get_running_loop().create_task(_invoke_async_delete_session()) event_queue.put(None) except RuntimeError as e: event_queue.put(e)