#!/usr/bin/env bash
set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
LOG_DIR="${ROOT_DIR}/logs"
mkdir -p "$LOG_DIR"

TIMESTAMP="$(date '+%Y%m%d_%H%M%S')"
LOG_FILE="${LOG_DIR}/model_probe_${TIMESTAMP}.log"
ENV_FILE="${PMFLOW_ENV_FILE:-${ROOT_DIR}/.env}"

echo "[模型检测] 开始执行独立连通性检测"
echo "[模型检测] 环境文件: ${ENV_FILE}"
echo "[模型检测] 日志文件: ${LOG_FILE}"

python3 "${ROOT_DIR}/scripts/model_probe.py" --env-file "${ENV_FILE}" "$@" 2>&1 | tee "${LOG_FILE}"
STATUS=${PIPESTATUS[0]}

if [[ "${STATUS}" -eq 0 ]]; then
  echo "[模型检测] 完成：全部模型连通正常"
else
  echo "[模型检测] 完成：存在连通性异常，请查看 ${LOG_FILE}"
fi

exit "${STATUS}"