#!/bin/bash

echo "🚀 Iniciando Ollama (llama3:1b)..."

ollama serve &

# Espera servidor subir
until curl -s http://localhost:7860 >/dev/null; do
  sleep 2
done

echo "📥 Baixando modelo llama3:1b..."
ollama pull llama3:1b

echo "🔥 Warmup do modelo..."

curl http://localhost:7860/api/generate -d '{
  "model": "llama3:1b",
  "prompt": "hello",
  "stream": false
}' >/dev/null

echo "✅ Modelo pronto!"

# Mantém container ativo
wait