Update tools/web_search.py

tools/web_search.py

@@ -44,44 +44,44 @@ class DuckDuckGoSearchTool(Tool):
 
     def forward(self, query: str) -> str:
 
+    queries = [
+        query,
+        query + " wikipedia",
+        query + " official",
+        query.replace("-", " "),
+    ]
+
+    seen = set()
+    formatted = []
+
+    for q in queries:
+
         try:
 
             results = list(
                 self.ddgs.text(
-                    query,
-                    max_results=self.max_results
+                    q,
+                    max_results=3
                 )
             )
 
-        except Exception as e:
-
-            return f"Search failed.\nReason: {e}"
+        except Exception:
+            continue
 
-        if not results:
+        for result in results:
 
-            return "No search results found."
-
-        formatted = []
+            url = result.get("href", "")
 
-        seen_urls = set()
+            if url in seen:
+                continue
 
-        for i, result in enumerate(results, start=1):
+            seen.add(url)
 
             title = result.get("title", "No title")
-
-            url = result.get("href", "")
-
             snippet = result.get("body", "")
 
-            if url in seen_urls:
-                continue
-
-            seen_urls.add(url)
-
             formatted.append(
                 f"""
-Result {i}
-
 Title:
 {title}
 
@@ -93,4 +93,10 @@ Snippet:
 """
             )
 
-        return "\n".join(formatted)
+        if len(formatted) >= 5:
+            break
+
+    if not formatted:
+        return "No useful search results were found."
+
+    return "\n".join(formatted[:5])