Update tools.py

tools.py

@@ -17,13 +17,55 @@ import re
 import time
 from typing import Optional, List, Dict, Any
 import re
-
-import wikipedia
+from datetime import datetime, timedelta
 from langchain_core.tools import tool
 
-import wikipedia
-from bs4 import BeautifulSoup
-from langchain_core.tools import tool
+@tool
+def search_featured_articles_by_date_range(start_date: str, end_date: str) -> list[str]:
+    """
+    Searches the English Wikipedia featured article archive and returns article titles
+    promoted between start_date and end_date.
+
+    Args:
+        start_date (str): Start date in YYYY-MM-DD format (e.g. '2016-11-01')
+        end_date (str): End date in YYYY-MM-DD format (e.g. '2016-11-30')
+
+    Returns:
+        list[str]: A list of article titles promoted as Featured Articles during that period.
+    """
+    try:
+        base_url = "https://en.wikipedia.org/wiki/Wikipedia:Featured_articles"
+        archive_url = "https://en.wikipedia.org/wiki/Wikipedia:Featured_articles_by_year"
+
+        start = datetime.strptime(start_date, "%Y-%m-%d")
+        end = datetime.strptime(end_date, "%Y-%m-%d")
+
+        # We'll collect year-specific pages
+        result_titles = []
+
+        for year in range(start.year, end.year + 1):
+            url = f"https://en.wikipedia.org/wiki/Wikipedia:Featured_articles_{year}"
+            response = requests.get(url)
+            if response.status_code != 200:
+                continue
+
+            soup = BeautifulSoup(response.text, "html.parser")
+            for li in soup.select("li"):
+                text = li.get_text()
+                date_matches = re.findall(r"\b(19|20)\d{2}-\d{2}-\d{2}\b", text)
+                for match in date_matches:
+                    try:
+                        d = datetime.strptime(match, "%Y-%m-%d")
+                        if start <= d <= end:
+                            a_tag = li.find("a")
+                            if a_tag:
+                                result_titles.append(a_tag.get_text(strip=True))
+                    except:
+                        continue
+
+        return sorted(set(result_titles))
+    except Exception as e:
+        return [f"Error: {str(e)}"]
 
 @tool
 def detect_non_commutative_subset(table_text: str) -> str:
@@ -157,6 +199,7 @@ def filter_vegetables(items: list[str]) -> list[str]:
 
 # List of all tools
 all_tools = [
+    search_featured_articles_by_date_range,
     detect_non_commutative_subset,
     reverse_sentence,
     filter_vegetables,