From fcfdd6f5b8ef145410749951f59c2cb058946abf Mon Sep 17 00:00:00 2001 From: gsd Date: Sun, 15 Feb 2026 21:35:23 +0300 Subject: [PATCH] ws --- transport_ws.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/transport_ws.py b/transport_ws.py index 2edf437..a8a38d0 100644 --- a/transport_ws.py +++ b/transport_ws.py @@ -13,16 +13,19 @@ class WSTransport: self.thread = None self.app = FastAPI() - + @self.app.websocket(self.path) async def websocket_endpoint(*, websocket: WebSocket, token: str = Query("")): if token != self.token: return await websocket.accept() - while True: - data = await websocket.receive_bytes() - await self._recv_q.put(data) + try: + while True: + data = await websocket.receive_bytes() + await self._recv_q.put(data) + except WebSocketDisconnect: + pass async def start(self): if self.thread != None: