github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-19 11:08:55 +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["name"] == "magic_function"
assert tool_call["args"] == {"input": 3} assert tool_call["args"] == {"input": 3}
assert tool_call["id"] is not None assert tool_call["id"] is not None
assert tool_call["type"] == "tool_call"
def _validate_tool_call_message_no_args(message: BaseMessage) -> None: 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["name"] == "magic_function_no_args"
assert tool_call["args"] == {} assert tool_call["args"] == {}
assert tool_call["id"] is not None assert tool_call["id"] is not None
assert tool_call["type"] == "tool_call"
class ChatModelIntegrationTests(ChatModelTests): class ChatModelIntegrationTests(ChatModelTests):
@ -212,6 +214,7 @@ class ChatModelIntegrationTests(ChatModelTests):
assert result.tool_calls assert result.tool_calls
tool_call = result.tool_calls[0] tool_call = result.tool_calls[0]
assert tool_call["args"].get("answer_style") assert tool_call["args"].get("answer_style")
assert tool_call["type"] == "tool_call"
def test_structured_output(self, model: BaseChatModel) -> None: def test_structured_output(self, model: BaseChatModel) -> None:
if not self.has_tool_calling: if not self.has_tool_calling:
@ -266,6 +269,7 @@ class ChatModelIntegrationTests(ChatModelTests):
"name": function_name, "name": function_name,
"args": function_args, "args": function_args,
"id": "abc123", "id": "abc123",
"type": "tool_call",
}, },
], ],
), ),
@ -310,6 +314,7 @@ class ChatModelIntegrationTests(ChatModelTests):
"name": function_name, "name": function_name,
"args": function_args, "args": function_args,
"id": "abc123", "id": "abc123",
"type": "tool_call",
}, },
], ],
), ),
@ -342,6 +347,7 @@ class ChatModelIntegrationTests(ChatModelTests):
"name": function_name, "name": function_name,
"args": function_args, "args": function_args,
"id": "abc123", "id": "abc123",
"type": "tool_call",
}, },
], ],
), ),