#!/usr/bin/env bash
set -euo pipefail
cd "$(dirname "${BASH_SOURCE[0]}")"
BASE_MODEL="${BASE_MODEL:-qwen3:8b}"
MODEL_NAME="${MODEL_NAME:-qwen3-8b-korean-coding-router:latest}"

echo "==== WAIT OLLAMA CONTAINER ===="
for i in $(seq 1 120); do
  if docker compose exec -T ollama ollama list >/tmp/qwen3_stack_ollama_list.txt 2>/tmp/qwen3_stack_ollama_err.txt; then
    echo "[OK] ollama ready"
    break
  fi
  echo "[WAIT] ollama not ready ($i/120)"
  sleep 2
done

if ! docker compose exec -T ollama ollama list | awk '{print $1}' | grep -qx "$BASE_MODEL"; then
  echo "==== PULL BASE MODEL: $BASE_MODEL ===="
  docker compose exec -T ollama ollama pull "$BASE_MODEL"
fi

echo "==== CREATE ROUTER MODEL: $MODEL_NAME ===="
docker compose exec -T ollama ollama create "$MODEL_NAME" -f /models/Modelfile

echo
echo "FINAL RESULT"
echo "================================================================================"
echo "status=SUCCESS"
echo "decision=DOCKER_OLLAMA_MODEL_CREATED"
echo "model_name=$MODEL_NAME"
echo "base_model=$BASE_MODEL"