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
class _ToolSchema(TypedDict):
name: str
description: str
parameters: dict
parameters: dict

View File

@ -272,7 +272,9 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
)
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."""
if not tools:
return tools

View File

@ -5,9 +5,10 @@ from __future__ import annotations
import copy
import logging
import warnings
from collections.abc import Sequence
from concurrent.futures import ThreadPoolExecutor
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 langsmith import Client