virattt / ai-hedge-fund

virattt/ai-hedge-fund

open_in_new前往仓库

一个AI对冲基金团队

项目概览

项目概述

AI Hedge Fund 是一个基于人工智能的对冲基金概念验证项目,旨在探索利用 AI 进行交易决策。该项目仅用于教育和研究目的,不用于真实交易或投资。项目正在演进为一个持久化、始终在线的 AI 对冲基金,支持回测、模拟交易和(可选)实盘运行,并将投资者代理重新构想为可插拔、可回测的“alpha 模型”。

核心功能

  • 交互式终端应用:构建基金(选择股票、策略、再平衡周期),回测已保存的基金并绘制权益曲线。
  • 非交互式运行:从授权文件(mandate file)运行一个基金周期,输出完整周期记录的 JSON 和人类可读摘要。
  • 回测功能:按再平衡周期对历史数据进行回测。
  • 支持多种 LLM 提供商:Anthropic、OpenAI、DeepSeek、Google、xAI、Kimi。
  • 自动保存 API 密钥到 ~/.hedge-fund/.env,无需预先配置。

适用与不适用场景

适用场景:

  • 教育用途:学习 AI 在金融决策中的应用。
  • 研究用途:探索 AI 驱动的交易策略。
  • 模拟交易和回测:在无风险环境中测试策略。

不适用场景:

  • 真实交易或投资:项目明确声明不用于真实交易,且系统实际上不会执行任何交易。
  • 投资建议:不提供任何投资建议或保证。
  • 生产环境:作为概念验证,不适合直接用于生产环境。

技术架构与依赖

  • 语言:Python
  • 包管理:使用 Poetry 进行开发,可通过 pipx、uv 或 pip 安装。
  • 外部依赖:
    • Financial Datasets API(用于价格、基本面、收益数据)
    • 至少一个 LLM API(Anthropic、OpenAI、DeepSeek、Google、xAI、Kimi)
  • 架构:基金作为一级实体,支持回测、模拟交易和可选实盘;投资者代理作为可插拔的 alpha 模型。

安装与快速开始

安装

pipx install aihf

或使用 uv tool install aihf,或 pip install aihf 到任意环境。

快速开始

运行交互式应用:

aihf

首次需要 API 密钥时,应用会提示输入并保存到 ~/.hedge-fund/.env。需要 Financial Datasets API 密钥和一个 LLM API 密钥。

典型使用方法

交互式应用

aihf

构建基金,选择股票、策略、再平衡周期,或回测已保存的基金。基金保存为 ~/.hedge-fund/mandates/ 下的授权文件。

非交互式运行

从授权文件运行一个基金周期:

aihf ~/.hedge-fund/mandates/example.yaml --tickers AAPL,MSFT

回测授权文件:

aihf ~/.hedge-fund/mandates/example.yaml --tickers AAPL,MSFT --backtest

授权文件定义策略、人员、风险、资本和节奏,不指定股票;--tickers 指定本次运行的股票。

配置与部署要点

  • API 密钥:首次使用时提示输入,保存到 ~/.hedge-fund/.env。环境变量中导出的密钥优先于保存的文件。
  • 授权文件:基金配置保存在 ~/.hedge-fund/mandates/ 下,可重复使用。
  • 开发环境:克隆仓库后使用 Poetry 安装依赖,运行测试:
git clone https://github.com/virattt/ai-hedge-fund.git
cd ai-hedge-fund
poetry install
poetry run aihf
poetry run pytest hedge_fund

限制、风险与许可证

  • 限制:项目为概念验证,不执行真实交易;仅用于教育和研究。
  • 风险:不提供投资建议或保证;创建者不对财务损失承担责任;过去的表现不代表未来结果。
  • 许可证:MIT License(详见 LICENSE 文件)。

官方链接

信息来源和分析时间