Update tools/search_tool.py

tools/search_tool.py

@@ -4,15 +4,15 @@ from duckduckgo_search import DDGS
 @tool
 def web_search(query: str) -> str:
     """
-    Returns top search results for a given query.
-
     Args:
-        query (str): The search string.
-
+        query (str): The factual question to answer.
     Returns:
-        str: Top search results from DuckDuckGo.
+        str: Extracted fact and URL.
     """
+    refined = f"{query} site:en.wikipedia.org"
     with DDGS() as ddgs:
-        results = ddgs.text(query)
-        output = [f"{r['title']} - {r['href']}" for r in results[:3]]
-        return "\n".join(output) if output else "No relevant results found."
+        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']})"
+        return "Could not find a direct answer from Wikipedia."