Fixing typecheck.

This commit is contained in:
filipi87
2026-05-12 18:00:07 -03:00
parent bea9e4b3ba
commit 7984556692

View File

@@ -39,8 +39,9 @@ from pipecat.frames.frames import (
) )
from pipecat.processors.frame_processor import FrameDirection from pipecat.processors.frame_processor import FrameDirection
from pipecat.services.aws.sagemaker.bidi_client import SageMakerBidiClient from pipecat.services.aws.sagemaker.bidi_client import SageMakerBidiClient
from pipecat.services.settings import STTSettings from pipecat.services.settings import STTSettings, assert_given
from pipecat.services.stt_service import STTService from pipecat.services.stt_service import STTService
from pipecat.transcriptions.language import 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
@@ -292,6 +293,9 @@ class NvidiaSageMakerSTTService(STTService):
): ):
logger.debug(f"{self}: received event: {event_type}") logger.debug(f"{self}: received event: {event_type}")
_lang = assert_given(self._settings.language)
language: Language | None = Language(_lang) if _lang is not None else None
if event_type == "conversation.item.input_audio_transcription.delta": if event_type == "conversation.item.input_audio_transcription.delta":
delta = msg.get("delta", "") delta = msg.get("delta", "")
if delta: if delta:
@@ -301,7 +305,7 @@ class NvidiaSageMakerSTTService(STTService):
delta, delta,
self._user_id, self._user_id,
time_now_iso8601(), time_now_iso8601(),
language=self._settings.language, language=language,
result=msg, result=msg,
) )
) )
@@ -315,7 +319,7 @@ class NvidiaSageMakerSTTService(STTService):
transcript, transcript,
self._user_id, self._user_id,
time_now_iso8601(), time_now_iso8601(),
language=self._settings.language, language=language,
result=msg, result=msg,
finalized=True, finalized=True,
) )