推荐一个基于大模型的交易工具。
珍惜大模型早鸟的短暂红利。一年后大家一起被claude 4吃掉。
这是一个AI驱动的对冲基金的概念验证。该项目的目标是探索使用AI进行交易决策。此项目仅用于教育目的,不适用于实际交易或投资。
该系统由多个协同工作的代理组成:
- Ben Graham Agent - 价值投资之父,只购买具有安全边际的隐藏宝石
- Bill Ackman Agent - 激进投资者,采取大胆立场并推动变革
- Cathie Wood Agent - 成长投资女王,相信创新和颠覆的力量
- Charlie Munger Agent - 沃伦·巴菲特的合作伙伴,只以公平价格购买优秀企业
- Stanley Druckenmiller Agent - 宏观传奇,寻找具有增长潜力的不对称机会
- Warren Buffett Agent - 奥马哈先知,以公平价格寻找优秀公司
- Valuation Agent - 计算股票的内在价值并生成交易信号
- Sentiment Agent - 分析市场情绪并生成交易信号
- Fundamentals Agent - 分析基本面数据并生成交易信号
- Technicals Agent - 分析技术指标并生成交易信号
- Risk Manager - 计算风险指标并设置头寸限制
- Portfolio Manager - 做出最终交易决策并生成订单
注意:该系统模拟交易决策,并不实际进行交易。
免责声明
此项目仅用于教育和研究目的。
- 不适用于实际交易或投资
- 不提供任何保证或担保
- 过去的表现并不预示未来的结果
- 创作者不对财务损失承担任何责任
- 投资决策请咨询财务顾问
使用此软件即表示您同意仅将其用于学习目的。
目录
设置
克隆仓库:
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
- 设置环境变量:
# 创建.env文件以存储API密钥
cp .env.example .env
- 设置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_KEY、GROQ_API_KEY、ANTHROPIC_API_KEY或DEEPSEEK_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
示例输出:
您还可以指定--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
示例输出:
您可以选择指定开始和结束日期,以在特定时间段内进行回测。
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
├── ...