agazo commited on
Commit
f4a881b
·
verified ·
1 Parent(s): ad0f6d8

Update langchain_agent.py

Browse files
Files changed (1) hide show
  1. langchain_agent.py +2 -2
langchain_agent.py CHANGED
@@ -20,6 +20,7 @@ from audio_to_text_tool import audio_to_text,audio_to_text_from_youtube
20
  from alphabetizer_tool import alphabetizer
21
  from nb_tool import get_team_players_by_season
22
  from nb_tool import get_npb_player_info
 
23
 
24
  class LangChainAgent:
25
  def __init__(self):
@@ -41,8 +42,7 @@ class LangChainAgent:
41
  audio_to_text,
42
  audio_to_text_from_youtube,
43
  alphabetizer,
44
- get_team_players_by_season,
45
- get_npb_player_info
46
  ]
47
 
48
  prompt = ChatPromptTemplate.from_messages([
 
20
  from alphabetizer_tool import alphabetizer
21
  from nb_tool import get_team_players_by_season
22
  from nb_tool import get_npb_player_info
23
+ from npb_tool import npb
24
 
25
  class LangChainAgent:
26
  def __init__(self):
 
42
  audio_to_text,
43
  audio_to_text_from_youtube,
44
  alphabetizer,
45
+ npb
 
46
  ]
47
 
48
  prompt = ChatPromptTemplate.from_messages([