keyword_solve

app.py

@@ -188,13 +188,31 @@ class BasicAgent:
         return state
 
     def _perform_search(self, state: AgentState) -> AgentState:
-        results = simple_search(state["search_query"], max_results=5)
-        print("\nSearch Results:")
-        for i, s in enumerate(results, 1):
-            print(f"[{i}] {s[:120]}…")
-        state["history"].append({"step": "search", "results": results})
-        state["logs"]["search"] = {"query": state["search_query"], "results": results}
-        state["needs_search"] = not results  # Set to True if no results found
+        try:
+            results = simple_search(state["search_query"], max_results=5)
+            print("\nSearch Results:")
+            for i, s in enumerate(results, 1):
+                print(f"[{i}] {s[:120]}…")
+            
+            if not results:
+                print("Warning: No search results found")
+                state["needs_search"] = True  # Try reformulating query
+            else:
+                state["needs_search"] = False
+                
+            state["history"].append({"step": "search", "results": results})
+            state["logs"]["search"] = {
+                "query": state["search_query"],
+                "results_count": len(results),
+                "results": results
+            }
+            
+        except Exception as e:
+            print(f"Search error: {str(e)}")
+            state["needs_search"] = True
+            state["history"].append({"step": "search", "error": str(e)})
+            state["logs"]["search_error"] = str(e)
+            
         state["current_step"] = "recheck"
         return state
 

tools.py

@@ -79,7 +79,40 @@ def retry_ddg(
 # -------- the only search function your agent will call
 def simple_search(query: str, max_results: int = 5) -> List[str]:
     """
-    Perform a DuckDuckGo search and return 'title – url' snippets.
+    Perform a web search using DuckDuckGo and return formatted results.
+    Includes retry logic and better error handling.
     """
-    query = tighten(query)           # optional heuristic cleaner
-    return retry_ddg(query, max_results)  # retry on failure
+    def _raw_search(q: str, max_results: int) -> List[str]:
+        try:
+            with DDGS() as ddgs:
+                results = []
+                for r in ddgs.text(q, max_results=max_results):
+                    # Format each result with title, link and snippet
+                    result = f"{r['title']} – {r['link']}\n{r['body']}"
+                    results.append(result)
+                return results
+        except Exception as e:
+            print(f"Search error: {str(e)}")
+            return []
+
+    # Retry logic with exponential backoff
+    max_attempts = 4
+    base_delay = 10  # seconds
+    
+    for attempt in range(max_attempts):
+        try:
+            results = _raw_search(query, max_results)
+            if results:
+                return results
+            print(f"Attempt {attempt + 1}/{max_attempts}: No results found")
+        except Exception as e:
+            print(f"Attempt {attempt + 1}/{max_attempts} failed: {str(e)}")
+            if attempt < max_attempts - 1:
+                delay = base_delay * (2 ** attempt)  # exponential backoff
+                print(f"Retrying in {delay}s...")
+                time.sleep(delay)
+            else:
+                print(f"All {max_attempts} attempts failed. Last exception: {str(e)}")
+                return []
+
+    return []