Function ws_serve

runtime/python/websocket/funasr_wss_server.py:342–534

(websocket, path=None)

async def ws_serve(websocket, path=None):
    # websockets 新版本不会传 path，这里做兼容
    if path is None:
        path = getattr(websocket, "path", None)
    frames = []
    frames_asr = []
    frames_asr_online = []
    global websocket_users
    websocket_users.add(websocket)

    websocket.status_dict_asr = {}  # hotword 等
    websocket.status_dict_asr_online = {"cache": {}, "is_final": False}
    websocket.status_dict_vad = {"cache": {}, "is_final": False}
    websocket.status_dict_punc = {"cache": {}}

    websocket.chunk_interval = 10
    websocket.vad_pre_idx = 0
    speech_start = False
    speech_end_i = -1

    websocket.wav_name = "microphone"
    websocket.mode = "2pass"
    websocket.is_speaking = True  # ✅ 默认初始化，避免 AttributeError

    # 保存离线片段
    websocket.audio_fs = 16000
    websocket.offline_seg_idx = 0
    websocket.save_offline_segments = bool(args.save_offline_segments)
    websocket.offline_save_dir = args.save_offline_segments_dir
    if websocket.save_offline_segments:
        _ensure_dir(websocket.offline_save_dir)
        print(f"[SAVE_OFFLINE_SEG] enabled, dir={websocket.offline_save_dir}")

    print("new user connected", flush=True)

    try:
        async for message in websocket:
            # ========== 1) 先处理“文本配置消息” ==========
            if isinstance(message, str):
                try:
                    messagejson = json.loads(message)
                except Exception as e:
                    print("bad json message:", e, message[:200])
                    continue

                print("=============messagejson============", messagejson)

                if "is_speaking" in messagejson:
                    websocket.is_speaking = bool(messagejson["is_speaking"])
                    websocket.status_dict_asr_online["is_final"] = (not websocket.is_speaking)

                if "chunk_interval" in messagejson:
                    websocket.chunk_interval = _safe_int(
                        messagejson["chunk_interval"], websocket.chunk_interval
                    )

                if "wav_name" in messagejson:
                    websocket.wav_name = messagejson.get("wav_name") or websocket.wav_name