98 lines
3.7 KiB
Python
98 lines
3.7 KiB
Python
from langchain_core.tools import tool
|
|
from typing import Annotated
|
|
from tradingagents.dataflows.interface import route_to_vendor
|
|
from tradingagents.utils.anonymizer import TickerAnonymizer
|
|
|
|
def _process_vendor_call(func_name, ticker, *args):
|
|
"""Helper to handle anonymization for vendor calls"""
|
|
# Initialize locally to ensure fresh state
|
|
anonymizer = TickerAnonymizer()
|
|
|
|
# 1. Deanonymize ticker
|
|
real_ticker = anonymizer.deanonymize_ticker(ticker)
|
|
if not real_ticker:
|
|
real_ticker = ticker
|
|
|
|
try:
|
|
# 2. Get Data
|
|
raw_data = route_to_vendor(func_name, real_ticker, *args)
|
|
|
|
# 3. Anonymize Output
|
|
return anonymizer.anonymize_text(raw_data, real_ticker)
|
|
except Exception as e:
|
|
# RETURN string error instead of raising.
|
|
# This ensures ToolNode generates a ToolMessage result, preventing "Dangling Tool Use" error.
|
|
return f"Error executing tool {func_name}: {str(e)}"
|
|
|
|
@tool
|
|
def get_fundamentals(
|
|
ticker: Annotated[str, "ticker symbol"],
|
|
curr_date: Annotated[str, "current date you are trading at, yyyy-mm-dd"],
|
|
) -> str:
|
|
"""
|
|
Retrieve comprehensive fundamental data for a given ticker symbol.
|
|
Uses the configured fundamental_data vendor.
|
|
Args:
|
|
ticker (str): Ticker symbol of the company
|
|
curr_date (str): Current date you are trading at, yyyy-mm-dd
|
|
Returns:
|
|
str: A formatted report containing comprehensive fundamental data
|
|
"""
|
|
return _process_vendor_call("get_fundamentals", ticker, curr_date)
|
|
|
|
|
|
@tool
|
|
def get_balance_sheet(
|
|
ticker: Annotated[str, "ticker symbol"],
|
|
freq: Annotated[str, "reporting frequency: annual/quarterly"] = "quarterly",
|
|
curr_date: Annotated[str, "current date you are trading at, yyyy-mm-dd"] = None,
|
|
) -> str:
|
|
"""
|
|
Retrieve balance sheet data for a given ticker symbol.
|
|
Uses the configured fundamental_data vendor.
|
|
Args:
|
|
ticker (str): Ticker symbol of the company
|
|
freq (str): Reporting frequency: annual/quarterly (default quarterly)
|
|
curr_date (str): Current date you are trading at, yyyy-mm-dd
|
|
Returns:
|
|
str: A formatted report containing balance sheet data
|
|
"""
|
|
return _process_vendor_call("get_balance_sheet", ticker, freq, curr_date)
|
|
|
|
|
|
@tool
|
|
def get_cashflow(
|
|
ticker: Annotated[str, "ticker symbol"],
|
|
freq: Annotated[str, "reporting frequency: annual/quarterly"] = "quarterly",
|
|
curr_date: Annotated[str, "current date you are trading at, yyyy-mm-dd"] = None,
|
|
) -> str:
|
|
"""
|
|
Retrieve cash flow statement data for a given ticker symbol.
|
|
Uses the configured fundamental_data vendor.
|
|
Args:
|
|
ticker (str): Ticker symbol of the company
|
|
freq (str): Reporting frequency: annual/quarterly (default quarterly)
|
|
curr_date (str): Current date you are trading at, yyyy-mm-dd
|
|
Returns:
|
|
str: A formatted report containing cash flow statement data
|
|
"""
|
|
return _process_vendor_call("get_cashflow", ticker, freq, curr_date)
|
|
|
|
|
|
@tool
|
|
def get_income_statement(
|
|
ticker: Annotated[str, "ticker symbol"],
|
|
freq: Annotated[str, "reporting frequency: annual/quarterly"] = "quarterly",
|
|
curr_date: Annotated[str, "current date you are trading at, yyyy-mm-dd"] = None,
|
|
) -> str:
|
|
"""
|
|
Retrieve income statement data for a given ticker symbol.
|
|
Uses the configured fundamental_data vendor.
|
|
Args:
|
|
ticker (str): Ticker symbol of the company
|
|
freq (str): Reporting frequency: annual/quarterly (default quarterly)
|
|
curr_date (str): Current date you are trading at, yyyy-mm-dd
|
|
Returns:
|
|
str: A formatted report containing income statement data
|
|
"""
|
|
return _process_vendor_call("get_income_statement", ticker, freq, curr_date) |