vrr/agent-zero
2
0
Fork 0
mirror of https://github.com/agent0ai/agent-zero.git synced 2026-05-07 17:22:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

nudge + move queue extension

Browse source
This commit is contained in:
frdel 2026-01-30 12:40:19 +01:00
parent 37cafbe86c
commit 3d8cc2a620
3 changed files with 19 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
python/extensions/monologue_end/_95_process_queue.py → python/extensions/process_chain_end/_50_process_queue.py
View file

@ -1,7 +1,7 @@
import asyncio
from python.helpers.extension import Extension
from python.helpers import message_queue as mq
from agent import LoopData
from agent import AgentContext, Agent, LoopData
class ProcessQueue(Extension):
@ -20,14 +20,15 @@ class ProcessQueue(Extension):
# This allows current monologue to fully complete first
asyncio.create_task(self._delayed_send(context))
async def _delayed_send(self, context):
async def _delayed_send(self, context: AgentContext):
"""Wait for task to complete, then send next queued message."""
# Small delay to ensure monologue fully completes
await asyncio.sleep(0.1)
# Wait for current task to finish
while context.task and context.task.is_alive():
# Wait for current task to finish, but no more than 1 minute to prevent hanging tasks
total_wait = 0
while context.is_running() and total_wait < 60:
await asyncio.sleep(0.1)
total_wait += 0.1
# Send next queued message
mq.send_next(context)
# Send next queued message if task is not running
if not context.is_running():
mq.send_next(context)