llm user and assistant aggregator renames

2024-04-09 08:31:48 -07:00
parent d7b2e67c35
commit 97b923e37e
4 changed files with 21 additions and 21 deletions
--- a/examples/starter-apps/chatbot.py
+++ b/examples/starter-apps/chatbot.py
@@ -6,8 +6,8 @@ from PIL import Image
 from typing import AsyncGenerator

 from dailyai.pipeline.aggregators import (
-    LLMResponseAggregator,
-    UserResponseAggregator,
+    LLMAssistantResponseAggregator,
+    LLMUserResponseAggregator,
 )
 from dailyai.pipeline.frames import (
    ImageFrame,
@@ -135,8 +135,8 @@ async def main(room_url: str, token):

            await transport.run_interruptible_pipeline(
                pipeline,
-                post_processor=LLMResponseAggregator(messages),
-                pre_processor=UserResponseAggregator(messages),
+                post_processor=LLMAssistantResponseAggregator(messages),
+                pre_processor=LLMUserResponseAggregator(messages),
            )

        transport.transcription_settings["extra"]["endpointing"] = True