Fix example 5

src/dailyai/services/ai_services.py

@@ -29,7 +29,6 @@ class AIService:
 
     async def run_to_queue(self, queue: asyncio.Queue, frames, add_end_of_stream=False) -> None:
         async for frame in self.run(frames):
-            print("got frame", frame.frame_type)
             await queue.put(frame)
 
         if add_end_of_stream:
@@ -48,29 +47,18 @@ class AIService:
         if not requested_frame_types:
             requested_frame_types = self.possible_output_frame_types()
 
-        print("running", self.__class__.__name__, "with frame types", requested_frame_types)
-
         if isinstance(frames, AsyncIterable):
             async for frame in frames:
                 async for output_frame in self.process_frame(requested_frame_types, frame):
-                    print(
-                        "yielding frame", self.__class__.__name__, output_frame.frame_type
-                    )
                     yield output_frame
         elif isinstance(frames, Iterable):
             for frame in frames:
                 async for output_frame in self.process_frame(requested_frame_types, frame):
-                    print(
-                        "yielding frame", self.__class__.__name__, output_frame.frame_type
-                    )
                     yield output_frame
         elif isinstance(frames, asyncio.Queue):
             while True:
                 frame = await frames.get()
                 async for output_frame in self.process_frame(requested_frame_types, frame):
-                    print(
-                        "yielding frame", self.__class__.__name__, output_frame.frame_type
-                    )
                     yield output_frame
                 if frame.frame_type == FrameType.END_STREAM:
                     break

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):

src/dailyai/services/to_be_updated/huggingface_ai_service.py

@@ -13,7 +13,6 @@ class HuggingFaceAIService(AIService):
     # available models at https://huggingface.co/Helsinki-NLP (**not all models use 2-character language codes**)
     def run_text_translation(self, sentence, source_language, target_language):
         translator = pipeline(f"translation", model=f"Helsinki-NLP/opus-mt-{source_language}-{target_language}")
-        print(translator(sentence))
 
         return translator(sentence)[0]["translation_text"]