ollama-init.sh

#!/bin/bash
set -e

# Start Ollama server in background
echo "Starting Ollama server..."
ollama serve > /tmp/ollama.log 2>&1 &
SERVER_PID=$!

# Wait for server to be ready
echo "Waiting for Ollama API to be ready..."
for i in {1..60}; do
  if curl -sf http://localhost:11434/api/tags > /dev/null 2>&1; then
    echo "Ollama server is ready!"
    break
  fi
  if [ $i -eq 60 ]; then
    echo "Timeout waiting for Ollama server" >&2
    cat /tmp/ollama.log >&2
    exit 1
  fi
  sleep 1
done

MODELS="${OLLAMA_MODELS:-llama3.2:3b den dengcao/Qwen3-Embedding-0.6B:Q8_0}"

echo "Pulling models: $MODELS"
for model in $MODELS; do
  echo "Pulling model: $model"
  if ollama pull "$model"; then
    echo "✓ Successfully pulled: $model"
  else
    echo "Failed to pull: $model" >&2
  fi
done

# Verify models are available
echo "Verifying models..."
ollama list

# Stop background server
echo "Stopping background server..."
kill $SERVER_PID 2>/dev/null || true
wait $SERVER_PID 2>/dev/null || true

# Start Ollama server in foreground
echo "Init complete. Starting Ollama server in foreground..."
exec ollama serve