Update sample 5!

src/dailyai/pipeline/aggregators.py

@@ -9,6 +9,7 @@ from dailyai.pipeline.frames import (
     EndParallelPipeQueueFrame,
     EndStreamQueueFrame,
     LLMMessagesQueueFrame,
+    LLMResponseEndQueueFrame,
     QueueFrame,
     TextQueueFrame,
     TranscriptionQueueFrame,
@@ -16,7 +17,7 @@ from dailyai.pipeline.frames import (
 from dailyai.pipeline.pipeline import Pipeline
 from dailyai.services.ai_services import AIService
 
-from typing import AsyncGenerator, Coroutine, List
+from typing import AsyncGenerator, Coroutine, List, Text
 
 
 class LLMContextAggregator(AIService):
@@ -122,6 +123,23 @@ class SentenceAggregator(FrameProcessor):
             yield frame
 
 
+class LLMFullResponseAggregator(FrameProcessor):
+    def __init__(self):
+        self.aggregation = ""
+
+    async def process_frame(
+        self, frame: QueueFrame
+    ) -> AsyncGenerator[QueueFrame, None]:
+        if isinstance(frame, TextQueueFrame):
+            self.aggregation += frame.text
+        elif isinstance(frame, LLMResponseEndQueueFrame):
+            yield TextQueueFrame(self.aggregation)
+            self.aggregation = ""
+        else:
+            yield frame
+
+
+
 class StatelessTextTransformer(FrameProcessor):
     def __init__(self, transform_fn):
         self.transform_fn = transform_fn
@@ -158,7 +176,7 @@ class ParallelPipeline(FrameProcessor):
             if not isinstance(frame, EndParallelPipeQueueFrame):
                 yield frame
 
-class GatedAccumulator(FrameProcessor):
+class GatedAggregator(FrameProcessor):
     def __init__(self, gate_open_fn, gate_close_fn, start_open):
         self.gate_open_fn = gate_open_fn
         self.gate_close_fn = gate_close_fn

src/dailyai/services/ai_services.py

@@ -12,6 +12,7 @@ from dailyai.pipeline.frames import (
     ImageQueueFrame,
     LLMMessagesQueueFrame,
     LLMResponseEndQueueFrame,
+    LLMResponseStartQueueFrame,
     QueueFrame,
     TextQueueFrame,
     TranscriptionQueueFrame,
@@ -78,6 +79,7 @@ class LLMService(AIService):
 
     async def process_frame(self, frame: QueueFrame) -> AsyncGenerator[QueueFrame, None]:
         if isinstance(frame, LLMMessagesQueueFrame):
+            yield LLMResponseStartQueueFrame()
             async for text_chunk in self.run_llm_async(frame.messages):
                 yield TextQueueFrame(text_chunk)
             yield LLMResponseEndQueueFrame()

src/dailyai/tests/test_aggregators.py

@@ -3,7 +3,7 @@ import functools
 import unittest
 
 from dailyai.pipeline.aggregators import (
-    GatedAccumulator,
+    GatedAggregator,
     ParallelPipeline,
     SentenceAggregator,
     StatelessTextTransformer,
@@ -43,7 +43,7 @@ class TestDailyFrameAggregators(unittest.IsolatedAsyncioTestCase):
         self.assertEqual(expected_sentences, [])
 
     async def test_gated_accumulator(self):
-        gated_accumulator = GatedAccumulator(
+        gated_aggregator = GatedAggregator(
             gate_open_fn=lambda frame: isinstance(frame, ImageQueueFrame),
             gate_close_fn=lambda frame: isinstance(frame, LLMResponseStartQueueFrame),
             start_open=False,
@@ -69,7 +69,7 @@ class TestDailyFrameAggregators(unittest.IsolatedAsyncioTestCase):
             LLMResponseEndQueueFrame(),
         ]
         for frame in frames:
-            async for out_frame in gated_accumulator.process_frame(frame):
+            async for out_frame in gated_aggregator.process_frame(frame):
                 self.assertEqual(out_frame, expected_output_frames.pop(0))
         self.assertEqual(expected_output_frames, [])