Merge pull request #2974 from pipecat-ai/mb/language-mapping-improvements

Improve language checking in STT and TTS services
This commit is contained in:
Mark Backman
2025-11-05 14:59:41 -05:00
committed by GitHub
26 changed files with 144 additions and 295 deletions

View File

@@ -53,6 +53,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
- Updated the `GoogleVertexLLMService` to use the `GoogleLLMService` as a base - Updated the `GoogleVertexLLMService` to use the `GoogleLLMService` as a base
class instead of the `OpenAILLMService`. class instead of the `OpenAILLMService`.
- Updated STT and TTS services to pass through unverified language codes with a
warning instead of returning None. This allows developers to use newly
supported languages before Pipecat's service classes are updated, while still
providing guidance on verified languages.
### Fixed ### Fixed
- Fixed an issue that would cause wrong user/assistant context ordering when - Fixed an issue that would cause wrong user/assistant context ordering when

View File

@@ -28,7 +28,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import InterruptibleTTSService, TTSService from pipecat.services.tts_service import InterruptibleTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
try: try:
@@ -50,7 +50,7 @@ def language_to_async_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding Async language code, or None if not supported. The corresponding Async language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.EN: "en", Language.EN: "en",
Language.FR: "fr", Language.FR: "fr",
Language.ES: "es", Language.ES: "es",
@@ -58,17 +58,7 @@ def language_to_async_language(language: Language) -> Optional[str]:
Language.IT: "it", Language.IT: "it",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. en-En -> en)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class AsyncAITTSService(InterruptibleTTSService): class AsyncAITTSService(InterruptibleTTSService):

View File

@@ -30,7 +30,7 @@ from pipecat.frames.frames import (
) )
from pipecat.services.aws.utils import build_event_message, decode_event, get_presigned_url from pipecat.services.aws.utils import build_event_message, decode_event, get_presigned_url
from pipecat.services.stt_service import STTService from pipecat.services.stt_service import STTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -326,7 +326,7 @@ class AWSTranscribeSTTService(STTService):
Returns: Returns:
AWS Transcribe compatible language code, or None if unsupported. AWS Transcribe compatible language code, or None if unsupported.
""" """
language_map = { LANGUAGE_MAP = {
# Afrikaans # Afrikaans
Language.AF: "af-ZA", Language.AF: "af-ZA",
Language.AF_ZA: "af-ZA", Language.AF_ZA: "af-ZA",
@@ -466,7 +466,7 @@ class AWSTranscribeSTTService(STTService):
Language.ZU_ZA: "zu-ZA", Language.ZU_ZA: "zu-ZA",
} }
return language_map.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
@traced_stt @traced_stt
async def _handle_transcription( async def _handle_transcription(

View File

@@ -26,7 +26,7 @@ from pipecat.frames.frames import (
TTSStoppedFrame, TTSStoppedFrame,
) )
from pipecat.services.tts_service import TTSService from pipecat.services.tts_service import TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
try: try:
@@ -47,7 +47,7 @@ def language_to_aws_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding AWS Polly language code, or None if not supported. The corresponding AWS Polly language code, or None if not supported.
""" """
language_map = { LANGUAGE_MAP = {
# Arabic # Arabic
Language.AR: "arb", Language.AR: "arb",
Language.AR_AE: "ar-AE", Language.AR_AE: "ar-AE",
@@ -119,7 +119,7 @@ def language_to_aws_language(language: Language) -> Optional[str]:
Language.CY_GB: "cy-GB", Language.CY_GB: "cy-GB",
} }
return language_map.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class AWSPollyTTSService(TTSService): class AWSPollyTTSService(TTSService):

View File

@@ -8,7 +8,9 @@
from typing import Optional from typing import Optional
from pipecat.transcriptions.language import Language from loguru import logger
from pipecat.transcriptions.language import Language, resolve_language
def language_to_azure_language(language: Language) -> Optional[str]: def language_to_azure_language(language: Language) -> Optional[str]:
@@ -20,7 +22,7 @@ def language_to_azure_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding Azure language code, or None if not supported. The corresponding Azure language code, or None if not supported.
""" """
language_map = { LANGUAGE_MAP = {
# Afrikaans # Afrikaans
Language.AF: "af-ZA", Language.AF: "af-ZA",
Language.AF_ZA: "af-ZA", Language.AF_ZA: "af-ZA",
@@ -341,4 +343,4 @@ def language_to_azure_language(language: Language) -> Optional[str]:
Language.ZU: "zu-ZA", Language.ZU: "zu-ZA",
Language.ZU_ZA: "zu-ZA", Language.ZU_ZA: "zu-ZA",
} }
return language_map.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)

