Compare commits

..
3 Commits
Author SHA1 Message Date
2d01812f8d Performance test (#118)
* update: optimize performance test files

* update:异步测试方法

* update:改回默认FunASR

---------

Co-authored-by: Alexisxty <alexisty233@gmail.com>
Co-authored-by: hrz <1710360675@qq.com>
2025-02-23 16:22:21 +08:00
3bc0b821a1 update: 修改了服务组件的性能测试代码 (#111)
* update: optimize performance test files

* update:异步测试方法

---------

Co-authored-by: hrz <1710360675@qq.com>
2025-02-23 16:07:33 +08:00
148578399f 使用配置文件设置日志参数 (#116)
* 使用配置文件设置日志参数

* update:优化

---------

Co-authored-by: 欣南科技 <huangrongzhuang@xin-nan.com>
Co-authored-by: hrz <1710360675@qq.com>
2025-02-23 15:18:59 +08:00
6 changed files with 480 additions and 238 deletions
+10 -16
View File
@@ -167,9 +167,10 @@ server:
### ASR ### ASR
| 类型 | 平台名称 | 使用方式 | 收费模式 | 备注 | | 类型 | 平台名称 | 使用方式 | 收费模式 | 备注 |
|:---:|:------:|:----:|:----:|:--:| |:---:|:---------:|:----:|:----:|:--:|
| ASR | FunASR | 本地使用 | 免费 | | | ASR | FunASR | 本地使用 | 免费 | |
| ASR | DoubaoASR | 接口调用 | 收费 | |
--- ---
@@ -179,22 +180,15 @@ server:
本项目支持以下三种部署方式,您可根据实际需求选择: 本项目支持以下三种部署方式,您可根据实际需求选择:
1. **[Docker 快速部署](./docs/Deployment.md)** 1. [Docker 快速部署](./docs/Deployment.md)
适合快速体验,不需过多环境配置。缺点是,拉取镜像有点慢。 适合快速体验,不需过多环境配置。缺点是,拉取镜像有点慢。
2.
*
*[借助 Docker 环境运行部署](./docs/Deployment.md#%E6%96%B9%E5%BC%8F%E4%BA%8C%E5%80%9F%E5%8A%A9docker%E7%8E%AF%E5%A2%83%E8%BF%90%E8%A1%8C%E9%83%A8%E7%BD%B2) 2. [借助 Docker 环境运行部署](./docs/Deployment.md#%E6%96%B9%E5%BC%8F%E4%BA%8C%E5%80%9F%E5%8A%A9docker%E7%8E%AF%E5%A2%83%E8%BF%90%E8%A1%8C%E9%83%A8%E7%BD%B2)
** 适用于已安装 Docker 且希望对代码进行自定义修改的用户。
适用于已安装 Docker 且希望对代码进行自定义修改的用户。
3. 3. [本地源码运行](./docs/Deployment.md#%E6%96%B9%E5%BC%8F%E4%B8%89%E6%9C%AC%E5%9C%B0%E6%BA%90%E7%A0%81%E8%BF%90%E8%A1%8C)
* 适合熟悉 Conda 环境或希望从零搭建运行环境的用户。
对于对响应速度要求较高的场景,推荐使用本地源码运行方式以降低额外开销。
*[本地源码运行](./docs/Deployment.md#%E6%96%B9%E5%BC%8F%E4%B8%89%E6%9C%AC%E5%9C%B0%E6%BA%90%E7%A0%81%E8%BF%90%E8%A1%8C)
**
适合熟悉 Conda 环境或希望从零搭建运行环境的用户。
对于对响应速度要求较高的场景,推荐使用本地源码运行方式以降低额外开销。
### 二、[固件编译](./docs/firmware-build.md) ### 二、[固件编译](./docs/firmware-build.md)
+14 -1
View File
@@ -22,6 +22,19 @@ server:
# 可选:设备白名单,如果设置了白名单,那么白名单的机器无论是什么token都可以连接。 # 可选:设备白名单,如果设置了白名单,那么白名单的机器无论是什么token都可以连接。
#allowed_devices: #allowed_devices:
# - "24:0A:C4:1D:3B:F0" # MAC地址列表 # - "24:0A:C4:1D:3B:F0" # MAC地址列表
log:
# 设置控制台输出的日志格式,时间、日志级别、标签、消息
log_format: "<green>{time:YY-MM-DD HH:mm:ss}</green>[<light-blue>{extra[tag]}</light-blue>] - <level>{level}</level> - <light-green>{message}</light-green>"
# 设置日志文件输出的格式,时间、日志级别、标签、消息
log_format_simple: "{time:YYYY-MM-DD HH:mm:ss} - {name} - {level} - {extra[tag]} - {message}"
# 设置日志等级:INFO、DEBUG
log_level: INFO
# 设置日志路径
log_dir: tmp
# 设置日志文件
log_file: "server.log"
# 设置数据文件路径
data_dir: data
manager: manager:
# 是否启用管理后台 # 是否启用管理后台
# 目前这个模块还在开发中,建议:不要修改enabled选项 # 目前这个模块还在开发中,建议:不要修改enabled选项
@@ -59,7 +72,7 @@ CMD_exit:
# 具体处理时选择的模块(The module selected for specific processing) # 具体处理时选择的模块(The module selected for specific processing)
selected_module: selected_module:
ASR: DoubaoASR ASR: FunASR
VAD: SileroVAD VAD: SileroVAD
# 将根据配置名称对应的type调用实际的LLM适配器 # 将根据配置名称对应的type调用实际的LLM适配器
LLM: ChatGLMLLM LLM: ChatGLMLLM
+14 -12
View File
@@ -1,27 +1,29 @@
import os import os
import sys import sys
from loguru import logger from loguru import logger
from config.settings import load_config
def setup_logging():
"""从配置文件中读取日志配置,并设置日志输出格式和级别"""
config = load_config()
log_config = config["log"]
log_format = log_config.get("log_format", "<green>{time:YY-MM-DD HH:mm:ss}</green>[<light-blue>{extra[tag]}</light-blue>] - <level>{level}</level> - <light-green>{message}</light-green>")
log_format_simple = log_config.get("log_format_file", "{time:YYYY-MM-DD HH:mm:ss} - {name} - {level} - {extra[tag]} - {message}")
log_level = log_config.get("log_level", "INFO")
log_dir = log_config.get("log_dir", "tmp")
log_file = log_config.get("log_file", "server.log")
data_dir = log_config.get("data_dir", "data")
def setup_logging(log_dir='tmp', data_dir='data'):
"""配置全局彩色日志(不同区块不同标签)"""
os.makedirs(log_dir, exist_ok=True) os.makedirs(log_dir, exist_ok=True)
os.makedirs(data_dir, exist_ok=True) os.makedirs(data_dir, exist_ok=True)
# 设置日志格式,时间、日志级别、标签、消息
log_format = (
"<green>{time:YY-MM-DD HH:mm:ss}</green>"
"[<light-blue>{extra[tag]}</light-blue>]"
" - <level>{level}</level> - "
"<light-green>{message}</light-green>"
)
# 配置日志输出 # 配置日志输出
logger.remove() logger.remove()
# 输出到控制台 # 输出到控制台
logger.add(sys.stdout, format=log_format, level="INFO") logger.add(sys.stdout, format=log_format, level=log_level)
# 输出到文件 # 输出到文件
logger.add(os.path.join(log_dir, "server.log"), format="{time:YYYY-MM-DD HH:mm:ss} - {name} - {level} - {extra[tag]} - {message}", level="INFO") logger.add(os.path.join(log_dir, log_file), format=log_format_simple, level=log_level)
return logger return logger
+112 -28
View File
@@ -1,5 +1,13 @@
import os import os
import sys import sys
import asyncio
from typing import List, Dict, Any
# 添加项目根目录到Python路径
current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.abspath(os.path.join(current_dir, "..", ".."))
sys.path.insert(0, project_root)
from config.logger import setup_logging from config.logger import setup_logging
import importlib import importlib
from datetime import datetime from datetime import datetime
@@ -21,39 +29,115 @@ def create_instance(class_name, *args, **kwargs):
raise ValueError(f"不支持的LLM类型: {class_name},请检查该配置的type是否设置正确") raise ValueError(f"不支持的LLM类型: {class_name},请检查该配置的type是否设置正确")
if __name__ == "__main__": async def test_single_model(llm_name: str, llm_config: Dict[str, Any], test_prompt: str, config: Dict[str, Any]) -> Dict[str, Any]:
"""异步测试单个模型"""
try:
# 获取实际的LLM类型
llm_type = llm_config["type"] if "type" in llm_config else llm_name
llm = create_instance(llm_type, llm_config)
# 开始测试
dialogue = []
dialogue.append({"role": "system", "content": config.get("prompt")})
dialogue.append({"role": "user", "content": test_prompt})
start_time = datetime.now()
llm_responses = llm.response("test", dialogue)
response_message = []
first_response_time = None
total_response_time = None
start = 0
full_response = ""
for content in llm_responses:
response_message.append(content)
full_response += content
if is_segment(response_message):
segment_text = "".join(response_message[start:])
segment_text = get_string_no_punctuation_or_emoji(segment_text)
if len(segment_text) > 0:
if first_response_time is None:
first_response_time = (datetime.now() - start_time).total_seconds()
start = len(response_message)
total_response_time = (datetime.now() - start_time).total_seconds()
return {
"name": llm_name,
"type": llm_type,
"first_response_time": first_response_time,
"total_response_time": total_response_time,
"response_length": len(full_response),
"status": "成功",
"response": full_response
}
except Exception as e:
print(f"测试 {llm_name} 时发生错误: {str(e)}")
return {
"name": llm_name,
"type": llm_config.get("type", llm_name),
"first_response_time": None,
"total_response_time": None,
"response_length": 0,
"status": f"失败 - {str(e)}",
"response": ""
}
async def main():
""" """
响应速度测试 LLM模型响应速度测试和排行(异步版本)
""" """
config = read_config(get_project_dir() + "config.yaml") config = read_config(get_project_dir() + "config.yaml")
llm = create_instance( test_prompt = "你好小智"
config["selected_module"]["LLM"]
if not "type" in config["LLM"][config["selected_module"]["LLM"]]
else
config["LLM"][config["selected_module"]["LLM"]]["type"],
config["LLM"][config["selected_module"]["LLM"]]
)
start_time = datetime.now() print("开始并发测试所有模型...")
dialogue = [] # 创建所有模型的测试任务
dialogue.append({"role": "system", "content": config.get("prompt")}) tasks = []
dialogue.append({"role": "user", "content": "你好小智"}) for llm_name, llm_config in config["LLM"].items():
llm_responses = llm.response("test", dialogue) task = asyncio.create_task(test_single_model(llm_name, llm_config, test_prompt, config))
response_message = [] tasks.append(task)
first_text = None
start = 0
for content in llm_responses: # 等待所有测试完成
response_message.append(content) test_results = await asyncio.gather(*tasks)
if is_segment(response_message): # 打印测试结果排行榜
segment_text = "".join(response_message[start:]) print("\n========= LLM模型性能测试排行榜 =========")
segment_text = get_string_no_punctuation_or_emoji(segment_text) print("测试提示词:", test_prompt)
if len(segment_text) > 0:
if first_text is None:
first_text = segment_text
print("大模型首次返回耗时:" + str(datetime.now() - start_time))
start = len(response_message)
print("大模型返回总耗时:" + str(datetime.now() - start_time)) # 过滤出成功的结果,并确保数值有效
successful_results = [r for r in test_results if r["status"] == "成功" and r["first_response_time"] is not None]
if successful_results:
print("\n1. 首次响应时间排行:")
sorted_by_first = sorted(successful_results, key=lambda x: x["first_response_time"])
for i, result in enumerate(sorted_by_first, 1):
print(f"{i}. {result['name']}({result['type']}) - {result['first_response_time']:.2f}")
print(f" 响应内容: {result['response'][:50]}...") # 只显示前50个字符
print("\n2. 总响应时间排行:")
sorted_by_total = sorted(successful_results, key=lambda x: x["total_response_time"] or float('inf'))
for i, result in enumerate(sorted_by_total, 1):
if result["total_response_time"] is not None:
print(f"{i}. {result['name']}({result['type']}) - {result['total_response_time']:.2f}")
print("\n3. 响应长度比较:")
sorted_by_length = sorted(successful_results, key=lambda x: x["response_length"], reverse=True)
for i, result in enumerate(sorted_by_length, 1):
print(f"{i}. {result['name']}({result['type']}) - {result['response_length']}字符")
else:
print("\n没有成功完成测试的模型。")
if len(test_results) != len(successful_results):
print("\n测试失败的模型:")
failed_results = [r for r in test_results if r["status"] != "成功" or r["first_response_time"] is None]
for result in failed_results:
print(f"- {result['name']}({result['type']}): {result['status']}")
if __name__ == "__main__":
# 运行异步主函数
asyncio.run(main())
+2 -2
View File
@@ -163,8 +163,8 @@ poetry run python app.py
conda remove -n xiaozhi-esp32-server --all -y conda remove -n xiaozhi-esp32-server --all -y
conda create -n xiaozhi-esp32-server python=3.10 -y conda create -n xiaozhi-esp32-server python=3.10 -y
conda activate xiaozhi-esp32-server conda activate xiaozhi-esp32-server
conda install conda-forge::libopus conda install conda-forge::libopus -y
conda install conda-forge::ffmpeg conda install conda-forge::ffmpeg -y
``` ```
## 2.安装本项目依赖 ## 2.安装本项目依赖
+312 -163
View File
@@ -1,24 +1,24 @@
import time import time
import aiohttp
import asyncio
from tabulate import tabulate from tabulate import tabulate
from typing import Dict from typing import Dict, List
from core.utils.llm import create_instance as create_llm_instance from core.utils.llm import create_instance as create_llm_instance
from core.utils.tts import create_instance as create_tts_instance from core.utils.tts import create_instance as create_tts_instance
from core.utils.util import read_config from core.utils.util import read_config
import statistics import statistics
from config.settings import get_config_file from config.settings import get_config_file
from concurrent.futures import ThreadPoolExecutor
import inspect import inspect
import os import os
import requests
import logging import logging
# 设置全局日志级别为WARNING,抑制INFO级别日志 # 设置全局日志级别为WARNING,抑制INFO级别日志
logging.basicConfig(level=logging.WARNING) logging.basicConfig(level=logging.WARNING)
class PerformanceTester:
class AsyncPerformanceTester:
def __init__(self): def __init__(self):
self.config = read_config(get_config_file()) self.config = read_config(get_config_file())
# 从配置读取测试句子,如果不存在则使用默认
self.test_sentences = self.config.get("module_test", {}).get( self.test_sentences = self.config.get("module_test", {}).get(
"test_sentences", "test_sentences",
["你好,请介绍一下你自己", "What's the weather like today?", ["你好,请介绍一下你自己", "What's the weather like today?",
@@ -30,13 +30,103 @@ class PerformanceTester:
"combinations": [] "combinations": []
} }
def _test_llm(self, llm_name: str, config: Dict) -> Dict: async def _check_ollama_service(self, base_url: str, model_name: str) -> bool:
"""测试单个LLM性能""" """异步检查Ollama服务状态"""
async with aiohttp.ClientSession() as session:
try:
# 检查服务是否可用
async with session.get(f"{base_url}/api/version") as response:
if response.status != 200:
print(f"🚫 Ollama服务未启动或无法访问: {base_url}")
return False
# 检查模型是否存在
async with session.get(f"{base_url}/api/tags") as response:
if response.status == 200:
data = await response.json()
models = data.get("models", [])
if not any(model["name"] == model_name for model in models):
print(f"🚫 Ollama模型 {model_name} 未找到,请先使用 ollama pull {model_name} 下载")
return False
else:
print(f"🚫 无法获取Ollama模型列表")
return False
return True
except Exception as e:
print(f"🚫 无法连接到Ollama服务: {str(e)}")
return False
async def _test_tts(self, tts_name: str, config: Dict) -> Dict:
"""异步测试单个TTS性能"""
try: try:
# 跳过未配置密钥的模块 logging.getLogger("core.providers.tts.base").setLevel(logging.WARNING)
if "api_key" in config and any(x in config["api_key"] for x in ["你的", "placeholder", "sk-xxx"]):
print(f"🚫 跳过未配置的LLM: {llm_name}") token_fields = ["access_token", "api_key", "token"]
return {"errors": 1} if any(field in config and any(x in config[field] for x in ["你的", "placeholder"]) for field in
token_fields):
print(f"⏭️ TTS {tts_name} 未配置access_token/api_key,已跳过")
return {"name": tts_name, "type": "tts", "errors": 1}
module_type = config.get('type', tts_name)
tts = create_tts_instance(
module_type,
config,
delete_audio_file=True
)
print(f"🎵 测试 TTS: {tts_name}")
tmp_file = tts.generate_filename()
await tts.text_to_speak("连接测试", tmp_file)
if not tmp_file or not os.path.exists(tmp_file):
print(f"{tts_name} 连接失败")
return {"name": tts_name, "type": "tts", "errors": 1}
total_time = 0
test_count = len(self.test_sentences[:2])
for i, sentence in enumerate(self.test_sentences[:2], 1):
start = time.time()
tmp_file = tts.generate_filename()
await tts.text_to_speak(sentence, tmp_file)
duration = time.time() - start
total_time += duration
if tmp_file and os.path.exists(tmp_file):
print(f"{tts_name} [{i}/{test_count}]")
else:
print(f"{tts_name} [{i}/{test_count}]")
return {"name": tts_name, "type": "tts", "errors": 1}
return {
"name": tts_name,
"type": "tts",
"avg_time": total_time / test_count,
"errors": 0
}
except Exception as e:
print(f"⚠️ {tts_name} 测试失败: {str(e)}")
return {"name": tts_name, "type": "tts", "errors": 1}
async def _test_llm(self, llm_name: str, config: Dict) -> Dict:
"""异步测试单个LLM性能"""
try:
# 对于Ollama,跳过api_key检查并进行特殊处理
if llm_name == "Ollama":
base_url = config.get('base_url', 'http://localhost:11434')
model_name = config.get('model_name')
if not model_name:
print(f"🚫 Ollama未配置model_name")
return {"name": llm_name, "type": "llm", "errors": 1}
if not await self._check_ollama_service(base_url, model_name):
return {"name": llm_name, "type": "llm", "errors": 1}
else:
if "api_key" in config and any(x in config["api_key"] for x in ["你的", "placeholder", "sk-xxx"]):
print(f"🚫 跳过未配置的LLM: {llm_name}")
return {"name": llm_name, "type": "llm", "errors": 1}
# 获取实际类型(兼容旧配置) # 获取实际类型(兼容旧配置)
module_type = config.get('type', llm_name) module_type = config.get('type', llm_name)
@@ -45,243 +135,302 @@ class PerformanceTester:
# 统一使用UTF-8编码 # 统一使用UTF-8编码
test_sentences = [s.encode('utf-8').decode('utf-8') for s in self.test_sentences] test_sentences = [s.encode('utf-8').decode('utf-8') for s in self.test_sentences]
total_time = 0 # 创建所有句子的测试任务
first_token_times = [] sentence_tasks = []
valid_times = []
for sentence in test_sentences: for sentence in test_sentences:
sentence_start = time.time() # 记录整句开始时间 sentence_tasks.append(self._test_single_sentence(llm_name, llm, sentence))
first_token_received = False
# 遍历响应流 # 并发执行所有句子测试
for chunk in llm.response("perf_test", [{"role": "user", "content": sentence}]): sentence_results = await asyncio.gather(*sentence_tasks)
if not first_token_received and chunk.strip() != '':
first_token_times.append(time.time() - sentence_start)
first_token_received = True
# 计算整句耗时 # 处理结果
sentence_duration = time.time() - sentence_start valid_results = [r for r in sentence_results if r is not None]
total_time += sentence_duration if not valid_results:
valid_times.append(sentence_duration)
# 新增有效性检查
if len(first_token_times) == 0 or len(valid_times) == 0:
print(f"⚠️ {llm_name} 无有效数据,可能配置错误") print(f"⚠️ {llm_name} 无有效数据,可能配置错误")
return {"errors": 1} return {"name": llm_name, "type": "llm", "errors": 1}
# 过滤异常数据(超过3倍标准差) first_token_times = [r["first_token_time"] for r in valid_results]
mean = statistics.mean(valid_times) response_times = [r["response_time"] for r in valid_results]
stdev = statistics.stdev(valid_times) if len(valid_times) > 1 else 0
filtered_times = [t for t in valid_times if t <= mean + 3*stdev] # 过滤异常数据
mean = statistics.mean(response_times)
stdev = statistics.stdev(response_times) if len(response_times) > 1 else 0
filtered_times = [t for t in response_times if t <= mean + 3 * stdev]
# 当有效数据不足时标记错误
if len(filtered_times) < len(test_sentences) * 0.5: if len(filtered_times) < len(test_sentences) * 0.5:
print(f"⚠️ {llm_name} 有效数据不足,可能网络不稳定") print(f"⚠️ {llm_name} 有效数据不足,可能网络不稳定")
return {"errors": 1} return {"name": llm_name, "type": "llm", "errors": 1}
return { return {
"avg_response": total_time / len(test_sentences), "name": llm_name,
"avg_first_token": sum(first_token_times)/len(first_token_times), "type": "llm",
"avg_response": sum(response_times) / len(response_times),
"avg_first_token": sum(first_token_times) / len(first_token_times),
"std_first_token": statistics.stdev(first_token_times) if len(first_token_times) > 1 else 0, "std_first_token": statistics.stdev(first_token_times) if len(first_token_times) > 1 else 0,
"std_response": statistics.stdev(valid_times) if len(valid_times) > 1 else 0, "std_response": statistics.stdev(response_times) if len(response_times) > 1 else 0,
"errors": 0 "errors": 0
} }
except Exception as e: except Exception as e:
print(f"LLM {llm_name} 测试失败: {str(e)}") print(f"LLM {llm_name} 测试失败: {str(e)}")
return {"errors": 1} return {"name": llm_name, "type": "llm", "errors": 1}
def _test_tts(self, tts_name: str, config: Dict) -> Dict: async def _test_single_sentence(self, llm_name: str, llm, sentence: str) -> Dict:
"""测试单个TTS性能""" """测试单个句子的性能"""
try: try:
# 关闭详细日志 print(f"📝 {llm_name} 开始测试: {sentence[:20]}...")
logging.getLogger("core.providers.tts.base").setLevel(logging.WARNING) sentence_start = time.time()
first_token_received = False
first_token_time = None
# 跳过未配置密钥的模块 async def process_response():
token_fields = ["access_token", "api_key", "token"] nonlocal first_token_received, first_token_time
if any(field in config and any(x in config[field] for x in ["你的", "placeholder"]) for field in token_fields): for chunk in llm.response("perf_test", [{"role": "user", "content": sentence}]):
print(f"⏭️ TTS {tts_name} 未配置access_token/api_key,已跳过") if not first_token_received and chunk.strip() != '':
return {"errors": 1} first_token_time = time.time() - sentence_start
first_token_received = True
print(f"{llm_name} 首个Token: {first_token_time:.3f}s")
yield chunk
# 获取实际类型(兼容旧配置) response_chunks = []
module_type = config.get('type', tts_name) async for chunk in process_response():
tts = create_tts_instance( response_chunks.append(chunk)
module_type,
config,
delete_audio_file=True # 确保参数名称正确
)
# 简化后的输出 response_time = time.time() - sentence_start
print(f"\n🎵 正在测试 TTS: {tts_name}") print(f"{llm_name} 完成响应: {response_time:.3f}s")
print(f"🔊 测试 {tts_name}", end="", flush=True)
# 连接测试 if first_token_time is None:
test_conn = tts.to_tts("连接测试") first_token_time = response_time # 如果没有检测到first token,使用总响应时间
if not os.path.exists(test_conn):
print("❌ 连接失败")
return {"errors": 1}
else:
print("")
total_time = 0 return {
test_count = len(self.test_sentences[:2]) "name": llm_name,
"type": "llm",
for i, sentence in enumerate(self.test_sentences[:2], 1): "first_token_time": first_token_time,
start = time.time() "response_time": response_time
file_path = tts.to_tts(sentence) }
duration = time.time() - start
total_time += duration
# 显示简单的进度标识
if os.path.exists(file_path):
print(f"✓[{i}/{test_count}]", end="", flush=True)
else:
print(f"✗[{i}/{test_count}]", end="", flush=True)
print() # 换行
return {"avg_time": total_time / test_count, "errors": 0}
except requests.exceptions.ConnectionError:
print(f"\n{tts_name} 无法连接服务端")
return {"errors": 1}
except Exception as e: except Exception as e:
print(f"\n⚠️ {tts_name} 测试失败: {str(e)}") print(f"⚠️ {llm_name} 句子测试失败: {str(e)}")
return {"errors": 1} return None
def run(self):
"""执行全量测试并自动跳过未配置的模块"""
print("🔍 开始自动检测已配置的模块...")
# 测试所有LLM
for llm_name, config in self.config.get("LLM", {}).items():
# 特殊处理CozeLLM的配置检查
if llm_name == "CozeLLM":
if any(x in config.get("bot_id", "") for x in ["你的"]) \
or any(x in config.get("user_id", "") for x in ["你的"]):
print(f"⏭️ LLM {llm_name} 未配置bot_id/user_id,已跳过")
continue
# 通用的api_key配置检查
if "api_key" in config and any(x in config["api_key"] for x in ["你的", "placeholder"]):
print(f"⏭️ LLM {llm_name} 未配置api_key,已跳过")
continue
print(f"🚀 正在测试 LLM: {llm_name}")
self.results["llm"][llm_name] = self._test_llm(llm_name, config)
# 测试所有TTS
for tts_name, config in self.config.get("TTS", {}).items():
# 根据不同服务的token字段检测
token_fields = ["access_token", "api_key", "token"]
if any(field in config and any(x in config[field] for x in ["你的", "placeholder"]) for field in token_fields):
print(f"⏭️ TTS {tts_name} 未配置access_token/api_key,已跳过")
continue
print(f"🎵 正在测试 TTS: {tts_name}")
self.results["tts"][tts_name] = self._test_tts(tts_name, config)
# 生成组合建议
self._generate_combinations()
self._print_results()
def _generate_combinations(self): def _generate_combinations(self):
"""生成最佳组合建议""" """生成最佳组合建议"""
# 调整过滤条件,例如设为 >= 0.05
valid_llms = [ valid_llms = [
k for k, v in self.results["llm"].items() k for k, v in self.results["llm"].items()
if v["errors"] == 0 and v["avg_first_token"] >= 0.05 if v["errors"] == 0 and v["avg_first_token"] >= 0.05
] ]
valid_tts = [k for k, v in self.results["tts"].items() if v["errors"] == 0] valid_tts = [k for k, v in self.results["tts"].items() if v["errors"] == 0]
# 找出基准值
min_first_token = min([self.results["llm"][llm]["avg_first_token"] for llm in valid_llms]) if valid_llms else 1
min_tts_time = min([self.results["tts"][tts]["avg_time"] for tts in valid_tts]) if valid_tts else 1
for llm in valid_llms: for llm in valid_llms:
for tts in valid_tts: for tts in valid_tts:
llm_weight = 0.8 if self.results["llm"][llm]["avg_first_token"] < 1.0 else 0.6 # 计算相对性能分数(越小越好)
tts_weight = 1 - llm_weight llm_score = self.results["llm"][llm]["avg_first_token"] / min_first_token
score = ( tts_score = self.results["tts"][tts]["avg_time"] / min_tts_time
self.results["llm"][llm]["avg_first_token"] * llm_weight +
self.results["tts"][tts]["avg_time"] * tts_weight # 计算稳定性分数(标准差/平均值,越小越稳定)
) llm_stability = self.results["llm"][llm]["std_first_token"] / self.results["llm"][llm][
"avg_first_token"]
# 综合得分(考虑性能和稳定性)
# 性能权重0.7,稳定性权重0.3
llm_final_score = llm_score * 0.7 + llm_stability * 0.3
# 总分 = LLM得分(70%) + TTS得分(30%)
total_score = llm_final_score * 0.7 + tts_score * 0.3
self.results["combinations"].append({ self.results["combinations"].append({
"llm": llm, "llm": llm,
"tts": tts, "tts": tts,
"score": score, "score": total_score,
"details": { "details": {
"llm_first_token": self.results["llm"][llm]["avg_first_token"], "llm_first_token": self.results["llm"][llm]["avg_first_token"],
"llm_stability": llm_stability,
"tts_time": self.results["tts"][tts]["avg_time"] "tts_time": self.results["tts"][tts]["avg_time"]
} }
}) })
# 按综合得分排序 # 分数越小越好
self.results["combinations"].sort(key=lambda x: x["score"]) self.results["combinations"].sort(key=lambda x: x["score"])
def _print_results(self): def _print_results(self):
"""控制台输出结果""" """打印测试结果"""
# LLM结果表格
llm_table = [] llm_table = []
for name, data in self.results["llm"].items(): for name, data in self.results["llm"].items():
if data["errors"] == 0: if data["errors"] == 0:
stability = data["std_first_token"] / data["avg_first_token"]
llm_table.append([ llm_table.append([
name, name, # 不需要固定宽度,让tabulate自己处理对齐
f"{data['avg_first_token']:.3f}s", f"{data['avg_first_token']:.3f}",
f"{data['avg_response']:.3f}s" f"{data['avg_response']:.3f}",
f"{stability:.3f}"
]) ])
if llm_table: if llm_table:
print("\nLLM 性能排行:") print("\nLLM 性能排行:")
print(tabulate( print(tabulate(
llm_table, llm_table,
headers=["名称", "平均首Token时间", "平均总响应时间"], headers=["名称", "首字耗时", "总耗时", "稳定性"],
tablefmt="github" tablefmt="github",
colalign=("left", "right", "right", "right"),
disable_numparse=True
)) ))
else: else:
print("\n⚠️ 没有可用的LLM模块进行测试。") print("\n⚠️ 没有可用的LLM模块进行测试。")
# TTS结果表格
tts_table = [] tts_table = []
for name, data in self.results["tts"].items(): for name, data in self.results["tts"].items():
if data["errors"] == 0: if data["errors"] == 0:
tts_table.append([ tts_table.append([
name, name, # 不需要固定宽度
f"{data['avg_time']:.3f}s" f"{data['avg_time']:.3f}"
]) ])
if tts_table: if tts_table:
print("\nTTS 性能排行:") print("\nTTS 性能排行:")
print(tabulate( print(tabulate(
tts_table, tts_table,
headers=["名称", "平均合成时间"], headers=["名称", "合成耗时"],
tablefmt="github" tablefmt="github",
colalign=("left", "right"),
disable_numparse=True
)) ))
else: else:
print("\n⚠️ 没有可用的TTS模块进行测试。") print("\n⚠️ 没有可用的TTS模块进行测试。")
# 最佳组合建议
if self.results["combinations"]: if self.results["combinations"]:
print("\n推荐配置组合 (综合响应速度):") print("\n推荐配置组合 (得分越小越好):")
combo_table = [] combo_table = []
for combo in self.results["combinations"][:5]: # 显示前5名 for combo in self.results["combinations"][:5]:
combo_table.append([ combo_table.append([
f"{combo['llm']} + {combo['tts']}", f"{combo['llm']} + {combo['tts']}", # 不需要固定宽度
f"{combo['score']:.3f}", f"{combo['score']:.3f}",
f"{combo['details']['llm_first_token']:.3f}s", f"{combo['details']['llm_first_token']:.3f}",
f"{combo['details']['tts_time']:.3f}s" f"{combo['details']['llm_stability']:.3f}",
f"{combo['details']['tts_time']:.3f}"
]) ])
print(tabulate( print(tabulate(
combo_table, combo_table,
headers=["组合方案", "综合得分", "LLM首Token", "TTS合成"], headers=["组合方案", "综合得分", "LLM首字耗时", "稳定性", "TTS合成耗时"],
tablefmt="github" tablefmt="github",
colalign=("left", "right", "right", "right", "right"),
disable_numparse=True
)) ))
else: else:
print("\n⚠️ 没有可用的模块组合建议。") print("\n⚠️ 没有可用的模块组合建议。")
def _execute_with_timeout(self, func, args=(), kwargs={}, timeout=None): def _process_results(self, all_results):
with ThreadPoolExecutor(max_workers=1) as executor: """处理测试结果"""
future = executor.submit(func, *args, **kwargs) for result in all_results:
try: if result["errors"] == 0:
result = future.result(timeout) if result["type"] == "llm":
return list(result) if inspect.isgenerator(result) else result self.results["llm"][result["name"]] = result
except TimeoutError: else:
raise Exception("操作超时") self.results["tts"][result["name"]] = result
async def run(self):
"""执行全量异步测试"""
print("🔍 开始筛选可用模块...")
# 创建所有测试任务
all_tasks = []
# LLM测试任务
for llm_name, config in self.config.get("LLM", {}).items():
# 检查配置有效性
if llm_name == "CozeLLM":
if any(x in config.get("bot_id", "") for x in ["你的"]) \
or any(x in config.get("user_id", "") for x in ["你的"]):
print(f"⏭️ LLM {llm_name} 未配置bot_id/user_id,已跳过")
continue
elif "api_key" in config and any(x in config["api_key"] for x in ["你的", "placeholder", "sk-xxx"]):
print(f"⏭️ LLM {llm_name} 未配置api_key,已跳过")
continue
# 对于Ollama,先检查服务状态
if llm_name == "Ollama":
base_url = config.get('base_url', 'http://localhost:11434')
model_name = config.get('model_name')
if not model_name:
print(f"🚫 Ollama未配置model_name")
continue
if not await self._check_ollama_service(base_url, model_name):
continue
print(f"📋 添加LLM测试任务: {llm_name}")
module_type = config.get('type', llm_name)
llm = create_llm_instance(module_type, config)
# 为每个句子创建独立任务
for sentence in self.test_sentences:
sentence = sentence.encode('utf-8').decode('utf-8')
all_tasks.append(self._test_single_sentence(llm_name, llm, sentence))
# TTS测试任务
for tts_name, config in self.config.get("TTS", {}).items():
token_fields = ["access_token", "api_key", "token"]
if any(field in config and any(x in config[field] for x in ["你的", "placeholder"]) for field in
token_fields):
print(f"⏭️ TTS {tts_name} 未配置access_token/api_key,已跳过")
continue
print(f"🎵 添加TTS测试任务: {tts_name}")
all_tasks.append(self._test_tts(tts_name, config))
print(
f"\n✅ 找到 {len([t for t in all_tasks if 'test_single_sentence' in str(t)]) / len(self.test_sentences):.0f} 个可用LLM模块")
print(f"✅ 找到 {len([t for t in all_tasks if '_test_tts' in str(t)])} 个可用TTS模块")
print("\n⏳ 开始并发测试所有模块...\n")
# 并发执行所有测试任务
all_results = await asyncio.gather(*all_tasks, return_exceptions=True)
# 处理LLM结果
llm_results = {}
for result in [r for r in all_results if r and isinstance(r, dict) and r.get("type") == "llm"]:
llm_name = result["name"]
if llm_name not in llm_results:
llm_results[llm_name] = {
"name": llm_name,
"type": "llm",
"first_token_times": [],
"response_times": [],
"errors": 0
}
llm_results[llm_name]["first_token_times"].append(result["first_token_time"])
llm_results[llm_name]["response_times"].append(result["response_time"])
# 计算LLM平均值和标准差
for llm_name, data in llm_results.items():
if len(data["first_token_times"]) >= len(self.test_sentences) * 0.5:
self.results["llm"][llm_name] = {
"name": llm_name,
"type": "llm",
"avg_response": sum(data["response_times"]) / len(data["response_times"]),
"avg_first_token": sum(data["first_token_times"]) / len(data["first_token_times"]),
"std_first_token": statistics.stdev(data["first_token_times"]) if len(
data["first_token_times"]) > 1 else 0,
"std_response": statistics.stdev(data["response_times"]) if len(data["response_times"]) > 1 else 0,
"errors": 0
}
# 处理TTS结果
for result in [r for r in all_results if r and isinstance(r, dict) and r.get("type") == "tts"]:
if result["errors"] == 0:
self.results["tts"][result["name"]] = result
# 生成组合建议并打印结果
print("\n📊 生成测试报告...")
self._generate_combinations()
self._print_results()
async def main():
tester = AsyncPerformanceTester()
await tester.run()
if __name__ == "__main__": if __name__ == "__main__":
tester = PerformanceTester() asyncio.run(main())
tester.run()