From 1981c010c659ec4792ec2c043d1ff3567b8f25c3 Mon Sep 17 00:00:00 2001 From: Freddy Boulton Date: Tue, 4 Apr 2023 06:57:44 -0700 Subject: [PATCH] Access http token for ws connection (#3735) * Access unsecure token * CHANGELOG --------- Co-authored-by: Abubakar Abid --- CHANGELOG.md | 1 + gradio/routes.py | 4 +++- 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 3a19c10889..817227e5ae 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -8,6 +8,7 @@ No changes to highlight. ## Bug Fixes: - Increase timeout for sending analytics data by [@dawoodkhan82](https://github.com/dawoodkhan82) in [PR 3647](https://github.com/gradio-app/gradio/pull/3647) +- Fix bug where http token was not accessed over websocket connections by [@freddyaboulton](https://github.com/freddyaboulton) in [PR 3735](https://github.com/gradio-app/gradio/pull/3735) ## Documentation Changes: diff --git a/gradio/routes.py b/gradio/routes.py index 9f3776547a..15b56bc041 100644 --- a/gradio/routes.py +++ b/gradio/routes.py @@ -167,7 +167,9 @@ class App(FastAPI): ) async def ws_login_check(websocket: WebSocket) -> Optional[str]: - token = websocket.cookies.get("access-token") + token = websocket.cookies.get("access-token") or websocket.cookies.get( + "access-token-unsecure" + ) return token # token is returned to allow request in queue @app.get("/token")