This commit is contained in:
Bagatur 2024-10-23 15:38:48 -07:00
parent 0c1a576218
commit d13d101c66
3 changed files with 7 additions and 3 deletions

View File

@ -1075,7 +1075,8 @@ Callbacks = Optional[Union[list[BaseCallbackHandler], BaseCallbackManager]]
from typing_extensions import TypedDict from typing_extensions import TypedDict
class _ToolSchema(TypedDict): class _ToolSchema(TypedDict):
name: str name: str
description: str description: str
parameters: dict parameters: dict

View File

@ -272,7 +272,9 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
) )
raise ValueError(msg) raise ValueError(msg)
def _standardize_tools(self, tools: Optional[Sequence]) -> Optional[List[_ToolSchema]]: def _standardize_tools(
self, tools: Optional[Sequence]
) -> Optional[List[_ToolSchema]]:
"""Convert tools to standard format for tracing.""" """Convert tools to standard format for tracing."""
if not tools: if not tools:
return tools return tools

View File

@ -5,9 +5,10 @@ from __future__ import annotations
import copy import copy
import logging import logging
import warnings import warnings
from collections.abc import Sequence
from concurrent.futures import ThreadPoolExecutor from concurrent.futures import ThreadPoolExecutor
from datetime import datetime, timezone from datetime import datetime, timezone
from typing import TYPE_CHECKING, Any, Optional, Union, Sequence from typing import TYPE_CHECKING, Any, Optional, Union
from uuid import UUID from uuid import UUID
from langsmith import Client from langsmith import Client