fix

langchain_agent.py

@@ -18,7 +18,7 @@ from convert_chessboard_image_to_fen_tool import convert_chessboard_image_to_fen
 from chess_image_to_fen_tool import chess_image_to_fen
 from audio_to_text_tool import audio_to_text,audio_to_text_from_youtube
 from alphabetizer_tool import alphabetizer
-from npb_tool import npb_players
+from npb_tool import npb_players,list_npb_players
 
 class LangChainAgent:
     def __init__(self):
@@ -40,7 +40,7 @@ class LangChainAgent:
             audio_to_text,
             audio_to_text_from_youtube,
             alphabetizer,
-            npb_players
+            list_npb_players
         ]
 
         prompt = ChatPromptTemplate.from_messages([

npb_tool.py

@@ -5,6 +5,7 @@ import os
 from constants import TAVILY_KEY
 from langchain.tools import tool
 from npb_agent import NpbAgent
+from typing import Optional
 
 # Assuming your NpbAgent class is defined as shown
 npb_agent = NpbAgent()
@@ -16,3 +17,28 @@ def npb_players(question: str) -> str:
     such as their team, jersey number, stats, or other biographical info.
     """
     return npb_agent(question)
+
+
+@tool
+def list_npb_players(player: str, season: str) -> str:
+    """
+    Retrieves a list of Nippon Professional Baseball (NPB) players for a specific team and season.
+    Each entry includes:
+    - Full name
+    - Jersey number
+    - Position/role (e.g., pitcher, catcher)
+    - Team name
+
+    Parameters:
+    - team (str): The name of the NPB team to query (e.g., "Hanshin Tigers").
+    - season (str): The target season/year (e.g., "2024").
+
+    Returns:
+    - str: A formatted list of player information, or an error message if no data is found.
+    """
+    
+    prompt = f""" 
+        I need a list containing the players for the same team of play {player} and season {season} in the NPB. Each entry must contain full name, jersey number, role and team name.  
+    """
+    
+    return npb_agent(prompt)