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

services: add pts to LLMFullResponseEndFrame in WordTTSService

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-10-14 10:40:41 -07:00
parent fc4fa2faaa
commit 6d778e0491
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/pipecat/services/ai_services.py
View File

@@ -427,15 +427,18 @@ class WordTTSService(TTSService):
self._words_task = None self._words_task = None
async def _words_task_handler(self): async def _words_task_handler(self):
last_pts = 0
while True: while True:
try: try:
(word, timestamp) = await self._words_queue.get() (word, timestamp) = await self._words_queue.get()
if word == "LLMFullResponseEndFrame" and timestamp == 0: if word == "LLMFullResponseEndFrame" and timestamp == 0:
await self.push_frame(LLMFullResponseEndFrame()) frame = LLMFullResponseEndFrame()
frame.pts = last_pts
else: else:
frame = TextFrame(word) frame = TextFrame(word)
frame.pts = self._initial_word_timestamp + timestamp frame.pts = self._initial_word_timestamp + timestamp
await self.push_frame(frame) last_pts = frame.pts
await self.push_frame(frame)
self._words_queue.task_done() self._words_queue.task_done()
except asyncio.CancelledError: except asyncio.CancelledError:
break break