From 041222604af614bdeb23bd6ea0c4744d91f9d733 Mon Sep 17 00:00:00 2001 From: rramakrishnan2 Date: Fri, 30 May 2025 12:42:12 -0700 Subject: [PATCH] feat: adding the reject method to task_updater Author: MadaraUchiha-314 --- src/a2a/server/tasks/task_updater.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/a2a/server/tasks/task_updater.py b/src/a2a/server/tasks/task_updater.py index c079edd4..ceced88b 100644 --- a/src/a2a/server/tasks/task_updater.py +++ b/src/a2a/server/tasks/task_updater.py @@ -96,6 +96,10 @@ def complete(self, message: Message | None = None): def failed(self, message: Message | None = None): """Marks the task as failed and publishes a final status update.""" self.update_status(TaskState.failed, message=message, final=True) + + def reject(self, message: Message | None = None): + """Marks the task as rejected and publishes a final status update.""" + self.update_status(TaskState.rejected, message=message, final=True) def submit(self, message: Message | None = None): """Marks the task as submitted and publishes a status update."""