Update tools/web_search.py

tools/web_search.py

@@ -2,15 +2,17 @@ from typing import Any, Optional
 from smolagents.tools import Tool
 import duckduckgo_search
 
+
 class DuckDuckGoSearchTool(Tool):
     name = "web_search"
     description = "Performs a duckduckgo web search based on your query (think a Google search) then 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):
+    def __init__(self, max_results=10, backend="auto", **kwargs):
         super().__init__()
         self.max_results = max_results
+        self.backend = backend
         try:
             from duckduckgo_search import DDGS
         except ImportError as e:
@@ -20,25 +22,29 @@ class DuckDuckGoSearchTool(Tool):
         self.ddgs = DDGS(**kwargs)
 
     def forward(self, query: str) -> str:
-        # results = self.ddgs.text(query, max_results=self.max_results)
-        # if len(results) == 0:
-        # results = list(self.ddgs.text(query, max_results=self.max_results))
-        # if not results:
-        #     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]
-        try:
-            results = list(self.ddgs.text(query, max_results=self.max_results))
-        except Exception as e:
+        backends_to_try = [self.backend] + [b for b in ("auto", "lite", "html", "api") if b != self.backend]
+        last_error = None
+        for backend in backends_to_try:
+            try:
+                results = list(
+                    self.ddgs.text(query, max_results=self.max_results, backend=backend)
+                )
+            except Exception as e:
+                last_error = e
+                continue
+            if results:
+                postprocessed_results = [
+                    f"[{result['title']}]({result['href']})\n{result['body']}"
+                    for result in results
+                ]
+                return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+        if last_error is not None:
             return (
                 "Web search failed. This can happen if outbound network access is blocked or the "
-                f"DuckDuckGo API is rate-limiting requests. Error: {e}"
+                f"DuckDuckGo API is rate-limiting requests. Error: {last_error}"
             )
-        if not results:
-            return (
-                "No results found. This may indicate network restrictions or a temporary "
-                "DuckDuckGo issue. Try a shorter query or retry later."
-            )
-        postprocessed_results = [
-            f"[{result['title']}]({result['href']})\n{result['body']}" for result in results
-        ]
-        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+        return (
+            "No results found. This may indicate network restrictions or a temporary "
+            "DuckDuckGo issue. Try a shorter query or retry later."
+        )
+