View File

@@ -28,7 +28,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import AudioContextWordTTSService, TTSService from pipecat.services.tts_service import AudioContextWordTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.text.base_text_aggregator import BaseTextAggregator from pipecat.utils.text.base_text_aggregator import BaseTextAggregator
from pipecat.utils.text.skip_tags_aggregator import SkipTagsAggregator from pipecat.utils.text.skip_tags_aggregator import SkipTagsAggregator
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
@@ -77,7 +77,7 @@ def language_to_cartesia_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding Cartesia language code, or None if not supported. The corresponding Cartesia language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AR: "ar", Language.AR: "ar",
Language.BG: "bg", Language.BG: "bg",
Language.BN: "bn", Language.BN: "bn",
@@ -122,17 +122,7 @@ def language_to_cartesia_language(language: Language) -> Optional[str]:
Language.ZH: "zh", Language.ZH: "zh",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class CartesiaTTSService(AudioContextWordTTSService): class CartesiaTTSService(AudioContextWordTTSService):

View File

@@ -20,7 +20,7 @@ from pydantic import BaseModel
from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
from pipecat.services.stt_service import SegmentedSTTService from pipecat.services.stt_service import SegmentedSTTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -37,7 +37,7 @@ def language_to_elevenlabs_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding ElevenLabs language code, or None if not supported. The corresponding ElevenLabs language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AF: "afr", # Afrikaans Language.AF: "afr", # Afrikaans
Language.AM: "amh", # Amharic Language.AM: "amh", # Amharic
Language.AR: "ara", # Arabic Language.AR: "ara", # Arabic
@@ -139,15 +139,7 @@ def language_to_elevenlabs_language(language: Language) -> Optional[str]:
Language.ZU: "zul", # Zulu Language.ZU: "zul", # Zulu
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
# If not found in base languages, try to find the base language from a variant
if not result:
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class ElevenLabsSTTService(SegmentedSTTService): class ElevenLabsSTTService(SegmentedSTTService):

View File

@@ -37,7 +37,7 @@ from pipecat.services.tts_service import (
AudioContextWordTTSService, AudioContextWordTTSService,
WordTTSService, WordTTSService,
) )
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
# See .env.example for ElevenLabs configuration needed # See .env.example for ElevenLabs configuration needed
@@ -72,7 +72,7 @@ def language_to_elevenlabs_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding ElevenLabs language code, or None if not supported. The corresponding ElevenLabs language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AR: "ar", Language.AR: "ar",
Language.BG: "bg", Language.BG: "bg",
Language.CS: "cs", Language.CS: "cs",
@@ -107,17 +107,7 @@ def language_to_elevenlabs_language(language: Language) -> Optional[str]:
Language.ZH: "zh", Language.ZH: "zh",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
def output_format_from_sample_rate(sample_rate: int) -> str: def output_format_from_sample_rate(sample_rate: int) -> str:

View File

@@ -18,7 +18,7 @@ from pydantic import BaseModel
from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
from pipecat.services.stt_service import SegmentedSTTService from pipecat.services.stt_service import SegmentedSTTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -41,7 +41,7 @@ def language_to_fal_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding Fal Wizper language code, or None if not supported. The corresponding Fal Wizper language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AF: "af", Language.AF: "af",
Language.AM: "am", Language.AM: "am",
Language.AR: "ar", Language.AR: "ar",
@@ -142,15 +142,7 @@ def language_to_fal_language(language: Language) -> Optional[str]:
Language.ZH: "zh", Language.ZH: "zh",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class FalSTTService(SegmentedSTTService): class FalSTTService(SegmentedSTTService):

