60 lines
1.8 KiB
Bash
Executable File
60 lines
1.8 KiB
Bash
Executable File
#!/bin/bash
|
|
# ── Ollama Initialization Script ──────────────────────────────────────────────
|
|
#
|
|
# Starts Ollama and pulls models on first run.
|
|
# Requires: curl (ships with the ollama image).
|
|
# jq is installed at runtime if missing so we can parse /api/tags reliably
|
|
# instead of fragile grep-based JSON extraction.
|
|
|
|
set -e
|
|
|
|
echo "🚀 Ollama startup — checking for models..."
|
|
|
|
# ── Ensure jq is available (ollama image is Debian-based) ────────────────────
|
|
if ! command -v jq &>/dev/null; then
|
|
echo "📦 Installing jq for reliable JSON parsing..."
|
|
apt-get update -qq && apt-get install -y -qq jq >/dev/null 2>&1 || true
|
|
fi
|
|
|
|
# Start Ollama in background
|
|
ollama serve &
|
|
OLLAMA_PID=$!
|
|
|
|
# Wait for Ollama to be ready
|
|
echo "⏳ Waiting for Ollama to be ready..."
|
|
for i in {1..60}; do
|
|
if curl -s http://localhost:11434/api/tags > /dev/null 2>&1; then
|
|
echo "✓ Ollama is ready"
|
|
break
|
|
fi
|
|
if [ "$i" -eq 60 ]; then
|
|
echo "❌ Ollama failed to start after 60 s"
|
|
exit 1
|
|
fi
|
|
echo " Attempt $i/60..."
|
|
sleep 1
|
|
done
|
|
|
|
# Check if models are already present (jq with grep fallback)
|
|
echo "📋 Checking available models..."
|
|
TAGS_JSON=$(curl -s http://localhost:11434/api/tags)
|
|
|
|
if command -v jq &>/dev/null; then
|
|
MODELS=$(echo "$TAGS_JSON" | jq '.models | length')
|
|
else
|
|
# Fallback: count "name" keys (less reliable but functional)
|
|
MODELS=$(echo "$TAGS_JSON" | grep -o '"name"' | wc -l)
|
|
fi
|
|
|
|
if [ "${MODELS:-0}" -eq 0 ]; then
|
|
echo "📥 No models found. Pulling llama3.2..."
|
|
ollama pull llama3.2 || echo "⚠️ Failed to pull llama3.2 (may already be pulling)"
|
|
else
|
|
echo "✓ Models available: $MODELS"
|
|
fi
|
|
|
|
echo "✓ Ollama initialization complete"
|
|
|
|
# Keep process running
|
|
wait $OLLAMA_PID
|