Consolidated local ollama model logic
This commit is contained in:
Executable
+41
@@ -0,0 +1,41 @@
|
||||
#!/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}"
|
||||
Reference in New Issue
Block a user