wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

processors(rtvi): use send a type=response as command responses

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-07-21 16:34:38 -07:00
parent be6995cfdf
commit 37b04ed283
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/pipecat/processors/frameworks/rtvi.py
View File

@@ -86,12 +86,16 @@ class RTVIMessage(BaseModel):
data: Optional[RTVIMessageData] = None data: Optional[RTVIMessageData] = None
class RTVIBasicResponse(BaseModel): class RTVIResponseData(BaseModel):
label: Literal["realtime-ai"] = "realtime-ai"
type: str type: str
success: bool success: bool
error: Optional[str] = None error: Optional[str] = None
class RTVIResponse(BaseModel):
label: Literal["realtime-ai"] = "realtime-ai"
type: Literal["response"] = "response"
data: RTVIResponseData
class RTVILLMContextMessageData(BaseModel): class RTVILLMContextMessageData(BaseModel):
messages: List[dict] messages: List[dict]
@@ -363,6 +367,6 @@ class RTVIProcessor(FrameProcessor):
if parent and self._start_frame: if parent and self._start_frame:
parent.link(pipeline) parent.link(pipeline)
message = RTVIBasicResponse(type=type, success=success, error=error) message = RTVIResponse(data=RTVIResponseData(type=type, success=success, error=error))
frame = TransportMessageFrame(message=message.model_dump(exclude_none=True)) frame = TransportMessageFrame(message=message.model_dump(exclude_none=True))
await self.push_frame(frame) await self.push_frame(frame)