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

ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
MODEL_DIR="${MODEL_DIR:-${ROOT}/models/Agents-A1-hf}"
PYTHON="${PYTHON:-/home/ripper/onepoint/.vllm-venv/bin/python}"
CONVERT="${CONVERT:-${ROOT}/vendor/llama.cpp/convert_hf_to_gguf.py}"
OUT_DIR="${OUT_DIR:-${ROOT}/artifacts/mmproj}"
OUT="${OUT:-${OUT_DIR}/mmproj-agents-a1-bf16.gguf}"
LOG="${LOG:-${ROOT}/runs/convert-mmproj-agents-a1-bf16.log}"

mkdir -p "${OUT_DIR}" "$(dirname "${LOG}")"
export PYTHONPATH="${ROOT}/vendor/llama.cpp:${ROOT}/vendor/llama.cpp/gguf-py:${PYTHONPATH:-}"

if [[ -s "${OUT}" ]]; then
  echo "exists: ${OUT}"
  exit 0
fi

{
  echo "started_at=$(date -u --iso-8601=seconds)"
  echo "model_dir=${MODEL_DIR}"
  echo "out=${OUT}"
  "${PYTHON}" "${CONVERT}" "${MODEL_DIR}" \
    --mmproj \
    --outtype bf16 \
    --outfile "${OUT}"
  echo "finished_at=$(date -u --iso-8601=seconds)"
} 2>&1 | tee "${LOG}"