aljaz/mcp-python-sdk
1
0
Fork 0
mirror of https://github.com/aljazceru/mcp-python-sdk.git synced 2025-12-20 15:24:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
parent f67c3ee6c0
commit fc021eea76
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