update

app.py

@@ -128,6 +128,66 @@ def audio_to_text(file_path: str) -> str:
     return json.dumps({"transcript": transcript_data})
 
 
+@tool
+def get_wikipedia_page_url_by_year(wikipedia_page_name: str, year: int) -> str:
+    """
+    Retrieve Wikipedia page URL for a specific year (latest revision in that year).
+
+    Args:
+        wikipedia_page_name (str): Name of the Wikipedia page
+        year (int): Year to get the page content from
+
+    Returns:
+        str: URL of the Wikipedia page from that year with revision included
+    """
+    import requests
+    import wikipediaapi
+
+    # Create Wikipedia API instance
+    wiki = wikipediaapi.Wikipedia(
+        user_agent="Final Project Agent Course (vthanhvinh@gmail.com)", language="en"
+    )
+
+    # Get the page
+    page = wiki.page(wikipedia_page_name)
+    if not page.exists():
+        raise ValueError(f"Wikipedia page '{wikipedia_page_name}' does not exist")
+
+    # Use Wikipedia API to get revisions from the specified year
+    api_url = "https://en.wikipedia.org/w/api.php"
+
+    # Get the latest revision from the specified year
+    params = {
+        "action": "query",
+        "format": "json",
+        "prop": "revisions",
+        "titles": wikipedia_page_name,
+        "rvprop": "ids|timestamp",
+        "rvstart": f"{year}-12-31T23:59:59Z",
+        "rvend": f"{year}-01-01T00:00:00Z",
+        "rvdir": "newer",
+        "rvlimit": 1,
+    }
+
+    response = requests.get(api_url, params=params)
+    data = response.json()
+
+    pages = data["query"]["pages"]
+    page_id = list(pages.keys())[0]
+    revisions = pages[page_id].get("revisions", [])
+
+    if not revisions:
+        raise ValueError(
+            f"No revisions found for '{wikipedia_page_name}' in year {year}"
+        )
+
+    # Get revision ID and construct URL
+    rev_id = revisions[0]["revid"]
+    url = f"https://en.wikipedia.org/w/index.php?title={wikipedia_page_name}&oldid={rev_id}"
+
+    return url
+
+
 @tool
 def get_file(question_id: str, file_name: str) -> str:
     """
@@ -195,6 +255,7 @@ class BasicAgent:
                 get_file,
                 audio_to_text,
                 WikipediaSearchTool(),
+                get_wikipedia_page_url_by_year,
             ],
             model=OpenAIServerModel(model_id="gpt-4o"),
             additional_authorized_imports=[
@@ -212,7 +273,8 @@ class BasicAgent:
                 "cv2",
                 "numpy",
                 "whisper",
-                "openpyxl" "json",
+                "openpyxl",
+                "json",
                 "wikipediaapi",
             ],
             name="multimodal_agent",
@@ -229,7 +291,7 @@ class BasicAgent:
             model=InferenceClientModel(
                 model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
             ),
-            tools=[get_file, audio_to_text],
+            tools=[get_file, audio_to_text, get_wikipedia_page_url_by_year],
             managed_agents=[self.multimodal_agent],
             additional_authorized_imports=[
                 "requests",