not exporting abstractions out

Author: asheshvidyut

src/mcp/__init__.py

@@ -1,10 +1,8 @@
 from .client.session import ClientSession
 from .client.session_group import ClientSessionGroup
 from .client.stdio import StdioServerParameters, stdio_client
-from .client.transport_session import ClientTransportSession
 from .server.session import ServerSession
 from .server.stdio import stdio_server
-from .server.transport_session import ServerTransportSession
 from .shared.exceptions import McpError
 from .types import (
     CallToolRequest,
@@ -115,6 +113,4 @@
     "stdio_server",
     "CompleteRequest",
     "JSONRPCResponse",
-    "ClientTransportSession",
-    "ServerTransportSession",
 ]

src/mcp/shared/context.py

@@ -3,14 +3,12 @@
 
 from typing_extensions import TypeVar
 
-from mcp.shared.session import BaseSession
+from mcp.client.transport_session import ClientTransportSession
+from mcp.server.transport_session import ServerTransportSession
 from mcp.types import RequestId, RequestParams
 
-if TYPE_CHECKING:
-    from mcp import ClientTransportSession, ServerTransportSession
-
 SessionT = TypeVar(
-    "SessionT", bound=BaseSession[Any, Any, Any, Any, Any] | "ClientTransportSession" | "ServerTransportSession"
+    "SessionT", bound=ClientTransportSession | ServerTransportSession
 )
 LifespanContextT = TypeVar("LifespanContextT")
 RequestT = TypeVar("RequestT", default=Any)