Updated get_top_ranked_team to refine the returned dictionary

app.py

@@ -21,7 +21,8 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
 
 @tool
 def get_top_ranked_team(division:str, gender:str)-> dict:
-    """A tool that returns a dictionary of information (e.g. name, ID, events won etc.) about the top ranked ream for a given division and gender.
+    """A tool that returns a dictionary of information about the top ranked ream 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"].
@@ -35,9 +36,10 @@ def get_top_ranked_team(division:str, gender:str)-> dict:
     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"] if k in school_info}
         return school_info
     else:
-        return "Error: Unable to fetch the school."
+        return "Error: Unable to fetch the school information."
 
 @tool
 def get_team_results(school_id: str) -> dict: