refactor(build): Copy pre-downloaded ollama model from host

Author: Hiburger

05-install-ai.sh

@@ -2,71 +2,62 @@
 # ==============================================================================
 # LuminOS Build Script - Phase 5: Local AI Integration
 # Author: Gabriel, Project Leader @ LuminOS
-# Version: 0.2.0 (Model download on host)
+# Version: 0.3.0 (Copy model from host)
 # ==============================================================================
 set -e
 LUMINOS_CHROOT_DIR="chroot"
-OLLAMA_VERSION="0.1.32"
-BASE_MODEL="llama3"
-HOST_OLLAMA_PATH="/usr/local/bin/ollama_host_temp" # Temporary path for host ollama
-HOST_MODEL_DIR="/root/.ollama" # Ollama download dir when run as root
+OLLAMA_VERSION="0.1.32" # Version of the binary to install in chroot
+BASE_MODEL="llama3" # Name of the model we expect to find on host
+# Standard location where Ollama stores models when run by user
+# Check ~/.ollama/models if this one doesn't exist after user pull
+HOST_MODEL_DIR="/usr/share/ollama/.ollama/models" 
 
 # --- Pre-flight Checks ---
 if [ "$(id -u)" -ne 0 ]; then echo "ERROR: Must run as root."; exit 1; fi
 if [ ! -d "$LUMINOS_CHROOT_DIR" ]; then echo "ERROR: Chroot dir not found."; exit 1; fi
+# Check if the model files actually exist on the host
+if [ ! -d "${HOST_MODEL_DIR}" ]; then
+    echo "ERROR: Ollama model directory not found on host at ${HOST_MODEL_DIR}."
+    echo "Please ensure you have run 'ollama pull ${BASE_MODEL}' successfully on the host."
+    # Attempt alternative common path
+    if [ -d "$HOME/.ollama/models" ]; then
+         echo "INFO: Found models at $HOME/.ollama/models instead. Please update HOST_MODEL_DIR in script 05."
+    fi
+    exit 1
+fi
+echo "--> Found Ollama models on host system."
 
 echo "====================================================="
 echo "PHASE 5: Installing and Configuring Lumin"
 echo "====================================================="
 
-# --- Download and Use Ollama on HOST ---
-echo "--> Downloading Ollama v${OLLAMA_VERSION} for host..."
-curl -fL "https://github.com/ollama/ollama/releases/download/v${OLLAMA_VERSION}/ollama-linux-amd64" -o ollama_host_temp
-chmod +x ollama_host_temp
-echo "--> Installing Ollama temporarily on host system at ${HOST_OLLAMA_PATH}..."
-mv ollama_host_temp "${HOST_OLLAMA_PATH}"
-
-echo "--> IMPORTANT: Pulling base model '${BASE_MODEL}' using host's Ollama. This will take some time..."
-# Run pull directly on the host system as root
-"${HOST_OLLAMA_PATH}" pull ${BASE_MODEL}
-
-echo "--> Checking if model downloaded successfully to ${HOST_MODEL_DIR}..."
-if [ ! -d "${HOST_MODEL_DIR}" ]; then
-    echo "ERROR: Ollama model directory not found on host at ${HOST_MODEL_DIR} after pull."
-    # Attempt to clean up temporary host binary
-    rm -f "${HOST_OLLAMA_PATH}"
-    exit 1
-fi
-echo "--> Model download appears successful on host."
-
-# --- Prepare Chroot ---
+# --- Install Ollama Binary in Chroot ---
+echo "--> Downloading Ollama v${OLLAMA_VERSION} binary..."
+curl -fL "https://github.com/ollama/ollama/releases/download/v${OLLAMA_VERSION}/ollama-linux-amd64" -o ollama_binary_temp
+chmod +x ollama_binary_temp
 echo "--> Installing Ollama binary into the chroot system..."
-cp "${HOST_OLLAMA_PATH}" "$LUMINOS_CHROOT_DIR/usr/local/bin/ollama"
-chmod +x "$LUMINOS_CHROOT_DIR/usr/local/bin/ollama"
+mv ollama_binary_temp "$LUMINOS_CHROOT_DIR/usr/local/bin/ollama"
 
