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

self._muted now returns from STT process_audio_frames

Browse Source
This commit is contained in:
Mark Backman
2025-02-13 07:00:44 -05:00
parent 1e8a86de63
commit 84143cc80c
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/pipecat/services/ai_services.py
View File

@@ -525,9 +525,13 @@ class STTService(AIService):
else: else:
logger.warning(f"Unknown setting for STT service: {key}") logger.warning(f"Unknown setting for STT service: {key}")
async def process_audio_frame(self, frame: AudioRawFrame): async def process_audio_frame(self, frame: AudioRawFrame, direction: FrameDirection):
if not self._muted: if self._muted:
await self.process_generator(self.run_stt(frame.audio)) return
await self.process_generator(self.run_stt(frame.audio))
if self._audio_passthrough:
await self.push_frame(frame, direction)
async def process_frame(self, frame: Frame, direction: FrameDirection): async def process_frame(self, frame: Frame, direction: FrameDirection):
"""Processes a frame of audio data, either buffering or transcribing it.""" """Processes a frame of audio data, either buffering or transcribing it."""
@@ -537,9 +541,7 @@ class STTService(AIService):
# In this service we accumulate audio internally and at the end we # In this service we accumulate audio internally and at the end we
# push a TextFrame. We also push audio downstream in case someone # push a TextFrame. We also push audio downstream in case someone
# else needs it. # else needs it.
await self.process_audio_frame(frame) await self.process_audio_frame(frame, direction)
if self._audio_passthrough:
await self.push_frame(frame, direction)
elif isinstance(frame, STTUpdateSettingsFrame): elif isinstance(frame, STTUpdateSettingsFrame):
await self._update_settings(frame.settings) await self._update_settings(frame.settings)
elif isinstance(frame, STTMuteFrame): elif isinstance(frame, STTMuteFrame):