Skip to content

Commit 5c2bf70

Browse files
yeesiancopybara-github
authored andcommitted
fix: Replace asyncio.run with create_task in ADK async thread mains.
PiperOrigin-RevId: 853388884
1 parent 65717fa commit 5c2bf70

File tree

1 file changed

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

1 file changed

+9
-5
lines changed

vertexai/preview/reasoning_engines/templates/adk.py

Lines changed: 9 additions & 5 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.run(_invoke_agent_async())
1066+
asyncio.get_running_loop().create_task(_invoke_agent_async())
10671067
except RuntimeError as e:
10681068
event_queue.put(e)
10691069
finally:
@@ -1225,7 +1225,9 @@ async def _invoke_async_get_session():
12251225

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

12801282
def _asyncio_thread_main():
12811283
try:
1282-
asyncio.run(_invoke_async_list_sessions())
1284+
asyncio.get_running_loop().create_task(_invoke_async_list_sessions())
12831285
finally:
12841286
event_queue.put(None)
12851287

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

13501352
def _asyncio_thread_main():
13511353
try:
1352-
result = asyncio.run(_invoke_async_create_session())
1354+
result = asyncio.get_running_loop().create_task(
1355+
_invoke_async_create_session()
1356+
)
13531357
event_queue.put(result)
13541358
except RuntimeError as e:
13551359
event_queue.put(e)
@@ -1411,7 +1415,7 @@ async def _invoke_async_delete_session():
14111415

14121416
def _asyncio_thread_main():
14131417
try:
1414-
asyncio.run(_invoke_async_delete_session())
1418+
asyncio.get_running_loop().create_task(_invoke_async_delete_session())
14151419
event_queue.put(None)
14161420
except RuntimeError as e:
14171421
event_queue.put(e)

0 commit comments

Comments
 (0)