rename classes

Author: asheshvidyut

src/mcp/__init__.py

@@ -1,4 +1,6 @@
 from .client.session import ClientSession
+from .client.transport_session import ClientTransportSession
+from .server.transport_session import ServerTransportSession
 from .client.session_group import ClientSessionGroup
 from .client.stdio import StdioServerParameters, stdio_client
 from .server.session import ServerSession
@@ -113,4 +115,6 @@
     "stdio_server",
     "CompleteRequest",
     "JSONRPCResponse",
+    "ClientTransportSession",
+    "ServerTransportSession",
 ]

src/mcp/client/session.py

@@ -9,7 +9,7 @@
 from typing_extensions import deprecated
 
 import mcp.types as types
-from mcp.client.transport_session import TransportSession
+from mcp.client.transport_session import ClientTransportSession
 from mcp.shared.context import RequestContext
 from mcp.shared.message import SessionMessage
 from mcp.shared.session import BaseSession, ProgressFnT, RequestResponder
@@ -101,7 +101,7 @@ async def _default_logging_callback(
 
 
 class ClientSession(
-    TransportSession,
+    ClientTransportSession,
     BaseSession[
         types.ClientRequest,
         types.ClientNotification,

src/mcp/client/transport_session.py

@@ -8,7 +8,7 @@
 from mcp.shared.session import ProgressFnT
 
 
-class TransportSession(ABC):
+class ClientTransportSession(ABC):
     """Abstract base class for communication transports."""
 
     @abstractmethod

src/mcp/server/session.py

@@ -47,7 +47,7 @@ async def handle_list_prompts(ctx: RequestContext) -> list[types.Prompt]:
 
 import mcp.types as types
 from mcp.server.models import InitializationOptions
-from mcp.server.transport_session import TransportSession
+from mcp.server.transport_session import ServerTransportSession
 from mcp.shared.message import ServerMessageMetadata, SessionMessage
 from mcp.shared.session import (
     BaseSession,
@@ -70,7 +70,7 @@ class InitializationState(Enum):
 
 
 class ServerSession(
-    TransportSession,
+    ServerTransportSession,
     BaseSession[
         types.ServerRequest,
         types.ServerNotification,

src/mcp/server/transport_session.py

@@ -1,17 +1,17 @@
 """Abstract base class for transport sessions."""
 
-import abc
+from abc import ABC, abstractmethod
 from typing import Any
 
 from pydantic import AnyUrl
 
 import mcp.types as types
 
 
-class TransportSession(abc.ABC):
+class ServerTransportSession(ABC):
     """Abstract base class for transport sessions."""
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_log_message(
         self,
         level: types.LoggingLevel,
@@ -22,17 +22,17 @@ async def send_log_message(
         """Send a log message notification."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_resource_updated(self, uri: AnyUrl) -> None:
         """Send a resource updated notification."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def list_roots(self) -> types.ListRootsResult:
         """Send a roots/list request."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def elicit(
         self,
         message: str,
@@ -42,12 +42,12 @@ async def elicit(
         """Send an elicitation/create request."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_ping(self) -> types.EmptyResult:
         """Send a ping request."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_progress_notification(
         self,
         progress_token: str | int,
@@ -59,17 +59,17 @@ async def send_progress_notification(
         """Send a progress notification."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_resource_list_changed(self) -> None:
         """Send a resource list changed notification."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_tool_list_changed(self) -> None:
         """Send a tool list changed notification."""
         raise NotImplementedError
 
-    @abc.abstractmethod
+    @abstractmethod
     async def send_prompt_list_changed(self) -> None:
         """Send a prompt list changed notification."""
         raise NotImplementedError