View File

@@ -31,7 +31,7 @@ from pipecat.frames.frames import (
) )
from pipecat.services.gladia.config import GladiaInputParams from pipecat.services.gladia.config import GladiaInputParams
from pipecat.services.stt_service import STTService from pipecat.services.stt_service import STTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -54,7 +54,7 @@ def language_to_gladia_language(language: Language) -> Optional[str]:
Returns: Returns:
The Gladia language code string or None if not supported. The Gladia language code string or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AF: "af", Language.AF: "af",
Language.AM: "am", Language.AM: "am",
Language.AR: "ar", Language.AR: "ar",
@@ -156,17 +156,7 @@ def language_to_gladia_language(language: Language) -> Optional[str]:
Language.ZH: "zh", Language.ZH: "zh",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
# Deprecation warning for nested InputParams # Deprecation warning for nested InputParams

View File

@@ -74,7 +74,7 @@ from pipecat.services.openai.llm import (
OpenAIAssistantContextAggregator, OpenAIAssistantContextAggregator,
OpenAIUserContextAggregator, OpenAIUserContextAggregator,
) )
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.string import match_endofsentence from pipecat.utils.string import match_endofsentence
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_gemini_live, traced_stt from pipecat.utils.tracing.service_decorators import traced_gemini_live, traced_stt
@@ -133,7 +133,7 @@ def language_to_gemini_language(language: Language) -> Optional[str]:
Returns: Returns:
The Gemini language code string, or None if the language is not supported. The Gemini language code string, or None if the language is not supported.
""" """
language_map = { LANGUAGE_MAP = {
# Arabic # Arabic
Language.AR: "ar-XA", Language.AR: "ar-XA",
# Bengali # Bengali
@@ -214,7 +214,8 @@ def language_to_gemini_language(language: Language) -> Optional[str]:
Language.VI: "vi-VN", Language.VI: "vi-VN",
Language.VI_VN: "vi-VN", Language.VI_VN: "vi-VN",
} }
return language_map.get(language)
return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class GeminiLiveContext(OpenAILLMContext): class GeminiLiveContext(OpenAILLMContext):

View File

@@ -36,7 +36,7 @@ from pipecat.frames.frames import (
TranscriptionFrame, TranscriptionFrame,
) )
from pipecat.services.stt_service import STTService from pipecat.services.stt_service import STTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
try: try:
@@ -64,7 +64,7 @@ def language_to_google_stt_language(language: Language) -> Optional[str]:
Returns: Returns:
Optional[str]: Google STT language code or None if not supported. Optional[str]: Google STT language code or None if not supported.
""" """
language_map = { LANGUAGE_MAP = {
# Afrikaans # Afrikaans
Language.AF: "af-ZA", Language.AF: "af-ZA",
Language.AF_ZA: "af-ZA", Language.AF_ZA: "af-ZA",
@@ -351,7 +351,7 @@ def language_to_google_stt_language(language: Language) -> Optional[str]:
Language.ZU_ZA: "zu-ZA", Language.ZU_ZA: "zu-ZA",
} }
return language_map.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class GoogleSTTService(STTService): class GoogleSTTService(STTService):

View File

@@ -36,7 +36,7 @@ from pipecat.frames.frames import (
TTSStoppedFrame, TTSStoppedFrame,
) )
from pipecat.services.tts_service import TTSService from pipecat.services.tts_service import TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
try: try:
from google.auth import default from google.auth import default
@@ -70,7 +70,7 @@ def language_to_google_tts_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding Google TTS language code, or None if not supported. The corresponding Google TTS language code, or None if not supported.
""" """
language_map = { LANGUAGE_MAP = {
# Afrikaans # Afrikaans
Language.AF: "af-ZA", Language.AF: "af-ZA",
Language.AF_ZA: "af-ZA", Language.AF_ZA: "af-ZA",
@@ -226,7 +226,7 @@ def language_to_google_tts_language(language: Language) -> Optional[str]:
Language.VI_VN: "vi-VN", Language.VI_VN: "vi-VN",
} }
return language_map.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class GoogleHttpTTSService(TTSService): class GoogleHttpTTSService(TTSService):

