wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix SpeechmaticsSTTService: use language checking for language and output_locale

Browse Source
This commit is contained in:
Mark Backman
2025-11-05 12:22:25 -05:00
parent 2ee54c985f
commit f6d7b6ae5f
1 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/pipecat/services/speechmatics/stt.py
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,7 +997,7 @@ 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
LANGUAGE_MAP = { LANGUAGE_MAP = {
@@ -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 = LANGUAGE_MAP.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: