Update agent.py

agent.py

@@ -13,7 +13,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from pathlib import Path
 import tempfile
 from dotenv import load_dotenv
-import wikipedia
+# import wikipedia
 
 # constants
 API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -203,27 +203,27 @@ def download_and_read_file(task_id: str) -> str:
         if os.path.exists(temp_path):
             os.remove(temp_path)
 
-@tool
-def wikipedia_search(query: str):
-    """
-    Search wikipedia for a query and return a max of three results.
-    Takes a string query as the search query, DO NOT include advanced search operators like "site:...", URLs, or file types, only accepts clean text keywords.
-    Args:
-        query (str): The clean search keywords without any search operators.
-    """
-    try:
-        search_results = WikipediaLoader(query=query, load_max_docs=3).load()
+# @tool
+# def wikipedia_search(query: str):
+#     """
+#     Search wikipedia for a query and return a max of three results.
+#     Takes a string query as the search query, DO NOT include advanced search operators like "site:...", URLs, or file types, only accepts clean text keywords.
+#     Args:
+#         query (str): The clean search keywords without any search operators.
+#     """
+#     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."
-        return "\n\n---\n\n".join(
-            f"Title: {doc.metadata.get('title', 'Unknown')}\n"
-            f"Content: {doc.page_content}"
-            for doc in search_results
-        )
-    except Exception as e:
-        print(f"Wikipedia search failed for {query}: {e}")
-        return f"Wikipedia search failed for {query} due to {str(e)}."
+#         if not search_results:
+#             return f"No Wikipedia results found for {query}. Consider another query or try a web search."
+#         return "\n\n---\n\n".join(
+#             f"Title: {doc.metadata.get('title', 'Unknown')}\n"
+#             f"Content: {doc.page_content}"
+#             for doc in search_results
+#         )
+#     except Exception as e:
+#         print(f"Wikipedia search failed for {query}: {e}")
+#         return f"Wikipedia search failed for {query} due to {str(e)}."
 
 
 model = build_gemini_llm()