aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-20 15:24:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Ruff

Browse Source
This commit is contained in:
Henry Mao
2025-03-12 13:51:31 -07:00
parent 2bcca39aae
commit ba184a2667
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/mcp/client/websocket.py
View File

@@ -4,9 +4,9 @@ from contextlib import asynccontextmanager
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 websocket_client(url: str) -> AsyncGenerator[
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: