from tavily import TavilyClient
from utils import clean_text
from langchain_core.tools import tool
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()

@tool
def npb(question: str) -> str:
    """
    Use this tool to answer questions about Nippon Professional Baseball (NPB) players,
    such as their team, jersey number, stats, or other biographical info.
    """
    return npb_agent(question)