diff --git a/app/controller/add_websocket_voice_recognise_route.py b/app/controller/add_websocket_voice_recognise_route.py index 6a2ab17..bd04b6e 100644 --- a/app/controller/add_websocket_voice_recognise_route.py +++ b/app/controller/add_websocket_voice_recognise_route.py @@ -6,21 +6,21 @@ from starlette.websockets import WebSocketState def add_websocket_voice_recognise_route(app: FastAPI): @app.websocket("/ws_voice_recognise") - async def ws_voice_recognise(websocket: WebSocket, user_id: str = Query(..., description="用户ID")): + async def ws_voice_recognise(receive_socket: WebSocket, user_id: str = Query(..., description="用户ID")): print("新用户连接", user_id) # 接收连接 - await websocket.accept() + await receive_socket.accept() try: while True: # 将接收得到消息发送供其他用户 - pcm_data_string = await websocket.receive_text() + pcm_data_string = await receive_socket.receive_text() pcm_data = json.loads(pcm_data_string) print("pcm_data", pcm_data) # await person_socket.send_text(data) except Exception as e: print(f"WebSocket error: {e}") finally: - if websocket.application_state == WebSocketState.CONNECTING: + if receive_socket.application_state == WebSocketState.CONNECTING: print("关闭websocket") - await websocket.close() + await receive_socket.close() print("websocket已经关闭", user_id)