LLMTask/LLMContextTask: fix LLMService type
This commit is contained in:
@@ -11,6 +11,8 @@ self-contained conversation context, removing the need for subclasses
|
|||||||
to manually wire `LLMContextAggregatorPair`.
|
to manually wire `LLMContextAggregatorPair`.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
from pipecat.pipeline.pipeline import Pipeline
|
from pipecat.pipeline.pipeline import Pipeline
|
||||||
from pipecat.processors.aggregators.llm_context import LLMContext
|
from pipecat.processors.aggregators.llm_context import LLMContext
|
||||||
from pipecat.processors.aggregators.llm_response_universal import (
|
from pipecat.processors.aggregators.llm_response_universal import (
|
||||||
@@ -49,7 +51,7 @@ class LLMContextTask(LLMTask):
|
|||||||
self,
|
self,
|
||||||
name: str,
|
name: str,
|
||||||
*,
|
*,
|
||||||
llm: LLMService,
|
llm: LLMService[Any],
|
||||||
active: bool = False,
|
active: bool = False,
|
||||||
bridged: tuple[str, ...] | None = None,
|
bridged: tuple[str, ...] | None = None,
|
||||||
defer_tool_frames: bool = True,
|
defer_tool_frames: bool = True,
|
||||||
|
|||||||
@@ -78,7 +78,7 @@ class LLMTask(PipelineTask):
|
|||||||
self,
|
self,
|
||||||
name: str,
|
name: str,
|
||||||
*,
|
*,
|
||||||
llm: LLMService,
|
llm: LLMService[Any],
|
||||||
pipeline: Pipeline | None = None,
|
pipeline: Pipeline | None = None,
|
||||||
active: bool = False,
|
active: bool = False,
|
||||||
bridged: tuple[str, ...] | None = None,
|
bridged: tuple[str, ...] | None = None,
|
||||||
|
|||||||
Reference in New Issue
Block a user