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)