Update nb_tool.py

nb_tool.py

@@ -5,18 +5,13 @@ from typing import Optional
 import re
 import sys
 from constants import TAVILY_KEY
-
-
-TAVILY_API_KEY = "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"
 import os
-import requests
-import re
-from bs4 import BeautifulSoup
 from tavily import TavilyClient
 
 # Initialize Tavily client (set your API key as an environment variable)
-tavily = TavilyClient(TAVILY_API_KEY)
+tavily = TavilyClient(TAVILY_KEY)
 
+@tool
 def get_npb_player_info(player_name: str) -> str:
     """
     Finds a Nippon Professional Baseball (NPB) player's jersey number and team name using Tavily search.
@@ -61,8 +56,7 @@ def get_npb_player_info(player_name: str) -> str:
 #if __name__ == "__main__":
 #    print(get_npb_player_info("Taisho Tamai"))
 
-
-
+@tool
 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.