From a432c295d3dc633cc56ed906c3131b035b1fd6d8 Mon Sep 17 00:00:00 2001 From: FAN-yeB <1442100690@qq.com> Date: Mon, 27 Apr 2026 15:06:40 +0800 Subject: [PATCH] =?UTF-8?q?update:=E8=A7=A3=E5=86=B3base=E7=B1=BB=E4=BF=AE?= =?UTF-8?q?=E6=94=B9=E5=90=8E=E6=B5=8B=E9=80=9F=E5=B7=A5=E5=85=B7=E5=85=BC?= =?UTF-8?q?=E5=AE=B9=E6=80=A7=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../performance_tester/performance_tester_asr.py | 2 +- .../performance_tester/performance_tester_tts.py | 16 ++++++++++++++++ 2 files changed, 17 insertions(+), 1 deletion(-) diff --git a/main/xiaozhi-server/performance_tester/performance_tester_asr.py b/main/xiaozhi-server/performance_tester/performance_tester_asr.py index db359e28..058fe4fd 100644 --- a/main/xiaozhi-server/performance_tester/performance_tester_asr.py +++ b/main/xiaozhi-server/performance_tester/performance_tester_asr.py @@ -68,7 +68,7 @@ class ASRPerformanceTester: """测试单个音频文件的性能""" try: start_time = time.time() - text, _ = await stt.speech_to_text([audio_data], "1", stt.audio_format) + text, _ = await stt.speech_to_text_wrapper([audio_data], "1", stt.audio_format) if text is None: return None diff --git a/main/xiaozhi-server/performance_tester/performance_tester_tts.py b/main/xiaozhi-server/performance_tester/performance_tester_tts.py index 61e49682..d0c47825 100644 --- a/main/xiaozhi-server/performance_tester/performance_tester_tts.py +++ b/main/xiaozhi-server/performance_tester/performance_tester_tts.py @@ -2,6 +2,7 @@ import asyncio import logging import os import time +import threading from typing import Dict import yaml from tabulate import tabulate @@ -44,6 +45,21 @@ class TTSPerformanceTester: module_type = config.get("type", tts_name) tts = create_tts_instance(module_type, config, delete_audio_file=True) + # 设置 mock conn 对象，避免 TTS 实现访问 self.conn.sample_rate 时为 None + class MockConn: + sample_rate = 16000 + audio_format = "pcm" + stop_event = threading.Event() # 需要是真正的 Event 对象 + client_abort = False + headers = {} + tts.conn = MockConn() + + # 设置 mock opus_encoder，避免某些 TTS 访问 self.opus_encoder 时为 None + class MockOpusEncoder: + pass + if not hasattr(tts, 'opus_encoder') or tts.opus_encoder is None: + tts.opus_encoder = MockOpusEncoder() + print(f"测试 TTS: {tts_name}") # 连接测试