#!/usr/bin/env bash
# Install Ollama and pull the best model for M5 24GB
set -e

echo "=== Ollama setup ==="

# Install Ollama if not present
if ! command -v ollama &> /dev/null; then
    echo "Installing Ollama..."
    curl -fsSL https://ollama.com/install.sh | sh
else
    echo "Ollama already installed: $(ollama --version)"
fi

# Start the Ollama daemon in background if not running
if ! pgrep -x "ollama" > /dev/null; then
    echo "Starting Ollama daemon..."
    ollama serve &
    sleep 3
fi

# Pull Mistral 7B (fits comfortably in 24GB)
echo "Pulling mistral..."
ollama pull mistral

# Verify
echo "Models available:"
ollama list

echo ""
echo "=== Ollama ready ==="
echo "Test it: ollama run mistral 'Hello'"