feat: Fixed build issues related to linting and formatting

Author: Rajesh Ramamoorthy

src/a2a/server/events/queue_lifecycle_manager.py

@@ -156,7 +156,7 @@ async def provision(self) -> QueueProvisionResult:
             self.region_name,
         )
 
-        async with self._resolved_session.client(
+        async with self._resolved_session.client(  # pyright: ignore[reportGeneralTypeIssues]
             'sqs', region_name=self.region_name
         ) as sqs:
             # Step 1: Create the SQS queue.
@@ -202,7 +202,7 @@ async def provision(self) -> QueueProvisionResult:
         # Step 4: Subscribe the SQS queue to the SNS topic.
         subscription_arn = ''
         try:
-            async with self._resolved_session.client(
+            async with self._resolved_session.client(  # pyright: ignore[reportGeneralTypeIssues]
                 'sns', region_name=self.region_name
             ) as sns:
                 sub_resp = await sns.subscribe(
@@ -220,7 +220,7 @@ async def provision(self) -> QueueProvisionResult:
                 queue_url,
             )
             try:
-                async with self._resolved_session.client(
+                async with self._resolved_session.client(  # pyright: ignore[reportGeneralTypeIssues]
                     'sqs', region_name=self.region_name
                 ) as sqs:
                     await sqs.delete_queue(QueueUrl=queue_url)
@@ -261,7 +261,7 @@ async def teardown(self) -> None:
 
         # Step 1: Unsubscribe from SNS (best-effort).
         try:
-            async with self._resolved_session.client(
+            async with self._resolved_session.client(  # pyright: ignore[reportGeneralTypeIssues]
                 'sns', region_name=self.region_name
             ) as sns:
                 await sns.unsubscribe(SubscriptionArn=result.subscription_arn)
@@ -276,7 +276,7 @@ async def teardown(self) -> None:
 
         # Step 2: Delete the SQS queue.
         try:
-            async with self._resolved_session.client(
+            async with self._resolved_session.client(  # pyright: ignore[reportGeneralTypeIssues]
                 'sqs', region_name=self.region_name
             ) as sqs:
                 await sqs.delete_queue(QueueUrl=result.queue_url)

src/a2a/server/events/sns_queue_manager.py

@@ -245,7 +245,7 @@ async def _sns_publish(self, message: str) -> None:
         Args:
             message: JSON string in the distributed wire format.
         """
-        async with self._session.client(
+        async with self._session.client(  # pyright: ignore[reportGeneralTypeIssues]
             'sns', region_name=self._region_name
         ) as sns:
             await sns.publish(TopicArn=self._topic_arn, Message=message)
@@ -268,7 +268,7 @@ async def _poll_loop(self) -> None:
             self._instance_id,
             self._sqs_queue_url,
         )
-        async with self._session.client(
+        async with self._session.client(  # pyright: ignore[reportGeneralTypeIssues]
             'sqs', region_name=self._region_name
         ) as sqs:
             while not self._stop_event.is_set():

src/a2a/server/tasks/dynamodb_task_store.py

@@ -84,7 +84,7 @@ async def save(
             'Saving task %s to DynamoDB table %s.', task.id, self._table_name
         )
         task_data = task.model_dump_json()
-        async with self._session.client(
+        async with self._session.client(  # pyright: ignore[reportGeneralTypeIssues]
             'dynamodb', region_name=self._region_name
         ) as client:
             await client.put_item(
@@ -113,7 +113,7 @@ async def get(
             task_id,
             self._table_name,
         )
-        async with self._session.client(
+        async with self._session.client(  # pyright: ignore[reportGeneralTypeIssues]
             'dynamodb', region_name=self._region_name
         ) as client:
             response = await client.get_item(
@@ -147,7 +147,7 @@ async def delete(
             task_id,
             self._table_name,
         )
-        async with self._session.client(
+        async with self._session.client(  # pyright: ignore[reportGeneralTypeIssues]
             'dynamodb', region_name=self._region_name
         ) as client:
             await client.delete_item(

tests/server/events/test_queue_lifecycle_manager.py

@@ -36,9 +36,7 @@ def _make_sns_client() -> AsyncMock:
     return client
 
 
-def _make_session(
-    sqs_client: AsyncMock, sns_client: AsyncMock
-) -> MagicMock:
+def _make_session(sqs_client: AsyncMock, sns_client: AsyncMock) -> MagicMock:
     session = MagicMock()
 
     def make_ctx(inner_client: AsyncMock) -> MagicMock:

tests/server/events/test_sns_queue_manager.py

@@ -292,13 +292,15 @@ async def test_poll_unwraps_sns_notification_envelope(
 ) -> None:
     """SNS notification envelope (non-raw delivery) is properly unwrapped."""
     task_data = json.loads(TASK_OBJ.model_dump_json())
-    inner = json.dumps({
-        'instance_id': 'instance-B',
-        'task_id': 'task-001',
-        'type': 'event',
-        'event_kind': 'task',
-        'event_data': task_data,
-    })
+    inner = json.dumps(
+        {
+            'instance_id': 'instance-B',
+            'task_id': 'task-001',
+            'type': 'event',
+            'event_kind': 'task',
+            'event_data': task_data,
+        }
+    )
     sns_envelope = {
         'Type': 'Notification',
         'MessageId': 'notif-001',
@@ -310,9 +312,7 @@ async def test_poll_unwraps_sns_notification_envelope(
         'ReceiptHandle': 'rh-002',
         'Body': json.dumps(sns_envelope),
     }
-    sqs_client.receive_message = AsyncMock(
-        return_value={'Messages': [sqs_msg]}
-    )
+    sqs_client.receive_message = AsyncMock(return_value={'Messages': [sqs_msg]})
 
     dist_queue = DistributedEventQueue(
         publish_fn=AsyncMock(), task_id='task-001', instance_id='instance-A'

tests/server/integration/test_distributed_stack.py

@@ -90,9 +90,7 @@ def factory(service: str, **kwargs):
     return _make(sqs_client_a), _make(sqs_client_b)
 
 
-def _wire_event(
-    instance_id: str, task_id: str, event: Task
-) -> dict[str, Any]:
+def _wire_event(instance_id: str, task_id: str, event: Task) -> dict[str, Any]:
     task_data = json.loads(event.model_dump_json())
     return {
         'instance_id': instance_id,

tests/server/tasks/test_dynamodb_task_store.py

@@ -48,6 +48,7 @@ def mock_session(dynamodb_client: AsyncMock) -> MagicMock:
 def store(mock_session: MagicMock):
     with patch.dict('sys.modules', {'aioboto3': MagicMock()}):
         from a2a.server.tasks.dynamodb_task_store import DynamoDBTaskStore
+
         s = DynamoDBTaskStore('test-table', session=mock_session)
         return s
 
@@ -145,6 +146,7 @@ async def fake_get_item(**kwargs):
 
     with patch.dict('sys.modules', {'aioboto3': MagicMock()}):
         from a2a.server.tasks.dynamodb_task_store import DynamoDBTaskStore
+
         s = DynamoDBTaskStore('test-table', session=session)
 
     await s.save(task)
@@ -162,6 +164,7 @@ def test_import_error_without_aioboto3() -> None:
         sys.modules['aioboto3'] = None  # type: ignore[assignment]
         from importlib import reload
         import a2a.server.tasks.dynamodb_task_store as mod
+
         reload(mod)
         with pytest.raises(ImportError, match='aws extra'):
             mod.DynamoDBTaskStore('test-table')