#!/usr/bin/with-contenv bashio
CONFIG_PATH=/data/options.json

GPU_LAYERS="$(bashio::config 'gpu_layers')"
TOKEN="$(bashio::config 'master_token' || echo '')"

if [ ! -z "$TOKEN" ]; then
    export LOCALAI_P2P_TOKEN="$TOKEN"
fi

# Créer le répertoire des modèles s'il n'existe pas
mkdir -p /share/localai/models

# Nettoyer les fichiers de configuration corrompus (contenant du code shell)
find /share/localai/models -name "*.yaml" -exec grep -l "#!/usr/bin/with-contenv bashio" {} \; | xargs rm -f 2>/dev/null || true

exec /build/local-ai run --models-path=/share/localai/models --gpu-layers="$GPU_LAYERS"