File size: 1,487 Bytes
4dcfab0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | #!/usr/bin/env bash
# pull-model.sh — pull an Ollama model with state-file lifecycle.
#
# Usage: pull-model.sh <model-id>
#
# State files written under $OPENJARVIS_HOME/.state/models/:
# <model>.downloading — created on start
# <model>.ready — on success (atomic rename)
# <model>.failed — on failure after retries
# <model>.log — captured stderr
set -euo pipefail
MODEL="${1:-}"
if [[ -z "$MODEL" ]]; then
echo "usage: pull-model.sh <model-id>" >&2
exit 2
fi
OPENJARVIS_HOME="${OPENJARVIS_HOME:-$HOME/.openjarvis}"
STATE_DIR="$OPENJARVIS_HOME/.state/models"
mkdir -p "$STATE_DIR"
DOWNLOADING="$STATE_DIR/${MODEL}.downloading"
READY="$STATE_DIR/${MODEL}.ready"
FAILED="$STATE_DIR/${MODEL}.failed"
LOG="$STATE_DIR/${MODEL}.log"
# Cleanup any prior state for this model.
rm -f "$READY" "$FAILED"
touch "$DOWNLOADING"
MAX_RETRIES=3
attempt=0
last_exit=0
while [[ $attempt -lt $MAX_RETRIES ]]; do
attempt=$((attempt + 1))
if ollama pull "$MODEL" >>"$LOG" 2>&1; then
# Atomic rename: write to .tmp then mv.
tmp="$STATE_DIR/${MODEL}.ready.tmp"
date -u +"%Y-%m-%dT%H:%M:%SZ" > "$tmp"
mv "$tmp" "$READY"
rm -f "$DOWNLOADING"
exit 0
else
last_exit=$?
fi
done
# All retries exhausted.
{
echo "pull-model.sh: $MODEL failed after $MAX_RETRIES attempts (exit=$last_exit)"
tail -n 50 "$LOG" 2>/dev/null || true
} > "$FAILED"
rm -f "$DOWNLOADING"
exit "$last_exit"
|