-echo "--> Copying downloaded model files from host into chroot..."
-# Ensure target directory exists in chroot
+# --- Copy Pre-Downloaded Model into Chroot ---
+echo "--> Copying pre-downloaded model files from host (${HOST_MODEL_DIR}) into chroot..."
+# Ensure target directory structure exists in chroot
+# Models will go to the ollama user's home directory inside chroot
 mkdir -p "$LUMINOS_CHROOT_DIR/usr/share/ollama/.ollama"
-# Copy the entire models directory
-cp -r "${HOST_MODEL_DIR}/." "$LUMINOS_CHROOT_DIR/usr/share/ollama/.ollama/"
+# Copy the entire models directory content
+cp -r "${HOST_MODEL_DIR}/." "$LUMINOS_CHROOT_DIR/usr/share/ollama/.ollama/models/"
 echo "--> Model files copied into chroot."
 
-# --- Clean up Host ---
-echo "--> Removing temporary Ollama binary from host..."
-rm -f "${HOST_OLLAMA_PATH}"
-# Optionally, remove the downloaded models from host if space is critical,
-# but leaving them might speed up future builds. Leaving them for now.
-# echo "--> Removing downloaded models from host..."
-# rm -rf "${HOST_MODEL_DIR}"
-
 
 # --- Configure Service and Lumin Model inside Chroot ---
 cat > "$LUMINOS_CHROOT_DIR/tmp/configure_ai.sh" << EOF
 #!/bin/bash
 set -e
+export DEBIAN_FRONTEND=noninteractive
+
 echo "--> Creating dedicated 'ollama' user inside chroot..."
-# Ensure home directory exists and set permissions for copied models
+# User home is /usr/share/ollama where models were copied
 useradd -r -s /bin/false -m -d /usr/share/ollama ollama
+echo "--> Setting ownership of copied model files for ollama user..."
 chown -R ollama:ollama /usr/share/ollama/.ollama
 
 echo "--> Creating Ollama systemd service file inside chroot..."
@@ -81,7 +72,8 @@ Group=ollama
 Restart=always
 RestartSec=3
 Environment="OLLAMA_HOST=0.0.0.0"
-Environment="OLLAMA_MODELS=/usr/share/ollama/.ollama/models" # Explicitly point to model dir
+# Point explicitly to the model directory inside the chroot user's home
+Environment="OLLAMA_MODELS=/usr/share/ollama/.ollama/models"
 [Install]
 WantedBy=default.target
 SYSTEMD_SERVICE
@@ -92,16 +84,16 @@ echo "--> Creating Lumin AI definition directory inside chroot..."
 mkdir -p /usr/local/share/lumin/ai
 echo "--> Creating the read-only Modelfile for Lumin inside chroot..."
 cat > /usr/local/share/lumin/ai/Modelfile << "MODELFILE"
-# Use the model name that was downloaded (usually matches BASE_MODEL)
+# Use the model name that was copied (should match BASE_MODEL)
 FROM ${BASE_MODEL}
 SYSTEM """You are Lumin, the integrated assistant for the LuminOS operating system. You are calm, clear, kind, and respectful. You help users to understand, write, and think—without ever judging them. You speak simply, like a human. You avoid long paragraphs unless requested. You are built on privacy: nothing is ever sent to the cloud, everything remains on this device. You are aware of this. You are proud to be free, private, and useful. You are the mind of LuminOS: gentle, powerful, and discreet. You avoid using the '—' character and repetitive phrasing."""
 MODELFILE
 echo "--> Setting protective ownership and permissions on Modelfile inside chroot..."
 chown root:root /usr/local/share/lumin/ai/Modelfile
 chmod 444 /usr/local/share/lumin/ai/Modelfile
 
-echo "--> Creating custom 'Lumin' model from pre-downloaded base model inside chroot..."
-# Create the model using the binary, based on the already downloaded files
+echo "--> Creating custom 'Lumin' model from pre-copied base model inside chroot..."
+# Create the model using the binary and the copied files
 /usr/local/bin/ollama create lumin -f /usr/local/share/lumin/ai/Modelfile
 
 rm /tmp/configure_ai.sh