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

processors(rtvi): send to the client if errors are fatal

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-08-16 23:17:55 -07:00
parent e22c80610e
commit 32daf65adc
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -135,6 +135,7 @@ class RTVIErrorResponse(BaseModel):
class RTVIErrorData(BaseModel): class RTVIErrorData(BaseModel):
message: str message: str
fatal: bool
class RTVIError(BaseModel): class RTVIError(BaseModel):
@@ -300,8 +301,8 @@ class RTVIProcessor(FrameProcessor):
async def interrupt_bot(self): async def interrupt_bot(self):
await self.push_frame(BotInterruptionFrame(), FrameDirection.UPSTREAM) await self.push_frame(BotInterruptionFrame(), FrameDirection.UPSTREAM)
async def send_error(self, error: str): async def send_error(self, frame: ErrorFrame):
message = RTVIError(data=RTVIErrorData(message=error)) message = RTVIError(data=RTVIErrorData(message=frame.error, fatal=frame.fatal))
await self._push_transport_message(message) await self._push_transport_message(message)
async def handle_function_call( async def handle_function_call(
@@ -337,7 +338,7 @@ class RTVIProcessor(FrameProcessor):
await self._cancel(frame) await self._cancel(frame)
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
elif isinstance(frame, ErrorFrame): elif isinstance(frame, ErrorFrame):
await self.send_error(frame.error) await self.send_error(frame)
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
# All other system frames # All other system frames
elif isinstance(frame, SystemFrame): elif isinstance(frame, SystemFrame):