Este es un prototipo conceptual de un fondo de cobertura impulsado por inteligencia artificial. El objetivo de este proyecto es explorar el uso de IA para tomar decisiones de trading. Este proyecto tiene fines educativos únicamente y no está destinado a operaciones reales o inversión.
El sistema emplea varios agentes que trabajan en conjunto:
Nota: el sistema simula decisiones de trading, no realiza operaciones reales.
Este proyecto es solo para fines educativos y de investigación.
Al usar este software, acepta utilizarlo únicamente con fines de aprendizaje.
Clona el repositorio:
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
Asegúrate de tener Docker instalado en tu sistema. Si no, puedes descargarlo desde el sitio oficial de Docker.
Clona el repositorio:
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
Edita el archivo .env para agregar tus claves API como se describió anteriormente.
Navega al directorio docker:
cd docker
# On Linux/Mac:
./run.sh build
# On Windows:
run.bat build
Importante: Debes configurar OPENAI_API_KEY
, GROQ_API_KEY
, ANTHROPIC_API_KEY
, o DEEPSEEK_API_KEY
para que el fondo funcione. Si deseas usar LLMs de todos los proveedores, necesitarás configurar todas las claves API.
Los datos financieros para AAPL, GOOGL, MSFT, NVDA y TSLA son gratuitos y no requieren clave API.
Para cualquier otro ticker, necesitarás establecer FINANCIAL_DATASETS_API_KEY
en el archivo .env.
poetry run python src/main.py --ticker AAPL,MSFT,NVDA
Nota: Todos los comandos de Docker deben ejecutarse desde el directorio 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
Ejemplo de Salida:
También puedes especificar un flag --ollama
para ejecutar el fondo usando LLMs locales.
# 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
Puedes especificar un flag --show-reasoning
para mostrar el razonamiento de cada agente en la consola.
# 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
Opcionalmente puedes especificar fechas de inicio y fin para tomar decisiones en un período específico.
# 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
Nota: Todos los comandos de Docker deben ejecutarse desde el directorio 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
Ejemplo de Salida:
Puedes especificar opcionalmente fechas de inicio y fin para realizar backtesting en un período específico.
# 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
También puedes especificar un flag --ollama
para ejecutar el backtester usando LLMs locales.
# 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
Importante: Por favor mantén tus pull requests pequeños y enfocados. Esto facilitará su revisión y fusión.
Si tienes una solicitud de funcionalidad, por favor abre un issue y asegúrate de etiquetarlo como enhancement
.
Este proyecto está licenciado bajo la Licencia MIT - consulta el archivo LICENSE para más detalles.