github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-09 15:03:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

docs, standard-tests: property tags, support tool decorator (#28234)

Browse Source
This commit is contained in:
Erick Friis
2024-11-20 09:19:03 -08:00
committed by GitHub
parent 4027da1b6e
commit 43e24cd4a1
2 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/standard-tests/langchain_tests/unit_tests/tools.py
View File

@@ -13,7 +13,7 @@ from langchain_tests.base import BaseStandardTests
class ToolsTests(BaseStandardTests):
@property
@abstractmethod
def tool_constructor(self) -> Union[Type[BaseTool], Callable]: ...
def tool_constructor(self) -> Union[Type[BaseTool], Callable, BaseTool]: ...
@property
def tool_constructor_params(self) -> dict:
@@ -31,6 +31,14 @@ class ToolsTests(BaseStandardTests):
@pytest.fixture
def tool(self) -> BaseTool:
if isinstance(self.tool_constructor, BaseTool):
if self.tool_constructor_params != {}:
msg = (
"If tool_constructor is an instance of BaseTool, "
"tool_constructor_params must be empty"
)
raise ValueError(msg)
return self.tool_constructor
return self.tool_constructor(**self.tool_constructor_params)