fix linting errors

Author: kthota-g

src/a2a/server/request_handlers/default_request_handler.py

@@ -20,13 +20,14 @@
 )
 from a2a.server.request_handlers.request_handler import RequestHandler
 from a2a.server.tasks import (
-    ResultAggregator,
     PushNotificationConfigStore,
     PushNotificationSender,
+    ResultAggregator,
     TaskManager,
     TaskStore,
 )
 from a2a.types import (
+    GetTaskPushNotificationConfigParams,
     InternalError,
     Message,
     MessageSendConfiguration,
@@ -38,7 +39,6 @@
     TaskPushNotificationConfig,
     TaskQueryParams,
     UnsupportedOperationError,
-    GetTaskPushNotificationConfigParams,
 )
 from a2a.utils.errors import ServerError
 from a2a.utils.telemetry import SpanKind, trace_class
@@ -58,7 +58,7 @@ class DefaultRequestHandler(RequestHandler):
 
     _running_agents: dict[str, asyncio.Task]
 
-    def __init__(
+    def __init__(  # noqa: PLR0913
         self,
         agent_executor: AgentExecutor,
         task_store: TaskStore,
@@ -202,7 +202,7 @@ async def on_message_send(
             context=context,
         )
 
-        task_id = cast(str, request_context.task_id)
+        task_id = cast('str', request_context.task_id)
         # Always assign a task ID. We may not actually upgrade to a task, but
         # dictating the task ID at this layer is useful for tracking running
         # agents.
@@ -240,7 +240,7 @@ async def on_message_send(
         finally:
             if interrupted:
                 # TODO: Track this disconnected cleanup task.
-                asyncio.create_task(
+                asyncio.create_task( # noqa: RUF006
                     self._cleanup_producer(producer_task, task_id)
                 )
             else:
@@ -292,7 +292,7 @@ async def on_message_send_stream(
             context=context,
         )
 
-        task_id = cast(str, request_context.task_id)
+        task_id = cast('str', request_context.task_id)
         queue = await self._queue_manager.create_or_tap(task_id)
         producer_task = asyncio.create_task(
             self._run_event_stream(
@@ -383,7 +383,7 @@ async def on_get_task_push_notification_config(
     ) -> TaskPushNotificationConfig:
         """Default handler for 'tasks/pushNotificationConfig/get'.
 
-        Requires a `PushNotifier` to be configured.
+        Requires a `PushConfigStore` to be configured.
         """
         if not self._push_config_store:
             raise ServerError(error=UnsupportedOperationError())
@@ -432,6 +432,7 @@ async def on_resubscribe_to_task(
             yield event
 
     def should_add_push_info(self, params: MessageSendParams) -> bool:
+        """Determines if push notification info should be set for a task."""
         return bool(
             self._push_config_store
             and params.configuration

src/a2a/server/request_handlers/jsonrpc_handler.py

@@ -34,7 +34,6 @@
     TaskPushNotificationConfig,
     TaskResubscriptionRequest,
     TaskStatusUpdateEvent,
-    GetTaskPushNotificationConfigParams
 )
 from a2a.utils.errors import ServerError
 from a2a.utils.helpers import validate

src/a2a/server/tasks/__init__.py

@@ -1,15 +1,22 @@
 """Components for managing tasks within the A2A server."""
 
-from a2a.server.tasks.base_push_notification_sender import BasePushNotificationSender
-from a2a.server.tasks.inmemory_push_notification_config_store import InMemoryPushNotificationConfigStore
+from a2a.server.tasks.base_push_notification_sender import (
+    BasePushNotificationSender,
+)
+from a2a.server.tasks.inmemory_push_notification_config_store import (
+    InMemoryPushNotificationConfigStore,
+)
 from a2a.server.tasks.inmemory_task_store import InMemoryTaskStore
-from a2a.server.tasks.push_notification_config_store import PushNotificationConfigStore
+from a2a.server.tasks.push_notification_config_store import (
+    PushNotificationConfigStore,
+)
 from a2a.server.tasks.push_notification_sender import PushNotificationSender
 from a2a.server.tasks.result_aggregator import ResultAggregator
 from a2a.server.tasks.task_manager import TaskManager
 from a2a.server.tasks.task_store import TaskStore
 from a2a.server.tasks.task_updater import TaskUpdater
 
+
 __all__ = [
     'BasePushNotificationSender',
     'InMemoryPushNotificationConfigStore',

src/a2a/server/tasks/base_push_notification_sender.py

@@ -2,16 +2,18 @@
 
 import httpx
 
-from a2a.server.tasks.push_notification_config_store import PushNotificationConfigStore
+from a2a.server.tasks.push_notification_config_store import (
+    PushNotificationConfigStore,
+)
 from a2a.server.tasks.push_notification_sender import PushNotificationSender
 from a2a.types import Task
 
+
 logger = logging.getLogger(__name__)
 
 
 class BasePushNotificationSender(PushNotificationSender):
-    """Base implementation of PushNotificationSender interface.
-    """
+    """Base implementation of PushNotificationSender interface."""
 
     def __init__(self, httpx_client: httpx.AsyncClient, config_store: PushNotificationConfigStore) -> None:
         """Initializes the BasePushNotificationSender.

src/a2a/server/tasks/inmemory_push_notification_config_store.py

@@ -1,9 +1,12 @@
 import asyncio
 import logging
 
-from a2a.server.tasks.push_notification_config_store import PushNotificationConfigStore
+from a2a.server.tasks.push_notification_config_store import (
+    PushNotificationConfigStore,
+)
 from a2a.types import PushNotificationConfig
 
+
 logger = logging.getLogger(__name__)
 
 
@@ -24,7 +27,7 @@ def __init__(self) -> None:
 
     async def set_info(
         self, task_id: str, notification_config: PushNotificationConfig
-    ):
+    ) -> None:
         """Sets or updates the push notification configuration for a task in memory."""
         async with self.lock:
             self._push_notification_infos[task_id] = notification_config
@@ -36,7 +39,7 @@ async def get_info(self, task_id: str) -> PushNotificationConfig | None:
 
 
 
-    async def delete_info(self, task_id: str):
+    async def delete_info(self, task_id: str) -> None:
         """Deletes the push notification configuration for a task from memory."""
         async with self.lock:
             if task_id in self._push_notification_infos:

src/a2a/server/tasks/push_notification_config_store.py

@@ -7,13 +7,13 @@ class PushNotificationConfigStore(ABC):
     """Interface for storing and retrieving push notification configurations for tasks."""
 
     @abstractmethod
-    async def set_info(self, task_id: str, notification_config: PushNotificationConfig):
+    async def set_info(self, task_id: str, notification_config: PushNotificationConfig) -> None:
         """Sets or updates the push notification configuration for a task."""
 
     @abstractmethod
     async def get_info(self, task_id: str) -> PushNotificationConfig | None:
         """Retrieves the push notification configuration for a task."""
 
     @abstractmethod
-    async def delete_info(self, task_id: str):
-        """Deletes the push notification configuration for a task."""
+    async def delete_info(self, task_id: str) -> None:
+        """Deletes the push notification configuration for a task."""

src/a2a/server/tasks/push_notification_sender.py

@@ -8,4 +8,4 @@ class PushNotificationSender(ABC):
 
     @abstractmethod
     async def send_notification(self, task: Task) -> None:
-        """Sends a push notification containing the latest task state."""
+        """Sends a push notification containing the latest task state."""