Update app.py

app.py

@@ -33,7 +33,29 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-search_tool = DuckDuckGoSearchTool()
+# Note: based on https://medium.com/@laurentkubaski/smolagents-duckduckgosearchtool-to-search-in-wikipedia-2578973bb131
+class CustomDuckDuckGoSearchTool(DuckDuckGoSearchTool):
+    name = "web_search"
+    description = "Performs a web search for a query and returns a list of the top search results formatted as markdown with page titles and urls."
+    inputs = {"query": {"type": "string", "description": "The search query to perform."}}
+    output_type = "string"
+
+    def __init__(self, max_results: int = 10, rate_limit: float | None = 1.0, backend: str = "auto", **kwargs):
+        super().__init__(max_results=max_results, rate_limit=rate_limit, **kwargs)
+        self.backend = backend # Add "backend" as new parameter
+
+    def forward(self, query: str) -> str:
+        self._enforce_rate_limit()
+        results = self.ddgs.text(
+            query=query,
+            max_results=self.max_results,
+            backend=self.backend) # there you go
+        if len(results) == 0:
+            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)
+
+search_tool = CustomDuckDuckGoSearchTool(backend="wikipedia")
 final_answer = FinalAnswerTool()
 
 # 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: