add handler for disconnect-bot message

2024-11-08 08:42:21 +01:00
parent 53f675f5cf
commit 5d7b3f2b38
1 changed files with 2 additions and 0 deletions
--- a/src/pipecat/processors/frameworks/rtvi.py
+++ b/src/pipecat/processors/frameworks/rtvi.py
@@ -743,6 +743,8 @@ class RTVIProcessor(FrameProcessor):
                case "update-config":
                    update_config = RTVIUpdateConfig.model_validate(message.data)
                    await self._handle_update_config(message.id, update_config)
+                case "disconnect-bot":
+                    await self.push_frame(EndFrame())
                case "action":
                    action = RTVIActionRun.model_validate(message.data)
                    action_frame = RTVIActionFrame(message_id=message.id, rtvi_action_run=action)