Modified files

agent.py

@@ -8,6 +8,7 @@ from smolagents import InferenceClientModel
 from dotenv import load_dotenv
 from markdownify import markdownify
 from requests.exceptions import RequestException
+from langchain_community.utilities import SerpAPIWrapper
 
 def initialize_agent():
     # Load environment variables from .env file
@@ -51,10 +52,26 @@ def initialize_agent():
         except Exception as e:
             return f"An unexpected error occurred: {str(e)}"
 
+    @tool
+    def google_search(query: str) -> str:
+        """Searches Google for the given query and returns the results.
+
+        Args:
+            query: The query to search for.
+
+        Returns:
+            The search results, or an error message if the search fails.
+        """
+        try:
+            serpapi = SerpAPIWrapper()
+            return serpapi.run(query)
+        except Exception as e:
+            return f"Error performing Google search: {str(e)}"
+
     # 3. Define the agents
     if model:
         web_agent = ToolCallingAgent(
-            tools=[WebSearchTool(), visit_webpage],
+            tools=[WebSearchTool(), visit_webpage, google_search],
             model=model,
             max_steps=10,
             name="web_search_agent",

requirements.txt

@@ -5,4 +5,5 @@ huggingface_hub
 gradio
 markdownify
 duckduckgo-search
-wikipedia
+wikipedia
+serpapi-python