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

Reraise exception and make websocket optional

Browse source
This commit is contained in:
Henry Mao 2025-03-08 10:52:10 +08:00
commit ea8a2dbd6d
No known key found for this signature in database
GPG key ID: 2B8D932B0174E933
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/mcp/client/websocket.py
View file

@ -56,8 +56,9 @@ async def websocket_client(url: str) -> AsyncGenerator[
except ValidationError as exc:
# If JSON parse or model validation fails, send the exception
await read_stream_writer.send(exc)
except (anyio.ClosedResourceError, Exception):
except (anyio.ClosedResourceError, Exception) as e:
await ws.close()
raise e
async def ws_writer():
"""
@ -71,8 +72,9 @@ async def websocket_client(url: str) -> AsyncGenerator[
by_alias=True, mode="json", exclude_none=True
)
await ws.send(json.dumps(msg_dict))
except (anyio.ClosedResourceError, Exception):
except (anyio.ClosedResourceError, Exception) as e:
await ws.close()
raise e
async with anyio.create_task_group() as tg:
# Start reader and writer tasks