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."""