diff --git a/gateway/src/desklock_gateway/main.py b/gateway/src/desklock_gateway/main.py index 1c4169d..fe0e1d3 100644 --- a/gateway/src/desklock_gateway/main.py +++ b/gateway/src/desklock_gateway/main.py @@ -54,6 +54,8 @@ async def voice(ws: WebSocket) -> None: try: while True: message = await ws.receive() + if message["type"] == "websocket.disconnect": + raise WebSocketDisconnect(message.get("code", 1000)) device["last_seen"] = _now() if "bytes" in message and message["bytes"] is not None: if recording: