Update agent.py

agent.py

@@ -13,6 +13,9 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from pathlib import Path
 import tempfile
 from dotenv import load_dotenv
+import traceback
+import time
+import random
 
 # constants
 API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -148,31 +151,97 @@ def build_gemini_llm():
 #         return f"Failed to read file: {e}"
 #     except Exception as e:
 #         return f"error downloading or reading file: {str(e)}"
-import traceback
-
 @tool
 def wikipedia_search(query: str) -> str:
     """
-    Search wikipedia for a query and return results.
-    Takes a string query as the keywords to search
+    Search Wikipedia for a query and return relevant results.
     Args:
-        query (str): Keywords you want to search.
+        query: Keywords to search on Wikipedia.
     """
-    try:
-        search_results = WikipediaLoader(query=query, load_max_docs=3).load()
-
-        if not search_results:
-            return f"No Wikipedia results found for {query}. Consider another query or try a web search."
-        print("wiki result:")
-        print(search_results)
-        return "---\n".join(
-            f"Title: {doc.metadata.get('title', 'Unknown')}\n"
-            f"Content: {doc.page_content}"
-            for doc in search_results
-        )
-    except Exception as e:
-        traceback.print_exc()
-        return f"Wikipedia search failed: {type(e).__name__}: {e}"
+
+    url = "https://en.wikipedia.org/w/api.php"
+    params = {
+        "action": "query",
+        "list": "search",
+        "srsearch": query,
+        "format": "json",
+        "srlimit": 3,
+        "utf8": 1,
+    }
+    headers = {
+        "User-Agent": "MyLangGraphAgent/1.0"
+    }
+
+    max_retries = 3
+    for attempt in range(max_retries):
+        try:
+            response = requests.get(
+                url,
+                params=params,
+                headers=headers,
+                timeout=10,
+            )
+            print(
+                f"Wikipedia status={response.status_code}, "
+                f"content-type={response.headers.get('content-type')}"
+            )
+
+            response.raise_for_status()
+            
+            data = response.json()
+            results = data.get("query", {}).get("search", [])
+            if not results:
+                return f"No Wikipedia results found for '{query}'."
+            return "\n\n---\n\n".join(
+                f"Title: {item['title']}\n"
+                f"Snippet: {item.get('snippet', '')}"
+                for item in results
+            )
+
+        except requests.exceptions.RequestException as e:
+            print(
+                f"Wikipedia request failed "
+                f"attempt {attempt + 1}/{max_retries}: {e}"
+            )
+        except requests.exceptions.JSONDecodeError:
+            print(
+                f"Wikipedia returned non-JSON response. "
+                f"Status={response.status_code}"
+            )
+            print("Response preview:")
+            print(response.text[:500])
+
+        if attempt < max_retries - 1:
+            delay = 2 ** attempt + random.random()
+            print(f"Retrying in {delay:.2f}s")
+            time.sleep(delay)
+    return (
+        f"Wikipedia search temporarily failed for '{query}'. "
+        "Please use another search source."
+    )
+# @tool
+# def wikipedia_search(query: str) -> str:
+#     """
+#     Search wikipedia for a query and return results.
+#     Takes a string query as the keywords to search
+#     Args:
+#         query (str): Keywords you want to search.
+#     """
+#     try:
+#         search_results = WikipediaLoader(query=query, load_max_docs=3).load()
+
+#         if not search_results:
+#             return f"No Wikipedia results found for {query}. Consider another query or try a web search."
+#         print("wiki result:")
+#         print(search_results)
+#         return "---\n".join(
+#             f"Title: {doc.metadata.get('title', 'Unknown')}\n"
+#             f"Content: {doc.page_content}"
+#             for doc in search_results
+#         )
+#     except Exception as e:
+#         traceback.print_exc()
+#         return f"Wikipedia search failed: {type(e).__name__}: {e}"
 
 # fix wikipedia engine builds invalid URL for region="wt-wt"(default) issue from duckducksearchrun
 from langchain_community.utilities import DuckDuckGoSearchAPIWrapper
@@ -186,7 +255,11 @@ search_ddgs = DuckDuckGoSearchRun(
 
 @tool
 def search_web(query: str) -> str:
-    """Search the web using DuckDuckGo."""
+    """
+    Search the web using DuckDuckGoSearchRun.
+    Args:
+        query: Keywords to search, only keywords and spaces.
+    """
     try:
         result = search_ddgs.invoke(query)
         if not result: