SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add duckduckgo search tool (#32882)

Browse Source 
* Add duckduckgo search tool
This commit is contained in:
Aymeric Roucher
2024-09-02 09:56:20 +02:00
committed by GitHub
parent b9bc691e8d
commit 1ca9ff5c91
14 changed files with 109 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/agents/test_tools_common.py
View File

@@ -90,8 +90,9 @@ class ToolTesterMixin:
def test_agent_type_output(self):
inputs = create_inputs(self.tool.inputs)
output = self.tool(**inputs)
agent_type = AGENT_TYPE_MAPPING[self.tool.output_type]
self.assertTrue(isinstance(output, agent_type))
if self.tool.output_type != "any":
agent_type = AGENT_TYPE_MAPPING[self.tool.output_type]
self.assertTrue(isinstance(output, agent_type))
def test_agent_types_inputs(self):
inputs = create_inputs(self.tool.inputs)
@@ -99,9 +100,3 @@ class ToolTesterMixin:
for _input, expected_input in zip(inputs, self.tool.inputs.values()):
input_type = expected_input["type"]
_inputs.append(AGENT_TYPE_MAPPING[input_type](_input))
output_type = AGENT_TYPE_MAPPING[self.tool.output_type]
# Should not raise an error
output = self.tool(**inputs)
self.assertTrue(isinstance(output, output_type))