fix: Event consumer should stop on input_required

kthota-g · kthota-g · commit 70ff81ae2998 · 2025-06-06T11:14:52.000-07:00
diff --git a/src/a2a/server/events/event_consumer.py b/src/a2a/server/events/event_consumer.py
@@ -112,6 +112,7 @@ async def consume_all(self) -> AsyncGenerator[Event]:
                             TaskState.failed,
                             TaskState.rejected,
                             TaskState.unknown,
+                            TaskState.input_required
                         )
                     )
                 )

Original file line number	Diff line number	Diff line change
`@@ -112,6 +112,7 @@ async def consume_all(self) -> AsyncGenerator[Event]:`
`112`	`112`	`TaskState.failed,`
`113`	`113`	`TaskState.rejected,`
`114`	`114`	`TaskState.unknown,`
	`115`	`+ TaskState.input_required`
`115`	`116`	`)`
`116`	`117`	`)`
`117`	`118`	`)`