Provision Netron and Lab 1 local assets

scripts/service_manager.sh

@@ -9,25 +9,6 @@ load_runtime_env
 
 mkdir -p "$STATE_DIR/run" "$STATE_DIR/logs"
 
-ensure_transformerlab_default_user() {
-  local helper_python="${TRANSFORMERLAB_DIR}/envs/transformerlab/bin/python"
-
-  if [ ! -x "$helper_python" ]; then
-    return 0
-  fi
-
-  if [ -z "${TRANSFORMERLAB_DEFAULT_USER_EMAIL:-}" ] || [ -z "${TRANSFORMERLAB_DEFAULT_USER_PASSWORD:-}" ]; then
-    return 0
-  fi
-
-  "$helper_python" "$SCRIPT_DIR/ensure_transformerlab_user.py" \
-    --transformerlab-dir "$TRANSFORMERLAB_DIR" \
-    --email "$TRANSFORMERLAB_DEFAULT_USER_EMAIL" \
-    --password "$TRANSFORMERLAB_DEFAULT_USER_PASSWORD" \
-    --first-name "${TRANSFORMERLAB_DEFAULT_USER_FIRST_NAME:-Student}" \
-    --last-name "${TRANSFORMERLAB_DEFAULT_USER_LAST_NAME:-}" >>"$STATE_DIR/logs/transformerlab_default_user.log" 2>&1 || true
-}
-
 check_wetty_prereqs() {
   if [ ! -x "$WETTY_BIN" ]; then
     echo "Missing WeTTY binary at $WETTY_BIN. Re-run ./labctl up." >&2
@@ -56,6 +37,40 @@ PY
   fi
 }
 
+ollama_version_gte_minimum() {
+  local version_output
+  local installed_version
+
+  if ! command -v "$OLLAMA_BIN" >/dev/null 2>&1; then
+    return 1
+  fi
+
+  version_output=$("$OLLAMA_BIN" --version 2>/dev/null || true)
+  installed_version=$(printf '%s' "$version_output" | grep -oE '[0-9]+\.[0-9]+\.[0-9]+' | head -n 1)
+
+  if [ -z "$installed_version" ]; then
+    return 1
+  fi
+
+  [ "$(printf '%s\n' "$COURSEWARE_OLLAMA_MIN_VERSION" "$installed_version" | sort -V | head -n 1)" = "$COURSEWARE_OLLAMA_MIN_VERSION" ]
+}
+
+assert_ollama_logprobs_support() {
+  if ollama_version_gte_minimum; then
+    return 0
+  fi
+
+  local version_output
+  version_output=$("$OLLAMA_BIN" --version 2>/dev/null || printf 'unknown')
+
+  cat <<EOF >&2
+Lab 1 requires Ollama ${COURSEWARE_OLLAMA_MIN_VERSION} or newer because the confidence visualizer depends on logprobs.
+Installed version: ${version_output}
+Re-run ./labctl up after upgrading Ollama.
+EOF
+  exit 1
+}
+
 resolve_targets() {
   if [ $# -eq 0 ]; then
     echo "No target specified." >&2
@@ -104,13 +119,10 @@ service_ready() {
     ollama)
       curl -fsS "$(service_url "$service")/api/tags" >/dev/null 2>&1
       ;;
-    transformerlab)
-      curl -fsS "$(service_url "$service")/healthz" >/dev/null 2>&1
-      ;;
     promptfoo)
       curl -fsS "$(service_url "$service")/health" >/dev/null 2>&1
       ;;
-    open-webui|chunkviz|embedding-atlas|unsloth|wiki|wetty)
+    open-webui|netron|chunkviz|embedding-atlas|unsloth|wiki|wetty)
       curl -fsS "$(service_url "$service")" >/dev/null 2>&1
       ;;
     *)
@@ -166,18 +178,16 @@ start_one() {
   local attempt
   local pid_grace_attempts=5
 
+  if [ "$service" = "ollama" ] || [ "$service" = "wiki" ]; then
+    assert_ollama_logprobs_support
+  fi
+
   if has_live_pid "$service"; then
-    if [ "$service" = "transformerlab" ]; then
-      ensure_transformerlab_default_user
-    fi
     echo "$service already running"
     return 0
   fi
 
   if service_ready "$service"; then
-    if [ "$service" = "transformerlab" ]; then
-      ensure_transformerlab_default_user
-    fi
     echo "$service already available"
     return 0
   fi
@@ -186,24 +196,6 @@ start_one() {
     open-webui)
       start_one ollama
       ;;
-    transformerlab)
-      if command -v python3 >/dev/null 2>&1; then
-        python3 "$SCRIPT_DIR/repair_transformerlab_plugin_supports.py" \
-          --transformerlab-dir "$TRANSFORMERLAB_DIR" \
-          --plugin "fastchat_server" \
-          --required-support "chat" \
-          --required-support "completion" \
-          --required-support "visualize_model" \
-          --required-support "model_layers" \
-          --required-support "rag" \
-          --required-support "tools" \
-          --required-support "template" \
-          --required-support "embeddings" \
-          --required-support "tokenize" \
-          --required-support "logprobs" \
-          --required-support "batched" >>"$STATE_DIR/logs/transformerlab_plugin_supports.log" 2>&1 || true
-      fi
-      ;;
     wetty)
       check_wetty_prereqs
       ;;
@@ -229,9 +221,6 @@ start_one() {
 
   for attempt in $(seq 1 60); do
     if service_ready "$service"; then
-      if [ "$service" = "transformerlab" ]; then
-        ensure_transformerlab_default_user
-      fi
       echo "started $service"
       return 0
     fi
@@ -313,7 +302,7 @@ urls() {
   cat <<EOF
 Ollama API:      $(service_url ollama)
 Open WebUI:      $(service_url open-webui)
-TransformerLab:  $(service_url transformerlab)
+Netron:          $(service_url netron)
 ChunkViz:        $(service_url chunkviz)
 Embedding Atlas: $(service_url embedding-atlas)
 Unsloth Studio:  $(service_url unsloth)