rtvi: reset bot transcription with interruptions

pipecat-ai · Oct 15, 2024 · 4a71eac · 4a71eac
1 parent f0d89e5
commit 4a71eac
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/src/pipecat/processors/frameworks/rtvi.py b/src/pipecat/processors/frameworks/rtvi.py
@@ -459,14 +459,18 @@ async def process_frame(self, frame: Frame, direction: FrameDirection):
 
         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):