Consolidated local ollama model logic

2026-04-07 15:42:00 +02:00
parent 9d8aa88bd0
commit 7309541c24
9 changed files with 1171 additions and 527 deletions
@@ -0,0 +1,54 @@
+#!/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 selftest-native
+acquire_lock jr-sql-ai-native-model.lock
+print_header "selftest-native: START"
+
+load_env
+need_cmd curl
+need_cmd python
+need_cmd "$OLLAMA_BIN"
+need_cmd grep
+check_for_legacy_docker_ollama
+[[ -x "$SQLAI_BIN" ]] || fail "bin/sqlai missing or not executable at ${SQLAI_BIN}" 70
+
+log "selftest-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}" 71
+ok "selftest-native: Ollama API reachable"
+
+tags="$(fetch_tags)"
+[[ -n "$tags" ]] || fail "/api/tags returned empty body" 72
+printf '%s' "$tags" | is_json || fail "/api/tags did not return valid JSON" 73
+ok "selftest-native: /api/tags returned valid JSON"
+
+model_in_tags "$EXPERT_MODEL" "$tags" || fail "expert model missing: ${EXPERT_MODEL}" 74
+ok "selftest-native: expert model present: ${EXPERT_MODEL}"
+
+if [[ -n "$BASE_MODEL" ]]; then
+  model_in_tags "$BASE_MODEL" "$tags" || fail "base model missing: ${BASE_MODEL}" 75
+  ok "selftest-native: base model present: ${BASE_MODEL}"
+fi
+
+log "selftest-native: warmup request via sqlai"
+set +e
+"$SQLAI_BIN" ask --text "Warmup: reply with exactly 'OK'." --no-metrics >/dev/null 2>&1
+rc=$?
+set -e
+[[ "$rc" -eq 0 ]] || warn "warmup failed rc=${rc}"
+
+log "selftest-native: real query via sqlai"
+set +e
+out="$("$SQLAI_BIN" ask --text "Give a concise checklist to troubleshoot parameter sniffing in SQL Server 2022. Keep it technical." --no-metrics 2>&1)"
+rc=$?
+set -e
+[[ "$rc" -eq 0 ]] || fail "real query failed rc=${rc}" 76
+[[ -n "${out//[[:space:]]/}" ]] || fail "real query returned empty output" 77
+ok "selftest-native: real query returned non-empty output"
+
+print_footer "selftest-native: END status=OK log=${LOG_FILE}"