github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

standard-tests[patch]: test that bind_tools can accept regular python function (#24135)

Browse Source
This commit is contained in:
Isaac Francisco 2024-07-11 10:42:17 -07:00 committed by GitHub
parent 2428984205
commit 7a62d3dbd6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
libs/standard-tests/langchain_standard_tests/unit_tests/chat_models.py
View File

@ -21,6 +21,11 @@ def my_adder_tool(a: int, b: int) -> int:
return a + b return a + b
def my_adder(a: int, b: int) -> int:
"""Takes two integers, a and b, and returns their sum."""
return a + b
class ChatModelTests(ABC): class ChatModelTests(ABC):
@property @property
@abstractmethod @abstractmethod
@ -108,7 +113,7 @@ class ChatModelUnitTests(ChatModelTests):
return return
tool_model = model.bind_tools( tool_model = model.bind_tools(
[Person, Person.schema(), my_adder_tool], tool_choice="any" [Person, Person.schema(), my_adder_tool, my_adder], tool_choice="any"
) )
assert isinstance(tool_model, RunnableBinding) assert isinstance(tool_model, RunnableBinding)