566 lines
22 KiB
Python
566 lines
22 KiB
Python
#
|
|
# Copyright (c) 2024-2025 Daily
|
|
#
|
|
# SPDX-License-Identifier: BSD 2-Clause License
|
|
#
|
|
|
|
import unittest
|
|
|
|
from pipecat.frames.frames import (
|
|
BotInterruptionFrame,
|
|
InterimTranscriptionFrame,
|
|
LLMFullResponseEndFrame,
|
|
LLMFullResponseStartFrame,
|
|
StartInterruptionFrame,
|
|
TextFrame,
|
|
TranscriptionFrame,
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
)
|
|
from pipecat.processors.aggregators.llm_response import (
|
|
LLMAssistantContextAggregator,
|
|
LLMUserContextAggregator,
|
|
)
|
|
from pipecat.processors.aggregators.openai_llm_context import (
|
|
OpenAILLMContext,
|
|
OpenAILLMContextFrame,
|
|
)
|
|
from pipecat.services.openai import OpenAIUserContextAggregator
|
|
from pipecat.tests.utils import SleepFrame, run_test
|
|
|
|
AGGREGATION_TIMEOUT = 0.1
|
|
AGGREGATION_SLEEP = 0.15
|
|
BOT_INTERRUPTION_TIMEOUT = 0.2
|
|
BOT_INTERRUPTION_SLEEP = 0.25
|
|
|
|
|
|
class BaseTestUserContextAggregator:
|
|
CONTEXT_CLASS = None # To be set in subclasses
|
|
AGGREGATOR_CLASS = None # To be set in subclasses
|
|
|
|
async def test_se(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [UserStartedSpeakingFrame(), UserStoppedSpeakingFrame()]
|
|
expected_down_frames = [UserStartedSpeakingFrame, UserStoppedSpeakingFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
|
|
async def test_ste(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
TranscriptionFrame(text="Hello!", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello!"
|
|
|
|
async def test_site(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="Hello", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat!"
|
|
|
|
async def test_st1iest2e(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
TranscriptionFrame(text="Hello Pipecat! ", user_id="cat", timestamp=""),
|
|
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
UserStartedSpeakingFrame(),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat! How are you?"
|
|
|
|
async def test_siet(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "How are you?"
|
|
|
|
async def test_sieit(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="are you?", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "How are you?"
|
|
|
|
async def test_set(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "How are you?"
|
|
|
|
async def test_seit(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "How are you?"
|
|
|
|
async def test_st1et2(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat!"
|
|
assert context.messages[1]["content"] == "How are you?"
|
|
|
|
async def test_set1t2(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
TranscriptionFrame(text="Hello Pipecat! ", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat! How are you?"
|
|
|
|
async def test_siet1it2(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="Hello ", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
TranscriptionFrame(text="Hello Pipecat! ", user_id="cat", timestamp=""),
|
|
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat! How are you?"
|
|
|
|
async def test_t(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
TranscriptionFrame(text="Hello!", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [OpenAILLMContextFrame]
|
|
expected_up_frames = [BotInterruptionFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
expected_up_frames=expected_up_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello!"
|
|
|
|
async def test_it(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, aggregation_timeout=AGGREGATION_TIMEOUT)
|
|
frames_to_send = [
|
|
InterimTranscriptionFrame(text="Hello ", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="Hello Pipecat!", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [OpenAILLMContextFrame]
|
|
expected_up_frames = [BotInterruptionFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
expected_up_frames=expected_up_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat!"
|
|
|
|
async def test_sie_delay_it(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(
|
|
context,
|
|
aggregation_timeout=AGGREGATION_TIMEOUT,
|
|
bot_interruption_timeout=BOT_INTERRUPTION_TIMEOUT,
|
|
)
|
|
frames_to_send = [
|
|
UserStartedSpeakingFrame(),
|
|
InterimTranscriptionFrame(text="How ", user_id="cat", timestamp=""),
|
|
SleepFrame(),
|
|
UserStoppedSpeakingFrame(),
|
|
SleepFrame(BOT_INTERRUPTION_SLEEP),
|
|
InterimTranscriptionFrame(text="are you?", user_id="cat", timestamp=""),
|
|
TranscriptionFrame(text="How are you?", user_id="cat", timestamp=""),
|
|
SleepFrame(sleep=AGGREGATION_SLEEP),
|
|
]
|
|
expected_down_frames = [
|
|
UserStartedSpeakingFrame,
|
|
UserStoppedSpeakingFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
expected_up_frames = [BotInterruptionFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
expected_up_frames=expected_up_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "How are you?"
|
|
|
|
|
|
class BaseTestAssistantContextAggreagator:
|
|
CONTEXT_CLASS = None # To be set in subclasses
|
|
AGGREGATOR_CLASS = None # To be set in subclasses
|
|
|
|
async def test_empty(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [LLMFullResponseStartFrame(), LLMFullResponseEndFrame()]
|
|
expected_down_frames = []
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
|
|
async def test_single_text(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="Hello Pipecat!"),
|
|
LLMFullResponseEndFrame(),
|
|
]
|
|
expected_down_frames = [OpenAILLMContextFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat!"
|
|
|
|
async def test_multiple_text(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, expect_stripped_words=False)
|
|
frames_to_send = [
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="Hello "),
|
|
TextFrame(text="Pipecat. "),
|
|
TextFrame(text="How are "),
|
|
TextFrame(text="you?"),
|
|
LLMFullResponseEndFrame(),
|
|
]
|
|
expected_down_frames = [OpenAILLMContextFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat. How are you?"
|
|
|
|
async def test_multiple_text_stripped(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context)
|
|
frames_to_send = [
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="Hello"),
|
|
TextFrame(text="Pipecat."),
|
|
TextFrame(text="How are"),
|
|
TextFrame(text="you?"),
|
|
LLMFullResponseEndFrame(),
|
|
]
|
|
expected_down_frames = [OpenAILLMContextFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat. How are you?"
|
|
|
|
async def test_multiple_llm_responses(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, expect_stripped_words=False)
|
|
frames_to_send = [
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="Hello "),
|
|
TextFrame(text="Pipecat."),
|
|
LLMFullResponseEndFrame(),
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="How are "),
|
|
TextFrame(text="you?"),
|
|
LLMFullResponseEndFrame(),
|
|
]
|
|
expected_down_frames = [OpenAILLMContextFrame, OpenAILLMContextFrame]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat."
|
|
assert context.messages[1]["content"] == "How are you?"
|
|
|
|
async def test_multiple_llm_responses_interruption(self):
|
|
assert self.CONTEXT_CLASS is not None, "CONTEXT_CLASS must be set in a subclass"
|
|
assert self.AGGREGATOR_CLASS is not None, "AGGREGATOR_CLASS must be set in a subclass"
|
|
|
|
context = self.CONTEXT_CLASS()
|
|
aggregator = self.AGGREGATOR_CLASS(context, expect_stripped_words=False)
|
|
frames_to_send = [
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="Hello "),
|
|
TextFrame(text="Pipecat."),
|
|
LLMFullResponseEndFrame(),
|
|
SleepFrame(AGGREGATION_SLEEP),
|
|
StartInterruptionFrame(),
|
|
LLMFullResponseStartFrame(),
|
|
TextFrame(text="How are "),
|
|
TextFrame(text="you?"),
|
|
LLMFullResponseEndFrame(),
|
|
]
|
|
expected_down_frames = [
|
|
OpenAILLMContextFrame,
|
|
StartInterruptionFrame,
|
|
OpenAILLMContextFrame,
|
|
]
|
|
await run_test(
|
|
aggregator,
|
|
frames_to_send=frames_to_send,
|
|
expected_down_frames=expected_down_frames,
|
|
)
|
|
assert context.messages[0]["content"] == "Hello Pipecat."
|
|
assert context.messages[1]["content"] == "How are you?"
|
|
|
|
|
|
#
|
|
# LLMUserContextAggregator, LLMAssistantContextAggregator
|
|
#
|
|
|
|
|
|
class TestLLMUserContextAggregator(BaseTestUserContextAggregator, unittest.IsolatedAsyncioTestCase):
|
|
CONTEXT_CLASS = OpenAILLMContext
|
|
AGGREGATOR_CLASS = LLMUserContextAggregator
|
|
|
|
|
|
class TestLLMAssistantContextAggregator(
|
|
BaseTestAssistantContextAggreagator, unittest.IsolatedAsyncioTestCase
|
|
):
|
|
CONTEXT_CLASS = OpenAILLMContext
|
|
AGGREGATOR_CLASS = LLMAssistantContextAggregator
|
|
|
|
|
|
#
|
|
# OpenAI
|
|
#
|
|
|
|
|
|
class TestOpenAIUserContextAggregator(
|
|
BaseTestUserContextAggregator, unittest.IsolatedAsyncioTestCase
|
|
):
|
|
CONTEXT_CLASS = OpenAILLMContext
|
|
AGGREGATOR_CLASS = OpenAIUserContextAggregator
|
|
|
|
|
|
class TestOpenAIAssistantContextAggregator(
|
|
BaseTestAssistantContextAggreagator, unittest.IsolatedAsyncioTestCase
|
|
):
|
|
CONTEXT_CLASS = OpenAILLMContext
|
|
AGGREGATOR_CLASS = LLMAssistantContextAggregator
|