mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-01 17:17:24 -04:00
Bill
fb9583b374
Major architecture transformation from batch-only to API service with
database persistence for Windmill integration.
## REST API Implementation
- POST /simulate/trigger - Start simulation jobs
- GET /simulate/status/{job_id} - Monitor job progress
- GET /results - Query results with filters (job_id, date, model)
- GET /health - Service health checks
## Database Layer
- SQLite persistence with 6 tables (jobs, job_details, positions,
holdings, reasoning_logs, tool_usage)
- Foreign key constraints with cascade deletes
- Replaces JSONL file storage
## Backend Components
- JobManager: Job lifecycle management with concurrency control
- RuntimeConfigManager: Thread-safe isolated runtime configs
- ModelDayExecutor: Single model-day execution engine
- SimulationWorker: Date-sequential, model-parallel orchestration
## Testing
- 102 unit and integration tests (85% coverage)
- Database: 98% coverage
- Job manager: 98% coverage
- API endpoints: 81% coverage
- Pydantic models: 100% coverage
- TDD approach throughout
## Docker Deployment
- Dual-mode: API server (persistent) + batch (one-time)
- Health checks with 30s interval
- Volume persistence for database and logs
- Separate entrypoints for each mode
## Validation Tools
- scripts/validate_docker_build.sh - Build validation
- scripts/test_api_endpoints.sh - Complete API testing
- scripts/test_batch_mode.sh - Batch mode validation
- DOCKER_API.md - Deployment guide
- TESTING_GUIDE.md - Testing procedures
## Configuration
- API_PORT environment variable (default: 8080)
- Backwards compatible with existing configs
- FastAPI, uvicorn, pydantic>=2.0 dependencies
Co-Authored-By: AI Assistant <noreply@example.com>
64 lines
1.6 KiB
Bash
64 lines
1.6 KiB
Bash
#!/bin/bash
|
|
set -e # Exit on any error
|
|
|
|
echo "🚀 Starting AI-Trader API Server..."
|
|
|
|
# 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 ""
|
|
exit 1
|
|
fi
|
|
|
|
echo "✅ Environment variables validated"
|
|
|
|
# Step 1: Initialize database
|
|
echo "📊 Initializing database..."
|
|
python -c "from api.database import initialize_database; initialize_database('data/jobs.db')"
|
|
echo "✅ Database initialized"
|
|
|
|
# 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: Start FastAPI server with uvicorn
|
|
echo "🌐 Starting FastAPI server on port ${API_PORT:-8080}..."
|
|
uvicorn api.main:app \
|
|
--host 0.0.0.0 \
|
|
--port ${API_PORT:-8080} \
|
|
--log-level info \
|
|
--access-log
|
|
|
|
# Cleanup on exit
|
|
trap "echo '🛑 Stopping services...'; kill $MCP_PID 2>/dev/null; exit 0" EXIT SIGTERM SIGINT
|