mirror of
https://github.com/QuentinFuxa/WhisperLiveKit.git
synced 2026-03-07 14:23:18 +00:00
103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
"""Typed configuration for the WhisperLiveKit pipeline."""
|
|
import logging
|
|
from dataclasses import dataclass, field, fields
|
|
from typing import Optional
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass
|
|
class WhisperLiveKitConfig:
|
|
"""Single source of truth for all WhisperLiveKit configuration.
|
|
|
|
Replaces the previous dict-based parameter system in TranscriptionEngine.
|
|
All fields have defaults matching the prior behaviour.
|
|
"""
|
|
|
|
# Server / global
|
|
host: str = "localhost"
|
|
port: int = 8000
|
|
diarization: bool = False
|
|
punctuation_split: bool = False
|
|
target_language: str = ""
|
|
vac: bool = True
|
|
vac_chunk_size: float = 0.04
|
|
log_level: str = "DEBUG"
|
|
ssl_certfile: Optional[str] = None
|
|
ssl_keyfile: Optional[str] = None
|
|
forwarded_allow_ips: Optional[str] = None
|
|
transcription: bool = True
|
|
vad: bool = True
|
|
pcm_input: bool = False
|
|
disable_punctuation_split: bool = False
|
|
diarization_backend: str = "sortformer"
|
|
backend_policy: str = "simulstreaming"
|
|
backend: str = "auto"
|
|
|
|
# Transcription common
|
|
warmup_file: Optional[str] = None
|
|
min_chunk_size: float = 0.1
|
|
model_size: str = "base"
|
|
model_cache_dir: Optional[str] = None
|
|
model_dir: Optional[str] = None
|
|
model_path: Optional[str] = None
|
|
lora_path: Optional[str] = None
|
|
lan: str = "auto"
|
|
direct_english_translation: bool = False
|
|
|
|
# LocalAgreement-specific
|
|
buffer_trimming: str = "segment"
|
|
confidence_validation: bool = False
|
|
buffer_trimming_sec: float = 15.0
|
|
|
|
# SimulStreaming-specific
|
|
disable_fast_encoder: bool = False
|
|
custom_alignment_heads: Optional[str] = None
|
|
frame_threshold: int = 25
|
|
beams: int = 1
|
|
decoder_type: Optional[str] = None
|
|
audio_max_len: float = 20.0
|
|
audio_min_len: float = 0.0
|
|
cif_ckpt_path: Optional[str] = None
|
|
never_fire: bool = False
|
|
init_prompt: Optional[str] = None
|
|
static_init_prompt: Optional[str] = None
|
|
max_context_tokens: Optional[int] = None
|
|
|
|
# Diarization (diart)
|
|
segmentation_model: str = "pyannote/segmentation-3.0"
|
|
embedding_model: str = "pyannote/embedding"
|
|
|
|
# Translation
|
|
nllb_backend: str = "transformers"
|
|
nllb_size: str = "600M"
|
|
|
|
def __post_init__(self):
|
|
# .en model suffix forces English
|
|
if self.model_size and self.model_size.endswith(".en"):
|
|
self.lan = "en"
|
|
# Normalize backend_policy aliases
|
|
if self.backend_policy == "1":
|
|
self.backend_policy = "simulstreaming"
|
|
elif self.backend_policy == "2":
|
|
self.backend_policy = "localagreement"
|
|
|
|
# ------------------------------------------------------------------
|
|
# Factory helpers
|
|
# ------------------------------------------------------------------
|
|
|
|
@classmethod
|
|
def from_namespace(cls, ns) -> "WhisperLiveKitConfig":
|
|
"""Create config from an argparse Namespace, ignoring unknown keys."""
|
|
known = {f.name for f in fields(cls)}
|
|
return cls(**{k: v for k, v in vars(ns).items() if k in known})
|
|
|
|
@classmethod
|
|
def from_kwargs(cls, **kwargs) -> "WhisperLiveKitConfig":
|
|
"""Create config from keyword arguments; warns on unknown keys."""
|
|
known = {f.name for f in fields(cls)}
|
|
unknown = set(kwargs.keys()) - known
|
|
if unknown:
|
|
logger.warning("Unknown config keys ignored: %s", unknown)
|
|
return cls(**{k: v for k, v in kwargs.items() if k in known})
|