From 85fe6c0580f78c39bf2e4a6914cd9718b66a320c Mon Sep 17 00:00:00 2001 From: Chad Bailey Date: Thu, 22 Feb 2024 16:22:41 +0000 Subject: [PATCH] more wip --- src/dailyai/services/base_transport_service.py | 18 +++++------------- ...6c-listen-respond-interruptible-refactor.py | 2 +- 2 files changed, 6 insertions(+), 14 deletions(-) diff --git a/src/dailyai/services/base_transport_service.py b/src/dailyai/services/base_transport_service.py index ef548b5ad..056e9be2b 100644 --- a/src/dailyai/services/base_transport_service.py +++ b/src/dailyai/services/base_transport_service.py @@ -140,18 +140,10 @@ class BaseTransportService(): else: self._context.append({"role": role, "content": text}) - async def run_pipeline(self, frame, context): + async def run_pipeline(self, frame): print(f"starting to speak_after_delay, {frame}") - print(f"past asyncio sleep, context is {context}") # TODO-CB: This exception for missing class gets eaten! - tma_in = LLMUserContextAggregator( - context, self._my_participant_id, complete_sentences=False - ) - tma_out = LLMAssistantContextAggregator( - context, self._my_participant_id - ) - print(f"about to call the runner, tma_in is {tma_in}") - await self._runner(frame, tma_in, tma_out) + await self._runner(frame) async def run_conversation(self, runner: Iterable[QueueFrame] | AsyncIterable[QueueFrame] @@ -176,13 +168,13 @@ class BaseTransportService(): self.interrupt() # self._current_phrase += " " + frame.text - current_llm_context = copy.deepcopy(self._context) + # current_llm_context = copy.deepcopy(self._context) current_response_task = asyncio.create_task( self.run_pipeline( - frame, current_llm_context) + frame) ) current_response_task.add_done_callback( - functools.partial(self.update_messages, current_llm_context) + functools.partial(self.update_messages, self._context) ) async def run(self): diff --git a/src/examples/foundational/06c-listen-respond-interruptible-refactor.py b/src/examples/foundational/06c-listen-respond-interruptible-refactor.py index ed6b9d1d7..4860fabe6 100644 --- a/src/examples/foundational/06c-listen-respond-interruptible-refactor.py +++ b/src/examples/foundational/06c-listen-respond-interruptible-refactor.py @@ -52,7 +52,7 @@ async def main(room_url: str, token): # voice_id=os.getenv("ELEVENLABS_VOICE_ID")) fl = FrameLogger("just outside the innermost layer") - async def run_response(in_frame, tma_in, tma_out): + async def run_response(in_frame): await tts.run_to_queue( transport.send_queue, # tma_out.run(