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) 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 self._aggregation += frame.text
if match_endofsentence(self._aggregation): if match_endofsentence(self._aggregation):
message = RTVIBotTranscriptionMessage( await self._push_aggregation()
data=RTVITextMessageData(text=self._aggregation)
) async def _push_aggregation(self):
await self._push_transport_message_urgent(message) if len(self._aggregation) > 0:
self._aggregation = "" message = RTVIBotTranscriptionMessage(data=RTVITextMessageData(text=self._aggregation))
await self._push_transport_message_urgent(message)
self._aggregation = ""
class RTVIBotLLMProcessor(RTVIFrameProcessor): class RTVIBotLLMProcessor(RTVIFrameProcessor):