Skip to content

Commit c027843

Browse files
authored
Handle reusing queue on task events (#32)
1 parent d94139a commit c027843

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
@@ -197,8 +197,10 @@ async def on_message_send_stream(
197197
consumer = EventConsumer(queue)
198198
producer_task.add_done_callback(consumer.agent_task_callback)
199199
async for event in result_aggregator.consume_and_emit(consumer):
200-
# Now we know we have a Task, register the queue
201-
if isinstance(event, Task):
200+
if isinstance(event, Task) and task_id != event.id:
201+
logger.warning(
202+
f'Agent generated task_id={event.id} does not match the RequestContext task_id={task_id}.'
203+
)
202204
try:
203205
await self._queue_manager.add(event.id, queue)
204206
task_id = event.id

0 commit comments

Comments
 (0)