apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-07-13 11:22:14 +02:00
Code Issues Projects Releases Packages Wiki Activity

chore: add language option for Rime

Browse source
This commit is contained in:
Abhishek Kumar 2026-04-07 18:32:09 +05:30
parent e255b33813
commit e04ce4e852
2 changed files with 17 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/services/configuration/registry.py
View file

@ -591,6 +591,7 @@ class CambTTSConfiguration(BaseTTSConfiguration):
RIME_TTS_MODELS = ["arcana", "mistv3", "mistv2", "mist"] RIME_TTS_MODELS = ["arcana", "mistv3", "mistv2", "mist"]
RIME_TTS_LANGUAGES = ["en", "de", "fr", "es", "hi"]
@register_tts @register_tts
@ -607,6 +608,10 @@ class RimeTTSConfiguration(BaseTTSConfiguration):
speed: float = Field( speed: float = Field(
default=1.0, ge=0.5, le=2.0, description="Speech speed multiplier" default=1.0, ge=0.5, le=2.0, description="Speech speed multiplier"
) )
language: str = Field(
default="en",
json_schema_extra={"examples": RIME_TTS_LANGUAGES, "allow_custom_input": True},
)
SPEACHES_TTS_MODELS = ["hexgrad/Kokoro-82M"] SPEACHES_TTS_MODELS = ["hexgrad/Kokoro-82M"]

12
api/services/pipecat/service_factory.py
View file

@ -325,10 +325,22 @@ def create_tts_service(user_config, audio_config: "AudioConfig"):
silence_time_s=1.0, silence_time_s=1.0,
) )
elif user_config.tts.provider == ServiceProviders.RIME.value: elif user_config.tts.provider == ServiceProviders.RIME.value:
from pipecat.transcriptions.language import Language
speed = getattr(user_config.tts, "speed", None) speed = getattr(user_config.tts, "speed", None)
language_code = getattr(user_config.tts, "language", None) or "en"
rime_language_mapping = {
"en": Language.EN,
"de": Language.DE,
"fr": Language.FR,
"es": Language.ES,
"hi": Language.HI,
}
pipecat_language = rime_language_mapping.get(language_code, Language.EN)
settings_kwargs = { settings_kwargs = {
"voice": user_config.tts.voice, "voice": user_config.tts.voice,
"model": user_config.tts.model, "model": user_config.tts.model,
"language": pipecat_language,
} }
if speed and speed != 1.0: if speed and speed != 1.0:
settings_kwargs["speedAlpha"] = speed settings_kwargs["speedAlpha"] = speed