web

agent.py

@@ -4,7 +4,7 @@ from langgraph.prebuilt import create_react_agent
 from tools import (
     wikipedia_search_tool, arxiv_search_tool,
     audio_transcriber_tool, excel_tool, analyze_code_tool, image_tool,
-    add_tool, subtract_tool, multiply_tool, divide_tool
+    add_tool, subtract_tool, multiply_tool, divide_tool, web_search_tool
 )
 
 # ──────────────────────────────── Config ────────────────────────────────
@@ -33,7 +33,7 @@ IMPORTANT:
 TOOLS = [
     wikipedia_search_tool, arxiv_search_tool,
     audio_transcriber_tool, excel_tool, analyze_code_tool, image_tool,
-    add_tool, subtract_tool, multiply_tool, divide_tool
+    add_tool, subtract_tool, multiply_tool, divide_tool, web_search_tool
 ]
 
 # ───────────────────────────── Agent Wrapper ─────────────────────────────

tools.py

@@ -262,13 +262,15 @@ def wikipedia_search_tool(wiki_query: str) -> str:
         summary_text = summary_data.get("extract")
         if not summary_text:
             summary_text = summary_data.get("description", "No summary available.")
-        print(f"Title: {first_title}\n\n{summary_text}")
+        print("Submitted wiki successfully")
         return f"Title: {first_title}\n\n{summary_text}"
         
 
     except requests.exceptions.RequestException as e:
+        print("Wikipedia search error: ", e)
         return f"Wikipedia search error: {e}"
     except Exception as e:
+        print("Unexpected error in wikipedia_search_tool: ", e)
         return f"Unexpected error in wikipedia_search_tool: {e}"
 
 @tool
@@ -442,54 +444,52 @@ def divide_tool(a: float, b: float) -> str:
     result = a / b
     return f"Division result: {a} ÷ {b} = {result}"
 
+@tool
+def web_search_tool(query: str) -> str:
+    """
+    TOOL NAME: Web Search Tool
 
+    Purpose: When the user asks for current information, recent news, or topics not covered by Wikipedia, use this tool.
 
+    Input: A string describing what to search for on the web.
 
-# 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>}.
-#     Retries up to 5 times on either a DuckDuckGo "202 Ratelimit" response or any exception (e.g. timeout).
-#     """
-#     print("reached web_search_tool")
-#     query = state.get("web_search_query", "")
-#     if not query:
-#         return {}  # nothing to do
-
-#     ddg = DDGS()
-#     max_retries = 5
-#     result_text = ""
-
-#     for attempt in range(1, max_retries + 1):
-#         try:
-#             result_text = str(ddg.text(query, max_results=5))
-#         except Exception as e:
-#             # Network error or timeout—retry up to max_retries
-#             if attempt < max_retries:
-#                 print(f"web_search_tool: exception '{e}', retrying in 4 seconds ({attempt}/{max_retries})")
-#                 time.sleep(4)
-#                 continue
-#             else:
-#                 # Final attempt failed
-#                 return {
-#                     "web_search_query": None,
-#                     "web_search_result": f"Error during DuckDuckGo search: {e}"
-#                 }
-
-#         # Check for DuckDuckGo rate‐limit indicator
-#         if "202 Ratelimit" in result_text:
-#             if attempt < max_retries:
-#                 print(f"web_search_tool: received '202 Ratelimit', retrying in 4 seconds ({attempt}/{max_retries})")
-#                 time.sleep(4)
-#                 continue
-#             else:
-#                 # Final attempt still rate‐limited
-#                 break
-
-#         # Successful response (no exception and no rate‐limit text)
-#         break
-
-#     return {
-#         "web_search_query": None,
-#         "web_search_result": result_text
-#     }
+    Example usage:
+    - "What's the latest news about AI?"
+    - "Current stock price of Tesla"
+    - "Recent developments in renewable energy"
+    """
+    print("reached web_search_tool")
+    if not query:
+        return "No search query provided."
+
+    ddg = DDGS()
+    max_retries = 5
+    result_text = ""
+
+    for attempt in range(1, max_retries + 1):
+        try:
+            result_text = str(ddg.text(query, max_results=5))
+        except Exception as e:
+            # Network error or timeout—retry up to max_retries
+            if attempt < max_retries:
+                print(f"web_search_tool: exception '{e}', retrying in 4 seconds ({attempt}/{max_retries})")
+                time.sleep(4)
+                continue
+            else:
+                # Final attempt failed
+                return f"Error during DuckDuckGo search: {e}"
+
+        # Check for DuckDuckGo rate‐limit indicator
+        if "202 Ratelimit" in result_text:
+            if attempt < max_retries:
+                print(f"web_search_tool: received '202 Ratelimit', retrying in 4 seconds ({attempt}/{max_retries})")
+                time.sleep(4)
+                continue
+            else:
+                # Final attempt still rate‐limited
+                break
+
+        # Successful response (no exception and no rate‐limit text)
+        break
+    print("Submitted web search successfully")
+    return result_text