A股多Agent投研框架 — 基于 TradingAgents 深度特化
4.6K
简体中文 | English
基于 TauricResearch/TradingAgents(65K ⭐)的 A 股深度特化 fork
全 Apache 2.0 开源 · pip install 即跑 · 零外部服务依赖
⚠️ 本项目是 TradingAgents 论文框架的工程实现与研究复现,面向研究与教学。
不构成任何投资建议,也不提供任何投资服务。
为什么做这个 Fork · 与上游对比 · 架构概览 · Analyst 角色 · 数据源 · 快速开始 · Web UI · 排错
作者目前正在寻找深圳的 AI 相关工作机会,重点关注腾讯等大型科技企业及金融机构的 AI 投研产品、FDE 与 AI 咨询 / 解决方案岗位。
兼具金融机构从业经历与 AI 产品实战,持续构建金融市场数据工具和多智能体系统,开源项目累计获得 17K+ GitHub Stars。
原版 TradingAgents 是一个出色的多 Agent 投研框架,但它针对美股设计:数据走 Yahoo Finance / Alpha Vantage,分析师不懂 A 股制度,辩论和决策完全面向美股市场。
本 Fork 的目标:把 TradingAgents 的多 Agent 辩论架构真正落地到 A 股,不是简单翻译,而是从数据层、Agent 角色、交易规则三个维度做深度特化。
| 维度 | 原版 | 本 Fork |
|---|---|---|
| 数据源 | Yahoo Finance / Alpha Vantage | mootdx + 东财 + 新浪 + 同花顺(全免费直连) |
| Analyst 角色 | 4 个(市场/情绪/新闻/基本面) | 7 个(+政策分析师/游资追踪/解禁监控) |
| 交易规则 | 美股(T+0、无涨跌停) | A 股(T+1、涨跌停、最小手数、交易时段) |
| 输出语言 | 英文 | 中文报告(内部辩论保持英文以保证推理质量) |
| Alpha 基准 | SPY | 沪深 300(CSI 300) |
| 特性 | 原版 TradingAgents | 本 Fork |
|---|---|---|
| 许可证 | Apache 2.0 | 全 Apache 2.0 |
| 部署依赖 | pip install | 开箱即用 |
| A 股数据 | ❌ | mootdx + 东财 + 新浪 + 同花顺(直连 HTTP) |
| A 股特化角色 | ❌ | 政策/游资/解禁 3 个深度角色 |
| A 股交易约束 | ❌ | T+1/涨跌停/手数/ST 全覆盖 |
┌─────────────────────────────────────────────────────────┐
│ 7 Analyst 研报生成 │
│ Market → Social → News → Fundamentals │
│ → Policy → Hot Money → Lockup │
│ (每个 Analyst 带工具循环) │
├─────────────────────────────────────────────────────────┤
│ Bull vs Bear 投研辩论 │
│ Bull Researcher ←→ Bear Researcher │
│ (最多 N 轮辩论) │
├─────────────────────────────────────────────────────────┤
│ Research Manager 综合研判 │
│ (深度思考 LLM,输出投资计划) │
├─────────────────────────────────────────────────────────┤
│ Trader 交易方案 │
│ (A 股约束:T+1/涨跌停/手数) │
├─────────────────────────────────────────────────────────┤
│ Aggressive ←→ Conservative ←→ Neutral │
│ 三方风险辩论 │
├─────────────────────────────────────────────────────────┤
│ Portfolio Manager 最终决策 │
│ (深度思考 LLM,输出评级 + 理由) │
└─────────────────────────────────────────────────────────┘
双 LLM 设计:
quick_think_llm:所有 Analyst、Researcher、Trader、Risk Debaterdeep_think_llm:Research Manager 和 Portfolio Manager(需要综合全局信息做决策)| 角色 | 职责 | 数据工具 |
|---|---|---|
| 🏪 市场分析师 | K 线形态、技术指标、量价分析 | get_stock_data, get_indicators |
| 💬 舆情分析师 | 社交媒体情绪、散户讨论热度 | get_news |
| 📰 新闻分析师 | 行业新闻、公告、宏观事件 | get_news, get_global_news, get_insider_transactions |
| 📊 基本面分析师 | 财报三表、盈利能力、估值 | get_fundamentals, get_balance_sheet, get_cashflow, get_income_statement |
| 角色 | 职责 | 数据工具 | 为什么需要 |
|---|---|---|---|
| 🏛️ 政策分析师 | 监管政策、产业政策、窗口指导 | get_news, get_global_news | A 股是政策市,政策变化直接影响板块轮动 |
| 🔥 游资追踪师 | 龙虎榜、大单流向、主力资金动态 | get_stock_data, get_news, get_insider_transactions | 游资是 A 股短线定价的核心力量 |
| 🔓 解禁监控师 | 限售股解禁、大股东减持、股权质押 | get_insider_transactions, get_news, get_fundamentals | 解禁是 A 股特有的重大供给冲击因素 |
所有 7 个 Analyst 的报告会流入后续的 Bull/Bear 辩论和三方风险辩论,确保 A 股特色因素贯穿整条决策链。
全部免费,无需 API Key,无积分墙:
| 来源 | 协议 | 提供内容 |
|---|---|---|
| mootdx | TCP 7709 | OHLCV K 线、财务快照、F10 文本 |
| 腾讯财经 | HTTP (qt.gtimg.cn) | PE / PB / 市值 / 换手率(实时) |
| 东方财富 | HTTP (datacenter / push2) | 龙虎榜、限售解禁、板块行情、个股信息 |
| 新浪财经 | HTTP | K 线历史、财报三表 |
| 同花顺 | HTTP (10jqka) | EPS 一致预期 |
| 财联社 | HTTP (cls.cn) | 全球财经快讯 |
| 百度股市通 | HTTP (finance.pae.baidu) | 概念板块分类、资金流向 |
完全不依赖 Tushare(积分墙)、Alpha Vantage(海外 API)、Yahoo Finance(不支持 A 股)。
数据源优先级 & 东财防封(v0.2.11):行情 / K线 / 市值 / 财务能从 mootdx(通达信 TCP,不封 IP)或腾讯拿到的,一律走它们;东财只用于它独有的数据(龙虎榜 / 解禁 / 资金流 / 板块 / 个股新闻等)。所有东财请求统一走内置节流入口
_em_get():串行限流(默认间隔 ≥1s + 0.1~0.5s 随机抖动)+ 复用 Keep-Alive 会话,多 Agent 跑批量分析不再触发临时封 IP(东财风控实测:每秒 >5 / 并发 ≥10 / 1 分钟 ≥200 触发封禁)。批量场景可设环境变量EM_MIN_INTERVAL=1.5~2进一步降速。仅东财限流,mootdx / 腾讯 / 新浪 / 同花顺 / 财联社 / 百度 不受影响。
# Python >= 3.10
git clone https://github.com/simonlin1212/tradingagents-astock.git
cd tradingagents-astock
pip install -e .
# 如需使用 Google Gemini 模型(无 [google] extra,需显式装,见下方 FAQ):
pip install --no-deps "langchain-google-genai>=4.0.0"
pip install "google-genai>=1.53.0" "httpx>=0.28.1"
# 如需让节点走你个人 Claude Pro/Max 订阅额度而非 API 计费(可选):
pip install -e ".[agentsdk]"
装完即可用,无需 Docker。 安装后直接跑
streamlit run web/app.py(Web UI)或tradingagents(CLI)即可,详见下方「Web UI」「CLI 方式」两节。Docker 仅是可选的部署方式,本地开发不需要。
默认走 API Key 计费。每次分析需 30-50 次 LLM 调用。
例外(v0.4.0 新增):装
[agentsdk]后可让部分或全部节点经 Claude Agent SDK 走你个人 Claude Pro/Max 订阅额度,不产生 API 账单。见下方「用个人 Claude 订阅额度」。
在项目根目录创建 .env 文件,按你选择的供应商配置:
# ── 方案 A:MiniMax(推荐,国内直连,性价比高)──────────
MINIMAX_API_KEY=sk-xxx
# 申请地址:https://platform.minimaxi.com/
# ── 方案 B:DeepSeek ─────────────────────────────────
DEEPSEEK_API_KEY=sk-xxx
# 申请地址:https://platform.deepseek.com/
# ── 方案 C:智谱 GLM ─────────────────────────────────
ZHIPU_API_KEY=xxx
# 申请地址:https://open.bigmodel.cn/
# ── 方案 D:通义千问 Qwen ────────────────────────────
DASHSCOPE_API_KEY=sk-xxx
# 申请地址:https://dashscope.console.aliyun.com/
# ── 方案 E:OpenAI ───────────────────────────────────
OPENAI_API_KEY=sk-xxx
# ── 方案 F:Anthropic ────────────────────────────────
ANTHROPIC_API_KEY=sk-ant-xxx
# ── 方案 G:Kimi(Anthropic 兼容 API)────────────────
ANTHROPIC_API_KEY=your-kimi-token
ANTHROPIC_BASE_URL=https://api.kimi.com/coding/
# ⚠️ 两个都要设。只给 key 不给端点,请求会发到 Anthropic 官方并报
# 「401 invalid x-api-key」。端点也可以写在 config 的 backend_url 里(见下)。
# ⚠️ 别用 ANTHROPIC_AUTH_TOKEN——那是 Claude Code CLI 的写法,本项目走 langchain,
# 只认 ANTHROPIC_API_KEY。
# ── 方案 H:任意 OpenAI 兼容网关(9Router / AI Router / 自建代理)──
OPENAI_COMPATIBLE_API_KEY=sk-xxx # 也接受 OPENAI_API_KEY
BACKEND_URL=https://your-relay.example/v1 # 你的网关地址(也可在 Web 侧栏「API Base URL」填)
新建一个 Python 文件(比如项目根目录下的 run.py),把下面这段粘进去,按你选的供应商改 config 后运行 uv run python run.py。根目录自带的 main.py 就是这个示例的可运行版本,直接 uv run python main.py 也行。
config不是仓库里的某个配置文件,而是传给TradingAgentsGraph(config=...)的一个字典:只写你要覆盖的项,其余项自动取tradingagents/default_config.py里的默认值(完整可选项见下文「配置说明」一节)。
from tradingagents.graph.trading_graph import TradingAgentsGraph
# ── MiniMax 示例(推荐)─────────────────────────────
config = {
"llm_provider": "minimax",
"deep_think_llm": "MiniMax-M2.7",
"quick_think_llm": "MiniMax-M2.7-highspeed",
"output_language": "Chinese",
}
# ── DeepSeek 示例 ───────────────────────────────────
# config = {
# "llm_provider": "deepseek",
# "deep_think_llm": "deepseek-chat",
# "quick_think_llm": "deepseek-chat",
# "output_language": "Chinese",
# }
# ── Anthropic + Kimi 示例 ───────────────────────────
# config = {
# "llm_provider": "anthropic",
# "deep_think_llm": "claude-sonnet-4-6",
# "quick_think_llm": "claude-sonnet-4-6",
# "backend_url": "https://api.kimi.com/coding/",
# "output_language": "Chinese",
# }
ta = TradingAgentsGraph(debug=True, config=config)
final_state, decision = ta.propagate("688017", "2026-05-12")
print(decision)
tradingagents # 交互式 CLI
tradingagents analyze # 同上(默认命令)
tradingagents performance # 决策绩效统计(见下)
tradingagents --help # 查看所有选项
想知道这套流程过往的判断准不准,跑:
tradingagents performance # 人读的报告
tradingagents performance --json # 机器读的 JSON
数据来自记忆日志:每次分析会落一条决策,下次分析同一只股票时自动拉真实行情回填收益与 alpha(对沪深 300)。统计本身零 LLM 调用,只读已经落盘的结果。
输出的核心指标是 direction_accuracy(方向正确率)——只有它衡量判断准不准:看多要跑赢、看空要跑输才算对,Hold 不表态不计入。另外给出 up_rate(标的上涨占比)与 outperform_rate(跑赢沪深300占比),这两个只描述标的怎么走,与判断对错无关:给出卖出评级后股价下跌是判断正确,但它不会计入「上涨占比」。
还有按评级、按标的分组,以及一项评级区分度检验——五档评级从 Buy 到 Sell,平均 alpha 是否真的单调递减。评级不单调,说明这套流程的评级没有实际区分能力。
几点务必注意:
内置 Streamlit 可视化界面,支持在侧边栏选择 LLM 供应商和模型,输入股票代码即可一键分析,适合不写代码的用户。
# 方式一:命令行启动(推荐)
tradingagents-web
# 方式二:直接运行
streamlit run web/app.py
打开浏览器访问 http://localhost:8501。
~/.tradingagents/llm_config.json,重开标签页或重启后自动恢复(v0.5.17)
所有配置通过 config 字典传入,完整选项:
| 参数 | 默认值 | 说明 |
|---|---|---|
llm_provider | "minimax" | LLM 提供商:minimax / deepseek / qwen / glm / openai / anthropic / google / xai / ollama |
deep_think_llm | "MiniMax-M2.7" | Research Manager + Portfolio Manager 用的模型 |
quick_think_llm | "MiniMax-M2.7-highspeed" | 所有 Analyst / Researcher / Trader 用的模型 |
backend_url | None | 自定义 API 端点 / 第三方中转网关。可在 Web UI 侧边栏填写,或用 .env 的 BACKEND_URL;方便国内通过代理访问 Claude / OpenAI |
role_llms | {} | 可选:给单个角色指定另一家模型(如多空辩手用不同厂商),留空 = 全部沿用 quick/deep 两档,行为不变。见下方「分角色模型」 #39 |
max_tokens | None | 单次回复的最大输出 token 数。None = 用 provider 默认值。报告写到一半就断,先调这里(不是上下文超长);也可用环境变量 TRADINGAGENTS_MAX_TOKENS。#91 |
output_language | "Chinese" | 报告输出语言(内部辩论始终英文) |
market_lookback_days | None | 技术分析回溯天数(分析区间 = 起始日期 → 分析日期)。Web/CLI 由「数据起始日期」自动算出;None = 模型自选(约 30 天)。#16 |
max_debate_rounds | 1 | Bull vs Bear 辩论轮数 |
max_risk_discuss_rounds | 1 | 风险三方辩论轮数 |
data_vendors | 全部 "a_stock" | 数据供应商路由 |
checkpoint_enabled | False | 启用 SQLite 断点续跑 |
memory_log_max_entries | None | 交易记忆最大条目数 |
默认所有角色共用 quick_think_llm / deep_think_llm 两档——大多数人只有一家模型,不需要碰这一项。
如果你手上有多家模型,可以给单个角色单独指定。最典型的用法是让多空辩手用不同厂商的模型:同一个模型分饰多角时倾向于互相附和,换成不同底座才会真的出现反驳。
config = {
"llm_provider": "deepseek", # 未单独配置的角色仍走这里
"deep_think_llm": "deepseek-chat",
"quick_think_llm": "deepseek-chat",
"role_llms": {
"bull": {"provider": "qwen", "model": "qwen-plus"},
"bear": {"provider": "glm", "model": "glm-4.6"},
# provider 省略则沿用 llm_provider,只换模型:
"portfolio_manager": {"model": "deepseek-reasoner"},
},
}
合法角色名(其余角色自动沿用两档默认):
| 分组 | 角色名 |
|---|---|
| 7 个分析师 | market social news fundamentals policy hot_money lockup |
| 辩论与决策 | bull bear research_manager trader |
| 风险三方 | risk_aggressive risk_neutral risk_conservative |
| 其他 | quality_gate portfolio_manager |
几点说明:
DEEPSEEK_API_KEY / DASHSCOPE_API_KEY / ZHIPU_API_KEY …),缺哪个会指名报出来。backend_url 带过去(那是给主 provider 配的端点),需要的话在该角色里单独写 backend_url。claude_agent_sdk 订阅覆盖时,role_llms 里配的角色会绕开订阅按 token 计费,启动时会点名警告是哪几个。Q: 用 DeepSeek/通义/智谱,却报 OpenAIError: The api_key client option must be set ... OPENAI_API_KEY?
每个供应商用各自的环境变量,不是 OPENAI_API_KEY:DeepSeek=DEEPSEEK_API_KEY、通义=DASHSCOPE_API_KEY、智谱=ZHIPU_API_KEY、MiniMax=MINIMAX_API_KEY、xAI=XAI_API_KEY、OpenRouter=OPENROUTER_API_KEY、OpenAI 兼容(自定义)=OPENAI_COMPATIBLE_API_KEY。在项目根目录 .env 里设置对应变量后重启程序。(v0.2.12 起缺 key 会直接提示该用哪个变量名。)
Q: 想接一个 OpenAI 兼容的第三方网关/中继(9Router、AI Router、自建代理),自定义 base_url + model?
用 「OpenAI 兼容(自定义 base_url)」 这一档(v0.2.20 新增)。Web 侧栏「LLM 供应商」选它 →「快速/深度思考模型 ID」手动填你网关支持的 model 名 →「API Base URL」填你的网关地址(如 https://your-relay.example/v1)→ .env 里设 OPENAI_COMPATIBLE_API_KEY=你的key(也接受 OPENAI_API_KEY)。CLI 方式选 OpenAI-Compatible 后会提示输入 Base URL。它走标准 Chat Completions(非 OpenAI Responses API,兼容性最好),model 名自由填、不受内置清单限制。配置方式等价:llm_provider="openai_compatible" + backend_url="<你的网关>" + deep_think_llm/quick_think_llm="<你的model>"。
Q: 明明装了 Python 3.12/3.14,pip install -e . 却报 requires a different Python: 3.9.6 not in '>=3.10'?
报错里的 3.9.6 就是当前这个 pip 绑定的解释器版本——你装的新版本没被它用上(macOS 自带的 pip3 常指向系统 3.9)。先确认是哪个解释器在跑:
pip3 -V # 末尾括号里就是它绑定的 Python
python3.12 -m pip -V # 换成你想用的版本再看
用 python -m pip 的写法就不会认错人,推荐配合虚拟环境:
python3.12 -m venv .venv && source .venv/bin/activate
python -m pip install -e .
Windows 用 py -3.12 -m venv .venv + .venv\Scripts\activate。(#92)
Q: 报告写到一半就结束了,上下文明明没超长?
撞的是输出上限,不是上下文上限——模型一次回复能吐多少 token 是另一个限制。v0.4.1 起,这种截断会在日志里明确告诉你(因为达到输出上限被截断),不再是默默给你半篇报告。调大即可:config 里设 max_tokens(例如 "max_tokens": 16000),或设环境变量 TRADINGAGENTS_MAX_TOKENS=16000。
另外,用 Kimi 等第三方模型名走 anthropic 通道时,langchain 认不出模型名,会默认一个很小的输出上限(旧版本表现为报告普遍偏短)。v0.4.1 起这种情况会自动放宽到 8192,仍不够就显式配 max_tokens。#91
Q: 接 Kimi 报 401 invalid x-api-key?
说明请求发到了 Anthropic 官方而不是 Kimi——光给 key 没给端点。两个都要给:
ANTHROPIC_API_KEY=你的kimi-token
ANTHROPIC_BASE_URL=https://api.kimi.com/coding/ # 或在 config 里写 backend_url
注意 ANTHROPIC_AUTH_TOKEN 在本项目里不生效(那是 Claude Code CLI 的写法),本项目走 langchain,只读 ANTHROPIC_API_KEY。v0.4.1 起,用非 Claude 模型名却没配端点会在启动时直接告诉你缺什么,而不是等 Anthropic 回一句看不懂的 401。#89
Q: 导出 PDF 报 UnicodeEncodeError: 'latin-1' codec can't encode?
你的环境里装了旧版 fpdf(pyfpdf),它和本项目
Content type
Image
Digest
sha256:1d93f23ff…
Size
337.5 MB
Last updated
23 days ago
docker pull birdxs/tradingagents-astock