112 lines
4.6 KiB
Python
112 lines
4.6 KiB
Python
"""
|
||
TradingAgents 分析脚本
|
||
用途：对持仓标的（NVDA / VOO）进行多 Agent 分析
|
||
用法：
|
||
python run_analysis.py NVDA # 分析 NVDA，使用今日日期
|
||
python run_analysis.py VOO # 分析 VOO
|
||
python run_analysis.py NVDA 2026-03-20 # 指定日期
|
||
python run_analysis.py NVDA 2026-03-20 "中东地缘冲突..." # 注入自定义分析视角
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import time
|
||
from datetime import date
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv()
|
||
|
||
from tradingagents.graph.trading_graph import TradingAgentsGraph
|
||
from tradingagents.default_config import DEFAULT_CONFIG
|
||
|
||
# ── 配置 ────────────────────────────────────────────────────────────────────
|
||
config = DEFAULT_CONFIG.copy()
|
||
config["llm_provider"] = "google"
|
||
config["deep_think_llm"] = "gemini-2.5-flash"
|
||
config["quick_think_llm"] = "gemini-2.5-flash"
|
||
config["max_debate_rounds"] = 1
|
||
config["max_risk_discuss_rounds"] = 1
|
||
config["data_vendors"] = {
|
||
"core_stock_apis": "yfinance",
|
||
"technical_indicators": "yfinance",
|
||
"fundamental_data": "yfinance",
|
||
"news_data": "yfinance",
|
||
}
|
||
|
||
# ── 入参解析 ─────────────────────────────────────────────────────────────────
|
||
ticker = sys.argv[1].upper() if len(sys.argv) > 1 else "NVDA"
|
||
analysis_date = sys.argv[2] if len(sys.argv) > 2 else str(date.today())
|
||
user_context = sys.argv[3] if len(sys.argv) > 3 else ""
|
||
|
||
# 如果没有传入上下文，交互式询问
|
||
if not user_context and sys.stdin.isatty():
|
||
print("\n💬 是否有自定义分析视角？（直接回车跳过）")
|
||
print(" 例：当前主导美股的核心因素是中东地缘冲突，请重点考虑此风险")
|
||
user_context = input(" > ").strip()
|
||
|
||
print(f"\n{'='*60}")
|
||
print(f"🤖 TradingAgents 多 Agent 分析")
|
||
print(f" 标的：{ticker}")
|
||
print(f" 日期：{analysis_date}")
|
||
print(f" 模型：Gemini 2.5 Flash")
|
||
if user_context:
|
||
print(f" 用户视角：{user_context[:60]}{'...' if len(user_context) > 60 else ''}")
|
||
print(f"{'='*60}\n")
|
||
|
||
# ── 执行分析（带 retry）────────────────────────────────────────────────────────
|
||
MAX_RETRIES = 3
|
||
for attempt in range(1, MAX_RETRIES + 1):
|
||
try:
|
||
print(f"[尝试 {attempt}/{MAX_RETRIES}]")
|
||
ta = TradingAgentsGraph(
|
||
selected_analysts=["market", "social", "news", "fundamentals"],
|
||
debug=False,
|
||
config=config,
|
||
)
|
||
final_state, decision = ta.propagate(ticker, analysis_date, user_context)
|
||
break
|
||
except Exception as e:
|
||
print(f"⚠️ 第 {attempt} 次失败: {type(e).__name__}: {str(e)[:120]}")
|
||
if attempt < MAX_RETRIES:
|
||
wait = 10 * attempt
|
||
print(f" 等待 {wait}s 后重试...")
|
||
time.sleep(wait)
|
||
else:
|
||
print("❌ 全部重试失败，退出。")
|
||
sys.exit(1)
|
||
|
||
# ── 输出结果 ──────────────────────────────────────────────────────────────────
|
||
print(f"\n{'='*60}")
|
||
print(f"📊 {ticker} 最终决策（{analysis_date}）：{decision}")
|
||
print(f"{'='*60}\n")
|
||
|
||
# 保存完整报告
|
||
output_dir = os.path.join(os.path.dirname(__file__), "results")
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
output_file = os.path.join(output_dir, f"{ticker}_{analysis_date}.txt")
|
||
|
||
with open(output_file, "w", encoding="utf-8") as f:
|
||
f.write(f"标的：{ticker}\n日期：{analysis_date}\n最终决策：{decision}\n")
|
||
if user_context:
|
||
f.write(f"用户视角：{user_context}\n")
|
||
f.write("\n" + "="*60 + "\n")
|
||
|
||
f.write("【交易员决策报告】\n")
|
||
f.write(str(final_state.get("trader_investment_plan", "N/A")))
|
||
f.write("\n\n" + "="*60 + "\n")
|
||
|
||
debate = final_state.get("investment_debate_state", {})
|
||
f.write("【多空辩论结论】\n")
|
||
f.write(str(debate.get("judge_decision", "N/A")))
|
||
f.write("\n\n" + "="*60 + "\n")
|
||
|
||
risk = final_state.get("risk_debate_state", {})
|
||
f.write("【风控审核结论】\n")
|
||
f.write(str(risk.get("judge_decision", "N/A")))
|
||
f.write("\n\n" + "="*60 + "\n")
|
||
|
||
f.write("【最终决策原文】\n")
|
||
f.write(str(final_state.get("final_trade_decision", "N/A")))
|
||
|
||
print(f"✅ 完整报告已保存：{output_file}")
|