Update app.py

app.py

@@ -83,6 +83,23 @@ def get_box_score_data(links: list) -> dict:
     except Exception as e:
         return {"Error": f"Error fetching boxScore data: {str(e)}"}
 
+@tool
+def get_stats_from_boxScore_data(box_scores: dict, stat: str) -> dict:
+    """A tool that fetches the player names and box score statistic for the provided stat and from the provided box score data
+    Args:
+        box_scores: A dictionary representing the box score data for all of last nights games
+        stat: A string representing the statistic that this function will return for each player in box_scores
+    """
+    try:
+        stats = {}
+        stat_key = ('Basic Box Score Stats', stat)
+        for teams in box_scores.keys():
+            for player in box_scores[teams]:
+                stats[player[list(player.keys())[0]]]=player[stat_key]
+        return stats
+    except Exception as e:
+        return {"Error": f"Error fetching boxScore data for given statistic: {str(e)}"}
+
 final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
 visit_webpage_tool = VisitWebpageTool()
@@ -115,7 +132,8 @@ agent = CodeAgent(
         visit_webpage_tool, 
         user_input_tool, 
         get_box_score_links, 
-        get_box_score_data
+        get_box_score_data,
+        get_stats_from_boxScore_data
     ],  # Add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,