Spaces:
Sleeping
Sleeping
Update npb_agent.py
Browse files- 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
|
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([
|