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

Refactoring to invoke append_to_audio_context instead of direct queue put.

Browse Source
This commit is contained in:
filipi87
2026-03-25 17:21:55 -03:00
parent 2441c4f801
commit c4253a7d98
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/pipecat/services/tts_service.py
View File

@@ -1137,12 +1137,13 @@ class TTSService(AIService):
""" """
if context_id and self.audio_context_available(context_id): if context_id and self.audio_context_available(context_id):
for word, timestamp in word_times: for word, timestamp in word_times:
await self._audio_contexts[context_id].put( await self.append_to_audio_context(
context_id,
_WordTimestampEntry( _WordTimestampEntry(
word=word, word=word,
timestamp=timestamp, timestamp=timestamp,
context_id=context_id, context_id=context_id,
) ),
) )
else: else:
await self._add_word_timestamps(word_times=word_times, context_id=context_id) await self._add_word_timestamps(word_times=word_times, context_id=context_id)
@@ -1207,7 +1208,9 @@ class TTSService(AIService):
self._audio_contexts[context_id] = asyncio.Queue() self._audio_contexts[context_id] = asyncio.Queue()
logger.trace(f"{self} created audio context {context_id}") logger.trace(f"{self} created audio context {context_id}")
async def append_to_audio_context(self, context_id: str, frame: Frame): async def append_to_audio_context(
self, context_id: str, frame: Frame | _WordTimestampEntry | None
):
"""Append audio or control frame to an existing context. """Append audio or control frame to an existing context.
Args: Args:
@@ -1241,7 +1244,7 @@ class TTSService(AIService):
# None. Once we reach None while handling audio we know we can # None. Once we reach None while handling audio we know we can
# safely remove the context. # safely remove the context.
logger.trace(f"{self} marking audio context {context_id} for deletion") logger.trace(f"{self} marking audio context {context_id} for deletion")
await self._audio_contexts[context_id].put(None) await self.append_to_audio_context(context_id, None)
else: else:
logger.warning(f"{self} unable to remove context {context_id}") logger.warning(f"{self} unable to remove context {context_id}")