#!/bin/bash
# Скрипт для загрузки моделей в Ollama
# Запустить ОДИН РАЗ после первого деплоя
# Модели сохраняются в PVC и не нужно скачивать повторно

set -e

NAMESPACE="${NAMESPACE:-gooseek}"
MODELS="${@:-llama3.2:3b}"

echo "=== Ollama Model Loader ==="
echo "Namespace: $NAMESPACE"
echo "Models:    $MODELS"

# Проверить что Ollama pod запущен
echo ""
echo "Checking Ollama pod status..."
kubectl -n $NAMESPACE wait --for=condition=ready pod -l app=ollama --timeout=120s

# Получить имя пода
POD=$(kubectl -n $NAMESPACE get pod -l app=ollama -o jsonpath='{.items[0].metadata.name}')
echo "Pod: $POD"

# Скачать модели
for MODEL in $MODELS; do
  echo ""
  echo "=== Pulling model: $MODEL ==="
  kubectl -n $NAMESPACE exec -it $POD -c ollama -- ollama pull $MODEL
done

# Показать список моделей
echo ""
echo "=== Installed models ==="
kubectl -n $NAMESPACE exec -it $POD -c ollama -- ollama list

echo ""
echo "=== Done! ==="
echo "Models are stored in PVC and will persist across restarts."