Update npb_agent.py

npb_agent.py

@@ -8,23 +8,12 @@ from langchain_core.messages import SystemMessage
 
 # --- Custom Tools ---
 from internet_search_tool import internet_search
+from nb_tool import get_team_players_by_season
+from nb_tool import get_npb_player_info
 
 PROMPT = """
-
     You are an agent specialized in answering questions related to Nippon Professional Baseball players. Report your thoughts, and 
     finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
-    
-    When you search the internet, consider the results of this site as the main reference: https://npb.jp.
-    
-    The first thing you have to do is to find out the player number and team.  To that end, search the web using the player name and
-    "ippon Professional Baseball". One of the results will have the prefix https://npb.jp/bis/eng/players. Go to that page and extract 
-    the player number and player team. They are contained in the html markups with id "pc_v_no" (the number) and "pc_v_team" (the team) of
-    the returned page.
-    
-    Then you can search the web by using the team name and the year of the season you're interested. You will get a page like 
-    this one: https://npb.jp/bis/eng/{year}. Go to that page and extract the names of the players. Then search each player for their numbers.
-         
-
 """
 
 class NpbAgent:
@@ -36,7 +25,8 @@ class NpbAgent:
             timeout=20)
 
         tools = [
-            internet_search
+            get_npb_player_info,
+            get_team_players_by_season
         ]
 
         prompt = ChatPromptTemplate.from_messages([