ai-hedge-fund
多智能体对冲基金模拟器,将巴菲特、芒格等 14 位投资大师的策略封装成独立 agent,协作分析股票并生成交易信号。亮点在于用 LLM 模拟不同投资风格,并支持回测与 Web 界面,适合学习 AI 在金融决策中的应用。注意:教育目的,非真实交易建议。
README
AI 对冲基金
这是一个 AI 驱动的对冲基金的概念验证。本项目的目标是探索使用 AI 做出交易决策。该项目仅供教育用途,不应用于真实交易或投资。
🚧 项目仍在演进中。 我们正在将其重建为一个持久运行、始终在线的 AI 对冲基金——一个作为一级实体的 基金,你可以进行回测、模拟交易,并(可选地)上线运行,而投资者代理则被重新构想为可插拔、可回测的“alpha 模型”。阅读 愿景 → 和 路线图 →。
该系统由多个代理协同工作:
- Aswath Damodaran 代理 —— 估值学术权威,专注于故事、数字和纪律性估值
- Ben Graham 代理 —— 价值投资之父,只买入具有安全边际的隐藏瑰宝
- Bill Ackman 代理 —— 激进投资者,敢于持仓并推动变革
- Cathie Wood 代理 —— 成长投资女王,坚信创新与颠覆的力量
- Charlie Munger 代理 —— Warren Buffett 的搭档,只以合理价格买入优质企业
- Michael Burry 代理 —— 《大空头》中的逆向投资者,搜寻深度价值
- Mohnish Pabrai 代理 —— Dhandho 投资者,寻找低风险翻倍的机会
- Nassim Taleb 代理 —— 黑天鹅风险分析师,专注于尾部风险、反脆弱性和非对称回报
- Peter Lynch 代理 —— 务实的投资者,在日常业务中寻找“十倍数股”
- Phil Fisher 代理 —— 细致入微的成长投资者,使用深入的“流言蜚语”调研
- Rakesh Jhunjhunwala 代理 —— 印度大牛市
- Stanley Druckenmiller 代理 —— 宏观传奇人物,寻找具有成长潜力的非对称机会
- Warren Buffett 代理 —— 奥马哈先知,以合理价格寻找优质公司
- 估值代理 —— 计算股票的内在价值并生成交易信号
- 情绪代理 —— 分析市场情绪并生成交易信号
- 基本面代理 —— 分析基本面数据并生成交易信号
- 技术面代理 —— 分析技术指标并生成交易信号
- 风险经理 —— 计算风险指标并设置仓位限制
- 投资组合经理 —— 做出最终交易决策并生成订单

注意:该系统实际上不执行任何交易。
免责声明
本项目仅用于教育和研究目的。
- 不用于真实交易或投资
- 不提供投资建议或保证
- 创建者对财务损失不承担任何责任
- 请咨询财务顾问以做出投资决策
- 过往业绩不代表未来结果
通过使用本软件,您同意仅将其用于学习目的。
目录
如何安装
在运行 AI 对冲基金之前,你需要先进行安装并设置 API 密钥。以下步骤对全栈 Web 应用程序和命令行界面均适用。
1. 克隆仓库
git clone https://github.com/virattt/ai-hedge-fund.git
cd ai-hedge-fund
2. 设置 API 密钥
创建一个 .env 文件用于存放 API 密钥:
# 为你的 API 密钥创建 .env 文件(在根目录下)
cp .env.example .env
打开并编辑 .env 文件,添加你的 API 密钥:
# 用于运行由 openai 托管的 LLM(gpt-4o、gpt-4o-mini 等)
OPENAI_API_KEY=your-openai-api-key
# 用于获取金融数据以驱动对冲基金
FINANCIAL_DATASETS_API_KEY=your-financial-datasets-api-key
重要:你必须至少设置一个 LLM API 密钥(例如 OPENAI_API_KEY、GROQ_API_KEY、ANTHROPIC_API_KEY 或 DEEPSEEK_API_KEY),对冲基金才能工作。
如何运行
⌨️ 命令行界面
你可以直接通过终端运行 AI 对冲基金。该方法提供更精细的控制,适用于自动化、脚本编写和集成场景。

快速开始
- 安装 Poetry(如果尚未安装):
curl -sSL https://install.python-poetry.org | python3 -
- 安装依赖:
poetry install
运行 AI 对冲基金
poetry run python src/main.py --ticker AAPL,MSFT,NVDA
你也可以指定 --ollama 标志,使用本地 LLM 运行 AI 对冲基金。
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --ollama
你还可以选择指定起始和结束日期,使其在特定时间段内做出决策。
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01
运行回测工具
poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA
示例输出:

注意:--ollama、--start-date 和 --end-date 标志同样适用于回测工具!
🖥️ Web 应用程序
运行 AI 对冲基金的新方式是通过我们的 Web 应用程序,它提供了用户友好的界面。对于偏好可视化界面而非命令行工具的用户,推荐使用此方式。
请参阅此处关于如何安装和运行 Web 应用程序的详细说明。

如何贡献
- Fork 本仓库
- 创建功能分支
- 提交你的更改
- 推送到分支
- 创建一个 Pull Request
重要:请保持你的 pull request 小而专注。这将使审查和合并更容易。
功能请求
如果你有功能请求,请提交一个 issue,并确保标记为 enhancement。
许可证
本项目采用 MIT 许可证 - 详情请参阅 LICENSE 文件。