d6b3aa9668d0211b8439fa8057b5295c1ab11f23,lib/streamlit/Proxy.py,Proxy,_client_ws_handler,#Proxy#Any#,130

Before Change

                // Watch for a CLOSE method as we sleep for throttle_secs.
                try:
                    msg = await ws.receive(timeout=throttle_secs)
                    if msg.type != WSMsgType.CLOSE:
                        raise RuntimeError(f"Unknown message type: {msg.type}")
                    break
                except asyncio.TimeoutError:
                    pass
        except concurrent.futures.CancelledError:

After Change

                    if msg.type == WSMsgType.TEXT:
                        payload = msg.json()
                        self.handle_payload(payload)
                    elif msg.type == WSMsgType.CLOSE:
                        break
                    else:
                        print("Unknown message type:", msg.type)
                except asyncio.TimeoutError:
                    pass
        except concurrent.futures.CancelledError:
            pass

Instances