View File

@@ -24,7 +24,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import InterruptibleTTSService from pipecat.services.tts_service import InterruptibleTTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
# See .env.example for LMNT configuration needed # See .env.example for LMNT configuration needed
@@ -46,7 +46,7 @@ def language_to_lmnt_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding LMNT language code, or None if not supported. The corresponding LMNT language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.DE: "de", Language.DE: "de",
Language.EN: "en", Language.EN: "en",
Language.ES: "es", Language.ES: "es",
@@ -68,17 +68,7 @@ def language_to_lmnt_language(language: Language) -> Optional[str]:
Language.ZH: "zh", Language.ZH: "zh",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class LmntTTSService(InterruptibleTTSService): class LmntTTSService(InterruptibleTTSService):

View File

@@ -26,7 +26,7 @@ from pipecat.frames.frames import (
TTSStoppedFrame, TTSStoppedFrame,
) )
from pipecat.services.tts_service import TTSService from pipecat.services.tts_service import TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
@@ -39,7 +39,7 @@ def language_to_minimax_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding MiniMax language name, or None if not supported. The corresponding MiniMax language name, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AR: "Arabic", Language.AR: "Arabic",
Language.CS: "Czech", Language.CS: "Czech",
Language.DE: "German", Language.DE: "German",
@@ -66,20 +66,7 @@ def language_to_minimax_language(language: Language) -> Optional[str]:
Language.ZH: "Chinese", Language.ZH: "Chinese",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Find matching language
for code, name in BASE_LANGUAGES.items():
if str(code.value).lower().startswith(base_code):
result = name
break
return result
class MiniMaxHttpTTSService(TTSService): class MiniMaxHttpTTSService(TTSService):

View File

@@ -35,7 +35,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import InterruptibleTTSService, TTSService from pipecat.services.tts_service import InterruptibleTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
try: try:
@@ -56,7 +56,7 @@ def language_to_neuphonic_lang_code(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding Neuphonic language code, or None if not supported. The corresponding Neuphonic language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.DE: "de", Language.DE: "de",
Language.EN: "en", Language.EN: "en",
Language.ES: "es", Language.ES: "es",
@@ -69,17 +69,7 @@ def language_to_neuphonic_lang_code(language: Language) -> Optional[str]:
Language.ZH: "zh", Language.ZH: "zh",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class NeuphonicTTSService(InterruptibleTTSService): class NeuphonicTTSService(InterruptibleTTSService):

View File

@@ -34,7 +34,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import InterruptibleTTSService, TTSService from pipecat.services.tts_service import InterruptibleTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
try: try:
@@ -55,7 +55,7 @@ def language_to_playht_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding PlayHT language code, or None if not supported. The corresponding PlayHT language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AF: "afrikans", Language.AF: "afrikans",
Language.AM: "amharic", Language.AM: "amharic",
Language.AR: "arabic", Language.AR: "arabic",
@@ -95,17 +95,7 @@ def language_to_playht_language(language: Language) -> Optional[str]:
Language.ZH: "mandarin", Language.ZH: "mandarin",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class PlayHTTTSService(InterruptibleTTSService): class PlayHTTTSService(InterruptibleTTSService):

View File

@@ -32,7 +32,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import AudioContextWordTTSService, TTSService from pipecat.services.tts_service import AudioContextWordTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.text.base_text_aggregator import BaseTextAggregator from pipecat.utils.text.base_text_aggregator import BaseTextAggregator
from pipecat.utils.text.skip_tags_aggregator import SkipTagsAggregator from pipecat.utils.text.skip_tags_aggregator import SkipTagsAggregator
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
@@ -61,7 +61,7 @@ def language_to_rime_language(language: Language) -> str:
Language.EN: "eng", Language.EN: "eng",
Language.ES: "spa", Language.ES: "spa",
} }
return LANGUAGE_MAP.get(language, "eng") return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class RimeTTSService(AudioContextWordTTSService): class RimeTTSService(AudioContextWordTTSService):

