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

Apply suggestions from code review

Browse source 
Co-authored-by: Marcelo Trylesinski <marcelotryle@gmail.com>
This commit is contained in:
Henry Mao 2025-03-06 20:12:51 +08:00 • committed by GitHub
commit fc021eea76
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -59,8 +59,7 @@ async def websocket_client(
try: try:
async for raw_text in ws: async for raw_text in ws:
try: try:
data = json.loads(raw_text) message = types.JSONRPCMessage.model_validate_json(raw_text)
message = types.JSONRPCMessage.model_validate(data)
await read_stream_send.send(message) await read_stream_send.send(message)
except Exception as exc: except Exception as exc:
# If JSON parse or model validation fails, send the exception # If JSON parse or model validation fails, send the exception