44

localai-p2p-worker/run.sh

@@ -1,14 +1,25 @@
-#!/usr/bin/with-contenv bashio
+#!/bin/bash
 
-# Les variables d'environnement sont automatiquement disponibles depuis les options
 echo "Starting LocalAI P2P Worker"
-echo "Master Token: $(bashio::config 'master_token')"
-echo "GPU Layers: $(bashio::config 'gpu_layers')"
-echo "Models Path: $(bashio::config 'models_path')"
+
+# Lire la configuration Home Assistant
+if [ -f /data/options.json ]; then
+    TOKEN="$(bashio::config 'master_token' 2>/dev/null || echo '')"
+    MODELS_PATH="$(bashio::config 'models_path' 2>/dev/null || echo '/share/localai/models')"
+    THREADS="$(bashio::config 'threads' 2>/dev/null || echo '8')"
+    
+    echo "Master Token: $TOKEN"
+    echo "Models Path: $MODELS_PATH"
+    echo "Threads: $THREADS"
+else
+    echo "No Home Assistant config found, using defaults"
+    TOKEN=""
+    MODELS_PATH="/share/localai/models"
+    THREADS="8"
+fi
 
 # Configurer le token P2P si fourni
-TOKEN="$(bashio::config 'master_token')"
-if [ -n "$TOKEN" ] && [ "$TOKEN" != "null" ]; then
+if [ -n "$TOKEN" ] && [ "$TOKEN" != "null" ] && [ "$TOKEN" != "" ]; then
     export LOCALAI_P2P_TOKEN="$TOKEN"
     export P2P_TOKEN="$TOKEN"
     echo "P2P Token configured"
@@ -17,19 +28,18 @@ else
 fi
 
 # Créer le répertoire des modèles
-MODELS_PATH="$(bashio::config 'models_path')"
 mkdir -p "$MODELS_PATH"
 
 # Nettoyer les fichiers corrompus
 find "$MODELS_PATH" -name "*.yaml" -exec grep -l "#!/usr/bin/with-contenv bashio\|mapping values are not allowed in this context" {} \; 2>/dev/null | xargs rm -f 2>/dev/null || true
 
-# Configurer les threads
-export THREADS="$(bashio::config 'threads' || echo '8')"
+# Configurer les variables d'environnement
+export THREADS="$THREADS"
 export OMP_NUM_THREADS="$THREADS"
 
 # Lancer LocalAI
 echo "Starting LocalAI with models path: $MODELS_PATH"
-exec /usr/local/bin/local-ai run \
+exec /build/local-ai run \
     --models-path="$MODELS_PATH" \
     --threads="$THREADS" \
     --address="0.0.0.0:8080" \