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

services(tavus): fix EndFrame and CancelFrame processing

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-01-27 14:36:30 -08:00
parent cca241a2b7
commit f09f4b8fc4
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/pipecat/services/tavus.py
View File

@@ -75,6 +75,14 @@ class TavusVideoService(AIService):
logger.debug(f"TavusVideoService persona grabbed {response_json}") logger.debug(f"TavusVideoService persona grabbed {response_json}")
return response_json["persona_name"] return response_json["persona_name"]
async def stop(self, frame: EndFrame):
await super().stop(frame)
await self._end_conversation()
async def cancel(self, frame: CancelFrame):
await super().cancel(frame)
await self._end_conversation()
async def _end_conversation(self) -> None: async def _end_conversation(self) -> None:
url = f"https://tavusapi.com/v2/conversations/{self._conversation_id}/end" url = f"https://tavusapi.com/v2/conversations/{self._conversation_id}/end"
headers = {"Content-Type": "application/json", "x-api-key": self._api_key} headers = {"Content-Type": "application/json", "x-api-key": self._api_key}
@@ -105,8 +113,6 @@ class TavusVideoService(AIService):
await self.stop_processing_metrics() await self.stop_processing_metrics()
elif isinstance(frame, StartInterruptionFrame): elif isinstance(frame, StartInterruptionFrame):
await self._send_interrupt_message() await self._send_interrupt_message()
elif isinstance(frame, (EndFrame, CancelFrame)):
await self._end_conversation()
else: else:
await self.push_frame(frame, direction) await self.push_frame(frame, direction)