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

Flush output from TTSService on LLMFullResponseEndFrame

Browse Source 
To cover cases when the LLM response does not end in punctuation.
This commit is contained in:
Lewis Wolfgang
2024-06-22 14:57:44 -04:00
committed by Ubuntu
parent 4c20ce20f6
commit e4967107b2
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,6 +23,7 @@ from pipecat.frames.frames import (
TTSStoppedFrame, TTSStoppedFrame,
TextFrame, TextFrame,
VisionImageRawFrame, VisionImageRawFrame,
LLMFullResponseEndFrame,
) )
from pipecat.processors.frame_processor import FrameDirection, FrameProcessor from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
from pipecat.utils.audio import calculate_audio_volume from pipecat.utils.audio import calculate_audio_volume
@@ -145,6 +146,11 @@ class TTSService(AIService):
self._current_sentence = "" self._current_sentence = ""
await self._push_tts_frames(self._current_sentence) await self._push_tts_frames(self._current_sentence)
await self.push_frame(frame) await self.push_frame(frame)
elif isinstance(frame, LLMFullResponseEndFrame):
if self._current_sentence:
await self._push_tts_frames(self._current_sentence.strip())
self._current_sentence = ""
await self.push_frame(frame)
else: else:
await self.push_frame(frame, direction) await self.push_frame(frame, direction)