Fix linting errors

Author: kthota-g

src/a2a/server/apps/jsonrpc/jsonrpc_app.py

@@ -25,6 +25,7 @@
     A2ARequest,
     AgentCard,
     CancelTaskRequest,
+    DeleteTaskPushNotificationConfigRequest,
     GetTaskPushNotificationConfigRequest,
     GetTaskRequest,
     InternalError,
@@ -33,14 +34,13 @@
     JSONRPCError,
     JSONRPCErrorResponse,
     JSONRPCResponse,
+    ListTaskPushNotificationConfigRequest,
     SendMessageRequest,
     SendStreamingMessageRequest,
     SendStreamingMessageResponse,
     SetTaskPushNotificationConfigRequest,
     TaskResubscriptionRequest,
     UnsupportedOperationError,
-    ListTaskPushNotificationConfigRequest,
-    DeleteTaskPushNotificationConfigRequest
 )
 from a2a.utils.errors import MethodNotImplementedError
 

src/a2a/server/request_handlers/default_request_handler.py

@@ -27,9 +27,11 @@
     TaskStore,
 )
 from a2a.types import (
+    DeleteTaskPushNotificationConfigParams,
     GetTaskPushNotificationConfigParams,
     InternalError,
     InvalidParamsError,
+    ListTaskPushNotificationConfigParams,
     Message,
     MessageSendConfiguration,
     MessageSendParams,
@@ -41,10 +43,6 @@
     TaskQueryParams,
     TaskState,
     UnsupportedOperationError,
-    ListTaskPushNotificationConfigParams,
-    DeleteTaskPushNotificationConfigParams,
-    DeleteTaskPushNotificationConfigResponse,
-    DeleteTaskPushNotificationConfigSuccessResponse
 )
 from a2a.utils.errors import ServerError
 from a2a.utils.telemetry import SpanKind, trace_class
@@ -487,7 +485,7 @@ async def on_list_task_push_notification_config(
                 task_push_notification_config.append(TaskPushNotificationConfig(
                     taskId=params.id, pushNotificationConfig=config
                 ))
-        
+
         return task_push_notification_config
 
     async def on_delete_task_push_notification_config(

src/a2a/server/request_handlers/jsonrpc_handler.py

@@ -10,6 +10,9 @@
     CancelTaskRequest,
     CancelTaskResponse,
     CancelTaskSuccessResponse,
+    DeleteTaskPushNotificationConfigRequest,
+    DeleteTaskPushNotificationConfigResponse,
+    DeleteTaskPushNotificationConfigSuccessResponse,
     GetTaskPushNotificationConfigRequest,
     GetTaskPushNotificationConfigResponse,
     GetTaskPushNotificationConfigSuccessResponse,
@@ -18,6 +21,9 @@
     GetTaskSuccessResponse,
     InternalError,
     JSONRPCErrorResponse,
+    ListTaskPushNotificationConfigRequest,
+    ListTaskPushNotificationConfigResponse,
+    ListTaskPushNotificationConfigSuccessResponse,
     Message,
     SendMessageRequest,
     SendMessageResponse,
@@ -34,12 +40,6 @@
     TaskPushNotificationConfig,
     TaskResubscriptionRequest,
     TaskStatusUpdateEvent,
-    ListTaskPushNotificationConfigRequest,
-    ListTaskPushNotificationConfigResponse,
-    ListTaskPushNotificationConfigSuccessResponse,
-    DeleteTaskPushNotificationConfigRequest,
-    DeleteTaskPushNotificationConfigResponse,
-    DeleteTaskPushNotificationConfigSuccessResponse
 )
 from a2a.utils.errors import ServerError
 from a2a.utils.helpers import validate
@@ -365,7 +365,7 @@ async def list_push_notification_config(
                     id=request.id, error=e.error if e.error else InternalError()
                 )
             )
-        
+
     async def delete_push_notification_config(
         self,
         request: DeleteTaskPushNotificationConfigRequest,

src/a2a/server/request_handlers/request_handler.py

@@ -4,16 +4,16 @@
 from a2a.server.context import ServerCallContext
 from a2a.server.events.event_queue import Event
 from a2a.types import (
+    DeleteTaskPushNotificationConfigParams,
     GetTaskPushNotificationConfigParams,
+    ListTaskPushNotificationConfigParams,
     Message,
     MessageSendParams,
     Task,
     TaskIdParams,
     TaskPushNotificationConfig,
     TaskQueryParams,
     UnsupportedOperationError,
-    ListTaskPushNotificationConfigParams,
-    DeleteTaskPushNotificationConfigParams
 )
 from a2a.utils.errors import ServerError
 

src/a2a/server/request_handlers/response_helpers.py

@@ -7,13 +7,17 @@
     A2AError,
     CancelTaskResponse,
     CancelTaskSuccessResponse,
+    DeleteTaskPushNotificationConfigResponse,
+    DeleteTaskPushNotificationConfigSuccessResponse,
     GetTaskPushNotificationConfigResponse,
     GetTaskPushNotificationConfigSuccessResponse,
     GetTaskResponse,
     GetTaskSuccessResponse,
     InvalidAgentResponseError,
     JSONRPCError,
     JSONRPCErrorResponse,
+    ListTaskPushNotificationConfigResponse,
+    ListTaskPushNotificationConfigSuccessResponse,
     Message,
     SendMessageResponse,
     SendMessageSuccessResponse,
@@ -25,10 +29,6 @@
     TaskArtifactUpdateEvent,
     TaskPushNotificationConfig,
     TaskStatusUpdateEvent,
-    ListTaskPushNotificationConfigResponse,
-    DeleteTaskPushNotificationConfigResponse,
-    ListTaskPushNotificationConfigSuccessResponse,
-    DeleteTaskPushNotificationConfigSuccessResponse
 )
 
 

src/a2a/server/tasks/base_push_notification_sender.py

@@ -6,7 +6,7 @@
     PushNotificationConfigStore,
 )
 from a2a.server.tasks.push_notification_sender import PushNotificationSender
-from a2a.types import Task, PushNotificationConfig
+from a2a.types import PushNotificationConfig, Task
 
 
 logger = logging.getLogger(__name__)
@@ -30,7 +30,7 @@ async def send_notification(self, task: Task) -> None:
         push_configs = await self._config_store.get_info(task.id)
         if not push_configs:
             return
-        
+
         for push_info in push_configs:
             await self._dispatch_notification(task, push_info)
 

src/a2a/server/tasks/inmemory_push_notification_config_store.py

@@ -27,10 +27,10 @@ async def set_info(
         async with self.lock:
             if task_id not in self._push_notification_infos:
                 self._push_notification_infos[task_id] = []
-            
+
             if notification_config.id is None:
                 notification_config.id = task_id
-            
+
             for config in self._push_notification_infos[task_id]:
                 if config.id == notification_config.id:
                     self._push_notification_infos[task_id].remove(config)
@@ -54,7 +54,7 @@ async def delete_info(self, task_id: str, config_id: str | None = None) -> None:
                 configurations = self._push_notification_infos[task_id]
                 if not configurations:
                     return
-                
+
                 for config in configurations:
                     if config.id == config_id:
                         configurations.remove(config)