From 8f9c252af9a16bde6f34d20fcc778325c2eeafb6 Mon Sep 17 00:00:00 2001 From: Moishe Lettvin Date: Tue, 27 Feb 2024 13:03:07 -0500 Subject: [PATCH] ... --- src/dailyai/services/ai_services.py | 4 ++++ src/examples/foundational/01-say-one-thing.py | 1 - src/examples/foundational/02-llm-say-one-thing.py | 3 +-- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/dailyai/services/ai_services.py b/src/dailyai/services/ai_services.py index 7b6535048..04a81f962 100644 --- a/src/dailyai/services/ai_services.py +++ b/src/dailyai/services/ai_services.py @@ -50,6 +50,10 @@ class PipeService(AbstractPipeService): self.source = source self.sink = sink + def chain(self, service: AbstractPipeService) -> AbstractPipeService: + self.source = service.sink + return self + async def process_queue(self): if not self.source: return diff --git a/src/examples/foundational/01-say-one-thing.py b/src/examples/foundational/01-say-one-thing.py index 872e78efb..cb063dfd2 100644 --- a/src/examples/foundational/01-say-one-thing.py +++ b/src/examples/foundational/01-say-one-thing.py @@ -35,7 +35,6 @@ async def main(room_url): voice_id=os.getenv("ELEVENLABS_VOICE_ID")) """ tts = PlayHTAIService( - source=None, sink=transport.send_queue, api_key=os.getenv("PLAY_HT_API_KEY"), user_id=os.getenv("PLAY_HT_USER_ID"), diff --git a/src/examples/foundational/02-llm-say-one-thing.py b/src/examples/foundational/02-llm-say-one-thing.py index f3a75a579..8d1ac1edd 100644 --- a/src/examples/foundational/02-llm-say-one-thing.py +++ b/src/examples/foundational/02-llm-say-one-thing.py @@ -30,12 +30,11 @@ async def main(room_url): ) tts = ElevenLabsTTSService( - source=llm, sink=transport.send_queue, aiohttp_session=session, api_key=os.getenv("ELEVENLABS_API_KEY"), voice_id=os.getenv("ELEVENLABS_VOICE_ID"), - ) + ).chain(llm) messages = [{ "role": "system",