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

Fix race condition where context times out after sending second transcript

Browse Source
This commit is contained in:
Daksh Dua
2026-02-20 15:37:14 -03:00
committed by filipi87
parent 5d07326e36
commit c49eda98e7
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1042,6 +1042,8 @@ class AudioContextTTSService(WebsocketTTSService):
audio from context ID "A" will be played first. audio from context ID "A" will be played first.
""" """
_CONTEXT_KEEPALIVE = object()
def __init__( def __init__(
self, self,
*, *,
@@ -1152,9 +1154,15 @@ class AudioContextTTSService(WebsocketTTSService):
A context ID string for the TTS request. A context ID string for the TTS request.
""" """
if self._reuse_context_id_within_turn and self._context_id: if self._reuse_context_id_within_turn and self._context_id:
self._refresh_active_audio_context()
return self._context_id return self._context_id
return super().create_context_id() return super().create_context_id()
def _refresh_active_audio_context(self):
"""Signal that the audio context is still in use, resetting the timeout."""
if self.has_active_audio_context():
self._contexts[self._context_id].put_nowait(AudioContextTTSService._CONTEXT_KEEPALIVE)
async def start(self, frame: StartFrame): async def start(self, frame: StartFrame):
"""Start the audio context TTS service. """Start the audio context TTS service.
@@ -1242,6 +1250,10 @@ class AudioContextTTSService(WebsocketTTSService):
while running: while running:
try: try:
frame = await asyncio.wait_for(queue.get(), timeout=AUDIO_CONTEXT_TIMEOUT) frame = await asyncio.wait_for(queue.get(), timeout=AUDIO_CONTEXT_TIMEOUT)
if frame is AudioContextTTSService._CONTEXT_KEEPALIVE:
# Context is still in use, reset the timeout.
continue
if frame: if frame:
await self.push_frame(frame) await self.push_frame(frame)
running = frame is not None running = frame is not None