Files
speech-to-text/lib/config.py

23 lines
1.1 KiB
Python

import os
HOST = os.environ.get("STT_HOST", "127.0.0.1")
PORT = int(os.environ.get("STT_PORT", "8080"))
BACKEND_URL = os.environ.get("STT_BACKEND_URL", "http://127.0.0.1:5092").rstrip("/")
API_KEY = os.environ.get("STT_API_KEY", "").strip()
DEFAULT_LANG = os.environ.get("STT_LANG", "fr").strip() or "fr"
MAX_MB = int(os.environ.get("STT_MAX_MB", "2000"))
MAX_BYTES = MAX_MB * 1024 * 1024
CHUNK_SECONDS = int(os.environ.get("STT_CHUNK_SECONDS", "120"))
CHUNK_WORKERS = int(os.environ.get("STT_WORKERS", "4"))
SILENCE_DB = os.environ.get("STT_SILENCE_DB", "-30")
SILENCE_DUR = os.environ.get("STT_SILENCE_DUR", "0.5")
MODEL_TYPE = os.environ.get("STT_MODEL", "whisper")
WHISPER_MODEL = os.environ.get("STT_WHISPER_MODEL", "large-v3")
WHISPER_DEVICE = os.environ.get("STT_WHISPER_DEVICE", "cuda")
WHISPER_COMPUTE = os.environ.get("STT_WHISPER_COMPUTE", "int8_float16")
LLM_URL = os.environ.get("OPENAI_COMPATIBLE_ENDPOINT", "").strip().rstrip("/")
LLM_API_KEY = os.environ.get("OPENAI_API_KEY", "").strip()
LLM_MODEL = os.environ.get("STT_LLM_MODEL", "gpt-4o-mini")
AVAILABLE_MODELS = ["large-v3", "large-v3-turbo", "medium", "small", "base", "tiny"]