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

Merge pull request #801 from pipecat-ai/aleix/rtvi-handle-transport-urgent-frames

Browse Source 
rtvi: handle transport urgent frames
This commit is contained in:
Mark Backman
2024-12-06 21:33:18 -05:00
committed by GitHub
parent 422ee98db0 d1a55c6063
commit 03a269fb87
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -657,6 +657,8 @@ class RTVIProcessor(FrameProcessor):
elif isinstance(frame, ErrorFrame): elif isinstance(frame, ErrorFrame):
await self._send_error_frame(frame) await self._send_error_frame(frame)
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
elif isinstance(frame, TransportMessageUrgentFrame):
await self._handle_transport_message(frame)
# All other system frames # All other system frames
elif isinstance(frame, SystemFrame): elif isinstance(frame, SystemFrame):
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
@@ -667,8 +669,6 @@ class RTVIProcessor(FrameProcessor):
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
await self._stop(frame) await self._stop(frame)
# Data frames # Data frames
elif isinstance(frame, TransportMessageFrame):
await self._handle_transport_message(frame)
elif isinstance(frame, RTVIActionFrame): elif isinstance(frame, RTVIActionFrame):
await self._action_queue.put(frame) await self._action_queue.put(frame)
# Other frames # Other frames
@@ -722,7 +722,7 @@ class RTVIProcessor(FrameProcessor):
except asyncio.CancelledError: except asyncio.CancelledError:
break break
async def _handle_transport_message(self, frame: TransportMessageFrame): async def _handle_transport_message(self, frame: TransportMessageUrgentFrame):
try: try:
message = RTVIMessage.model_validate(frame.message) message = RTVIMessage.model_validate(frame.message)
await self._message_queue.put(message) await self._message_queue.put(message)