mirror of
https://github.com/linyqh/NarratoAI.git
synced 2026-05-02 06:38:37 +00:00
99dd4193ae
- 在 WebUI 中增加 Fun-ASR 转录界面，支持上传多种音视频格式并生成 SRT 字幕 - 新增 `app/services/fun_asr_subtitle.py` 服务模块，实现完整的 REST API 调用流程，包括获取上传凭证、文件上传、提交任务、轮询结果和 SRT 格式转换 - 在配置文件中增加 `[fun_asr]` 配置段，支持保存 API Key - 添加完整的单元测试，覆盖核心转换逻辑和服务流程 - 为兼容 Python 3.11 以下版本，将 `tomllib` 导入改为尝试导入并回退到 `tomli` - 在 `defaults.py` 中添加 `from __future__ import annotations` 以支持类型注解
87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
try:
|
|
import tomllib
|
|
except ModuleNotFoundError: # Python < 3.11
|
|
import tomli as tomllib
|
|
|
|
from app.config import config as cfg
|
|
from app.config.defaults import (
|
|
get_openai_compatible_ui_values,
|
|
normalize_openai_compatible_model_name,
|
|
)
|
|
|
|
|
|
class ConfigBootstrapDefaultsTests(unittest.TestCase):
|
|
def test_load_config_bootstraps_webui_llm_defaults(self):
|
|
original_root_dir = cfg.root_dir
|
|
original_config_file = cfg.config_file
|
|
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
tmp_path = Path(tmp_dir)
|
|
example_file = tmp_path / "config.example.toml"
|
|
example_file.write_text(
|
|
"""
|
|
[app]
|
|
vision_llm_provider = "openai"
|
|
vision_openai_model_name = "gemini/gemini-2.0-flash-lite"
|
|
vision_openai_api_key = ""
|
|
vision_openai_base_url = ""
|
|
text_llm_provider = "openai"
|
|
text_openai_model_name = "deepseek/deepseek-chat"
|
|
text_openai_api_key = ""
|
|
text_openai_base_url = ""
|
|
hide_config = true
|
|
""".strip()
|
|
+ "\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
config_path = tmp_path / "config.toml"
|
|
try:
|
|
cfg.root_dir = str(tmp_path)
|
|
cfg.config_file = str(config_path)
|
|
|
|
config_data = cfg.load_config()
|
|
saved_config = tomllib.loads(config_path.read_text(encoding="utf-8"))
|
|
finally:
|
|
cfg.root_dir = original_root_dir
|
|
cfg.config_file = original_config_file
|
|
|
|
self.assertEqual("openai", config_data["app"]["vision_llm_provider"])
|
|
self.assertEqual("Qwen/Qwen3.5-122B-A10B", config_data["app"]["vision_openai_model_name"])
|
|
self.assertEqual("https://api.siliconflow.cn/v1", config_data["app"]["vision_openai_base_url"])
|
|
self.assertEqual("openai", config_data["app"]["text_llm_provider"])
|
|
self.assertEqual("Pro/zai-org/GLM-5", config_data["app"]["text_openai_model_name"])
|
|
self.assertEqual("https://api.siliconflow.cn/v1", config_data["app"]["text_openai_base_url"])
|
|
self.assertEqual("Qwen/Qwen3.5-122B-A10B", saved_config["app"]["vision_openai_model_name"])
|
|
self.assertEqual("Pro/zai-org/GLM-5", saved_config["app"]["text_openai_model_name"])
|
|
self.assertTrue(saved_config["app"]["hide_config"])
|
|
|
|
|
|
class OpenAICompatibleModelDefaultsTests(unittest.TestCase):
|
|
def test_ui_keeps_full_model_name_and_openai_provider(self):
|
|
provider, model_name = get_openai_compatible_ui_values(
|
|
"Qwen/Qwen3.5-122B-A10B",
|
|
"fallback-model",
|
|
)
|
|
|
|
self.assertEqual("openai", provider)
|
|
self.assertEqual("Qwen/Qwen3.5-122B-A10B", model_name)
|
|
|
|
def test_normalize_only_strips_openai_prefix(self):
|
|
self.assertEqual(
|
|
"Qwen/Qwen3.5-122B-A10B",
|
|
normalize_openai_compatible_model_name("openai/Qwen/Qwen3.5-122B-A10B"),
|
|
)
|
|
self.assertEqual(
|
|
"Qwen/Qwen3.5-122B-A10B",
|
|
normalize_openai_compatible_model_name("Qwen/Qwen3.5-122B-A10B"),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|