Update tools.py

tools.py

@@ -36,6 +36,35 @@ def _download_file_for_task(task_id: str, ext: str) -> str:
     return ""
 
 
+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
+    
+    ddg = DuckDuckGoSearchRun()
+    max_retries = 5
+    for attempt in range(max_retries):
+        result_text = ddg.run(query)
+        if "202 Ratelimit" not in result_text:
+            break
+        if attempt < max_retries - 1:
+            print(f"web_search_result: rate limit error, retrying in 10 seconds")
+            time.sleep(3)
+    print(f"web_search_result reached ")
+    return {
+        "web_search_query": None,
+        "web_search_result": result_text
+    }
+
+
+
 def ocr_image_tool(state: AgentState) -> AgentState:
     """
     Expects state["ocr_path"] to be either: