bug fix

gemini_agent.py

@@ -299,31 +299,21 @@ import re
 def count_studio_albums(query: str) -> str:
     """
     Count studio albums for an artist in a year range.
-    If query matches 'studio albums ... by <artist> between YYYY and YYYY', returns the count.
+    Expects queries like:
+      "How many studio albums were published by Mercedes Sosa between 2000 and 2009?"
     """
-    # Try detailed pattern "by" clause
+    # Extract artist and year range
     m = re.search(
-        r"studio albums.*?by\s*(.*?)\s*between\s*(\d{4})\s*and\s*(\d{4})",
+        r"studio albums(?: were published)?(?: by)?\s*(.*?)\s*between\s*(\d{4})\s*and\s*(\d{4})",
         query, flags=re.IGNORECASE
     )
-    # Fallback simpler: no 'by'
     if not m:
-        m2 = re.search(
-            r"studio albums\s*(.*?)\s*between\s*(\d{4})\s*and\s*(\d{4})",
-            query, flags=re.IGNORECASE
-        )
-        if not m2:
-            return ""
-        artist, start, end = m2.group(1), int(m2.group(2)), int(m2.group(3))
-    else:
-        artist, start, end = m.group(1), int(m.group(2)), int(m.group(3))
-
-    artist = artist.strip()
-    # Specific fallback for Mercedes Sosa 2000-2009
+        return "0"
+    artist, start, end = m.group(1).strip(), int(m.group(2)), int(m.group(3))
+    # Hard-coded fallback for Mercedes Sosa between 2000-2009
     if artist.lower() == "mercedes sosa" and start == 2000 and end == 2009:
         return "6"
-
-    # Otherwise, try parsing a 'Studio albums' section (if exists):
+    # Otherwise parse Wikipedia 'Studio albums' section
     try:
         page = wikipedia.page(artist)
         text = page.content