33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
from langchain_core.tools import tool
|
|
from typing import Annotated
|
|
from datetime import datetime, timedelta
|
|
from tradingagents.dataflows.interface import route_to_vendor
|
|
|
|
|
|
@tool
|
|
def get_stock_data(
|
|
symbol: Annotated[str, "公司的股票代碼"],
|
|
start_date: Annotated[str, "開始日期，格式為 yyyy-mm-dd"],
|
|
end_date: Annotated[str, "結束日期，格式為 yyyy-mm-dd"],
|
|
) -> str:
|
|
"""
|
|
檢索給定股票代碼的股價數據 (OHLCV)。
|
|
使用設定的核心股票 API 供應商。
|
|
Args:
|
|
symbol (str): 公司的股票代碼，例如 AAPL, TSM
|
|
start_date (str): 開始日期，格式為 yyyy-mm-dd
|
|
end_date (str): 結束日期，格式為 yyyy-mm-dd
|
|
Returns:
|
|
str: 一個格式化的數據框，包含指定股票代碼在指定日期範圍內的股價數據。
|
|
"""
|
|
# 強制至少 365 天的資料範圍，確保分析品質一致
|
|
try:
|
|
start_dt = datetime.strptime(start_date, "%Y-%m-%d")
|
|
end_dt = datetime.strptime(end_date, "%Y-%m-%d")
|
|
if (end_dt - start_dt).days < 365:
|
|
start_dt = end_dt - timedelta(days=365)
|
|
start_date = start_dt.strftime("%Y-%m-%d")
|
|
print(f"[get_stock_data] 日期範圍不足 1 年，已自動調整 start_date 為 {start_date}")
|
|
except ValueError:
|
|
pass # 日期格式錯誤時使用原始值
|
|
return route_to_vendor("get_stock_data", symbol, start_date, end_date) |