Fix example 5

2024-01-17 18:58:03 -05:00
parent 13f2f792af
commit 0d21768d00
4 changed files with 19 additions and 28 deletions
--- a/src/dailyai/services/daily_transport_service.py
+++ b/src/dailyai/services/daily_transport_service.py
@@ -200,10 +200,10 @@ class DailyTransportService(EventHandler):

    async def marshal_frames(self):
        while True:
-            frame = await self.send_queue.get()
+            frame: QueueFrame | list = await self.send_queue.get()
            self.threadsafe_send_queue.put(frame)
            self.send_queue.task_done()
-            if frame.frame_type == FrameType.END_STREAM:
+            if type(frame) == QueueFrame and frame.frame_type == FrameType.END_STREAM:
                break

    def wait_for_send_queue_to_empty(self):