175 lines
5.3 KiB
Markdown
175 lines
5.3 KiB
Markdown
# TradingAgents 中文版
|
||
|
||
> 基于 [TauricResearch/TradingAgents](https://github.com/TauricResearch/TradingAgents) 定制，新增：
|
||
> - 🇨🇳 全部 Agent 输出改为简体中文
|
||
> - 💬 支持注入自定义分析视角
|
||
> - 🔄 Gemini API 自动重试（SSL 断连容错）
|
||
> - 📝 中文分析入口脚本 `run_analysis.py`
|
||
> - 🖥️ TUI 界面新增用户视角输入步骤
|
||
|
||
**⚠️ 声明：本项目仅供研究和学习使用，不构成任何投资建议。**
|
||
|
||
---
|
||
|
||
## Agent 架构
|
||
|
||
```
|
||
📊 分析师团队
|
||
├── 市场分析师 → 技术指标（MACD/RSI/均线等）
|
||
├── 情绪分析师 → 社交媒体舆情
|
||
├── 新闻分析师 → 宏观新闻 & 行业动态
|
||
└── 基本面分析师 → 财报、估值、公司健康度
|
||
|
||
🔬 研究员团队
|
||
├── 多头研究员 → 寻找买入理由
|
||
└── 空头研究员 → 寻找做空理由（结构化辩论）
|
||
|
||
💼 决策层
|
||
├── 研究主管 → 综合多空辩论，得出投资建议
|
||
├── 交易员 → 制定具体交易计划
|
||
├── 激进风控 → 支持高风险高收益策略
|
||
├── 保守风控 → 强调风险控制
|
||
├── 中性风控 → 中立评估
|
||
└── 风控主管 → 最终审批，输出 BUY / SELL / HOLD
|
||
```
|
||
|
||
---
|
||
|
||
## 快速开始
|
||
|
||
### 1. 克隆并安装
|
||
|
||
```bash
|
||
git clone https://github.com/dw1161/TradingAgents.git
|
||
cd TradingAgents
|
||
|
||
# 创建虚拟环境（推荐 Python 3.11-3.13）
|
||
python3 -m venv .venv
|
||
source .venv/bin/activate # Windows: .venv\Scripts\activate
|
||
|
||
# 安装依赖
|
||
pip install -r requirements.txt
|
||
```
|
||
|
||
### 2. 配置 API Key
|
||
|
||
创建 `.env` 文件（已被 gitignore，不会上传）：
|
||
|
||
```bash
|
||
# 三选一，用哪个填哪个
|
||
GOOGLE_API_KEY=你的Gemini_Key
|
||
ANTHROPIC_API_KEY=你的Claude_Key
|
||
OPENAI_API_KEY=你的OpenAI_Key
|
||
```
|
||
|
||
### 3. 运行分析
|
||
|
||
**TUI 交互界面（推荐，有实时进度展示）：**
|
||
```bash
|
||
python -m cli.main
|
||
```
|
||
|
||
**命令行脚本：**
|
||
```bash
|
||
# 基础用法
|
||
python run_analysis.py NVDA 2026-03-20
|
||
|
||
# 注入自定义分析视角（核心功能）
|
||
python run_analysis.py NVDA 2026-03-20 "中东地缘冲突升级是当前美股主要风险，请重点评估"
|
||
|
||
# 交互式输入视角
|
||
python run_analysis.py NVDA
|
||
```
|
||
|
||
结果保存至 `results/NVDA_日期.txt`。
|
||
|
||
---
|
||
|
||
## 自定义分析视角（新功能）
|
||
|
||
支持在分析开始前注入你的判断框架，**所有 Agent 都会优先考虑此视角**：
|
||
|
||
```bash
|
||
# 地缘政治视角
|
||
python run_analysis.py NVDA 2026-03-20 "中东伊朗局势升级正在推高油价并压制科技股估值"
|
||
|
||
# 宏观政策视角
|
||
python run_analysis.py NVDA 2026-03-20 "美联储降息预期升温，流动性改善利好成长股"
|
||
|
||
# 产业趋势视角
|
||
python run_analysis.py NVDA 2026-03-20 "AI算力需求超预期，数据中心扩张带来订单增量"
|
||
```
|
||
|
||
TUI 界面（`cli.main`）在 Step 8 也支持输入视角。
|
||
|
||
---
|
||
|
||
## 配置说明
|
||
|
||
修改 `run_analysis.py` 顶部调整行为：
|
||
|
||
```python
|
||
config["llm_provider"] = "google" # google / anthropic / openai
|
||
config["deep_think_llm"] = "gemini-2.5-flash" # 深度推理（研究员辩论/风控）
|
||
config["quick_think_llm"] = "gemini-2.5-flash" # 快速任务（情绪/新闻分析）
|
||
config["max_debate_rounds"] = 1 # 辩论轮数（越多越慢越贵）
|
||
```
|
||
|
||
### 支持的模型提供商
|
||
|
||
| 提供商 | `llm_provider` | 环境变量 | 推荐模型 |
|
||
|--------|---------------|---------|---------|
|
||
| Google Gemini | `google` | `GOOGLE_API_KEY` | `gemini-2.5-flash` |
|
||
| Anthropic Claude | `anthropic` | `ANTHROPIC_API_KEY` | `claude-sonnet-4-6` |
|
||
| OpenAI | `openai` | `OPENAI_API_KEY` | `gpt-5.4` |
|
||
|
||
---
|
||
|
||
## 日期参数说明
|
||
|
||
传入日期为**截止日期**，Agent 用该日期及之前的数据分析：
|
||
- 推荐传**昨天或前天**（今天数据不完整）
|
||
- 可传历史日期做回测验证
|
||
|
||
---
|
||
|
||
## 支持的标的
|
||
|
||
| 标的 | 支持情况 |
|
||
|------|---------|
|
||
| 美股个股（NVDA/AAPL 等） | ✅ 完整支持 |
|
||
| 美股 ETF（VOO/SPY 等） | ✅ 支持（基本面分析意义较小） |
|
||
| 国内 A 股 / 商品 | ⚠️ 需自定义数据适配层 |
|
||
|
||
---
|
||
|
||
## 已知问题
|
||
|
||
1. **Gemini 偶发断连**：已在 LLM 客户端层加入指数退避自动重试（最多 5 次）
|
||
2. **Python 3.14 Pydantic 警告**：Warning 非 Error，不影响运行，忽略即可
|
||
3. **回测前视偏差**：Agent 联网获取信息可能导致回测数据失真，历史结果仅供参考
|
||
|
||
---
|
||
|
||
## 本版改动说明
|
||
|
||
| 文件 | 改动 |
|
||
|------|------|
|
||
| `tradingagents/agents/analysts/*.py` | 所有分析师 prompt 加中文输出指令 |
|
||
| `tradingagents/agents/researchers/*.py` | 多空研究员 prompt 加中文指令 |
|
||
| `tradingagents/agents/managers/*.py` | 研究主管/风控主管加中文指令 |
|
||
| `tradingagents/agents/trader/trader.py` | 交易员加中文指令 |
|
||
| `tradingagents/agents/risk_mgmt/*.py` | 风控辩手加中文指令 |
|
||
| `tradingagents/llm_clients/google_client.py` | 加入 SSL 断连自动重试 |
|
||
| `tradingagents/graph/propagation.py` | 支持 `user_context` 参数 |
|
||
| `tradingagents/graph/trading_graph.py` | `propagate()` 支持用户视角注入 |
|
||
| `cli/main.py` | TUI 新增 Step 8 用户视角输入 |
|
||
| `run_analysis.py` | 新增中文分析入口脚本 |
|
||
|
||
---
|
||
|
||
## 参考
|
||
|
||
- 原项目：https://github.com/TauricResearch/TradingAgents
|
||
- 论文：https://arxiv.org/abs/2412.20138
|