Update tools/web_search.py

tools/web_search.py

@@ -1,6 +1,8 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
-import duckduckgo_search
+# import duckduckgo_search
+import ddgs
+
 
 class DuckDuckGoSearchTool(Tool):
     name = "web_search"
@@ -12,7 +14,8 @@ class DuckDuckGoSearchTool(Tool):
         super().__init__()
         self.max_results = max_results
         try:
-            from duckduckgo_search import DDGS
+            # from duckduckgo_search import DDGS
+            from ddgs import DDGS
         except ImportError as e:
             raise ImportError(
                 "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
@@ -20,8 +23,12 @@ 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]
-        return "## Search Results\n\n" + "\n\n".join(postprocessed_results)
+
+        post = [
+            f"[{r.get('title')}]({r.get('href')})\n{r.get('body')}"
+            for r in results
+        ]
+        return "## Search Results\n\n" + "\n\n".join(post)