Create scripts/deploy.sh

scripts/deploy.sh

@@ -0,0 +1,128 @@
+#!/bin/bash
+
+# Script de despliegue completo para Diamond AI Platform
+
+set -e  # Exit on error
+
+echo "🚀 Iniciando despliegue de Diamond AI Platform..."
+
+# Variables de configuración
+ENVIRONMENT=${1:-production}
+MODEL_NAME="wizardlm-7b-uncensored"
+SOLANA_NETWORK="mainnet-beta"
+
+# Colores para output
+GREEN='\033[0;32m'
+RED='\033[0;31m'
+NC='\033[0m' # No Color
+
+# Función para verificar comandos
+check_command() {
+    if ! command -v $1 &> /dev/null; then
+        echo -e "${RED}Error: $1 no está instalado${NC}"
+        exit 1
+    fi
+}
+
+# Verificar dependencias
+echo "📦 Verificando dependencias..."
+check_command docker
+check_command docker-compose
+check_command nvidia-docker
+check_command solana
+check_command python3
+
+# Verificar GPUs disponibles
+echo "🎮 Verificando GPUs..."
+if ! nvidia-smi &> /dev/null; then
+    echo -e "${RED}Error: No se detectaron GPUs NVIDIA${NC}"
+    exit 1
+fi
+
+# Configurar entorno
+echo "⚙️ Configurando entorno $ENVIRONMENT..."
+export ENVIRONMENT=$ENVIRONMENT
+export SOLANA_NETWORK=$SOLANA_NETWORK
+
+# Crear directorios necesarios
+echo "📁 Creando estructura de directorios..."
+mkdir -p data/{models,datasets,logs}
+mkdir -p triton_models/$MODEL_NAME
+mkdir -p prometheus_data grafana_data
+
+# Descargar modelo WizardLM
+echo "🤖 Descargando modelo $MODEL_NAME..."
+if [ ! -d "data/models/$MODEL_NAME" ]; then
+    python3 scripts/download_model.py \
+        --model-name $MODEL_NAME \
+        --output-dir data/models/$MODEL_NAME \
+        --quantization fp16
+fi
+
+# Convertir modelo a formato Tritón
+echo "🔄 Convirtiendo modelo a formato Tritón..."
+python3 scripts/convert_to_triton.py \
+    --input data/models/$MODEL_NAME \
+    --output triton_models/$MODEL_NAME \
+    --model-type gpt \
+    --dtype fp16
+
+# Configurar Tritón
+echo "🎯 Configurando Tritón Inference Server..."
+cp config/triton/* triton_models/
+
+# Inicializar base de datos
+echo "🗄️ Inicializando base de datos..."
+docker-compose up -d postgres
+sleep 10  # Esperar que PostgreSQL inicie
+
+# Ejecutar migraciones
+echo "🔄 Ejecutando migraciones de base de datos..."
+docker-compose exec backend python -m alembic upgrade head
+
+# Verificar balance de tokens de la plataforma
+echo "💰 Verificando tokens DIAMANTE de la plataforma..."
+SOLANA_BALANCE=$(solana balance --url $SOLANA_NETWORK 2>/dev/null || echo "0 SOL")
+echo "Balance de plataforma: $SOLANA_BALANCE"
+
+# Iniciar todos los servicios
+echo "🚀 Iniciando todos los servicios..."
+docker-compose up -d
+
+# Esperar que los servicios estén listos
+echo "⏳ Esperando que los servicios estén listos..."
+sleep 30
+
+# Verificar salud de los servicios
+echo "🏥 Verificando salud de los servicios..."
+for service in triton-server backend postgres redis; do
+    if docker-compose ps $service | grep -q "Up"; then
+        echo -e "${GREEN}✓ $service está en ejecución${NC}"
+    else
+        echo -e "${RED}✗ $service no está ejecutándose${NC}"
+        exit 1
+    fi
+done
+
+# Inicializar tokens de la plataforma
+echo "💎 Inicializando tokens DIAMANTE..."
+docker-compose exec backend python scripts/init_tokens.py
+
+# Ejecutar tests
+echo "🧪 Ejecutando tests..."
+docker-compose exec backend python -m pytest tests/ -v
+
+echo -e "${GREEN}✅ Despliegue completado exitosamente!${NC}"
+echo ""
+echo "🌐 URLs de acceso:"
+echo "   Frontend: http://localhost:3000"
+echo "   API: http://localhost:8080"
+echo "   Tritón: http://localhost:8000"
+echo "   Grafana: http://localhost:3001"
+echo "   Prometheus: http://localhost:9090"
+echo ""
+echo "📋 Comandos útiles:"
+echo "   Ver logs: docker-compose logs -f"
+echo "   Detener: docker-compose down"
+echo "   Reiniciar: docker-compose restart"
+echo "   Actualizar: ./scripts/update.sh"