Update tools/search_tool.py

tools/search_tool.py

@@ -1,18 +1,23 @@
 from smolagents import tool
 from duckduckgo_search import DDGS
 
+
 @tool
 def web_search(query: str) -> str:
     """
+    Search the web for factual answers about the query using DuckDuckGo.
+
     Args:
-        query (str): The factual question to answer.
+        query (str): The question or search term to look up.
+
     Returns:
-        str: Extracted fact and URL.
+        str: Relevant information extracted from search results, preferably from Wikipedia.
     """
     refined = f"{query} site:en.wikipedia.org"
     with DDGS() as ddgs:
         results = ddgs.text(refined)
         for r in results[:5]:
-            if "Mercedes Sosa" in r['body'] or "albums" in r['body']:
-                return f"{r['body']}\\n\\nSource: [{r['href']}]({r['href']})"
+            snippet = r.get("body") or r.get("content") or ""
+            if snippet:
+                return f"{snippet}\n\nSource: [{r['href']}]({r['href']})"
         return "Could not find a direct answer from Wikipedia."