diff --git a/src/mcp/client/__init__.py b/src/mcp/client/__init__.py index 7b9464710..446d2109b 100644 --- a/src/mcp/client/__init__.py +++ b/src/mcp/client/__init__.py @@ -3,7 +3,4 @@ from mcp.client.client import Client from mcp.client.session import ClientSession -__all__ = [ - "Client", - "ClientSession", -] +__all__ = ["Client", "ClientSession"] diff --git a/src/mcp/client/session_group.py b/src/mcp/client/session_group.py index 825e6d66e..3ae8fa1cb 100644 --- a/src/mcp/client/session_group.py +++ b/src/mcp/client/session_group.py @@ -30,7 +30,7 @@ class SseServerParameters(BaseModel): - """Parameters for intializing a sse_client.""" + """Parameters for initializing a sse_client.""" # The endpoint URL. url: str @@ -46,7 +46,7 @@ class SseServerParameters(BaseModel): class StreamableHttpParameters(BaseModel): - """Parameters for intializing a streamable_http_client.""" + """Parameters for initializing a streamable_http_client.""" # The endpoint URL. url: str diff --git a/src/mcp/client/stdio/__init__.py b/src/mcp/client/stdio.py similarity index 100% rename from src/mcp/client/stdio/__init__.py rename to src/mcp/client/stdio.py diff --git a/src/mcp/client/websocket.py b/src/mcp/client/websocket.py index d9d0aa497..cf4b86e99 100644 --- a/src/mcp/client/websocket.py +++ b/src/mcp/client/websocket.py @@ -8,7 +8,7 @@ from websockets.asyncio.client import connect as ws_connect from websockets.typing import Subprotocol -import mcp.types as types +from mcp import types from mcp.shared.message import SessionMessage