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