Update tools.py

tools.py

@@ -1,6 +1,7 @@
 from langchain_core.tools import tool
 import pandas as pd
 import os
+import re
 import wikipedia
 
 
@@ -51,20 +52,38 @@ def divide(a: int, b: int) -> int:
     return a / b
 
 @tool
-def wikipedia_search_tool(query: str) -> str:
+def wikipedia_search_tool(query: str, year_range: str = None) -> str:
     """
-        Search wikipedia for articles
+        Deep Wikipedia search with optional year filtering.
+        - Retrieves full text from multiple related pages.
+        - Filters for year mentions if year_range is specified (e.g. '1965-1968').
 
         Args:
             query: The query to search in wikipedia
+            year_range: The range of years to search information for. Only use if necessary
 
         Example:
             {"message": "How many albums did the Beatles produce between 1965 and 1968?"}
-            wikipedia_search_tool("The Beatles")
+            wikipedia_search_tool("The Beatles albums", "1965-1968")
     """
     try:
-        page_title = wikipedia.search(query)[0]
-        summary = wikipedia.summary(page_title, sentences=5)
-        return f"Title: {page.title}\n\n{page.content}"
+        results = wikipedia.search(query, results=5)
+        if not results:
+            return "No relevant pages found."
+
+        combined = ""
+        for title in results:
+            try:
+                page = wikipedia.page(title)
+                content = clean_and_truncate(page.content)
+                if year_range:
+                    y1, y2 = map(int, year_range.split('-'))
+                    content = extract_year_range(content, y1, y2)
+                if content:
+                    combined += f"\n\n## {title}\n{content}"
+            except:
+                continue
+
+        return combined.strip() if combined else "No content matched the year filter."
     except Exception as e:
         return f"Error: {str(e)}"