fix: support both old and new websockets request headers format (#1588)

Co-authored-by: xujiayu <[email protected]>
k2-fsa · Dec 3, 2024 · 0d6bf52 · 0d6bf52
1 parent dc3287f
commit 0d6bf52
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/python-api-examples/non_streaming_server.py b/python-api-examples/non_streaming_server.py
@@ -641,7 +641,11 @@ async def process_request(
         path: str,
         request_headers: websockets.Headers,
     ) -> Optional[Tuple[http.HTTPStatus, websockets.Headers, bytes]]:
-        if "sec-websocket-key" not in request_headers:
+        if "sec-websocket-key" not in (
+            request_headers.headers  # For new request_headers
+            if hasattr(request_headers, "headers")
+            else request_headers  # For old request_headers
+        ):
             # This is a normal HTTP request
             if path == "/":
                 path = "/index.html"

diff --git a/python-api-examples/streaming_server.py b/python-api-examples/streaming_server.py
@@ -584,7 +584,11 @@ async def process_request(
         path: str,
         request_headers: websockets.Headers,
     ) -> Optional[Tuple[http.HTTPStatus, websockets.Headers, bytes]]:
-        if "sec-websocket-key" not in request_headers:
+        if "sec-websocket-key" not in (
+            request_headers.headers  # For new request_headers
+            if hasattr(request_headers, "headers")
+            else request_headers  # For old request_headers
+        ):
             # This is a normal HTTP request
             if path == "/":
                 path = "/index.html"