Apply suggestions from code review

Co-authored-by: Marcelo Trylesinski <marcelotryle@gmail.com>
2025-03-06 20:12:51 +08:00 · 2025-03-06 20:12:51 +08:00 · fc021eea76
commit fc021eea76
parent f67c3ee6c0
1 changed files with 1 additions and 2 deletions
--- a/src/mcp/client/websocket.py
+++ b/src/mcp/client/websocket.py
@ -59,8 +59,7 @@ async def websocket_client(
            try:
                async for raw_text in ws:
                    try:
-                        data = json.loads(raw_text)
-                        message = types.JSONRPCMessage.model_validate(data)
+                        message = types.JSONRPCMessage.model_validate_json(raw_text)
                        await read_stream_send.send(message)
                    except Exception as exc:
                        # If JSON parse or model validation fails, send the exception