ruff format

Author: asheshvidyut

src/mcp/server/experimental/task_result_handler.py

@@ -10,11 +10,11 @@
 """
 
 import logging
-from typing import Any
+from typing import cast, Any
 
 import anyio
 
-from mcp.server.session import ServerTransportSession
+from mcp.server.session import ServerSession, ServerTransportSession
 from mcp.shared.exceptions import McpError
 from mcp.shared.experimental.tasks.helpers import RELATED_TASK_METADATA_KEY, is_terminal
 from mcp.shared.experimental.tasks.message_queue import TaskMessageQueue
@@ -76,6 +76,7 @@ async def send_message(
 
         This is a helper for delivering queued task messages.
         """
+        session = cast(ServerSession, session)
         await session.send_message(message)
 
     async def handle(

src/mcp/shared/session.py

@@ -155,14 +155,16 @@ def cancelled(self) -> bool:  # pragma: no cover
         return self._cancel_scope.cancel_called
 
 
-class Session(ABC,
+class Session(
+    ABC,
     Generic[
         SendRequestT,
         SendNotificationT,
         SendResultT,
         ReceiveRequestT,
         ReceiveNotificationT,
-    ]):
+    ],
+):
     """
     Base class for a session that could be inherited by
     BaseSessions for JSON-RPC (read-write stream dependent)
@@ -202,7 +204,7 @@ async def send_notification(
         """Emits a notification, which is a one-way message that does not expect a response."""
 
 
-class BaseSession(Session):
+class BaseSession(Session[SendRequestT, SendNotificationT, SendResultT, ReceiveRequestT, ReceiveNotificationT]):
     """Implements an MCP "session" on top of read/write streams, including features
     like request/response linking, notifications, and progress.
 

tests/client/test_resource_cleanup.py

@@ -7,7 +7,15 @@
 
 from mcp.shared.message import SessionMessage
 from mcp.shared.session import BaseSession, RequestId, SendResultT
-from mcp.types import ClientNotification, ClientRequest, ClientResult, EmptyResult, ErrorData, PingRequest
+from mcp.types import (
+    ProgressToken,
+    ClientNotification,
+    ClientRequest,
+    ClientResult,
+    EmptyResult,
+    ErrorData,
+    PingRequest,
+)
 
 
 @pytest.mark.anyio
@@ -24,6 +32,11 @@ async def _send_response(
         ) -> None:  # pragma: no cover
             pass
 
+        async def send_progress_notification(
+            self, progress_token: ProgressToken, progress: float, total: float | None = None, message: str | None = None
+        ) -> None:  # pragma: no cover
+            pass
+
         @property
         def _receive_request_adapter(self) -> TypeAdapter[Any]:
             return TypeAdapter(object)  # pragma: no cover