Apply suggestions from code review

calclavia · Kludex · web-flow · commit fc021eea76ef · 2025-03-06T20:12:51.000+08:00
Co-authored-by: Marcelo Trylesinski &lt;marcelotryle@gmail.com&gt;
diff --git a/src/mcp/client/websocket.py b/src/mcp/client/websocket.py
@@ -59,8 +59,7 @@ async def ws_reader():
             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
