web

agent.py

@@ -11,17 +11,13 @@ from tools import (
 SYSTEM_PROMPT = """
 You are a helpful AI assistant. You will answer questions by thinking step-by-step, taking actions using tools when necessary, and finishing with a final answer.
 
-When you want to use a tool, respond *exactly* in the following format:
 
-Thought: [your reasoning]
-Action: [tool_name]
+
+
 Action Input: [input]
 
 When you receive an observation, continue reasoning. Write:
 
-Thought: [final reasoning]
-FINAL ANSWER: [your answer]
-
 IMPORTANT:
 - YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
 - If using tools that require a `task_id`, only use the value provided.
@@ -31,6 +27,10 @@ IMPORTANT:
 - Try to use the web search tool first before using wiki and arxiv  tools.
 - If wiki tool doesnt give you enough information, try again with a shorter query, or a more genral query. Dont use a query that is too specific.
 - DOnt user wiki tool, web search tool and arxiv tool more than 3 times.
+
+Return the final answer in the following format:
+Thought: [final reasoning]
+FINAL ANSWER: [your answer]
 """
 
 TOOLS = [

tools.py

@@ -219,12 +219,21 @@ def wikipedia_search_tool(wiki_query: str) -> str:
     - "What is the capital of France?"
     - "Find information about quantum computing"
     - "What is the history of the internet?"
-    If no valid wiki_query is provided, returns {}.
+    If no valid wiki_query is provided, returns an empty string.
     """
     print("reached wikipedia search tool")
-    query = wiki_query
+
+    # --- Simple in-memory cache to avoid repeated look-ups in a single session
+    if not hasattr(wikipedia_search_tool, "_cache"):
+        wikipedia_search_tool._cache = {}
+
+    query = wiki_query.strip()
     if not query:
-        return {}
+        return ""
+
+    if query in wikipedia_search_tool._cache:
+        print("Returning cached Wikipedia result for query:", query)
+        return wikipedia_search_tool._cache[query]
 
     try:
         # 1) Use the MediaWiki API to search for page titles matching the query
@@ -240,16 +249,17 @@ def wikipedia_search_tool(wiki_query: str) -> str:
         search_data = search_resp.json()
 
         search_results = search_data.get("query", {}).get("search", [])
-        # print("wikipedia: search_results",search_results)
         if not search_results:
-            print(f"No Wikipedia page found for '{query}'.")
-            return f"No Wikipedia page found for '{query}'."
+            msg = f"No Wikipedia page found for '{query}'. [END_OF_SEARCH]"
+            wikipedia_search_tool._cache[query] = msg
+            return msg
 
         # 2) Take the first search result's title
         first_title = search_results[0].get("title", "")
         if not first_title:
-            print("Unexpected format from Wikipedia search.")
-            return "Unexpected format from Wikipedia search."
+            msg = "Unexpected format from Wikipedia search. [END_OF_SEARCH]"
+            wikipedia_search_tool._cache[query] = msg
+            return msg
 
         # 3) Fetch the page summary for that title via the REST summary endpoint
         title_for_url = requests.utils.requote_uri(first_title)
@@ -262,16 +272,18 @@ 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.")
+
+        result = f"Title: {first_title}\n\n{summary_text}\n\n[END_OF_SEARCH]"
+        wikipedia_search_tool._cache[query] = result
         print("Submitted wiki successfully")
-        return f"Title: {first_title}\n\n{summary_text}"
-        
+        return result
 
     except requests.exceptions.RequestException as e:
         print("Wikipedia search error: ", e)
-        return f"Wikipedia search error: {e}"
+        return f"Wikipedia search error: {e} [END_OF_SEARCH]"
     except Exception as e:
         print("Unexpected error in wikipedia_search_tool: ", e)
-        return f"Unexpected error in wikipedia_search_tool: {e}"
+        return f"Unexpected error in wikipedia_search_tool: {e} [END_OF_SEARCH]"
 
 @tool
 def arxiv_search_tool(query: str) -> str:
@@ -452,16 +464,20 @@ def web_search_tool(query: str) -> str:
     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.
-
-    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 hasattr(web_search_tool, "_cache"):
+        web_search_tool._cache = {}
+
+    query = query.strip()
     if not query:
         return "No search query provided."
 
+    if query in web_search_tool._cache:
+        print("Returning cached web search result for query:", query)
+        return web_search_tool._cache[query]
+
     ddg = DDGS()
     max_retries = 5
     result_text = ""
@@ -470,26 +486,23 @@ def web_search_tool(query: str) -> str:
         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}"
+                return f"Error during DuckDuckGo search: {e} [END_OF_SEARCH]"
 
-        # 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
+        break  # Successful
 
-        # Successful response (no exception and no rate‐limit text)
-        break
+    result_text += "\n\n[END_OF_SEARCH]"
+    web_search_tool._cache[query] = result_text
     print("Submitted web search successfully")
     return result_text