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()
@@ -235,7 +235,8 @@ tools = [
     divide_numbers,
     extract_text_from_image,
     download_and_read_file,
-    web_search
+    search_web,
+    wikipedia_search
 ]
 
 model_with_tools = model.bind_tools(tools)