Update web_search.py

web_search.py

@@ -1,17 +1,20 @@
-# web_search.py — DuckDuckGo basic web RAG
+# web_search.py
 import requests
 
 def web_search(query, max_results=3):
-    headers = {"User-Agent": "Mozilla/5.0"}
-    url = f"https://duckduckgo.com/html/?q={query}"
-    resp = requests.get(url, headers=headers)
-
-    snippets = []
-    for line in resp.text.splitlines():
-        if "result__snippet" in line:
-            text = line.split(">")[-1].split("<")[0]
-            snippets.append(text)
-        if len(snippets) >= max_results:
-            break
-
-    return "\n".join(snippets) if snippets else "No relevant information found."
+    try:
+        url = f"https://duckduckgo.com/html/?q={query}"
+        headers = {'User-Agent': 'Mozilla/5.0'}
+        response = requests.get(url, headers=headers)
+        results = []
+        for line in response.text.split("\n"):
+            if "<a rel=\"nofollow\"" in line and 'href="' in line:
+                start = line.find('href="') + 6
+                end = line.find('"', start)
+                link = line[start:end]
+                results.append(link)
+            if len(results) >= max_results:
+                break
+        return "\n".join(results)
+    except Exception as e:
+        return f"[Web search failed: {e}]"