modelcontextprotocol/python-sdk
Watch
1
0
Fork
You've already forked python-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ruff format

Browse source
This commit is contained in:
TerminalMan 2024-12-29 17:08:09 +00:00
commit 99727a97ae
1 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/mcp/client/stdio.py
View file

@ -77,7 +77,8 @@ class StdioServerParameters(BaseModel):
""" """
The text encoding error handler. The text encoding error handler.
See https://docs.python.org/3/library/codecs.html#codec-base-classes for explanations of possible values See https://docs.python.org/3/library/codecs.html#codec-base-classes for
explanations of possible values
""" """
@ -108,7 +109,11 @@ async def stdio_client(server: StdioServerParameters):
try: try:
async with read_stream_writer: async with read_stream_writer:
buffer = "" buffer = ""
async for chunk in TextReceiveStream(process.stdout, encoding=server.encoding, errors=server.encoding_error_handler): async for chunk in TextReceiveStream(
process.stdout,
encoding=server.encoding,
errors=server.encoding_error_handler,
):
lines = (buffer + chunk).split("\n") lines = (buffer + chunk).split("\n")
buffer = lines.pop() buffer = lines.pop()
@ -130,7 +135,12 @@ async def stdio_client(server: StdioServerParameters):
async with write_stream_reader: async with write_stream_reader:
async for message in write_stream_reader: async for message in write_stream_reader:
json = message.model_dump_json(by_alias=True, exclude_none=True) json = message.model_dump_json(by_alias=True, exclude_none=True)
await process.stdin.send((json + "\n").encode(encoding=server.encoding, errors=server.encoding_error_handler)) await process.stdin.send(
(json + "\n").encode(
encoding=server.encoding,
errors=server.encoding_error_handler,
)
)
except anyio.ClosedResourceError: except anyio.ClosedResourceError:
await anyio.lowlevel.checkpoint() await anyio.lowlevel.checkpoint()