Search api tool

agents/BasicAgent.py

@@ -1,7 +1,8 @@
 import os
 
-from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, HfApiModel, LiteLLMModel, Tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, VisitWebpageTool, HfApiModel, LiteLLMModel, Tool, GoogleSearchTool
 from langchain_community.tools.wikidata.tool import WikidataAPIWrapper, WikidataQueryRun
+from langchain_community.tools.brave_search.tool import BraveSearch
 
 class BasicAgent(CodeAgent):
     def __init__(self):
@@ -27,7 +28,7 @@ class BasicAgent(CodeAgent):
         # TOOLS
         print("Initializing tools...")
         try:
-            search_tool = DuckDuckGoSearchTool()
+            search_tool = Tool.from_langchain(BraveSearch.from_api_key(os.environ.get("SEARCH_API_KEY", "")))
             visit_tool = VisitWebpageTool()
             wikidata = Tool.from_langchain(WikidataQueryRun(api_wrapper=WikidataAPIWrapper())) 
             tools = [search_tool, visit_tool, wikidata]