ruff fixes

Author: asheshvidyut

examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/server.py

@@ -13,7 +13,6 @@
 from starlette.routing import Mount
 from starlette.types import Receive, Scope, Send
 
-
 from .event_store import InMemoryEventStore
 
 # Configure logging

examples/snippets/clients/stdio_client.py

@@ -5,8 +5,6 @@
 import asyncio
 import os
 
-from mcp import ClientSession, StdioServerParameters, types
-
 from mcp import ClientSession, ClientTransportSession, StdioServerParameters, types
 from mcp.client.stdio import stdio_client
 from mcp.shared.context import RequestContext

examples/snippets/servers/elicitation.py

@@ -11,9 +11,9 @@
 
 from mcp.server.fastmcp import Context, FastMCP
 from mcp.server.session import ServerSession
+from mcp.server.transport_session import ServerTransportSession
 from mcp.shared.exceptions import UrlElicitationRequiredError
 from mcp.types import ElicitRequestURLParams
-from mcp.server.transport_session import ServerTransportSession
 
 mcp = FastMCP(name="Elicitation Example")
 

src/mcp/__init__.py

@@ -5,9 +5,8 @@
 from .client.transport_session import ClientTransportSession
 from .server.session import ServerSession
 from .server.stdio import stdio_server
-from .shared.exceptions import McpError, UrlElicitationRequiredError
 from .server.transport_session import ServerTransportSession
-from .shared.exceptions import McpError
+from .shared.exceptions import McpError, UrlElicitationRequiredError
 from .types import (
     CallToolRequest,
     ClientCapabilities,

src/mcp/server/lowlevel/server.py

@@ -97,10 +97,10 @@ async def main():
 from mcp.server.lowlevel.helper_types import ReadResourceContents
 from mcp.server.models import InitializationOptions
 from mcp.server.session import ServerSession
-from mcp.server.transport_session import ServerTransportSession
 from mcp.server.streamable_http import EventStore
 from mcp.server.streamable_http_manager import StreamableHTTPASGIApp, StreamableHTTPSessionManager
 from mcp.server.transport_security import TransportSecuritySettings
+from mcp.server.transport_session import ServerTransportSession
 from mcp.shared.context import RequestContext
 from mcp.shared.exceptions import McpError, UrlElicitationRequiredError
 from mcp.shared.message import ServerMessageMetadata, SessionMessage

src/mcp/server/session.py

@@ -47,11 +47,11 @@ async def handle_list_prompts(ctx: RequestContext) -> list[types.Prompt]:
 import mcp.types as types
 from mcp.server.experimental.session_features import ExperimentalServerSessionFeatures
 from mcp.server.models import InitializationOptions
+from mcp.server.transport_session import ServerTransportSession
 from mcp.server.validation import validate_sampling_tools, validate_tool_use_result_messages
 from mcp.shared.exceptions import StatelessModeNotSupported
 from mcp.shared.experimental.tasks.capabilities import check_tasks_capability
 from mcp.shared.experimental.tasks.helpers import RELATED_TASK_METADATA_KEY
-from mcp.server.transport_session import ServerTransportSession
 from mcp.shared.message import ServerMessageMetadata, SessionMessage
 from mcp.shared.session import (
     BaseSession,

src/mcp/server/transport_session.py

@@ -3,13 +3,13 @@
 from abc import ABC, abstractmethod
 from typing import Any
 
-
 import mcp.types as types
 from mcp.shared.message import SessionMessage
 
 
 class ServerTransportSession(ABC):
     """Abstract base class for transport sessions."""
+
     @abstractmethod
     async def send_message(self, message: SessionMessage) -> None:
         """Send a raw session message."""

src/mcp/shared/context.py

@@ -12,8 +12,8 @@
 if TYPE_CHECKING:
     from mcp import ClientTransportSession, ServerTransportSession
 
-SessionT = TypeVar(
-    "SessionT",
+SessionT_co = TypeVar(
+    "SessionT_co",
     bound=BaseSession[Any, Any, Any, Any, Any] | "ClientTransportSession" | "ServerTransportSession",
     covariant=True,
 )
@@ -22,10 +22,10 @@
 
 
 @dataclass
-class RequestContext(Generic[SessionT, LifespanContextT, RequestT]):
+class RequestContext(Generic[SessionT_co, LifespanContextT, RequestT]):
     request_id: RequestId
     meta: RequestParamsMeta | None
-    session: SessionT
+    session: SessionT_co
     lifespan_context: LifespanContextT
     # NOTE: This is typed as Any to avoid circular imports. The actual type is
     # mcp.server.experimental.request_context.Experimental, but importing it here

src/mcp/shared/session.py

@@ -153,8 +153,10 @@ def in_flight(self) -> bool:  # pragma: no cover
     def cancelled(self) -> bool:  # pragma: no cover
         return self._cancel_scope.cancel_called
 
+
 class Session:
     """Base class for a session that can send progress notifications."""
+
     async def send_progress_notification(
         self,
         progress_token: ProgressToken,
@@ -164,6 +166,7 @@ async def send_progress_notification(
     ) -> None:
         """Sends a progress notification for a request that is currently being processed."""
 
+
 class BaseSession(
     Session,
     Generic[

tests/client/test_list_roots_callback.py

@@ -1,8 +1,8 @@
 import pytest
 from pydantic import FileUrl
 
-from mcp.client.session import ClientSession
 from mcp import Client
+from mcp.client.session import ClientSession
 from mcp.server.fastmcp import FastMCP
 from mcp.server.fastmcp.server import Context
 from mcp.server.session import ServerSession