agazo commited on
Commit
0e96680
·
verified ·
1 Parent(s): 7d55abc

Update npb_agent.py

Browse files
Files changed (1) hide show
  1. npb_agent.py +2 -5
npb_agent.py CHANGED
@@ -7,7 +7,6 @@ from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
7
  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_roster
12
  from nb_tool import get_npb_player_info
13
 
@@ -17,8 +16,7 @@ PROMPT = """
17
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
18
 
19
  If you need to identify players that come before or after other players in NPB, you can first get the player number and team.
20
- Then you get the players of the team for that season. Then, for each of those players, you get their player number. With that you
21
- are all set to answer the question. Consider for now that all players returned by the tool get_team_players_by_season are pitchers.
22
 
23
  """
24
 
@@ -32,8 +30,7 @@ class NpbAgent:
32
 
33
  tools = [
34
  get_npb_player_info,
35
- get_team_roster,
36
- internet_search
37
  ]
38
 
39
  prompt = ChatPromptTemplate.from_messages([
 
7
  from langchain_core.messages import SystemMessage
8
 
9
  # --- Custom Tools ---
 
10
  from nb_tool import get_team_roster
11
  from nb_tool import get_npb_player_info
12
 
 
16
  finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
17
 
18
  If you need to identify players that come before or after other players in NPB, you can first get the player number and team.
19
+ Then you get the players of the team for that season. Then you just find what you need out of that list.
 
20
 
21
  """
22
 
 
30
 
31
  tools = [
32
  get_npb_player_info,
33
+ get_team_roster
 
34
  ]
35
 
36
  prompt = ChatPromptTemplate.from_messages([