codebytemirza
diff --git a/‎src/mcp/client/websocket.py‎
Lines changed: 3 additions & 2 deletions b/‎src/mcp/client/websocket.py‎
Lines changed: 3 additions & 2 deletions
diff --git a/‎tests/shared/test_ws.py‎
Lines changed: 0 additions & 1 deletion b/‎tests/shared/test_ws.py‎
Lines changed: 0 additions & 1 deletion
@@ -4,9 +4,9 @@
 from typing import AsyncGenerator
 
 import anyio
+from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from pydantic import ValidationError
 from websockets.asyncio.client import connect as ws_connect
-from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from websockets.typing import Subprotocol
 
 import mcp.types as types
@@ -59,7 +59,8 @@ async def ws_reader():
 
         async def ws_writer():
             """
-            Reads JSON-RPC messages from write_stream_reader and sends them to the server.
+            Reads JSON-RPC messages from write_stream_reader and 
+            sends them to the server.
             """
             async with write_stream_reader:
                 async for message in write_stream_reader:
 
@@ -8,7 +8,6 @@
 import uvicorn
 from pydantic import AnyUrl
 from starlette.applications import Starlette
-from starlette.requests import Request
 from starlette.routing import WebSocketRoute
 
 from mcp.client.session import ClientSession