From 3cfdea0e975c261c93b8776318e891b8609c07d7 Mon Sep 17 00:00:00 2001 From: Marcelo Trylesinski Date: Fri, 23 Jan 2026 18:35:05 +0100 Subject: [PATCH] refactor: move stdio package to its own module (#1942) --- src/mcp/client/__init__.py | 5 +---- src/mcp/client/session_group.py | 4 ++-- src/mcp/client/{stdio/__init__.py => stdio.py} | 0 src/mcp/client/websocket.py | 2 +- 4 files changed, 4 insertions(+), 7 deletions(-) rename src/mcp/client/{stdio/__init__.py => stdio.py} (100%) diff --git a/src/mcp/client/__init__.py b/src/mcp/client/__init__.py index 7b946471..446d2109 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 825e6d66..3ae8fa1c 100644 --- a/src/mcp/client/session_group.py +++ b/src/mcp/client/session_group.py @@ -30,7 +30,7 @@ from mcp.shared.session import ProgressFnT 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 d9d0aa49..cf4b86e9 100644 --- a/src/mcp/client/websocket.py +++ b/src/mcp/client/websocket.py @@ -8,7 +8,7 @@ from pydantic import ValidationError 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