From 6fbf98d8e24adfbcd1231c840e5747a61476e941 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= Date: Fri, 19 Jul 2024 17:40:20 -0700 Subject: [PATCH] processors(realtime-ai): llm-context now uses a data field --- src/pipecat/processors/frameworks/realtimeai.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/src/pipecat/processors/frameworks/realtimeai.py b/src/pipecat/processors/frameworks/realtimeai.py index c76d7e74d..4eed08154 100644 --- a/src/pipecat/processors/frameworks/realtimeai.py +++ b/src/pipecat/processors/frameworks/realtimeai.py @@ -89,10 +89,14 @@ class RealtimeAIBasicResponse(BaseModel): error: Optional[str] = None -class RealtimeAILLMContextResponse(BaseModel): +class RealtimeAILLMContextMessageData(BaseModel): + messages: List[dict] + + +class RealtimeAILLMContextMessage(BaseModel): tag: Literal["realtime-ai"] = "realtime-ai" type: Literal["llm-context"] = "llm-context" - messages: List[dict] + data: RealtimeAILLMContextMessageData class RealtimeAITranscriptionMessageData(BaseModel): @@ -279,10 +283,10 @@ class RealtimeAIProcessor(FrameProcessor): await self.push_frame(frame) async def _handle_llm_get_context(self): - messages = self._tma_in.messages - response = RealtimeAILLMContextResponse(messages=messages) - message = TransportMessageFrame(message=response.model_dump(exclude_none=True)) - await self.push_frame(message) + data = RealtimeAILLMContextMessageData(messages=self._tma_in.messages) + message = RealtimeAILLMContextMessage(data=data) + frame = TransportMessageFrame(message=message.model_dump(exclude_none=True)) + await self.push_frame(frame) async def _handle_llm_append_context(self, data: RealtimeAILLMMessageData): if data and data.messages: