Update tools.py

tools.py

@@ -1,6 +1,7 @@
 from langchain_core.tools import tool
 import pandas as pd
 import os
+import wikipedia
 
 
 @tool
@@ -47,4 +48,22 @@ def divide(a: int, b: int) -> int:
     """
     if b == 0:
         raise ValueError("Cannot divide by zero.")
-    return a / b
+    return a / b
+
+@tool
+def wikipedia_search_tool(query: str) -> str:
+    """
+        Search wikipedia for articles
+
+        Args:
+            query: The query to search in wikipedia
+
+        Example:
+            wikipedia_search_tool("How many albums did the Beatles produce between 1965 and 1968?")
+    """
+    try:
+        page_title = wikipedia.search(query)[0]
+        summary = wikipedia.summary(page_title, sentences=5)
+        return f"Page Title: {page_title}\n\n{summary}"
+    except Exception as e:
+        return f"Error: {str(e)}"