View File

@@ -23,7 +23,7 @@ from pipecat.frames.frames import (
TranscriptionFrame, TranscriptionFrame,
) )
from pipecat.services.stt_service import SegmentedSTTService, STTService from pipecat.services.stt_service import SegmentedSTTService, STTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -48,7 +48,7 @@ def language_to_riva_language(language: Language) -> Optional[str]:
Returns: Returns:
Optional[str]: Riva language code or None if not supported. Optional[str]: Riva language code or None if not supported.
""" """
language_map = { LANGUAGE_MAP = {
# Arabic # Arabic
Language.AR: "ar-AR", Language.AR: "ar-AR",
# English # English
@@ -85,7 +85,7 @@ def language_to_riva_language(language: Language) -> Optional[str]:
Language.ES_US: "es-US", # US Spanish Language.ES_US: "es-US", # US Spanish
} }
return language_map.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class RivaSTTService(STTService): class RivaSTTService(STTService):

View File

@@ -19,7 +19,7 @@ from pipecat.frames.frames import (
TranscriptionFrame, TranscriptionFrame,
) )
from pipecat.services.stt_service import STTService from pipecat.services.stt_service import STTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -43,7 +43,7 @@ def language_to_sarvam_language(language: Language) -> str:
The Sarvam language code string. The Sarvam language code string.
""" """
# Mapping of pipecat Language enum to Sarvam language codes # Mapping of pipecat Language enum to Sarvam language codes
SARVAM_LANGUAGES = { LANGUAGE_MAP = {
Language.BN_IN: "bn-IN", Language.BN_IN: "bn-IN",
Language.GU_IN: "gu-IN", Language.GU_IN: "gu-IN",
Language.HI_IN: "hi-IN", Language.HI_IN: "hi-IN",
@@ -58,9 +58,7 @@ def language_to_sarvam_language(language: Language) -> str:
Language.AS_IN: "as-IN", Language.AS_IN: "as-IN",
} }
return SARVAM_LANGUAGES.get( return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
language, "unknown"
) # Default to unknown (Sarvam models auto-detect the language)
class SarvamSTTService(STTService): class SarvamSTTService(STTService):

View File

@@ -29,7 +29,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.tts_service import InterruptibleTTSService, TTSService from pipecat.services.tts_service import InterruptibleTTSService, TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
try: try:
@@ -64,7 +64,7 @@ def language_to_sarvam_language(language: Language) -> Optional[str]:
Language.TE: "te-IN", # Telugu Language.TE: "te-IN", # Telugu
} }
return LANGUAGE_MAP.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=False)
class SarvamHttpTTSService(TTSService): class SarvamHttpTTSService(TTSService):

View File

