aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-19 14:54:24 +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 from typing import AsyncGenerator
import anyio import anyio
from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
from pydantic import ValidationError from pydantic import ValidationError
from websockets.asyncio.client import connect as ws_connect from websockets.asyncio.client import connect as ws_connect
from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
from websockets.typing import Subprotocol from websockets.typing import Subprotocol
import mcp.types as types import mcp.types as types
@@ -59,7 +59,8 @@ async def websocket_client(url: str) -> AsyncGenerator[
async def ws_writer(): 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 with write_stream_reader:
async for message in write_stream_reader: async for message in write_stream_reader:

1
tests/shared/test_ws.py
View File

@@ -8,7 +8,6 @@ import pytest
import uvicorn import uvicorn
from pydantic import AnyUrl from pydantic import AnyUrl
from starlette.applications import Starlette from starlette.applications import Starlette
from starlette.requests import Request
from starlette.routing import WebSocketRoute from starlette.routing import WebSocketRoute
from mcp.client.session import ClientSession from mcp.client.session import ClientSession