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

Merge pull request #704 from pipecat-ai/aleix/cartesia-tts-stopped-frame

Browse Source 
services(cartesia): generated TTSStoppedFrame after no more audio
This commit is contained in:
Aleix Conchillo Flaqué
2024-11-10 05:17:36 -08:00
committed by GitHub
parent 446bb0aeaf d125a38737
commit 379750df91
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/pipecat/services/cartesia.py
View File

@@ -228,14 +228,13 @@ class CartesiaTTSService(WordTTSService):
if not msg or msg["context_id"] != self._context_id: if not msg or msg["context_id"] != self._context_id:
continue continue
if msg["type"] == "done": if msg["type"] == "done":
await self.push_frame(TTSStoppedFrame())
await self.stop_ttfb_metrics() await self.stop_ttfb_metrics()
# Unset _context_id but not the _context_id_start_timestamp # Unset _context_id but not the _context_id_start_timestamp
# because we are likely still playing out audio and need the # because we are likely still playing out audio and need the
# timestamp to set send context frames. # timestamp to set send context frames.
self._context_id = None self._context_id = None
await self.add_word_timestamps( await self.add_word_timestamps([("LLMFullResponseEndFrame", 0), ("Reset", 0)])
[("TTSStoppedFrame", 0), ("LLMFullResponseEndFrame", 0), ("Reset", 0)]
)
elif msg["type"] == "timestamps": elif msg["type"] == "timestamps":
await self.add_word_timestamps( await self.add_word_timestamps(
list(zip(msg["word_timestamps"]["words"], msg["word_timestamps"]["start"])) list(zip(msg["word_timestamps"]["words"], msg["word_timestamps"]["start"]))