Update nb_tool.py

nb_tool.py

@@ -22,6 +22,9 @@ def get_npb_player_info(player_name: str) -> str:
     Returns:
     - str: The player's jersey number and team name, or an error message if not found.
     """
+    return get_npb_player_info_(player_name)
+
+def get_npb_player_info_(player_name: str) -> str:
     try:
         # Search using Tavily for the NPB English profile page
         query = f"site:npb.jp/bis/eng/players {player_name}"
@@ -57,6 +60,43 @@ def get_npb_player_info(player_name: str) -> str:
 #    print(get_npb_player_info("Taisho Tamai"))
 
 @tool
+def get_team_roster(team_name: str, year: int) -> list[dict]:
+    """
+    Returns the team roster for a given NPB team and season.
+    
+    Each player is represented as a dict with:
+    - name: "Last, First"
+    - number: jersey number (str or None if not found)
+    - role (pitcher, etc.)
+
+    Parameters:
+    - team_name (str): e.g., "Yomiuri Giants"
+    - year (int): e.g., 2023
+
+    Returns:
+    - List of dicts representing players.
+    """
+    roster = []
+    player_names = get_team_players_by_season(team_name, year)
+
+    for player_name in player_names:
+        # Get player info
+        info = get_npb_player_info_(player_name)
+        
+        # Try to extract number from the response
+        match = re.search(r"#(\d+)", info)
+        number = match.group(1) if match else None
+
+        roster.append({
+            "name": player_name,
+            "number": number,
+            "role": "Pitcher"  # Static for now
+        })
+
+    return roster
+
+
+
 def get_team_players_by_season(team_name: str, year: int) -> list[str]:
     """
     Given a team name and year, returns the list of player names for that NPB season.