Skip to content

Commit 0621aea

Browse files
committed
Handle reusing queue on task events
1 parent 16dbf22 commit 0621aea

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/a2a/server/request_handlers/default_request_handler.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -195,8 +195,10 @@ async def on_message_send_stream(
195195
consumer = EventConsumer(queue)
196196
producer_task.add_done_callback(consumer.agent_task_callback)
197197
async for event in result_aggregator.consume_and_emit(consumer):
198-
# Now we know we have a Task, register the queue
199-
if isinstance(event, Task):
198+
if isinstance(event, Task) and task_id != event.id:
199+
logger.warning(
200+
f'Agent generated task_id={event.id} does not match the RequestContext task_id={task_id}.'
201+
)
200202
try:
201203
await self._queue_manager.add(event.id, queue)
202204
task_id = event.id

0 commit comments

Comments
 (0)