wehub-resource-sync
62 lines
1.7 KiB
Python
62 lines
1.7 KiB
Python
# Copyright 2023 LiveKit, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Literal
|
|
|
|
# Speech models supported by Camb.ai MARS series
|
|
SpeechModel = Literal[
|
|
"mars-flash", # Faster inference, 22.05kHz
|
|
"mars-pro", # Higher quality, 48kHz
|
|
"mars-instruct", # Supports user_instructions, 22.05kHz
|
|
]
|
|
|
|
# Sample rates per model
|
|
MODEL_SAMPLE_RATES: dict[str, int] = {
|
|
"mars-flash": 22050,
|
|
"mars-pro": 48000,
|
|
"mars-instruct": 22050,
|
|
}
|
|
|
|
# Audio output formats
|
|
OutputFormat = Literal[
|
|
"pcm_s16le", # 16-bit PCM (recommended for LiveKit)
|
|
"pcm_s32le", # 32-bit PCM (highest quality)
|
|
"wav", # WAV with headers
|
|
"flac", # Lossless compression
|
|
"adts", # Streaming format
|
|
]
|
|
|
|
|
|
@dataclass
|
|
class _TTSOptions:
|
|
"""Internal TTS configuration options."""
|
|
|
|
voice_id: int
|
|
language: str
|
|
speech_model: str
|
|
output_format: str
|
|
user_instructions: str | None
|
|
enhance_named_entities: bool
|
|
|
|
|
|
# Constants
|
|
DEFAULT_VOICE_ID = 147320
|
|
DEFAULT_LANGUAGE = "en-us"
|
|
DEFAULT_MODEL: SpeechModel = "mars-flash"
|
|
DEFAULT_OUTPUT_FORMAT: OutputFormat = "pcm_s16le"
|
|
NUM_CHANNELS = 1
|