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

rtvi: reset bot transcription with interruptions

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-10-15 14:54:18 -07:00
parent f0d89e57ad
commit 4a71eacac3
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/pipecat/processors/frameworks/rtvi.py
View File

@@ -459,14 +459,18 @@ class RTVIBotTranscriptionProcessor(RTVIFrameProcessor):
await self.push_frame(frame, direction)
if isinstance(frame, TextFrame):
if isinstance(frame, UserStartedSpeakingFrame):
await self._push_aggregation()
elif isinstance(frame, TextFrame):
self._aggregation += frame.text
if match_endofsentence(self._aggregation):
message = RTVIBotTranscriptionMessage(
data=RTVITextMessageData(text=self._aggregation)
)
await self._push_transport_message_urgent(message)
self._aggregation = ""
await self._push_aggregation()
async def _push_aggregation(self):
if len(self._aggregation) > 0:
message = RTVIBotTranscriptionMessage(data=RTVITextMessageData(text=self._aggregation))
await self._push_transport_message_urgent(message)
self._aggregation = ""
class RTVIBotLLMProcessor(RTVIFrameProcessor):