diff --git a/src/pipecat/tasks/llm/llm_context_task.py b/src/pipecat/tasks/llm/llm_context_task.py index 6a570e4fb..189fd0044 100644 --- a/src/pipecat/tasks/llm/llm_context_task.py +++ b/src/pipecat/tasks/llm/llm_context_task.py @@ -11,6 +11,8 @@ self-contained conversation context, removing the need for subclasses to manually wire `LLMContextAggregatorPair`. """ +from typing import Any + from pipecat.pipeline.pipeline import Pipeline from pipecat.processors.aggregators.llm_context import LLMContext from pipecat.processors.aggregators.llm_response_universal import ( @@ -49,7 +51,7 @@ class LLMContextTask(LLMTask): self, name: str, *, - llm: LLMService, + llm: LLMService[Any], active: bool = False, bridged: tuple[str, ...] | None = None, defer_tool_frames: bool = True, diff --git a/src/pipecat/tasks/llm/llm_task.py b/src/pipecat/tasks/llm/llm_task.py index 7b05fd5e5..c652ba60e 100644 --- a/src/pipecat/tasks/llm/llm_task.py +++ b/src/pipecat/tasks/llm/llm_task.py @@ -78,7 +78,7 @@ class LLMTask(PipelineTask): self, name: str, *, - llm: LLMService, + llm: LLMService[Any], pipeline: Pipeline | None = None, active: bool = False, bridged: tuple[str, ...] | None = None,