Update tools.py

tools.py

@@ -1,6 +1,5 @@
 from langchain_core.tools import tool
-import requests
-from datetime import datetime
+import wikipediaapi
 
 
 @tool
@@ -50,56 +49,23 @@ def divide(a: int, b: int) -> int:
     return a / b
 
 @tool
-def get_studio_album_count(artist_name: str, start_year: int, end_year: int) -> dict:
+def search_wikipedia(page_title: str, language: str) -> str:
     """
-    Search music APIs for relevant information related to music artists.
-    It returns a dictionary with the number of albums produced during that interval and the names.
-    
+    This tool allows searching wikipedia for information on a specific topic. It will return the resulting information in HTML format
+
     Args:
-        artist_name: Name of the artist to search
-        start_year: The start year for searching information
-        end_year: The end year for searching information
+        page_title: The page title to search for
+        language: The language for searching the content. For example, "en" for english, "es" for spanish,  "fr" for french, etc.
 
-    Exampe:
-        {"message": "How many albums did The Beatles produce between 1965 and 1968 ?"}
-        get_studio_album_count("The Beatles", 1965, 1968)
+    Example:
+        {"message": "How many albums did Juanes produce between 2000 and 2005 ?"}
+        search_wikipedia("Juanes", "es")
     """
-    # Step 1: Get artist MBID
-    search_url = f"https://musicbrainz.org/ws/2/artist/?query=artist:{artist_name}&fmt=json"
-    artist_resp = requests.get(search_url).json()
-    if not artist_resp["artists"]:
-        return {"error": f"Artist '{artist_name}' not found."}
-    
-    artist_mbid = artist_resp["artists"][0]["id"]
-
-    # Step 2: Get release-groups (albums)
-    album_url = f"https://musicbrainz.org/ws/2/release-group?artist={artist_mbid}&type=album&fmt=json&limit=100"
-    albums_resp = requests.get(album_url).json()
-    
-    studio_albums = []
-    for release_group in albums_resp.get("release-groups", []):
-        title = release_group.get("title", "")
-        first_date = release_group.get("first-release-date", "")
-        
-        # Filter by date
-        try:
-            year = int(first_date[:4])
-        except:
-            continue
-        if not (start_year <= year <= end_year):
-            continue
-        
-        # Filter out secondary types (e.g. Live, Compilation)
-        secondary_types = release_group.get("secondary-types", [])
-        if secondary_types:
-            continue  # Not a studio album
 
-        studio_albums.append({
-            "title": title,
-            "release_year": year,
-        })
+    try:
+        wiki_wiki = wikipediaapi.Wikipedia('AIAgent (gabriel_abilleira@tutanota.com)', language, extract_format=wikipediaapi.ExtractFormat.HTML)
+        page = wiki_wiki.page(page_title)
     
-    return {
-        "studio_album_count": len(studio_albums),
-        "studio_album_list": sorted(studio_albums, key=lambda x: x["release_year"]),
-    }
+        return f"Title: {page.title}\n\n{page.text}"
+    except Exception as e:
+        return f"Error: {str(e)}"