mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-02 09:37:23 -04:00
Bill
9e05ce0891
Preserve existing merged.jsonl to avoid data loss and API rate limits. Only fetch new data if merged.jsonl is missing or empty. Problem: - Entrypoint always fetched fresh data from Alpha Vantage on every start - Overwrote existing mounted data directory - Caused API rate limit issues and data inconsistencies - Lost historical data needed for backtesting specific date ranges Solution: - Check if merged.jsonl exists and has content before fetching - Display stock count when using existing data - Provide manual refresh instructions for when updates are needed Benefits: - Faster container startup (no API calls if data exists) - Avoids Alpha Vantage rate limits (5 calls/min, 500/day) - Preserves user's existing historical datasets - Enables reliable backtesting with consistent data To refresh data: rm data/merged.jsonl && docker-compose restart
83 lines
2.3 KiB
Bash
Executable File
83 lines
2.3 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e # Exit on any error
|
|
|
|
echo "🚀 Starting AI-Trader..."
|
|
|
|
# Validate required environment variables
|
|
echo "🔍 Validating environment variables..."
|
|
MISSING_VARS=()
|
|
|
|
if [ -z "$OPENAI_API_KEY" ]; then
|
|
MISSING_VARS+=("OPENAI_API_KEY")
|
|
fi
|
|
|
|
if [ -z "$ALPHAADVANTAGE_API_KEY" ]; then
|
|
MISSING_VARS+=("ALPHAADVANTAGE_API_KEY")
|
|
fi
|
|
|
|
if [ -z "$JINA_API_KEY" ]; then
|
|
MISSING_VARS+=("JINA_API_KEY")
|
|
fi
|
|
|
|
if [ ${#MISSING_VARS[@]} -gt 0 ]; then
|
|
echo ""
|
|
echo "❌ ERROR: Missing required environment variables:"
|
|
for var in "${MISSING_VARS[@]}"; do
|
|
echo " - $var"
|
|
done
|
|
echo ""
|
|
echo "Please set these variables in your .env file:"
|
|
echo " 1. Copy .env.example to .env"
|
|
echo " 2. Edit .env and add your API keys"
|
|
echo " 3. Restart the container"
|
|
echo ""
|
|
echo "See docs/DOCKER.md for more information."
|
|
exit 1
|
|
fi
|
|
|
|
echo "✅ Environment variables validated"
|
|
|
|
# Step 1: Data preparation
|
|
echo "📊 Checking price data..."
|
|
if [ -f "/app/data/merged.jsonl" ] && [ -s "/app/data/merged.jsonl" ]; then
|
|
echo "✅ Using existing price data ($(wc -l < /app/data/merged.jsonl) stocks)"
|
|
echo " To refresh data, delete /app/data/merged.jsonl and restart"
|
|
else
|
|
echo "📊 Fetching and merging price data..."
|
|
# Run scripts from /app/scripts but output to /app/data
|
|
cd /app/data
|
|
python /app/scripts/get_daily_price.py
|
|
python /app/scripts/merge_jsonl.py
|
|
cd /app
|
|
fi
|
|
|
|
# Step 2: Start MCP services in background
|
|
echo "🔧 Starting MCP services..."
|
|
cd /app
|
|
python agent_tools/start_mcp_services.py &
|
|
MCP_PID=$!
|
|
|
|
# Step 3: Wait for services to initialize
|
|
echo "⏳ Waiting for MCP services to start..."
|
|
sleep 3
|
|
|
|
# Step 4: Run trading agent with config file
|
|
echo "🤖 Starting trading agent..."
|
|
|
|
# Smart config selection: custom_config.json takes precedence if it exists
|
|
if [ -f "configs/custom_config.json" ]; then
|
|
CONFIG_FILE="configs/custom_config.json"
|
|
echo "✅ Using custom configuration: configs/custom_config.json"
|
|
elif [ -n "$1" ]; then
|
|
CONFIG_FILE="$1"
|
|
echo "✅ Using specified configuration: $CONFIG_FILE"
|
|
else
|
|
CONFIG_FILE="configs/default_config.json"
|
|
echo "✅ Using default configuration: configs/default_config.json"
|
|
fi
|
|
|
|
python main.py "$CONFIG_FILE"
|
|
|
|
# Cleanup on exit
|
|
trap "echo '🛑 Stopping MCP services...'; kill $MCP_PID 2>/dev/null; exit 0" EXIT SIGTERM SIGINT
|