@@ -31,7 +31,7 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.stt_service import STTService from pipecat.services.stt_service import STTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
try: try:
@@ -604,11 +604,21 @@ class SpeechmaticsSTTService(STTService):
Creates a transcription config object based on the service parameters. Aligns Creates a transcription config object based on the service parameters. Aligns
with the Speechmatics RT API transcription config. with the Speechmatics RT API transcription config.
""" """
# Convert language if it's a Language enum
language = self._params.language
if isinstance(language, Language):
language = _language_to_speechmatics_language(language)
# Convert output locale if it's a Language enum
output_locale = self._params.output_locale
if isinstance(output_locale, Language):
output_locale = _locale_to_speechmatics_locale(language, output_locale)
# Transcription config # Transcription config
transcription_config = TranscriptionConfig( transcription_config = TranscriptionConfig(
language=self._params.language, language=language,
domain=self._params.domain, domain=self._params.domain,
output_locale=self._params.output_locale, output_locale=output_locale,
operating_point=self._params.operating_point, operating_point=self._params.operating_point,
diarization="speaker" if self._params.enable_diarization else None, diarization="speaker" if self._params.enable_diarization else None,
enable_partials=self._params.enable_partials, enable_partials=self._params.enable_partials,
@@ -987,10 +997,10 @@ def _language_to_speechmatics_language(language: Language) -> str:
language: The Language enum to convert. language: The Language enum to convert.
Returns: Returns:
str: The Speechmatics language code, if found. str: The Speechmatics language code.
""" """
# List of supported input languages # List of supported input languages
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AR: "ar", Language.AR: "ar",
Language.BA: "ba", Language.BA: "ba",
Language.EU: "eu", Language.EU: "eu",
@@ -1047,15 +1057,7 @@ def _language_to_speechmatics_language(language: Language) -> str:
Language.CY: "cy", Language.CY: "cy",
} }
# Get the language code return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
result = BASE_LANGUAGES.get(language)
# Fail if language is not supported
if not result:
raise ValueError(f"Unsupported language: {language}")
# Return the language code
return result
def _locale_to_speechmatics_locale(language_code: str, locale: Language) -> str | None: def _locale_to_speechmatics_locale(language_code: str, locale: Language) -> str | None:

View File

@@ -18,7 +18,7 @@ from openai.types.audio import Transcription
from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
from pipecat.services.stt_service import SegmentedSTTService from pipecat.services.stt_service import SegmentedSTTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -35,7 +35,7 @@ def language_to_whisper_language(language: Language) -> Optional[str]:
Returns: Returns:
str or None: The corresponding Whisper language code, or None if not supported. str or None: The corresponding Whisper language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.AF: "af", Language.AF: "af",
Language.AR: "ar", Language.AR: "ar",
Language.HY: "hy", Language.HY: "hy",
@@ -95,15 +95,7 @@ def language_to_whisper_language(language: Language) -> Optional[str]:
Language.CY: "cy", Language.CY: "cy",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class BaseWhisperSTTService(SegmentedSTTService): class BaseWhisperSTTService(SegmentedSTTService):

View File

