agazo commited on
Commit
6b58720
·
verified ·
1 Parent(s): 1edc9d5

Update npb_agent.py

Browse files
Files changed (1) hide show
  1. npb_agent.py +2 -1
npb_agent.py CHANGED
@@ -8,9 +8,10 @@ from langchain_core.messages import SystemMessage
8
 
9
  # --- Custom Tools ---
10
  from internet_search_tool import internet_search
11
- from nb_tool import get_team_players_by_season
12
  from nb_tool import get_npb_player_info
13
 
 
14
  PROMPT = """
15
  You are an agent specialized in answering questions related to Nippon Professional Baseball players. Report your thoughts, and
16
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
 
8
 
9
  # --- Custom Tools ---
10
  from internet_search_tool import internet_search
11
+ from nb_tool import get_team_roster
12
  from nb_tool import get_npb_player_info
13
 
14
+
15
  PROMPT = """
16
  You are an agent specialized in answering questions related to Nippon Professional Baseball players. Report your thoughts, and
17
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].