Update tools.py

tools.py

@@ -8,20 +8,28 @@ import pytesseract
 from state import AgentState
 from langchain.schema import HumanMessage 
 import regex as re
+import time
+
 def web_search_tool(state: AgentState) -> AgentState:
     """
     Expects: state["web_search_query"] is a non‐empty string.
     Returns: {"web_search_query": None, "web_search_result": <string>}
     We also clear web_search_query so we don’t loop forever.
+    If the result is a DuckDuckGo 202 Ratelimit error, retry up to 5 times with a 5 second sleep between attempts.
     """
     # print("reached web search tool")
     query = state.get("web_search_query", "")
     if not query:
         return {}  # nothing to do
     
-    # Run DuckDuckGo
     ddg = DuckDuckGoSearchRun()
-    result_text = ddg.run(query)
+    max_retries = 5
+    for attempt in range(max_retries):
+        result_text = ddg.run(query)
+        if result_text.strip() != "https://lite.duckduckgo.com/lite/ 202 Ratelimit":
+            break
+        if attempt < max_retries - 1:
+            time.sleep(5)
     print(f"web_search_result: {result_text}")
     return {
         "web_search_query": None,
@@ -214,7 +222,7 @@ def wikipedia_search_tool(state: AgentState) -> AgentState:
         search_data = search_resp.json()
 
         search_results = search_data.get("query", {}).get("search", [])
-        print("wikipedia: search_results",search_results)
+        # print("wikipedia: search_results",search_results)
         if not search_results:
             return {"wiki_query": None, "wiki_result": f"No Wikipedia page found for '{query}'."}