Added custom google search tool using googlesearch-python.

functions/agent.py

@@ -1,13 +1,14 @@
 '''Agent definition for GAIA question answering system.'''
 
 # Imports for agent creation
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, VisitWebpageTool, Tool
+from smolagents import CodeAgent, InferenceClientModel, VisitWebpageTool, Tool
 from langchain_community.agent_toolkits.load_tools import load_tools
+from functions.tools import google_search
 
 def create_agent():
     '''Creates agent for GAIA question answering system.'''
 
-    wikipedia_tool = Tool.from_langchain(
+    wikipedia = Tool.from_langchain(
         load_tools(["wikipedia"])[0]
     )
 
@@ -16,7 +17,7 @@ def create_agent():
     )
 
     agent = CodeAgent(
-        tools=[wikipedia_tool, DuckDuckGoSearchTool(), VisitWebpageTool()],
+        tools=[wikipedia, google_search, VisitWebpageTool()],
         model=model
     )
 

functions/tools.py

@@ -1 +1,20 @@
 '''Tools for GAIA question answering agent.'''
+
+from smolagents import tool
+from googlesearch import search
+
+@tool
+def google_search(query: str) -> str:
+    """
+    Perform a Google search and return the top 10 results.
+    
+    Args:
+        query (str): The search query.
+        
+    Returns:
+        str: The URLs of the top search results, separated by newlines.
+    """
+
+    results = list(search(query, num_results=10, advanced=True))
+
+    return results

requirements.txt

@@ -1,4 +1,5 @@
 duckduckgo-search
+googlesearch-python
 gradio[oauth]
 markdownify
 requests