chore: lint

Signed-off-by: Shingo OKAWA <[email protected]>

Author: ognis1205

src/a2a/server/events/event_consumer.py

@@ -112,7 +112,7 @@ async def consume_all(self) -> AsyncGenerator[Event]:
                             TaskState.failed,
                             TaskState.rejected,
                             TaskState.unknown,
-                            TaskState.input_required
+                            TaskState.input_required,
                         )
                     )
                 )

src/a2a/server/request_handlers/grpc_handler.py

@@ -55,7 +55,7 @@ def __init__(
         self,
         agent_card: AgentCard,
         request_handler: RequestHandler,
-        context_builder: CallContextBuilder | None = None
+        context_builder: CallContextBuilder | None = None,
     ):
         """Initializes the GrpcHandler.
 

tests/server/events/test_event_consumer.py

@@ -223,6 +223,7 @@ async def mock_dequeue() -> Any:
     assert consumed_events[0] == events[0]
     assert mock_event_queue.task_done.call_count == 1
 
+
 @pytest.mark.asyncio
 async def test_consume_task_input_required(
     event_consumer: MagicMock,
@@ -236,8 +237,8 @@ async def mock_dequeue() -> Any:
 
     mock_event_queue.dequeue_event = mock_dequeue
     consumed_events: list[Any] = []
-    #consumer should terminate on input_required task
+    # consumer should terminate on input_required task
     async for event in event_consumer.consume_all():
         consumed_events.append(event)
     assert len(consumed_events) == 1
-    assert consumed_events[0] == task
+    assert consumed_events[0] == task