64 lines
1.7 KiB
Bash
Executable File
64 lines
1.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
|
|
# shellcheck disable=SC1090
|
|
source "${ROOT}/.env"
|
|
|
|
ts(){ date -Is; }
|
|
|
|
log_dir="${ROOT}/logs"
|
|
mkdir -p "$log_dir"
|
|
log_file="${log_dir}/update-$(date -Iseconds).log"
|
|
|
|
# log everything (stdout+stderr)
|
|
exec > >(tee -a "$log_file") 2>&1
|
|
|
|
echo "[$(ts)] update: starting (ROOT=$ROOT)"
|
|
|
|
echo "[$(ts)] update: pulling docker image(s)"
|
|
docker compose -f "${ROOT}/docker-compose.yml" pull
|
|
|
|
echo "[$(ts)] update: restarting services"
|
|
docker compose -f "${ROOT}/docker-compose.yml" up -d
|
|
|
|
echo "[$(ts)] update: waiting for Ollama API at ${OLLAMA_URL} ..."
|
|
for i in {1..120}; do
|
|
if curl -sS "${OLLAMA_URL}/api/tags" >/dev/null 2>&1; then
|
|
echo "[$(ts)] update: Ollama API is up."
|
|
break
|
|
fi
|
|
if [[ $i -eq 120 ]]; then
|
|
echo "[$(ts)] update: ERROR: API did not come up in time."
|
|
exit 1
|
|
fi
|
|
sleep 1
|
|
done
|
|
|
|
echo "[$(ts)] update: pulling base model: ${BASE_MODEL}"
|
|
docker exec -it ollama ollama pull "${BASE_MODEL}"
|
|
|
|
if [[ -n "${EXTRA_MODELS:-}" ]]; then
|
|
for m in ${EXTRA_MODELS}; do
|
|
echo "[$(ts)] update: pulling extra model: $m"
|
|
docker exec -it ollama ollama pull "$m"
|
|
done
|
|
fi
|
|
|
|
echo "[$(ts)] update: rebuilding expert model: ${EXPERT_MODEL}"
|
|
|
|
tmp="$(mktemp)"
|
|
sed "s/\${BASE_MODEL}/${BASE_MODEL}/g" "${ROOT}/Modelfile" > "$tmp"
|
|
|
|
docker cp "$tmp" ollama:/tmp/Modelfile.jr-sql-expert
|
|
docker exec -it ollama ollama create "${EXPERT_MODEL}" -f /tmp/Modelfile.jr-sql-expert
|
|
|
|
rm -f "$tmp"
|
|
|
|
echo "[$(ts)] update: verifying model exists..."
|
|
docker exec -it ollama ollama list | grep -F "${EXPERT_MODEL}" >/dev/null && \
|
|
echo "[$(ts)] update: OK: ${EXPERT_MODEL} is available."
|
|
|
|
echo "[$(ts)] update: complete"
|