Skip to content

Commit 83f4076

Browse files
yeesiancopybara-github
authored andcommitted
fix: Replace asyncio.run with create_task in ADK async thread mains.
PiperOrigin-RevId: 853794408
1 parent 937d5af commit 83f4076

File tree

1 file changed

+5
-9
lines changed
  • vertexai/preview/reasoning_engines/templates

1 file changed

+5
-9
lines changed

vertexai/preview/reasoning_engines/templates/adk.py

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1063,7 +1063,7 @@ async def _invoke_agent_async():
10631063

10641064
def _asyncio_thread_main():
10651065
try:
1066-
asyncio.get_running_loop().create_task(_invoke_agent_async())
1066+
asyncio.run(_invoke_agent_async())
10671067
except RuntimeError as e:
10681068
event_queue.put(e)
10691069
finally:
@@ -1225,9 +1225,7 @@ async def _invoke_async_get_session():
12251225

12261226
def _asyncio_thread_main():
12271227
try:
1228-
result = asyncio.get_running_loop().create_task(
1229-
_invoke_async_get_session()
1230-
)
1228+
result = asyncio.run(_invoke_async_get_session())
12311229
event_queue.put(result)
12321230
except RuntimeError as e:
12331231
event_queue.put(e)
@@ -1281,7 +1279,7 @@ async def _invoke_async_list_sessions():
12811279

12821280
def _asyncio_thread_main():
12831281
try:
1284-
asyncio.get_running_loop().create_task(_invoke_async_list_sessions())
1282+
asyncio.run(_invoke_async_list_sessions())
12851283
finally:
12861284
event_queue.put(None)
12871285

@@ -1351,9 +1349,7 @@ async def _invoke_async_create_session():
13511349

13521350
def _asyncio_thread_main():
13531351
try:
1354-
result = asyncio.get_running_loop().create_task(
1355-
_invoke_async_create_session()
1356-
)
1352+
result = asyncio.run(_invoke_async_create_session())
13571353
event_queue.put(result)
13581354
except RuntimeError as e:
13591355
event_queue.put(e)
@@ -1415,7 +1411,7 @@ async def _invoke_async_delete_session():
14151411

14161412
def _asyncio_thread_main():
14171413
try:
1418-
asyncio.get_running_loop().create_task(_invoke_async_delete_session())
1414+
asyncio.run(_invoke_async_delete_session())
14191415
event_queue.put(None)
14201416
except RuntimeError as e:
14211417
event_queue.put(e)

0 commit comments

Comments
 (0)