EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: xujiayu <xujiayu@kaihong.com>
This commit is contained in:
JiayuXu
2024-12-03 17:22:12 +08:00
committed by GitHub
parent dc3287f3a8
commit 0d6bf52844
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python-api-examples/non_streaming_server.py
View File

@@ -641,7 +641,11 @@ class NonStreamingServer:
path: str, path: str,
request_headers: websockets.Headers, request_headers: websockets.Headers,
) -> Optional[Tuple[http.HTTPStatus, websockets.Headers, bytes]]: ) -> 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 # This is a normal HTTP request
if path == "/": if path == "/":
path = "/index.html" path = "/index.html"

6
python-api-examples/streaming_server.py
View File

@@ -584,7 +584,11 @@ class StreamingServer(object):
path: str, path: str,
request_headers: websockets.Headers, request_headers: websockets.Headers,
) -> Optional[Tuple[http.HTTPStatus, websockets.Headers, bytes]]: ) -> 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 # This is a normal HTTP request
if path == "/": if path == "/":
path = "/index.html" path = "/index.html"