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

LLMUserAggregator: fix backwards compatibility with ExternalTurnStartStrategies

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-12-28 15:03:43 -08:00
parent e757b4bf6f
commit 46db8e58d6
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/pipecat/processors/aggregators/llm_response_universal.py
View File

@@ -65,7 +65,7 @@ from pipecat.processors.aggregators.llm_context import (
from pipecat.processors.frame_processor import FrameDirection, FrameProcessor from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
from pipecat.turns.bot import BaseBotTurnStartStrategy from pipecat.turns.bot import BaseBotTurnStartStrategy
from pipecat.turns.mute import BaseUserMuteStrategy from pipecat.turns.mute import BaseUserMuteStrategy
from pipecat.turns.turn_start_strategies import TurnStartStrategies from pipecat.turns.turn_start_strategies import ExternalTurnStartStrategies, TurnStartStrategies
from pipecat.turns.user import BaseUserTurnStartStrategy from pipecat.turns.user import BaseUserTurnStartStrategy
from pipecat.utils.string import TextPartForConcatenation, concatenate_aggregated_text from pipecat.utils.string import TextPartForConcatenation, concatenate_aggregated_text
from pipecat.utils.time import time_now_iso8601 from pipecat.utils.time import time_now_iso8601
@@ -359,9 +359,14 @@ class LLMUserAggregator(LLMContextAggregator):
self._user_turn_end_timeout_task_handler() self._user_turn_end_timeout_task_handler()
) )
await self._setup_turn_start_strategies()
await self._setup_user_mute_strategies()
async def _setup_user_mute_strategies(self):
for s in self._params.user_mute_strategies: for s in self._params.user_mute_strategies:
await s.setup(self.task_manager) await s.setup(self.task_manager)
async def _setup_turn_start_strategies(self):
if self._turn_start_strategies.user: if self._turn_start_strategies.user:
for s in self._turn_start_strategies.user: for s in self._turn_start_strategies.user:
await s.setup(self.task_manager) await s.setup(self.task_manager)
@@ -387,9 +392,10 @@ class LLMUserAggregator(LLMContextAggregator):
await self.cancel_task(self._user_turn_end_timeout_task) await self.cancel_task(self._user_turn_end_timeout_task)
self._user_turn_end_timeout_task = None self._user_turn_end_timeout_task = None
for s in self._params.user_mute_strategies: await self._cleanup_turn_start_strategies()
await s.cleanup() await self._cleanup_user_mute_strategies()
async def _cleanup_turn_start_strategies(self):
if self._turn_start_strategies.user: if self._turn_start_strategies.user:
for s in self._turn_start_strategies.user: for s in self._turn_start_strategies.user:
await s.cleanup() await s.cleanup()
@@ -398,6 +404,10 @@ class LLMUserAggregator(LLMContextAggregator):
for s in self._turn_start_strategies.bot: for s in self._turn_start_strategies.bot:
await s.cleanup() await s.cleanup()
async def _cleanup_user_mute_strategies(self):
for s in self._params.user_mute_strategies:
await s.cleanup()
async def _maybe_mute_frame(self, frame: Frame): async def _maybe_mute_frame(self, frame: Frame):
should_mute_frame = self._user_is_muted and isinstance( should_mute_frame = self._user_is_muted and isinstance(
frame, frame,
@@ -472,6 +482,10 @@ class LLMUserAggregator(LLMContextAggregator):
" )" " )"
) )
await self._cleanup_turn_start_strategies()
self._turn_start_strategies = ExternalTurnStartStrategies()
await self._setup_turn_start_strategies()
async def _handle_vad_user_started_speaking(self, frame: VADUserStartedSpeakingFrame): async def _handle_vad_user_started_speaking(self, frame: VADUserStartedSpeakingFrame):
self._vad_user_speaking = True self._vad_user_speaking = True