@@ -20,7 +20,7 @@ from typing_extensions import TYPE_CHECKING, override
from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
from pipecat.services.stt_service import SegmentedSTTService from pipecat.services.stt_service import SegmentedSTTService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
from pipecat.utils.tracing.service_decorators import traced_stt from pipecat.utils.tracing.service_decorators import traced_stt
@@ -106,159 +106,70 @@ def language_to_whisper_language(language: Language) -> Optional[str]:
Note: Note:
Only includes languages officially supported by Whisper. Only includes languages officially supported by Whisper.
""" """
language_map = { LANGUAGE_MAP = {
# Arabic # Arabic
Language.AR: "ar", Language.AR: "ar",
Language.AR_AE: "ar",
Language.AR_BH: "ar",
Language.AR_DZ: "ar",
Language.AR_EG: "ar",
Language.AR_IQ: "ar",
Language.AR_JO: "ar",
Language.AR_KW: "ar",
Language.AR_LB: "ar",
Language.AR_LY: "ar",
Language.AR_MA: "ar",
Language.AR_OM: "ar",
Language.AR_QA: "ar",
Language.AR_SA: "ar",
Language.AR_SY: "ar",
Language.AR_TN: "ar",
Language.AR_YE: "ar",
# Bengali # Bengali
Language.BN: "bn", Language.BN: "bn",
Language.BN_BD: "bn",
Language.BN_IN: "bn",
# Czech # Czech
Language.CS: "cs", Language.CS: "cs",
Language.CS_CZ: "cs",
# Danish # Danish
Language.DA: "da", Language.DA: "da",
Language.DA_DK: "da",
# German # German
Language.DE: "de", Language.DE: "de",
Language.DE_AT: "de",
Language.DE_CH: "de",
Language.DE_DE: "de",
# Greek # Greek
Language.EL: "el", Language.EL: "el",
Language.EL_GR: "el",
# English # English
Language.EN: "en", Language.EN: "en",
Language.EN_AU: "en",
Language.EN_CA: "en",
Language.EN_GB: "en",
Language.EN_HK: "en",
Language.EN_IE: "en",
Language.EN_IN: "en",
Language.EN_KE: "en",
Language.EN_NG: "en",
Language.EN_NZ: "en",
Language.EN_PH: "en",
Language.EN_SG: "en",
Language.EN_TZ: "en",
Language.EN_US: "en",
Language.EN_ZA: "en",
# Spanish # Spanish
Language.ES: "es", Language.ES: "es",
Language.ES_AR: "es",
Language.ES_BO: "es",
Language.ES_CL: "es",
Language.ES_CO: "es",
Language.ES_CR: "es",
Language.ES_CU: "es",
Language.ES_DO: "es",
Language.ES_EC: "es",
Language.ES_ES: "es",
Language.ES_GQ: "es",
Language.ES_GT: "es",
Language.ES_HN: "es",
Language.ES_MX: "es",
Language.ES_NI: "es",
Language.ES_PA: "es",
Language.ES_PE: "es",
Language.ES_PR: "es",
Language.ES_PY: "es",
Language.ES_SV: "es",
Language.ES_US: "es",
Language.ES_UY: "es",
Language.ES_VE: "es",
# Persian # Persian
Language.FA: "fa", Language.FA: "fa",
Language.FA_IR: "fa",
# Finnish # Finnish
Language.FI: "fi", Language.FI: "fi",
Language.FI_FI: "fi",
# French # French
Language.FR: "fr", Language.FR: "fr",
Language.FR_BE: "fr",
Language.FR_CA: "fr",
Language.FR_CH: "fr",
Language.FR_FR: "fr",
# Hindi # Hindi
Language.HI: "hi", Language.HI: "hi",
Language.HI_IN: "hi",
# Hungarian # Hungarian
Language.HU: "hu", Language.HU: "hu",
Language.HU_HU: "hu",
# Indonesian # Indonesian
Language.ID: "id", Language.ID: "id",
Language.ID_ID: "id",
# Italian # Italian
Language.IT: "it", Language.IT: "it",
Language.IT_IT: "it",
# Japanese # Japanese
Language.JA: "ja", Language.JA: "ja",
Language.JA_JP: "ja",
# Korean # Korean
Language.KO: "ko", Language.KO: "ko",
Language.KO_KR: "ko",
# Dutch # Dutch
Language.NL: "nl", Language.NL: "nl",
Language.NL_BE: "nl",
Language.NL_NL: "nl",
# Polish # Polish
Language.PL: "pl", Language.PL: "pl",
Language.PL_PL: "pl",
# Portuguese # Portuguese
Language.PT: "pt", Language.PT: "pt",
Language.PT_BR: "pt",
Language.PT_PT: "pt",
# Romanian # Romanian
Language.RO: "ro", Language.RO: "ro",
Language.RO_RO: "ro",
# Russian # Russian
Language.RU: "ru", Language.RU: "ru",
Language.RU_RU: "ru",
# Slovak # Slovak
Language.SK: "sk", Language.SK: "sk",
Language.SK_SK: "sk",
# Swedish # Swedish
Language.SV: "sv", Language.SV: "sv",
Language.SV_SE: "sv",
# Thai # Thai
Language.TH: "th", Language.TH: "th",
Language.TH_TH: "th",
# Turkish # Turkish
Language.TR: "tr", Language.TR: "tr",
Language.TR_TR: "tr",
# Ukrainian # Ukrainian
Language.UK: "uk", Language.UK: "uk",
Language.UK_UA: "uk",
# Urdu # Urdu
Language.UR: "ur", Language.UR: "ur",
Language.UR_IN: "ur",
Language.UR_PK: "ur",
# Vietnamese # Vietnamese
Language.VI: "vi", Language.VI: "vi",
Language.VI_VN: "vi",
# Chinese # Chinese
Language.ZH: "zh", Language.ZH: "zh",
Language.ZH_CN: "zh",
Language.ZH_HK: "zh",
Language.ZH_TW: "zh",
} }
return language_map.get(language)
return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
class WhisperSTTService(SegmentedSTTService): class WhisperSTTService(SegmentedSTTService):

