feat: add diarization config defaults (enabled=false, hf_token)

This commit is contained in:
2026-04-02 00:53:53 +02:00
parent 7cd6c2a848
commit 7dfc0e0c5f
2 changed files with 19 additions and 1 deletions
+6 -1
View File
@@ -28,6 +28,10 @@ DEFAULTS = {
"network": {
"host": "127.0.0.1",
},
"diarization": {
"enabled": False,
"hf_token": "",
},
"pid_file": os.path.expanduser("~/.local/run/tueit-transcriber.pid"),
}
@@ -63,4 +67,5 @@ def _write_defaults():
f.write('[whisper]\nmodel = "large-v3"\nlanguage = "de"\ndevice = "auto"\nbase_url = ""\n\n')
f.write('[audio]\ndevice = ""\n\n')
f.write('[server]\nport = 8765\n\n')
f.write(f'[output]\npath = "{DEFAULTS["output"]["path"]}"\n')
f.write(f'[output]\npath = "{DEFAULTS["output"]["path"]}"\n\n')
f.write('[diarization]\nenabled = false\nhf_token = ""\n\n')
+13
View File
@@ -35,3 +35,16 @@ def test_config_has_audio_and_whisper_base_url():
assert "audio" in cfg
assert cfg["audio"]["device"] == ""
assert cfg["whisper"]["base_url"] == ""
def test_config_has_diarization_defaults():
from unittest.mock import patch
import tempfile, os
with tempfile.TemporaryDirectory() as tmpdir:
cfg_path = os.path.join(tmpdir, "config.toml")
with patch("config.CONFIG_PATH", cfg_path):
import config
cfg = config.load()
assert "diarization" in cfg
assert cfg["diarization"]["enabled"] is False
assert cfg["diarization"]["hf_token"] == ""