综合介绍
AI Hedge Fund 是一个利用多代理系统进行交易决策的人工智能对冲基金。该系统通过多个专门的代理协同工作,包括市场数据代理、量化代理、风险管理代理和投资组合管理代理,以实现复杂的交易决策和自动化交易。
这个 Python 开源项目,通过四个专业 AI 智能体 (市场数据、量化分析、风险管理、投资组合管理) 协同工作,自动完成从数据收集、分析到交易决策的全过程,让普通开发者也能构建自己的算法交易系统。
功能列表
- 多代理系统架构:多个专门代理协同工作,进行复杂的交易决策。
- 技术分析:使用MACD、RSI、布林带和OBV等技术指标进行分析。
- 风险管理:提供头寸大小建议,评估投资组合风险。
- 投资组合管理:自动化交易决策,生成订单。
- 回测功能:提供性能分析,支持多股票代码。
使用帮助
安装流程
- 克隆仓库:
git clone https://github.com/virattt/ai-hedge-fund.git cd ai-hedge-fund
- 安装Poetry(如果尚未安装):
curl -sSL https://install.python-poetry.org | python3 -
- 安装依赖:
poetry install
- 设置环境变量:
cp .env.example .env export OPENAI_API_KEY='your-api-key-here' export FINANCIAL_DATASETS_API_KEY='your-api-key-here'
使用说明
运行对冲基金
使用以下命令运行对冲基金:
poetry run python src/agents.py --ticker AAPL --start-date 2024-01-01 --end-date 2024-03-01
示例输出:
{
"action": "buy",
"quantity": 50000
}
运行回测工具
使用以下命令运行回测工具:
poetry run python src/backtester.py --ticker AAPL --start-date 2024-01-01 --end-date 2024-03-01
示例输出:
Starting backtest...
Date Ticker Action Quantity Price Cash Stock Total Value
----------------------------------------------------------------------
2024-01-01 AAPL buy 519.0 192.53 76.93 519.0 100000.00
2024-01-02 AAPL hold 0 185.64 76.93 519.0 96424.09
2024-01-03 AAPL hold 0 184.25 76.93 519.0 95702.68
2024-01-04 AAPL hold 0 181.91 76.93 519.0 94488.22
2024-01-05 AAPL hold 0 181.18 76.93 519.0 94109.35
2024-01-08 AAPL sell 519 185.56 96382.57 0.0 96382.57
2024-01-09 AAPL buy 520.0 185.14 109.77 520.0 96382.57
项目结构
src/agents.py
:主要代理定义和工作流程src/backtester.py
:回测功能src/tools.py
:技术分析工具pyproject.toml
:Poetry 配置.env.example
:环境变量示例README.md
:文档