diff --git a/src/pipecat/services/assemblyai/stt.py b/src/pipecat/services/assemblyai/stt.py index bf036eb62..7e9fd0853 100644 --- a/src/pipecat/services/assemblyai/stt.py +++ b/src/pipecat/services/assemblyai/stt.py @@ -26,6 +26,7 @@ from pipecat.processors.frame_processor import FrameDirection from pipecat.services.stt_service import STTService from pipecat.transcriptions.language import Language from pipecat.utils.time import time_now_iso8601 +from pipecat.utils.tracing.service_decorators import traced_stt from .models import ( AssemblyAIConnectionParams, @@ -108,6 +109,11 @@ class AssemblyAISTTService(STTService): await self._websocket.send(json.dumps({"type": "ForceEndpoint"})) await self.start_processing_metrics() + @traced_stt + async def _trace_transcription(self, transcript: str, is_final: bool, language: Language): + """Record transcription event for tracing.""" + pass + def _build_ws_url(self) -> str: """Build WebSocket URL with query parameters using urllib.parse.urlencode.""" params = { @@ -252,6 +258,7 @@ class AssemblyAISTTService(STTService): message, ) ) + await self._trace_transcription(message.transcript, True, self._language) await self.stop_processing_metrics() else: await self.push_frame(