Formatting

Author: holtskinner

examples/google_adk/birthday_planner/adk_agent_executor.py

@@ -99,7 +99,7 @@ def _run_agent(
         session_id,
         new_message: types.Content,
         task_updater: TaskUpdater,
-    ) -> AsyncGenerator[Event, None]:
+    ) -> AsyncGenerator[Event]:
         return self.runner.run_async(
             session_id=session_id,
             user_id='self',

src/a2a/server/events/event_consumer.py

@@ -85,7 +85,7 @@ async def consume_all(self) -> AsyncGenerator[Event]:
                     logger.debug('Stopping event consumption in consume_all.')
                     self.queue.close()
                     break
-            except asyncio.TimeoutError:
+            except TimeoutError:
                 # continue polling until there is a final event
                 continue
             except asyncio.QueueShutDown:

src/a2a/server/request_handlers/default_request_handler.py

@@ -161,7 +161,7 @@ async def on_message_send(
 
     async def on_message_send_stream(
         self, params: MessageSendParams
-    ) -> AsyncGenerator[Event, None]:
+    ) -> AsyncGenerator[Event]:
         """Default handler for 'message/stream'."""
         task_manager = TaskManager(
             task_id=params.message.taskId,
@@ -232,7 +232,7 @@ async def on_get_task_push_notification_config(
 
     async def on_resubscribe_to_task(
         self, params: TaskIdParams
-    ) -> AsyncGenerator[Event, None]:
+    ) -> AsyncGenerator[Event]:
         """Default handler for 'tasks/resubscribe'."""
         task: Task | None = await self.task_store.get(params.id)
         if not task:

src/a2a/server/tasks/result_aggregator.py

@@ -36,7 +36,7 @@ async def current_result(self) -> Task | Message | None:
 
     async def consume_and_emit(
         self, consumer: EventConsumer
-    ) -> AsyncGenerator[Event, None]:
+    ) -> AsyncGenerator[Event]:
         """Processes the event stream and emits the same event stream out."""
         async for event in consumer.consume_all():
             await self.task_manager.process(event)