Updated get_top_ranked_team to return multiple teams

app.py

@@ -3,20 +3,21 @@ from bs4 import BeautifulSoup
 import datetime
 import requests
 import pytz
+from typing import List
 import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 @tool
-def get_top_ranked_team(division:str, gender:str)-> dict:
-    """A tool that returns a dictionary of information about the top ranked ream for a given division and gender.
+def get_top_ranked_teams(division:str, gender:str, num_teams:int)-> List[dict]:
+    """A tool that returns a list of dictionaries containing information about the top "num_teams" ranked teams for a given division and gender.
     This dictionary includes the team name, their board name (i.e. the short form of their name) and their school ID.
     Args:
         division: A string representing the division. This can only take values from ["NCAA Division I", "NCAA Division II", "NCAA Division III].
         gender: A string representing the gender. This can only take values from ["Women", "Men"].
     """
-    api_url = f"https://scoreboard.clippd.com/api/rankings/leaderboard?rankingType=Team&gender={gender}&division={division}&sortField=rank&season=2025&limit=1&offset=0"
+    api_url = f"https://scoreboard.clippd.com/api/rankings/leaderboard?rankingType=Team&gender={gender}&division={division}&sortField=rank&season=2025&limit={num_teams}&offset=0"
     headers = {
         'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7)',
         'Content-Type': 'application/json'
@@ -24,8 +25,11 @@ def get_top_ranked_team(division:str, gender:str)-> dict:
     response = requests.get(api_url, headers=headers)
     if response.status_code == 200:
         data = response.json()
-        school_info = data.get("results", [{}])[0]
-        school_info = {k: school_info[k] for k in ["schoolName", "boardName", "schoolId", "eventsWon", "eventsTop3", "strokePlayEvents", "matchPlayEvents"] if k in school_info}
+        schools = data.get("results", [{}])
+        school_info = []
+        for school in schools:
+            school_data = {k: school[k] for k in ["schoolName", "boardName", "schoolId", "eventsWon", "eventsTop3", "strokePlayEvents", "matchPlayEvents"] if k in school}
+            school_info.append(school_data)
         return school_info
     else:
         return "Error: Unable to fetch the school information."
@@ -126,7 +130,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_top_ranked_team, get_team_results], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_top_ranked_teams, get_team_results], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,