Skip to content
Merged
6 changes: 6 additions & 0 deletions src/a2a/server/request_handlers/default_request_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@
TaskState.rejected,
}


@trace_class(kind=SpanKind.SERVER)
class DefaultRequestHandler(RequestHandler):
"""Default request handler for all incoming requests.
Expand Down Expand Up @@ -249,6 +250,11 @@ async def on_message_send(
InternalError(message='Task ID mismatch in agent response')
)

if self._push_notifier and task_id:
latest_task = await result_aggregator.current_result
if isinstance(latest_task, Task):
await self._push_notifier.send_notification(latest_task)

finally:
if interrupted:
# TODO: Track this disconnected cleanup task.
Expand Down
Loading