这是一个由人工智能驱动的对冲基金概念验证项目。该项目旨在探索利用AI进行交易决策的可能性,仅用于教育目的,不适用于实际交易或投资。
本系统采用多智能体协作模式:
注意:本系统仅模拟交易决策,不进行实际交易。
本项目仅用于教育与研究目的:
使用本软件即表示您同意仅将其用于学习目的。
克隆仓库:
git clone https://github.com/virattt/ai-hedge-fund.git
cd ai-hedge-fund
curl -sSL https://install.python-poetry.org | python3 -
poetry install
# Create .env file for your API keys
cp .env.example .env
# For running LLMs hosted by openai (gpt-4o, gpt-4o-mini, etc.)
# Get your OpenAI API key from https://platform.openai.com/
OPENAI_API_KEY=your-openai-api-key
# For running LLMs hosted by groq (deepseek, llama3, etc.)
# Get your Groq API key from https://groq.com/
GROQ_API_KEY=your-groq-api-key
# For getting financial data to power the hedge fund
# Get your Financial Datasets API key from https://financialdatasets.ai/
FINANCIAL_DATASETS_API_KEY=your-financial-datasets-api-key
确保系统已安装Docker,如未安装请从Docker官网下载
克隆仓库:
git clone https://github.com/virattt/ai-hedge-fund.git
cd ai-hedge-fund
# Create .env file for your API keys
cp .env.example .env
编辑.env文件添加API密钥(方法同上)
进入docker目录:
cd docker
# On Linux/Mac:
./run.sh build
# On Windows:
run.bat build
重要提示:必须设置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
注意:所有Docker命令必须在docker/
目录下执行
# Navigate to the docker directory first
cd docker
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA main
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA main
示例输出:
可通过--ollama
参数使用本地LLM运行AI对冲基金:
# With Poetry:
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --ollama
# With Docker (from docker/ directory):
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA --ollama main
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA --ollama main
通过--show-reasoning
参数可打印各智能体的决策逻辑:
# With Poetry:
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --show-reasoning
# With Docker (from docker/ directory):
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA --show-reasoning main
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA --show-reasoning main
可选指定起止日期分析特定时段:
# With Poetry:
poetry run python src/main.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01
# With Docker (from docker/ directory):
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01 main
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01 main
poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA
注意:所有Docker命令必须在docker/
目录下执行
# Navigate to the docker directory first
cd docker
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA backtest
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA backtest
示例输出:
可选指定起止日期进行特定时段回测:
# With Poetry:
poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01
# With Docker (from docker/ directory):
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01 backtest
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01 backtest
通过--ollama
参数可使用本地LLM运行回测:
# With Poetry:
poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA --ollama
# With Docker (from docker/ directory):
# On Linux/Mac:
./run.sh --ticker AAPL,MSFT,NVDA --ollama backtest
# On Windows:
run.bat --ticker AAPL,MSFT,NVDA --ollama backtest
重要:请保持PR精简专注,便于审核合并。
如有功能需求,请提交issue并标记enhancement
标签。
本项目采用MIT许可证,详见LICENSE文件。