fix: add remaining terminal states (#43)

kthota-g · web-flow · commit 039bf01c67a1 · 2025-05-19T09:23:08.000-05:00
diff --git 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,
                         )
                     )
                 )

Original file line number	Diff line number	Diff line change
`@@ -79,6 +79,8 @@ async def consume_all(self) -> AsyncGenerator[Event]:`
`79`	`79`	`TaskState.completed,`
`80`	`80`	`TaskState.canceled,`
`81`	`81`	`TaskState.failed,`
	`82`	`+ TaskState.rejected,`
	`83`	`+ TaskState.unknown,`
`82`	`84`	`)`
`83`	`85`	`)`
`84`	`86`	`)`