fixes as noted by aleix

pipecat-ai · Sep 25, 2024 · 3621fce · 3621fce
1 parent e123f33
commit 3621fce
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/pipecat/services/ai_services.py b/src/pipecat/services/ai_services.py
@@ -259,6 +259,10 @@ def __init__(
     async def flush_audio(self):
         pass
 
+    async def say(self, text: str):
+        await super.say(text)
+        await self.flush_audio()
+
     async def start(self, frame: StartFrame):
         await super().start(frame)
         if self._push_stop_frames:
@@ -278,6 +282,11 @@ async def cancel(self, frame: CancelFrame):
             await self._stop_frame_task
             self._stop_frame_task = None
 
+    async def process_frame(self, frame: Frame, direction: FrameDirection):
+        super().process_frame(frame, direction)
+        if isinstance(frame, TTSSpeakFrame):
+            await self.flush_audio()
+
     async def push_frame(self, frame: Frame, direction: FrameDirection = FrameDirection.DOWNSTREAM):
         await super().push_frame(frame, direction)