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

LLMUserContextAggregator: reset user speaking time after bot interruption

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-02-12 23:54:59 -08:00
parent b602e78625
commit 9f6a1c093a
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/pipecat/processors/aggregators/llm_response.py
View File

@@ -326,6 +326,8 @@ class LLMUserContextAggregator(LLMContextResponseAggregator):
diff_time = time.time() - self._last_user_speaking_time diff_time = time.time() - self._last_user_speaking_time
if diff_time > self._bot_interruption_timeout: if diff_time > self._bot_interruption_timeout:
await self.push_frame(BotInterruptionFrame(), FrameDirection.UPSTREAM) await self.push_frame(BotInterruptionFrame(), FrameDirection.UPSTREAM)
# Reset time so we don't interrupt again right away.
self._last_user_speaking_time = time.time()
class LLMAssistantContextAggregator(LLMContextResponseAggregator): class LLMAssistantContextAggregator(LLMContextResponseAggregator):

1
tests/test_llm_response.py
View File

@@ -343,6 +343,7 @@ class BaseTestUserContextAggregator:
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT) aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
frames_to_send = [ frames_to_send = [
InterimTranscriptionFrame(text="Hello ", user_id="cat", timestamp=""), InterimTranscriptionFrame(text="Hello ", user_id="cat", timestamp=""),
SleepFrame(),
TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""), TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
SleepFrame(sleep=AGGREGATION_SLEEP), SleepFrame(sleep=AGGREGATION_SLEEP),
] ]