#!/usr/bin/env bash

set -euo pipefail

API_KEY=$(grep '^nvapi-' /home/azurian/.nvidia | tr -d '\r\n ')

CONTAINER_NAME="nim-qwen3.8-27b-spark"
IMAGE="nvcr.io/nim/qwen/qwen3.8-27b:2.1.1-variant"
CACHE_DIR="/home/azurian/.cache/nim/qwen3.8-27b"
PORT="8001"

mkdir -p "$CACHE_DIR"
chmod -R a+rwX "$CACHE_DIR"

docker rm -f "$CONTAINER_NAME" 2>/dev/null || true

echo "Avvio Qwen3.8-27B NIM su DGX Spark..."

docker run -d \
  --name "$CONTAINER_NAME" \
  --restart unless-stopped \
  --runtime=nvidia \
  --gpus all \
  --ipc=host \
  --shm-size=16G \
  -e NGC_API_KEY="$API_KEY" \
  -v "$CACHE_DIR:/opt/nim/.cache" \
  -p "${PORT}:8000" \
  "$IMAGE"

echo
echo "Container avviato: $CONTAINER_NAME"
echo
echo "Log:"
echo "docker logs -f $CONTAINER_NAME"
echo
echo "Health:"
echo "curl http://localhost:${PORT}/v1/health/ready"
echo
echo "Models:"
echo "curl -s http://localhost:${PORT}/v1/models | jq"