#!/usr/bin/env bash
# Pulls a starter set of Ollama models. Edit MODELS below to change what gets
# downloaded - pick sizes that match your RAM (see README.md section 2).
# Usage: bash scripts/pull-models.sh
set -euo pipefail

MODELS=(
  "llama3.2"        # general-purpose, ~3B, good default for 8-16GB machines
  "qwen2.5-coder"    # code-focused model
)

if ! command -v ollama >/dev/null 2>&1; then
  echo "Ollama is not installed. Run scripts/install-mac.sh first (or see"
  echo "docs/windows-linux.md on Windows/Linux)."
  exit 1
fi

for model in "${MODELS[@]}"; do
  echo "Pulling ${model}..."
  ollama pull "${model}"
done

echo "Done. Installed models:"
ollama list