diff --git a/.github/workflows/coverage.yaml b/.github/workflows/coverage.yaml index ae0cb9c57..852611eba 100644 --- a/.github/workflows/coverage.yaml +++ b/.github/workflows/coverage.yaml @@ -39,6 +39,7 @@ jobs: --extra google \ --extra langchain \ --extra livekit \ + --extra local-smart-turn-v3 \ --extra piper \ --extra websocket diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml index cb35a169c..54495911b 100644 --- a/.github/workflows/tests.yaml +++ b/.github/workflows/tests.yaml @@ -43,6 +43,7 @@ jobs: --extra google \ --extra langchain \ --extra livekit \ + --extra local-smart-turn-v3 \ --extra piper \ --extra websocket diff --git a/tests/test_langchain.py b/tests/test_langchain.py index 8ec8905b4..cc5c8f030 100644 --- a/tests/test_langchain.py +++ b/tests/test_langchain.py @@ -24,10 +24,15 @@ from pipecat.frames.frames import ( ) from pipecat.pipeline.pipeline import Pipeline from pipecat.processors.aggregators.llm_context import LLMContext -from pipecat.processors.aggregators.llm_response_universal import LLMContextAggregatorPair +from pipecat.processors.aggregators.llm_response_universal import ( + LLMContextAggregatorPair, + LLMUserAggregatorParams, +) from pipecat.processors.frame_processor import FrameProcessor from pipecat.processors.frameworks.langchain import LangchainProcessor from pipecat.tests.utils import SleepFrame, run_test +from pipecat.turns.user_stop import SpeechTimeoutUserTurnStopStrategy +from pipecat.turns.user_turn_strategies import UserTurnStrategies class TestLangchain(unittest.IsolatedAsyncioTestCase): @@ -65,7 +70,12 @@ class TestLangchain(unittest.IsolatedAsyncioTestCase): self.mock_proc = self.MockProcessor("token_collector") context = LLMContext() - context_aggregator = LLMContextAggregatorPair(context) + context_aggregator = LLMContextAggregatorPair( + context, + user_params=LLMUserAggregatorParams( + user_turn_strategies=UserTurnStrategies(stop=[SpeechTimeoutUserTurnStopStrategy()]) + ), + ) pipeline = Pipeline( [context_aggregator.user(), proc, self.mock_proc, context_aggregator.assistant()]