brain / scripts /preload_ollama_models.sh
Leon4gr45's picture
Deploy complete BrainAPI source 53d7c6447a51
5583f46 verified
Raw History Blame Contribute Delete
839 Bytes
#!/usr/bin/env bash
set -euo pipefail
export OLLAMA_HOST=127.0.0.1:11434
export OLLAMA_MODELS=${OLLAMA_MODELS:-/opt/ollama-models}
ollama serve > /tmp/ollama-preload.log 2>&1 &
ollama_pid=$!
trap 'kill "$ollama_pid" 2>/dev/null || true; wait "$ollama_pid" 2>/dev/null || true' EXIT
ready=false
for attempt in {1..60}; do
if curl --noproxy '*' --fail --silent http://127.0.0.1:11434/api/version >/dev/null; then
ready=true
break
fi
if ! kill -0 "$ollama_pid" 2>/dev/null; then
cat /tmp/ollama-preload.log >&2
exit 1
fi
sleep 1
done
if [ "$ready" != true ]; then
echo 'Ollama did not become ready during model preload' >&2
exit 1
fi
ollama pull "${OLLAMA_EMBEDDING_MODEL:-embeddinggemma-2:270m}"
if [ -n "${OLLAMA_CHAT_MODEL:-}" ]; then
ollama pull "${OLLAMA_CHAT_MODEL:-}"
fi