agazo commited on
Commit
0e64283
·
verified ·
1 Parent(s): 2a0c42f

Update langchain_agent.py

Browse files
Files changed (1) hide show
  1. langchain_agent.py +4 -2
langchain_agent.py CHANGED
@@ -18,7 +18,8 @@ from convert_chessboard_image_to_fen_tool import convert_chessboard_image_to_fen
18
  from chess_image_to_fen_tool import chess_image_to_fen
19
  from audio_to_text_tool import audio_to_text,audio_to_text_from_youtube
20
  from alphabetizer_tool import alphabetizer
21
- from npb_tool import npb_players,list_npb_players
 
22
 
23
  class LangChainAgent:
24
  def __init__(self):
@@ -40,7 +41,8 @@ class LangChainAgent:
40
  audio_to_text,
41
  audio_to_text_from_youtube,
42
  alphabetizer,
43
- list_npb_players
 
44
  ]
45
 
46
  prompt = ChatPromptTemplate.from_messages([
 
18
  from chess_image_to_fen_tool import chess_image_to_fen
19
  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
  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([