Increase websocket messaging timeout (#4235)

* increase websocket timeout * changelog
gradio-app · Jun 2, 2023 · 858f202 · 858f202
1 parent c095cf7
commit 858f202
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -20,6 +20,8 @@
 
 - Update the js client by [@pngwn](https://github.com/pngwn) in [PR 3899](https://github.com/gradio-app/gradio/pull/3899)
 - Fix documentation for the shape of the numpy array produced by the `Image` component by [@der3318](https://github.com/der3318) in [PR 4204](https://github.com/gradio-app/gradio/pull/4204).
+- Updates the timeout for websocket messaging from 1 second to 5 seconds by [@abidlabs](https://github.com/abidlabs) in [PR 4235](https://github.com/gradio-app/gradio/pull/4235)
+
 
 ## Breaking Changes:
 

diff --git a/gradio/routes.py b/gradio/routes.py
@@ -514,14 +514,14 @@ async def join_queue(
             # to create a unique id for each job
             try:
                 await asyncio.wait_for(
-                    websocket.send_json({"msg": "send_hash"}), timeout=1
+                    websocket.send_json({"msg": "send_hash"}), timeout=5
                 )
             except AsyncTimeOutError:
                 return
 
             try:
                 session_info = await asyncio.wait_for(
-                    websocket.receive_json(), timeout=1
+                    websocket.receive_json(), timeout=5
                 )
             except AsyncTimeOutError:
                 return