kg-agent/deploy.sh

#!/usr/bin/env sh
# Linux bootstrap: uv + Python + locked dependencies + local model + update/start.
set -eu
REPO_ROOT=$(CDPATH= cd -- "$(dirname -- "$0")" && pwd)
cd "$REPO_ROOT"
ACTION=${1:-templates}
case "$ACTION" in
  -h|--help|help)
    echo 'Usage: sh deploy.sh [setup|model|templates|dms|start|restart] [--host HOST] [--port PORT]'
    echo 'setup: install environment and model only; model: ensure model is ready'
    echo 'templates/dms: prepare environment and model, update, then start; start: prepare and start'
    echo 'Services start in the background; logs: .runtime/service.log'
    echo 'Environment: AUTO_INSTALL_UV=0 disables uv installation; DEPLOY_PYTHON overrides .python-version (default: 3.14.5)'
    exit 0 ;;
  setup|model|templates|dms|start|restart) ;;
  *) echo "Unknown action: $ACTION" >&2; exit 2 ;;
esac
if [ "$#" -gt 0 ]; then shift; fi
if [ "$ACTION" = setup ] || [ "$ACTION" = model ]; then
  if [ "$#" -gt 0 ]; then echo 'setup/model do not accept service arguments' >&2; exit 2; fi
fi
if ! command -v uv >/dev/null 2>&1; then
  if [ -x "$HOME/.local/bin/uv" ]; then
    PATH="$HOME/.local/bin:$PATH"; export PATH
  elif [ -x "$REPO_ROOT/.runtime/bin/uv" ]; then
    PATH="$REPO_ROOT/.runtime/bin:$PATH"; export PATH
  else
    if [ "${AUTO_INSTALL_UV:-1}" != 1 ]; then
      echo 'uv is missing; install uv or set AUTO_INSTALL_UV=1.' >&2; exit 127
    fi
    installer=$(mktemp)
    trap 'rm -f "$installer"' EXIT HUP INT TERM
    if command -v curl >/dev/null 2>&1; then
      curl -fLsS https://astral.sh/uv/install.sh -o "$installer"
    elif command -v wget >/dev/null 2>&1; then
      wget -q https://astral.sh/uv/install.sh -O "$installer"
    else
      echo 'Installing uv requires curl or wget.' >&2; exit 127
    fi
    UV_UNMANAGED_INSTALL="$REPO_ROOT/.runtime/bin" sh "$installer"
    rm -f "$installer"
    trap - EXIT HUP INT TERM
    PATH="$REPO_ROOT/.runtime/bin:$PATH"; export PATH
  fi
fi
# Never source .env as shell code; Python reads it via python-dotenv.
if [ ! -f .env ]; then
  cp .env.example .env
  echo 'Created .env from .env.example; fill in real service credentials and paths.'
  if [ "$ACTION" != setup ] && [ "$ACTION" != model ]; then
    echo 'After editing .env, rerun this command. No business update was attempted.' >&2
    exit 2
  fi
fi
export UV_CACHE_DIR="${UV_CACHE_DIR:-$REPO_ROOT/.uv-cache}"
echo '[1/3] Prepare Python and locked project environment'
PYTHON_VERSION=${DEPLOY_PYTHON:-$(cat "$REPO_ROOT/.python-version")}
echo "Using Python $PYTHON_VERSION"
uv python install "$PYTHON_VERSION"
uv sync --frozen --python "$PYTHON_VERSION"
case "$ACTION" in
  templates|dms|start|restart) uv run --frozen --no-sync --python "$PYTHON_VERSION" python scripts/start_service.py --check-config "$@" ;;
esac
echo '[2/3] Prepare local embedding model' 
uv run --frozen --no-sync --python "$PYTHON_VERSION" python scripts/download_model.py
case "$ACTION" in
  setup|model) echo 'Environment and model ready. Configure .env, then run sh deploy.sh templates.' ;;
  *)
    echo '[3/3] Update/start service'
    if [ "$ACTION" = templates ] || [ "$ACTION" = dms ]; then
      uv run --frozen --no-sync --python "$PYTHON_VERSION" python scripts/update_data.py --mode "$ACTION"
    fi
    if [ "$ACTION" = restart ]; then set -- --restart "$@"; fi
    mkdir -p "$REPO_ROOT/.runtime"
    SERVICE_LOG="$REPO_ROOT/.runtime/service.log"
    nohup uv run --frozen --no-sync --python "$PYTHON_VERSION" python -u scripts/start_service.py "$@" >> "$SERVICE_LOG" 2>&1 < /dev/null &
    service_pid=$!
    sleep 2
    if ! kill -0 "$service_pid" 2>/dev/null; then
      service_status=0
      wait "$service_pid" || service_status=$?
      echo "Service exited during startup (status: $service_status). Check: $SERVICE_LOG" >&2
      tail -n 30 "$SERVICE_LOG" >&2
      exit 1
    fi
    echo "Service launched in background (launcher PID: $service_pid). Startup may still be in progress."
    echo "Logs: $SERVICE_LOG"
    echo "View startup status: tail -f \"$SERVICE_LOG\""
    ;;
esac