From 8b6a1f967d6268b78412f31e605a1a55d15616d3 Mon Sep 17 00:00:00 2001 From: dongruixiao Date: Thu, 19 Jun 2025 12:07:21 +0800 Subject: [PATCH] fix: comsume_all not catch asyncio.TimeoutError in python 3.10 --- src/a2a/server/events/event_consumer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/a2a/server/events/event_consumer.py b/src/a2a/server/events/event_consumer.py index 71d7bd55..56f4c66f 100644 --- a/src/a2a/server/events/event_consumer.py +++ b/src/a2a/server/events/event_consumer.py @@ -130,6 +130,9 @@ async def consume_all(self) -> AsyncGenerator[Event]: except TimeoutError: # continue polling until there is a final event continue + except asyncio.TimeoutError: + # This class was made an alias of build-in TimeoutError after 3.11 + continue except QueueClosed: # Confirm that the queue is closed, e.g. we aren't on # python 3.12 and get a queue empty error on an open queue