View File

@@ -25,7 +25,7 @@ from pipecat.frames.frames import (
TTSStoppedFrame, TTSStoppedFrame,
) )
from pipecat.services.tts_service import TTSService from pipecat.services.tts_service import TTSService
from pipecat.transcriptions.language import Language from pipecat.transcriptions.language import Language, resolve_language
from pipecat.utils.tracing.service_decorators import traced_tts from pipecat.utils.tracing.service_decorators import traced_tts
# The server below can connect to XTTS through a local running docker # The server below can connect to XTTS through a local running docker
@@ -45,7 +45,7 @@ def language_to_xtts_language(language: Language) -> Optional[str]:
Returns: Returns:
The corresponding XTTS language code, or None if not supported. The corresponding XTTS language code, or None if not supported.
""" """
BASE_LANGUAGES = { LANGUAGE_MAP = {
Language.CS: "cs", Language.CS: "cs",
Language.DE: "de", Language.DE: "de",
Language.EN: "en", Language.EN: "en",
@@ -65,22 +65,7 @@ def language_to_xtts_language(language: Language) -> Optional[str]:
Language.ZH: "zh-cn", Language.ZH: "zh-cn",
} }
result = BASE_LANGUAGES.get(language) return resolve_language(language, LANGUAGE_MAP, use_base_code=True)
# If not found in base languages, try to find the base language from a variant
if not result:
# Convert enum value to string and get the base language part (e.g. es-ES -> es)
lang_str = str(language.value)
base_code = lang_str.split("-")[0].lower()
# Special handling for Chinese variants
if base_code == "zh":
result = "zh-cn"
else:
# Look up the base code in our supported languages
result = base_code if base_code in BASE_LANGUAGES.values() else None
return result
class XTTSService(TTSService): class XTTSService(TTSService):

View File

@@ -14,6 +14,8 @@ combinations for various speech and text processing services.
import sys import sys
from enum import Enum from enum import Enum
from loguru import logger
if sys.version_info < (3, 11): if sys.version_info < (3, 11):
class StrEnum(str, Enum): class StrEnum(str, Enum):
@@ -569,3 +571,53 @@ class Language(StrEnum):
# Zulu # Zulu
ZU = "zu" ZU = "zu"
ZU_ZA = "zu-ZA" ZU_ZA = "zu-ZA"
def resolve_language(
language: Language, language_map: dict[Language, str], use_base_code: bool = True
) -> str:
"""Resolve a Language enum to a service-specific language code.
Checks the language map first, then falls back to extracting the appropriate
code format with a warning if not found in the verified list.
Args:
language: The Language enum value to convert.
language_map: Dictionary mapping Language enums to service language codes.
use_base_code: If True, extracts base code (e.g., 'en' from 'en-US').
If False, uses full language code as-is.
Returns:
The resolved language code for the service.
Examples::
# Service expecting base codes (e.g., Cartesia)
>>> LANGUAGE_MAP = {Language.EN: "en", Language.ES: "es"}
>>> resolve_language(Language.EN_US, LANGUAGE_MAP, use_base_code=True)
# Logs: "Language en-US not verified. Using base code 'en'."
"en"
# Service expecting full codes (e.g., AWS)
>>> LANGUAGE_MAP = {Language.EN_US: "en-US", Language.ES_ES: "es-ES"}
>>> resolve_language(Language.EN_GB, LANGUAGE_MAP, use_base_code=False)
# Logs: "Language en-GB not verified. Using 'en-GB'."
"en-GB"
"""
# Check if language is in the verified map
result = language_map.get(language)
if result is not None:
return result
# Not in map - fall back with warning
lang_str = str(language.value)
if use_base_code:
# Extract base code (e.g., "en" from "en-US")
base_code = lang_str.split("-")[0].lower()
logger.warning(f"Language {language.value} not verified. Using base code '{base_code}'.")
return base_code
else:
logger.warning(f"Language {language.value} not verified. Using '{lang_str}'.")
return lang_str