xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-03 04:10:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add timeout to asyncio.wait

Browse Source
This commit is contained in:
dst03106
2025-05-13 14:46:50 +09:00
parent 943fe57231
commit e7317ce99f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/cli.py
View File

@ -86,7 +86,9 @@ def run(inargs=None, args=None):
if get_settings().litellm.get("enable_callbacks", False):
# There may be additional events on the event queue from the run above. If there are give them time to complete.
get_logger().debug("Waiting for event queue to complete")
await asyncio.wait([task for task in asyncio.all_tasks() if task is not asyncio.current_task()])
tasks = [task for task in asyncio.all_tasks() if task is not asyncio.current_task()]
if tasks:
await asyncio.wait(tasks, timeout=30)
return result