50 lines
1.4 KiB
Bash
Executable File
50 lines
1.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
cp -n "${ROOT}/.env.example" "${ROOT}/.env" || true
|
|
source "${ROOT}/.env"
|
|
|
|
ts(){ date -Is; }
|
|
log_dir="${ROOT}/logs"
|
|
mkdir -p "$log_dir"
|
|
log_file="${log_dir}/bootstrap-$(date -Iseconds).log"
|
|
exec > >(tee -a "$log_file") 2>&1
|
|
|
|
echo "[$(ts)] bootstrap: starting (ROOT=$ROOT)"
|
|
echo "[$(ts)] bootstrap: docker compose up -d"
|
|
docker compose -f "${ROOT}/docker-compose.yml" up -d
|
|
|
|
echo "[$(ts)] bootstrap: waiting for Ollama API at ${OLLAMA_URL} ..."
|
|
for i in {1..90}; do
|
|
if curl -sS "${OLLAMA_URL}/api/tags" >/dev/null 2>&1; then
|
|
echo "[$(ts)] bootstrap: Ollama API is up."
|
|
break
|
|
fi
|
|
if [[ $i -eq 90 ]]; then
|
|
echo "[$(ts)] bootstrap: ERROR: API did not come up in time."
|
|
exit 1
|
|
fi
|
|
sleep 1
|
|
done
|
|
|
|
echo "[$(ts)] bootstrap: 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)] bootstrap: pulling extra model: $m"
|
|
docker exec -it ollama ollama pull "$m"
|
|
done
|
|
fi
|
|
|
|
echo "[$(ts)] bootstrap: building expert model: ${EXPERT_MODEL}"
|
|
tmp="$(mktemp)"
|
|
sed "s/\${BASE_MODEL}/${BASE_MODEL}/g" "${ROOT}/Modelfile" > "$tmp"
|
|
docker exec -i ollama ollama create "${EXPERT_MODEL}" -f - < "$tmp"
|
|
rm -f "$tmp"
|
|
|
|
echo "[$(ts)] bootstrap: done"
|
|
echo "[$(ts)] bootstrap: test:"
|
|
echo " echo "SELECT 1;" | ${ROOT}/bin/sqlai analyze-tsql"
|