From 9d65e5ac03733964feb620259af1e70e252387b7 Mon Sep 17 00:00:00 2001
From: Henry Mao <1828968+calclavia@users.noreply.github.com>
Date: Fri, 7 Mar 2025 22:02:16 +0800
Subject: [PATCH] Remove optional check

---
 src/mcp/client/websocket.py | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/src/mcp/client/websocket.py b/src/mcp/client/websocket.py
index d8a726a..62c777f 100644
--- a/src/mcp/client/websocket.py
+++ b/src/mcp/client/websocket.py
@@ -42,12 +42,6 @@ async def websocket_client(url: str) -> AsyncGenerator[
 
     # Connect using websockets, requesting the "mcp" subprotocol
     async with ws_connect(url, subprotocols=[Subprotocol("mcp")]) as ws:
-        # Optional check to ensure the server actually accepted "mcp"
-        if ws.subprotocol != "mcp":
-            raise ValueError(
-                f"Server did not accept subprotocol 'mcp'. Actual subprotocol: {ws.subprotocol}"
-            )
-
         async def ws_reader():
             """
             Reads text messages from the WebSocket, parses them as JSON-RPC messages,