ddadmin
b1cbdc3e0b
- Зачем: - при smoke-тесте выявлены расхождения между onnx-asr API и мок-тестами. - Что: - убран deprecated cpu_preprocessing из load_model. - sample_rate передан как keyword-аргумент в recognize(). - исправлены имена полей SegmentResult: start_ts→start, end_ts→end. - добавлен device "onnx" в DEVICE_DEFAULTS и _VALID_DEVICES. - Проверка: - uv run pytest -q (172 passed). - uv run transcribe file.mp4 --device onnx --model gigaam-v3.
125 lines
4.3 KiB
Python
125 lines
4.3 KiB
Python
"""Загрузка конфигурации из ``.transcriber.toml`` и каскад приоритетов."""
|
|
|
|
import sys
|
|
import warnings
|
|
from pathlib import Path
|
|
|
|
if sys.version_info >= (3, 11):
|
|
import tomllib
|
|
else:
|
|
import tomli as tomllib
|
|
|
|
HARDCODED_DEFAULTS: dict[str, str] = {
|
|
"model": "medium",
|
|
"language": "ru",
|
|
"device": "auto",
|
|
"compute_type": "float32",
|
|
}
|
|
|
|
DEVICE_DEFAULTS: dict[str, dict[str, str]] = {
|
|
"cuda": {"model": "medium", "compute_type": "float16"},
|
|
"cpu": {"model": "medium", "compute_type": "float32"},
|
|
"openvino": {"model": "medium", "compute_type": "int8"},
|
|
"openvino-gpu": {"model": "medium", "compute_type": "int8"},
|
|
"openvino-cpu": {"model": "medium", "compute_type": "int8"},
|
|
"onnx": {"model": "gigaam-v3", "compute_type": "int8"},
|
|
}
|
|
|
|
# Одно место правды для допустимых ключей конфига
|
|
_VALID_KEYS = set(HARDCODED_DEFAULTS)
|
|
_VALID_DEVICES = {"auto", "cpu", "cuda", "openvino", "openvino-gpu", "openvino-cpu", "onnx"}
|
|
|
|
|
|
def find_config_file() -> Path | None:
|
|
"""Ищет конфиг: сначала ``.transcriber.toml`` в cwd, затем ``~/.config/transcriber/config.toml``."""
|
|
cwd_config = Path.cwd() / ".transcriber.toml"
|
|
if cwd_config.is_file():
|
|
return cwd_config
|
|
|
|
global_config = Path.home() / ".config" / "transcriber" / "config.toml"
|
|
if global_config.is_file():
|
|
return global_config
|
|
|
|
return None
|
|
|
|
|
|
def load_config(path: Path | None = None) -> dict[str, str]:
|
|
"""Загружает и валидирует TOML-конфиг.
|
|
|
|
Неизвестные ключи вызывают предупреждение (а не ошибку) для forward
|
|
compatibility: новые версии могут добавить ключи, которых ещё нет в текущей.
|
|
"""
|
|
if path is None:
|
|
path = find_config_file()
|
|
if path is None:
|
|
return {}
|
|
|
|
try:
|
|
raw = path.read_bytes()
|
|
data = tomllib.loads(raw.decode("utf-8"))
|
|
except Exception as exc:
|
|
raise ValueError(f"Ошибка чтения конфига {path}: {exc}") from exc
|
|
|
|
unknown = set(data) - _VALID_KEYS
|
|
if unknown:
|
|
warnings.warn(
|
|
f"Неизвестные ключи в {path}: {', '.join(sorted(unknown))}",
|
|
stacklevel=2,
|
|
)
|
|
|
|
result: dict[str, str] = {}
|
|
for key in _VALID_KEYS:
|
|
if key not in data:
|
|
continue
|
|
value = data[key]
|
|
if not isinstance(value, str):
|
|
raise ValueError(
|
|
f"Значение '{key}' в {path} должно быть строкой, получено {type(value).__name__}"
|
|
)
|
|
if key == "device" and value not in _VALID_DEVICES:
|
|
raise ValueError(
|
|
f"Недопустимое значение device = '{value}' в {path}. "
|
|
f"Ожидается: {', '.join(sorted(_VALID_DEVICES))}"
|
|
)
|
|
if key == "language" and not value:
|
|
raise ValueError(f"Значение 'language' в {path} не может быть пустым")
|
|
result[key] = value
|
|
|
|
return result
|
|
|
|
|
|
def resolve_defaults(
|
|
cli_values: dict[str, str | None], config: dict[str, str]
|
|
) -> dict[str, str]:
|
|
"""Каскад приоритетов: CLI > конфиг-файл > hardcoded-дефолты."""
|
|
result: dict[str, str] = {}
|
|
for key in HARDCODED_DEFAULTS:
|
|
cli_val = cli_values.get(key)
|
|
if cli_val is not None:
|
|
result[key] = cli_val
|
|
elif key in config:
|
|
result[key] = config[key]
|
|
else:
|
|
result[key] = HARDCODED_DEFAULTS[key]
|
|
return result
|
|
|
|
|
|
def apply_device_defaults(
|
|
defaults: dict[str, str],
|
|
resolved_device: str,
|
|
cli_values: dict[str, str | None],
|
|
config: dict[str, str],
|
|
) -> dict[str, str]:
|
|
"""Применяет device-aware дефолты для model и compute_type,
|
|
если они не были явно заданы через CLI или конфиг."""
|
|
device_defs = DEVICE_DEFAULTS.get(resolved_device, {})
|
|
if not device_defs:
|
|
return defaults
|
|
|
|
result = dict(defaults)
|
|
for key in ("model", "compute_type"):
|
|
if cli_values.get(key) is None and key not in config:
|
|
if key in device_defs:
|
|
result[key] = device_defs[key]
|
|
return result
|