Minor fix: add a type annotation

This commit is contained in:
Paul Kompfner
2025-08-15 17:07:24 -04:00
parent d50922cdcd
commit 337f00c16c
2 changed files with 4 additions and 4 deletions

View File

@@ -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:

View File

@@ -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]):