clientsession and server session to implement abstract classes

Author: asheshvidyut

src/mcp/client/session.py

@@ -14,6 +14,8 @@
 from mcp.shared.session import BaseSession, ProgressFnT, RequestResponder
 from mcp.shared.version import SUPPORTED_PROTOCOL_VERSIONS
 
+from src.mcp.client.transport_session import TransportSession
+
 DEFAULT_CLIENT_INFO = types.Implementation(name="mcp", version="0.1.0")
 
 logger = logging.getLogger("client")
@@ -100,6 +102,7 @@ async def _default_logging_callback(
 
 
 class ClientSession(
+    TransportSession,
     BaseSession[
         types.ClientRequest,
         types.ClientNotification,

src/mcp/server/session.py

@@ -54,6 +54,8 @@ async def handle_list_prompts(ctx: RequestContext) -> list[types.Prompt]:
 )
 from mcp.shared.version import SUPPORTED_PROTOCOL_VERSIONS
 
+from src.mcp.server.transport_session import TransportSession
+
 
 class InitializationState(Enum):
     NotInitialized = 1
@@ -69,6 +71,7 @@ class InitializationState(Enum):
 
 
 class ServerSession(
+    TransportSession,
     BaseSession[
         types.ServerRequest,
         types.ServerNotification,