これはAIを活用したヘッジファンドのコンセプト実証です。このプロジェクトの目的は、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
出力例:
ローカルLLMを使用してAIヘッジファンドを実行するには--ollama
フラグを指定できます。
# 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
ローカルLLMを使用してバックテスターを実行するには--ollama
フラグを指定できます。
# 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
重要: プルリクエストは小さく焦点を絞ったものにしてください。これによりレビューとマージが容易になります。
機能リクエストがある場合は、issueを開き、enhancement
タグを付けてください。
このプロジェクトはMITライセンスの下で公開されています。詳細はLICENSEファイルを参照してください。