AI Hedge Fund:投资大师们一起为你赚钱

814 阅读4分钟

推荐一个基于大模型的交易工具。

珍惜大模型早鸟的短暂红利。一年后大家一起被claude 4吃掉。

github.com/virattt/ai-…

这是一个AI驱动的对冲基金的概念验证。该项目的目标是探索使用AI进行交易决策。此项目仅用于教育目的,不适用于实际交易或投资。

该系统由多个协同工作的代理组成:

  1. Ben Graham Agent - 价值投资之父,只购买具有安全边际的隐藏宝石
  2. Bill Ackman Agent - 激进投资者,采取大胆立场并推动变革
  3. Cathie Wood Agent - 成长投资女王,相信创新和颠覆的力量
  4. Charlie Munger Agent - 沃伦·巴菲特的合作伙伴,只以公平价格购买优秀企业
  5. Stanley Druckenmiller Agent - 宏观传奇,寻找具有增长潜力的不对称机会
  6. Warren Buffett Agent - 奥马哈先知,以公平价格寻找优秀公司
  7. Valuation Agent - 计算股票的内在价值并生成交易信号
  8. Sentiment Agent - 分析市场情绪并生成交易信号
  9. Fundamentals Agent - 分析基本面数据并生成交易信号
  10. Technicals Agent - 分析技术指标并生成交易信号
  11. Risk Manager - 计算风险指标并设置头寸限制
  12. Portfolio Manager - 做出最终交易决策并生成订单

Screenshot 2025-03-08 at 4 45 22 PM转存失败,建议直接上传图片文件

注意:该系统模拟交易决策,并不实际进行交易。

Twitter Follow

免责声明

此项目仅用于教育和研究目的

  • 不适用于实际交易或投资
  • 不提供任何保证或担保
  • 过去的表现并不预示未来的结果
  • 创作者不对财务损失承担任何责任
  • 投资决策请咨询财务顾问

使用此软件即表示您同意仅将其用于学习目的。

目录

设置

克隆仓库:

git clone https://github.com/virattt/ai-hedge-fund.git
cd ai-hedge-fund
  1. 安装Poetry(如果尚未安装):
curl -sSL https://install.python-poetry.org | python3 -
  1. 安装依赖:
poetry install
  1. 设置环境变量:
# 创建.env文件以存储API密钥
cp .env.example .env
  1. 设置API密钥:
# 对于运行由openai托管的LLM(gpt-4o, gpt-4o-mini等)
# 从https://platform.openai.com/获取您的OpenAI API密钥
OPENAI_API_KEY=your-openai-api-key

# 对于运行由groq托管的LLM(deepseek, llama3等)
# 从https://groq.com/获取您的Groq API密钥
GROQ_API_KEY=your-groq-api-key

# 获取为对冲基金提供动力的财务数据
# 从https://financialdatasets.ai/获取您的Financial Datasets API密钥
FINANCIAL_DATASETS_API_KEY=your-financial-datasets-api-key

重要:您必须设置OPENAI_API_KEYGROQ_API_KEYANTHROPIC_API_KEYDEEPSEEK_API_KEY才能使对冲基金工作。如果您想使用所有提供商的LLM,则需要设置所有API密钥。

AAPL、GOOGL、MSFT、NVDA和TSLA的财务数据是免费的,不需要API密钥。

对于任何其他股票代码,您需要在.env文件中设置FINANCIAL_DATASETS_API_KEY

使用

运行对冲基金

poetry run python src/main.py --ticker AAPL,MSFT,NVDA

示例输出: Screenshot 2025-01-06 at 5 50 17 PM

您还可以指定--show-reasoning标志,将每个代理的推理打印到控制台。

poetry run python src/main.py --ticker AAPL,MSFT,NVDA --show-reasoning

您可以选择指定开始和结束日期,以在特定时间段内做出决策。

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

示例输出: Screenshot 2025-01-06 at 5 47 52 PM转存失败,建议直接上传图片文件

您可以选择指定开始和结束日期,以在特定时间段内进行回测。

poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01

项目结构

ai-hedge-fund/
├── src/
│   ├── agents/                   # 代理定义和工作流程
│   │   ├── bill_ackman.py        # Bill Ackman代理
│   │   ├── fundamentals.py       # 基本面分析代理
│   │   ├── portfolio_manager.py  # 投资组合管理代理
│   │   ├── risk_manager.py       # 风险管理代理
│   │   ├── sentiment.py          # 情绪分析代理
│   │   ├── technicals.py         # 技术分析代理
│   │   ├── valuation.py          # 估值分析代理
│   │   ├── warren_buffett.py     # 沃伦·巴菲特代理
│   ├── tools/                    # 代理工具
│   │   ├── api.py                # API工具
│   ├── backtester.py             # 回测工具
│   ├── main.py # 主入口点
├── pyproject.toml
├── ...