added google search tool

.gitignore

@@ -0,0 +1 @@
+.DS_Store

app.py

@@ -34,10 +34,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-
-final_answer = FinalAnswerTool()
-websearch = DuckDuckGoSearchTool()
-
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
@@ -57,7 +53,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_current_time_in_timezone, websearch, final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_current_time_in_timezone, GoogleSearchTool(), DuckDuckGoSearchTool(), FinalAnswerTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,

requirements.txt

@@ -2,4 +2,5 @@ markdownify
 smolagents
 requests
 duckduckgo_search
+googlesearch-python
 pandas

tools/web_search.py

@@ -1,6 +1,7 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 import duckduckgo_search
+import googlesearch
 
 class DuckDuckGoSearchTool(Tool):
     name = "web_search"
@@ -25,3 +26,39 @@ class DuckDuckGoSearchTool(Tool):
             raise Exception("No results found! Try a less restrictive/shorter query.")
         postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
         return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+
+class GoogleSearchTool(Tool):
+    name = "google_search"
+    description = "Performs a Google web search based on your query and returns the top search results."
+    inputs = {'query': {'type': 'string', 'description': 'The search query to perform.'}}
+    output_type = "string"
+
+    def __init__(self, max_results=10, **kwargs):
+        super().__init__()
+        self.max_results = max_results
+        try:
+            from googlesearch import search
+        except ImportError as e:
+            raise ImportError(
+                "You must install package `googlesearch-python` to run this tool: for instance run `pip install googlesearch-python`."
+            ) from e
+        self.search = search
+
+    def forward(self, query: str) -> str:
+        results = []
+        try:
+            # Get search results (URLs only)
+            search_results = list(self.search(query, num_results=self.max_results))
+            
+            # If no results found
+            if len(search_results) == 0:
+                raise Exception("No results found! Try a less restrictive/shorter query.")
+            
+            # Format the results
+            for i, url in enumerate(search_results):
+                title = f"Result {i+1}"  # Basic title since we don't get title data from this library
+                results.append(f"[{title}]({url})\n{url}")
+                
+            return "## Search Results\n\n" + "\n\n".join(results)
+        except Exception as e:
+            return f"Error performing search: {str(e)}"