Update tools.py

tools.py

@@ -32,8 +32,30 @@ WIKIPEDIA_PAGES = {
     # ➕ add more GAIA topics here
 }
 
+@tool
+def handle_question(question: str) -> str:
+    """
+    Dynamically handle a question by routing to appropriate tools and combining results.
+    """
+    strategy = route_question.run(question)
+    
+    if strategy == "extract_structured_facts_from_url":
+        wiki_url = resolve_wikipedia_url.run(question)
+        if not wiki_url:
+            return "❌ Could not find Wikipedia page."
+        return extract_structured_facts_from_url.run(wiki_url)
+
+    if strategy == "search_featured_articles_by_date_range":
+        return search_featured_articles_by_date_range.run("2016-11-01", "2016-11-30")
+
+    return "🤔 I will use internal reasoning."
+
+
 @tool
 def resolve_wikipedia_url(question: str) -> Optional[str]:
+    """
+    Returns a known Wikipedia URL if the question contains a known entity.
+    """
     q = question.lower()
     for key, url in WIKIPEDIA_PAGES.items():
         if key in q:
@@ -50,12 +72,12 @@ def route_question(question: str) -> str:
     """
     q = question.lower()
 
-    if any(keyword in q for keyword in ["how many", "list", "albums", "awards", "published", "release"]):
-        return "search_web"
-
-    if any(keyword in q for keyword in ["table", "section", "discography", "infobox", "html"]):
+    if "wikipedia" in q and any(k in q for k in ["how many", "list", "albums", "awards", "release"]):
         return "extract_structured_facts_from_url"
 
+    if "featured article" in q and any(k in q for k in ["promoted", "in", "nominated"]):
+        return "search_featured_articles_by_date_range"
+
     # Default to internal logic (math, logic puzzles, wordplay)
     return "use_internal_logic"
 
@@ -171,53 +193,54 @@ def categorize_grocery_items(items: list[str]) -> dict:
     return result
 
 
-# @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.
-#     """
-#     print(f"🛠️ search_featured_articles_by_date_range called with: {start_date} , {end_date}")
-#     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 = date_matches = re.findall(r"\b(19\d{2}|20\d{2})-\d{2}-\d{2}\b", text)
-#                 print("🔍 Date matches:", date_matches)
+@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.
+    """
+    print(f"🛠️ search_featured_articles_by_date_range called with: {start_date} , {end_date}")
+    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 = date_matches = re.findall(r"\b(19\d{2}|20\d{2})-\d{2}-\d{2}\b", text)
+                print("🔍 Date matches:", date_matches)
                 
-#                 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)}"]
+                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:
     """
@@ -382,14 +405,13 @@ def filter_vegetables(items: list[str]) -> list[str]:
 
 # List of all tools
 all_tools = [
-    resolve_wikipedia_url,
-    route_question,
-    extract_structured_facts_from_url,
-    search_web,
-    # search_featured_articles_by_date_range,
-    categorize_grocery_items,
-    detect_non_commutative_subset,
-    reverse_sentence,
-    filter_vegetables,
-    # extract_wikipedia_section_html
+    handle_question,                          # Master router for high-level logic
+    resolve_wikipedia_url,                   # Used internally by handle_question
+    extract_structured_facts_from_url,       # For Wikipedia fact extraction (e.g., albums)
+    search_featured_articles_by_date_range,  # For Featured Article date lookups (e.g., dinosaur)
+    search_web,                               # Fallback if Wikipedia fails
+    detect_non_commutative_subset,            # For math table puzzles
+    reverse_sentence,                         # For reverse decoding puzzles
+    categorize_grocery_items,                 # For classification questions
+    filter_vegetables                         # For food-type filtering questions
 ]