AGAZO_Final_Assignment / npb_tool.py
agazo's picture
Update npb_tool.py
ad0f6d8 verified
raw
history blame contribute delete
556 Bytes
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)