#!/usr/bin/env bash
# Stop SGLang multi-node server and clean up.
set -euo pipefail
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "${SCRIPT_DIR}/config.env"

log() { echo "[$(date '+%H:%M:%S')] $*"; }

log "Stopping ${EXPERIMENT} ..."
ssh -o StrictHostKeyChecking=no "${HEAD_NODE}" \
  "docker rm -f ${EXPERIMENT}_head 2>/dev/null && echo 'Head stopped' || echo 'Head already stopped'"
ssh -o StrictHostKeyChecking=no "${WORKER_NODE}" \
  "docker rm -f ${EXPERIMENT}_worker 2>/dev/null && echo 'Worker stopped' || echo 'Worker already stopped'"
log "Done."