This commit is contained in:
Chad Bailey
2024-02-22 16:22:41 +00:00
parent ae7482ed18
commit 85fe6c0580
2 changed files with 6 additions and 14 deletions

View File

@@ -140,18 +140,10 @@ class BaseTransportService():
else: else:
self._context.append({"role": role, "content": text}) 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"starting to speak_after_delay, {frame}")
print(f"past asyncio sleep, context is {context}")
# TODO-CB: This exception for missing class gets eaten! # TODO-CB: This exception for missing class gets eaten!
tma_in = LLMUserContextAggregator( await self._runner(frame)
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)
async def run_conversation(self, runner: Iterable[QueueFrame] async def run_conversation(self, runner: Iterable[QueueFrame]
| AsyncIterable[QueueFrame] | AsyncIterable[QueueFrame]
@@ -176,13 +168,13 @@ class BaseTransportService():
self.interrupt() self.interrupt()
# self._current_phrase += " " + frame.text # 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( current_response_task = asyncio.create_task(
self.run_pipeline( self.run_pipeline(
frame, current_llm_context) frame)
) )
current_response_task.add_done_callback( 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): async def run(self):

View File

@@ -52,7 +52,7 @@ async def main(room_url: str, token):
# voice_id=os.getenv("ELEVENLABS_VOICE_ID")) # voice_id=os.getenv("ELEVENLABS_VOICE_ID"))
fl = FrameLogger("just outside the innermost layer") 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( await tts.run_to_queue(
transport.send_queue, transport.send_queue,
# tma_out.run( # tma_out.run(