Files
sql-ai-gui/scripts/bootstrap-native.sh

42 lines
1.1 KiB
Bash
Executable File

#!/usr/bin/env bash
set -euo pipefail
ROOT_OVERRIDE="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
export ROOT_OVERRIDE
# shellcheck disable=SC1091
source "${ROOT_OVERRIDE}/scripts/common.sh"
init_log bootstrap-native
acquire_lock jr-sql-ai-native-model.lock
print_header "bootstrap-native: START"
need_cmd curl
need_cmd python
load_env
need_cmd "$OLLAMA_BIN"
check_for_legacy_docker_ollama
[[ -n "$BASE_MODEL" ]] || fail "BASE_MODEL is empty" 50
log "bootstrap-native: ensuring local Ollama runtime is ready at ${OLLAMA_URL}"
ensure_ollama_ready "$OLLAMA_API_MAX_WAIT" "$OLLAMA_API_START_WAIT" || fail "Ollama API not reachable at ${OLLAMA_URL}" 51
ok "bootstrap-native: Ollama API reachable"
if fetch_version >/dev/null 2>&1; then
ok "bootstrap-native: Ollama version endpoint reachable"
fi
pull_configured_models
tmp="$(mktemp)"
trap 'rm -f "$tmp"' EXIT
render_modelfile "$tmp"
log "bootstrap-native: building expert model ${EXPERT_MODEL}"
ollama_cmd create "${EXPERT_MODEL}" -f "$tmp"
verify_models_available
ok "bootstrap-native: base and expert models are available"
warmup_sqlai
print_footer "bootstrap-native: END status=OK log=${LOG_FILE}"