KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename web search tool gto WebSearchTool

This commit is contained in:
Aymeric 2024-12-24 12:27:10 +01:00
parent 775f6e8d7f
commit 48b7f53a54
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/smolagents/default_tools.py
View File

@ -139,9 +139,9 @@ class UserInputTool(Tool):
return user_input return user_input
class DuckDuckGoSearchTool(Tool): class WebSearchTool(Tool):
name = "web_search" name = "web_search"
description = """Performs a web search based on your query (think a Google search) then returns the top search results as a list of dict elements. description = """Performs a duckduckgo web search based on your query (think a Google search) then returns the top search results as a list of dict elements.
Each result has keys 'title', 'href' and 'body'.""" Each result has keys 'title', 'href' and 'body'."""
inputs = { inputs = {
"query": {"type": "string", "description": "The search query to perform."} "query": {"type": "string", "description": "The search query to perform."}
@ -223,7 +223,7 @@ __all__ = [
"PythonInterpreterTool", "PythonInterpreterTool",
"FinalAnswerTool", "FinalAnswerTool",
"UserInputTool", "UserInputTool",
"DuckDuckGoSearchTool", "WebSearchTool",
"VisitWebpageTool", "VisitWebpageTool",
"SpeechToTextTool", "SpeechToTextTool",
] ]