Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-10 01:10:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

standard-tests[patch]: check for ToolCall["type"] (#24209)

Browse Source
This commit is contained in:
Bagatur 2024-07-12 16:17:34 -07:00 committed by GitHub
parent 3f1cf00d97
commit bdc03997c9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py
View File

@ -44,6 +44,7 @@ def _validate_tool_call_message(message: BaseMessage) -> None:
assert tool_call["name"] == "magic_function"
assert tool_call["args"] == {"input": 3}
assert tool_call["id"] is not None
assert tool_call["type"] == "tool_call"
def _validate_tool_call_message_no_args(message: BaseMessage) -> None:
@ -53,6 +54,7 @@ def _validate_tool_call_message_no_args(message: BaseMessage) -> None:
assert tool_call["name"] == "magic_function_no_args"
assert tool_call["args"] == {}
assert tool_call["id"] is not None
assert tool_call["type"] == "tool_call"
class ChatModelIntegrationTests(ChatModelTests):
@ -212,6 +214,7 @@ class ChatModelIntegrationTests(ChatModelTests):
assert result.tool_calls
tool_call = result.tool_calls[0]
assert tool_call["args"].get("answer_style")
assert tool_call["type"] == "tool_call"
def test_structured_output(self, model: BaseChatModel) -> None:
if not self.has_tool_calling:
@ -266,6 +269,7 @@ class ChatModelIntegrationTests(ChatModelTests):
"name": function_name,
"args": function_args,
"id": "abc123",
"type": "tool_call",
},
],
),
@ -310,6 +314,7 @@ class ChatModelIntegrationTests(ChatModelTests):
"name": function_name,
"args": function_args,
"id": "abc123",
"type": "tool_call",
},
],
),
@ -342,6 +347,7 @@ class ChatModelIntegrationTests(ChatModelTests):
"name": function_name,
"args": function_args,
"id": "abc123",
"type": "tool_call",
},
],
),