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

processors(rtvi): add interrupt_bot()

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-08-13 09:22:43 -07:00
parent 8bc6ceaa3d
commit a42d0c9907
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,13 +10,12 @@ from typing import Any, Awaitable, Callable, Dict, List, Literal, Optional, Unio
from pydantic import BaseModel, Field, PrivateAttr, ValidationError from pydantic import BaseModel, Field, PrivateAttr, ValidationError
from pipecat.frames.frames import ( from pipecat.frames.frames import (
BotInterruptionFrame,
CancelFrame, CancelFrame,
EndFrame, EndFrame,
Frame, Frame,
InterimTranscriptionFrame, InterimTranscriptionFrame,
StartFrame, StartFrame,
StartInterruptionFrame,
StopInterruptionFrame,
SystemFrame, SystemFrame,
TranscriptionFrame, TranscriptionFrame,
TransportMessageFrame, TransportMessageFrame,
@@ -233,6 +232,9 @@ class RTVIProcessor(FrameProcessor):
def register_service(self, service: RTVIService): def register_service(self, service: RTVIService):
self._registered_services[service.name] = service self._registered_services[service.name] = service
async def interrupt_bot(self):
await self.push_frame(BotInterruptionFrame(), FrameDirection.UPSTREAM)
async def push_frame(self, frame: Frame, direction: FrameDirection = FrameDirection.DOWNSTREAM): async def push_frame(self, frame: Frame, direction: FrameDirection = FrameDirection.DOWNSTREAM):
if isinstance(frame, SystemFrame): if isinstance(frame, SystemFrame):
await super().push_frame(frame, direction) await super().push_frame(frame, direction)
@@ -414,8 +416,7 @@ class RTVIProcessor(FrameProcessor):
# config. Let's interrupt it for now and update the config. Another # config. Let's interrupt it for now and update the config. Another
# solution is to wait until the bot stops speaking and then apply the # solution is to wait until the bot stops speaking and then apply the
# config, but this definitely is more complicated to achieve. # config, but this definitely is more complicated to achieve.
await self.push_frame(StartInterruptionFrame()) await self.interrupt_bot()
await self.push_frame(StopInterruptionFrame())
await self._update_config(data) await self._update_config(data)
await self._handle_get_config(request_id) await self._handle_get_config(request_id)