84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
from pathlib import Path
|
|
|
|
from pydantic import Field, field_validator
|
|
from pydantic_settings import BaseSettings, SettingsConfigDict
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parent
|
|
|
|
|
|
def asr_bundle_dir() -> Path:
|
|
return PROJECT_ROOT / "models" / "asr" / "SenseVoiceSmall"
|
|
|
|
|
|
def kokoro_bundle_dir() -> Path:
|
|
return PROJECT_ROOT / "models" / "tts" / "Kokoro-82M"
|
|
|
|
|
|
def vad_bundle_path() -> Path:
|
|
return PROJECT_ROOT / "models" / "vad" / "silero_vad.jit"
|
|
|
|
|
|
class Settings(BaseSettings):
|
|
model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8", extra="ignore")
|
|
|
|
@field_validator("llm_api_key", "llm_base_url", "llm_model", mode="before")
|
|
@classmethod
|
|
def llm_required_nonempty(cls, value: str) -> str:
|
|
text = str(value or "").strip()
|
|
if not text:
|
|
raise ValueError("LLM_API_KEY, LLM_BASE_URL and LLM_MODEL must be set (non-empty).")
|
|
return text
|
|
|
|
@field_validator("http_host", mode="before")
|
|
@classmethod
|
|
def normalize_listen_host(cls, value: str) -> str:
|
|
host = str(value or "0.0.0.0").strip().lower()
|
|
if host in {"127.0.0.1", "localhost", "::1"}:
|
|
return "0.0.0.0"
|
|
return host or "0.0.0.0"
|
|
|
|
@field_validator("ssl_certfile", "ssl_keyfile", mode="after")
|
|
@classmethod
|
|
def resolve_ssl_path(cls, value: str) -> str:
|
|
text = (value or "").strip()
|
|
if not text:
|
|
return ""
|
|
p = Path(text)
|
|
if not p.is_absolute():
|
|
p = PROJECT_ROOT / p
|
|
return str(p.resolve())
|
|
|
|
llm_api_key: str = Field(..., validation_alias="LLM_API_KEY")
|
|
llm_base_url: str = Field(..., validation_alias="LLM_BASE_URL")
|
|
llm_model: str = Field(..., validation_alias="LLM_MODEL")
|
|
llm_timeout: int = 8
|
|
llm_max_tokens: int = 150
|
|
llm_temperature: float = 0.55
|
|
llm_system_prompt: str = (
|
|
"你是一个自然、友好的中文语音助手。"
|
|
"请像真人当面聊天一样回答，口语化、简洁、自然，通常 1 到 2 句。"
|
|
"不要使用书面公告腔，不要分点，不要长段解释，不要夸张语气词。"
|
|
"不要虚构身份关系，不要自称家人、恋人或其他私人身份。"
|
|
"避免客服套话，比如‘随时为你服务’、‘很高兴为你服务’、‘请问有什么可以帮您’。"
|
|
"优先使用短句和自然停顿，适合直接语音播报。"
|
|
)
|
|
|
|
tts_voice: str = "zf_xiaoxiao"
|
|
tts_speed: float = 0.9
|
|
tts_sentence_max_len: int = 64
|
|
tts_segment_pause_ms: int = 90
|
|
tts_fade_ms: int = 12
|
|
|
|
http_host: str = Field(default="0.0.0.0", validation_alias="HTTP_HOST")
|
|
http_port: int = Field(default=8080, validation_alias="HTTP_PORT")
|
|
avatar_fps: int = 25
|
|
avatar_driver_mode: str = "blendshape_stream"
|
|
avatar_control_protocol: str = "ws"
|
|
avatar_blendshape_schema: str = "arkit"
|
|
cors_origins: str = "*"
|
|
ssl_certfile: str = ""
|
|
ssl_keyfile: str = ""
|
|
|
|
|
|
settings = Settings()
|