Minor fix: add a type annotation
This commit is contained in:
@@ -16,7 +16,7 @@ from typing import Any, Generic, List, TypeVar, Union, cast
|
|||||||
from loguru import logger
|
from loguru import logger
|
||||||
|
|
||||||
from pipecat.adapters.schemas.tools_schema import ToolsSchema
|
from pipecat.adapters.schemas.tools_schema import ToolsSchema
|
||||||
from pipecat.processors.aggregators.llm_context import LLMContext
|
from pipecat.processors.aggregators.llm_context import LLMContext, NotGiven
|
||||||
|
|
||||||
# Should be a TypedDict
|
# Should be a TypedDict
|
||||||
TLLMInvocationParams = TypeVar("TLLMInvocationParams", bound=dict[str, Any])
|
TLLMInvocationParams = TypeVar("TLLMInvocationParams", bound=dict[str, Any])
|
||||||
@@ -75,8 +75,7 @@ class BaseLLMAdapter(ABC, Generic[TLLMInvocationParams]):
|
|||||||
"""
|
"""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
# TODO: should this also be able to return NotGiven?
|
def from_standard_tools(self, tools: Any) -> List[Any] | NotGiven:
|
||||||
def from_standard_tools(self, tools: Any) -> List[Any]:
|
|
||||||
"""Convert tools from standard format to provider format.
|
"""Convert tools from standard format to provider format.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
|
|||||||
@@ -12,6 +12,7 @@ from dataclasses import dataclass
|
|||||||
from typing import Any, List, Optional, TypedDict
|
from typing import Any, List, Optional, TypedDict
|
||||||
|
|
||||||
from loguru import logger
|
from loguru import logger
|
||||||
|
from openai import NotGiven
|
||||||
|
|
||||||
from pipecat.adapters.base_llm_adapter import BaseLLMAdapter
|
from pipecat.adapters.base_llm_adapter import BaseLLMAdapter
|
||||||
from pipecat.adapters.schemas.tools_schema import AdapterType, ToolsSchema
|
from pipecat.adapters.schemas.tools_schema import AdapterType, ToolsSchema
|
||||||
@@ -41,7 +42,7 @@ class GeminiLLMInvocationParams(TypedDict):
|
|||||||
|
|
||||||
system_instruction: Optional[str]
|
system_instruction: Optional[str]
|
||||||
messages: List[Content]
|
messages: List[Content]
|
||||||
tools: List[Any]
|
tools: List[Any] | NotGiven
|
||||||
|
|
||||||
|
|
||||||
class GeminiLLMAdapter(BaseLLMAdapter[GeminiLLMInvocationParams]):
|
class GeminiLLMAdapter(BaseLLMAdapter[GeminiLLMInvocationParams]):
|
||||||
|
|||||||
Reference in New Issue
Block a user