From 0b237fc0333e4d7732fb312cca2d5319a335a63a Mon Sep 17 00:00:00 2001 From: Krishna Thota Date: Fri, 16 May 2025 15:55:35 -0700 Subject: [PATCH] fix: add remaining terminal states --- src/a2a/server/events/event_consumer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/a2a/server/events/event_consumer.py b/src/a2a/server/events/event_consumer.py index 05f1dde46..03ff278ae 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, ) ) )