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

LLMUserContextAggregator: add space between transcriptions

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-02-13 07:24:05 -08:00
parent 9f6a1c093a
commit 84510fd521
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -287,7 +287,7 @@ class LLMUserContextAggregator(LLMContextResponseAggregator):
await self.push_aggregation() await self.push_aggregation()
async def _handle_transcription(self, frame: TranscriptionFrame): async def _handle_transcription(self, frame: TranscriptionFrame):
self._aggregation += frame.text self._aggregation += f" {frame.text}" if self._aggregation else frame.text
# We just got a final result, so let's reset interim results. # We just got a final result, so let's reset interim results.
self._seen_interim_results = False self._seen_interim_results = False
# Reset aggregation timer. # Reset aggregation timer.

6
tests/test_llm_response.py → tests/test_context_aggregators.py
View File

@@ -109,7 +109,7 @@ class BaseTestUserContextAggregator:
aggregator = self.AGGREGATOR_CLASS(context) aggregator = self.AGGREGATOR_CLASS(context)
frames_to_send = [ frames_to_send = [
UserStartedSpeakingFrame(), UserStartedSpeakingFrame(),
TranscriptionFrame(text="Hello Pipecat! ", user_id="cat", timestamp=""), TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""), InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
SleepFrame(), SleepFrame(),
UserStoppedSpeakingFrame(), UserStoppedSpeakingFrame(),
@@ -271,7 +271,7 @@ class BaseTestUserContextAggregator:
frames_to_send = [ frames_to_send = [
UserStartedSpeakingFrame(), UserStartedSpeakingFrame(),
UserStoppedSpeakingFrame(), UserStoppedSpeakingFrame(),
TranscriptionFrame(text="Hello Pipecat! ", user_id="cat", timestamp=""), TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""), TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
SleepFrame(sleep=AGGREGATION_SLEEP), SleepFrame(sleep=AGGREGATION_SLEEP),
] ]
@@ -298,7 +298,7 @@ class BaseTestUserContextAggregator:
InterimTranscriptionFrame(text="Hello ", user_id="cat", timestamp=""), InterimTranscriptionFrame(text="Hello ", user_id="cat", timestamp=""),
SleepFrame(), SleepFrame(),
UserStoppedSpeakingFrame(), UserStoppedSpeakingFrame(),
TranscriptionFrame(text="Hello Pipecat! ", user_id="cat", timestamp=""), TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""), InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""), TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
SleepFrame(sleep=AGGREGATION_SLEEP), SleepFrame(sleep=AGGREGATION_SLEEP),