diff --git a/src/a2a/server/events/event_consumer.py b/src/a2a/server/events/event_consumer.py index 05f1dde4..03ff278a 100644 --- a/src/a2a/server/events/event_consumer.py +++ b/src/a2a/server/events/event_consumer.py @@ -79,6 +79,8 @@ async def consume_all(self) -> AsyncGenerator[Event]: TaskState.completed, TaskState.canceled, TaskState.failed, + TaskState.rejected, + TaskState.unknown, ) ) )