mirror of
https://github.com/TauricResearch/TradingAgents.git
synced 2026-09-27 06:56:39 +03:00
Yijia-Xiao
c42a2f2c61
- interface -> router; symbol_utils -> symbols, which also takes safe_ticker_component - utils is split: get_current_date to date_window, the HTTP helpers to net - dataflows imports are absolute; the NoMarketDataError re-export from symbols is gone
86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
from typing import Annotated
|
|
|
|
from langchain_core.tools import tool
|
|
from langgraph.prebuilt import InjectedState
|
|
|
|
from tradingagents.dataflows.date_window import as_of
|
|
from tradingagents.dataflows.router import route_to_vendor
|
|
|
|
|
|
@tool
|
|
def get_fundamentals(
|
|
ticker: Annotated[str, "ticker symbol"],
|
|
curr_date: Annotated[str, "current date you are trading at, yyyy-mm-dd"],
|
|
trade_date: Annotated[str, InjectedState("trade_date")] = "",
|
|
) -> 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 route_to_vendor("get_fundamentals", ticker, as_of(curr_date, trade_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,
|
|
trade_date: Annotated[str, InjectedState("trade_date")] = "",
|
|
) -> 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 route_to_vendor("get_balance_sheet", ticker, freq, as_of(curr_date, trade_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,
|
|
trade_date: Annotated[str, InjectedState("trade_date")] = "",
|
|
) -> 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 route_to_vendor("get_cashflow", ticker, freq, as_of(curr_date, trade_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,
|
|
trade_date: Annotated[str, InjectedState("trade_date")] = "",
|
|
) -> 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 route_to_vendor("get_income_statement", ticker, freq, as_of(curr_date, trade_date))
|