test(langchain): Use new PipelineParams in test

This commit is contained in:
TomTom101
2024-06-02 20:19:55 +02:00
parent 410dbd3dfc
commit 89673a4040

View File

@@ -10,7 +10,7 @@ from pipecat.frames.frames import (LLMFullResponseEndFrame,
UserStoppedSpeakingFrame) UserStoppedSpeakingFrame)
from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.pipeline import Pipeline
from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.runner import PipelineRunner
from pipecat.pipeline.task import PipelineTask from pipecat.pipeline.task import PipelineParams, PipelineTask
from pipecat.processors.aggregators.llm_response import ( from pipecat.processors.aggregators.llm_response import (
LLMAssistantResponseAggregator, LLMUserResponseAggregator) LLMAssistantResponseAggregator, LLMUserResponseAggregator)
from pipecat.processors.frame_processor import FrameProcessor from pipecat.processors.frame_processor import FrameProcessor
@@ -63,7 +63,7 @@ class TestLangchain(unittest.IsolatedAsyncioTestCase):
] ]
) )
task = PipelineTask(pipeline) task = PipelineTask(pipeline, PipelineParams(allow_interruptions=False))
await task.queue_frames( await task.queue_frames(
[ [
UserStartedSpeakingFrame(), UserStartedSpeakingFrame(),