MirrorMirror
/
01
mirror of https://github.com/OpenInterpreter/01.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: don't output if websocket disconnected normally

This commit is contained in:
Shiven Mian 2024-02-13 20:09:08 -08:00
parent 86975c4026
commit 818597caa9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
01OS/01OS/server/server.py
View File

@ -9,7 +9,7 @@ import traceback
import re import re
from fastapi import FastAPI from fastapi import FastAPI
from fastapi.responses import PlainTextResponse from fastapi.responses import PlainTextResponse
from starlette.websockets import WebSocket from starlette.websockets import WebSocket, WebSocketDisconnect
from .stt.stt import stt_bytes from .stt.stt import stt_bytes
from .tts.tts import tts from .tts.tts import tts
from pathlib import Path from pathlib import Path
@ -99,6 +99,8 @@ async def websocket_endpoint(websocket: WebSocket):
send_task = asyncio.create_task(send_messages(websocket)) send_task = asyncio.create_task(send_messages(websocket))
try: try:
await asyncio.gather(receive_task, send_task) await asyncio.gather(receive_task, send_task)
except WebSocketDisconnect:
pass
except Exception as e: except Exception as e:
traceback.print_exc() traceback.print_exc()
logger.info(f"Connection lost. Error: {e}") logger.info(f"Connection lost. Error: {e}")