ws

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:
+            try:
-                data = await websocket.receive_bytes()
+                while True:
-                await self._recv_q.put(data)
+                    data = await websocket.receive_bytes()
+                    await self._recv_q.put(data)
+            except WebSocketDisconnect:
+                pass
 
     async def start(self):
         if self.thread != None: