Update tools.py

tools.py

@@ -37,6 +37,41 @@ def handle_question(question: str) -> str:
         return ", ".join(filter_vegetables.run(question.split(", ")))
     return web_lookup.run(question)
 
+@tool
+def count_albums_by_year_range(
+    artist_name: str, start_year: int, end_year: int
+) -> Optional[int]:
+    """
+    Counts how many albums (typically studio albums) an artist released between two years by parsing their Wikipedia discography page.
+
+    Args:
+        artist_name (str): Name of the artist (e.g., "Mercedes Sosa").
+        start_year (int): Start of the year range (inclusive).
+        end_year (int): End of the year range (inclusive).
+
+    Returns:
+        Optional[int]: Number of albums released in the year range, or None if parsing failed.
+    """
+    print("🛠️ count_albums_by_year_range")
+    try:
+        title = artist_name.strip().replace(" ", "_") + "_discography"
+        url = f"https://en.wikipedia.org/wiki/{title}"
+        res = requests.get(url, timeout=10)
+        if res.status_code != 200:
+            return None
+
+        soup = BeautifulSoup(res.text, "html.parser")
+        text = soup.get_text()
+
+        # Match years in parentheses like (2003), (2005)
+        years = re.findall(r"\((19|20)\d{2}\)", text)
+        year_nums = [int(y) for y in years]
+        count = sum(start_year <= y <= end_year for y in year_nums)
+        return count
+
+    except Exception as e:
+        return None
+
 
 @tool
 def web_lookup(query: str) -> str:
@@ -52,6 +87,7 @@ def web_lookup(query: str) -> str:
     Returns:
         str: A concise factual answer extracted from Tavily or Wikipedia.
     """
+    print("🛠️ web_lookup")
     try:
         # Step 1: Tavily search
         response = client.search(query=query, search_depth="advanced", max_results=5)
@@ -109,6 +145,7 @@ def extract_number_from_snippets(snippets: list[str]) -> Optional[int]:
         >>> extract_number_from_snippets(["The Eiffel Tower was built in 1889.", "Height is 324m"])
         1889
     """
+    print("🛠️ extract_number_from_snippets")
     for s in snippets:
         match = re.search(r"\b\d{1,4}\b", s)
         if match:
@@ -121,6 +158,7 @@ def get_article_nominator_from_fac_page(title: str) -> str:
     """
     Get the nominator of a Featured Article by scanning the main FAC page (not just archives).
     """
+    print("🛠️ get_article_nominator_from_fac_page")
     base = "https://en.wikipedia.org/wiki/Wikipedia:Featured_article_candidates"
     url = f"{base}/{title}"
     res = requests.get(url)
@@ -155,6 +193,7 @@ def extract_structured_facts_from_url(url: str, selector: Optional[str] = None)
     Returns:
         str: Cleaned structured data from the page.
     """
+    print("🛠️ extract_structured_facts_from_url")
     try:
         response = requests.get(url, timeout=10)
         response.raise_for_status()
@@ -357,6 +396,7 @@ def filter_vegetables(items: list[str]) -> list[str]:
 
 # List of all tools
 all_tools = [
+    count_albums_by_year_range,
     web_lookup,
     extract_number_from_snippets,
     detect_non_commutative_subset,