up:修改性能测试工具读取配置的目录

This commit is contained in:
FAN-yeB
2025-08-14 10:31:31 +08:00
parent 768d2b82d6
commit 1cfb8de3a0
2 changed files with 19 additions and 5 deletions
@@ -4,11 +4,11 @@ import os
import statistics
import time
from typing import Dict
import yaml
import aiohttp
from tabulate import tabulate
from config.settings import load_config
from core.utils.asr import create_instance as create_stt_instance
from core.utils.llm import create_instance as create_llm_instance
from core.utils.tts import create_instance as create_tts_instance
@@ -19,7 +19,14 @@ description = "基础性能测试工具"
class AsyncPerformanceTester:
def __init__(self):
self.config = load_config()
# 从data/.config.yaml读取配置
config_path = os.path.join("data", ".config.yaml")
if not os.path.exists(config_path):
raise FileNotFoundError(f"配置文件 {config_path} 不存在")
with open(config_path, "r", encoding="utf-8") as f:
self.config = yaml.safe_load(f) or {}
self.test_sentences = self.config.get("module_test", {}).get(
"test_sentences",
[
@@ -3,18 +3,25 @@ import asyncio
import logging
import statistics
import base64
import yaml
from typing import Dict
from tabulate import tabulate
from config.settings import load_config
from core.utils.vllm import create_instance
# 设置全局日志级别为WARNING,抑制INFO级别日志
logging.basicConfig(level=logging.WARNING)
description = "视觉识别模型性能测试"
description = "视觉识别模型性能测试"
class AsyncVisionPerformanceTester:
def __init__(self):
self.config = load_config()
# 从data/.config.yaml读取配置
config_path = os.path.join("data", ".config.yaml")
if not os.path.exists(config_path):
raise FileNotFoundError(f"配置文件 {config_path} 不存在")
with open(config_path, "r", encoding="utf-8") as f:
self.config = yaml.safe_load(f) or {}
self.test_images = [
"../../docs/images/demo1.png",
"../../docs/images/demo2.png",