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

examples: make sure unhandled frames are always pushed

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-01-30 13:15:49 -08:00
parent 642affb2fe
commit 0131d0a531
4 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/foundational/22b-natural-conversation-proposal.py
View File

@@ -117,12 +117,15 @@ class CompletenessCheck(FrameProcessor):
async def process_frame(self, frame: Frame, direction: FrameDirection): async def process_frame(self, frame: Frame, direction: FrameDirection):
await super().process_frame(frame, direction) await super().process_frame(frame, direction)
if isinstance(frame, TextFrame) and frame.text == "YES": if isinstance(frame, TextFrame) and frame.text == "YES":
logger.debug("Completeness check YES") logger.debug("Completeness check YES")
await self.push_frame(UserStoppedSpeakingFrame()) await self.push_frame(UserStoppedSpeakingFrame())
await self._notifier.notify() await self._notifier.notify()
elif isinstance(frame, TextFrame) and frame.text == "NO": elif isinstance(frame, TextFrame) and frame.text == "NO":
logger.debug("Completeness check NO") logger.debug("Completeness check NO")
else:
await self.push_frame(frame, direction)
class OutputGate(FrameProcessor): class OutputGate(FrameProcessor):

2
examples/foundational/22c-natural-conversation-mixed-llms.py
View File

@@ -328,6 +328,8 @@ class CompletenessCheck(FrameProcessor):
await self._notifier.notify() await self._notifier.notify()
elif isinstance(frame, TextFrame) and frame.text == "NO": elif isinstance(frame, TextFrame) and frame.text == "NO":
logger.debug("!!! Completeness check NO") logger.debug("!!! Completeness check NO")
else:
await self.push_frame(frame, direction)
class OutputGate(FrameProcessor): class OutputGate(FrameProcessor):

2
examples/foundational/22d-natural-conversation-gemini-audio.py
View File

@@ -456,6 +456,8 @@ class CompletenessCheck(FrameProcessor):
# logger.debug("!!! CompletenessCheck idle wait START") # logger.debug("!!! CompletenessCheck idle wait START")
self._wakeup_time = time.time() + self.wait_time self._wakeup_time = time.time() + self.wait_time
self._idle_task = self.create_task(self._idle_task_handler()) self._idle_task = self.create_task(self._idle_task_handler())
else:
await self.push_frame(frame, direction)
async def _idle_task_handler(self): async def _idle_task_handler(self):
try: try:

2
examples/foundational/25-google-audio-in.py
View File

@@ -212,7 +212,7 @@ class InputTranscriptionFrameEmitter(FrameProcessor):
elif isinstance(frame, LLMFullResponseEndFrame): elif isinstance(frame, LLMFullResponseEndFrame):
await self.push_frame(LLMDemoTranscriptionFrame(text=self._aggregation.strip())) await self.push_frame(LLMDemoTranscriptionFrame(text=self._aggregation.strip()))
self._aggregation = "" self._aggregation = ""
elif isinstance(frame, MetricsFrame): else:
await self.push_frame(frame, direction) await self.push_frame(frame, direction)