Update entrypoint.sh

entrypoint.sh

@@ -3,6 +3,7 @@ set -e
 
 SNAPSHOT_DIR="/data/qdrant_snapshots"
 QDRANT_URL="http://localhost:6333"
+API_KEY="${QDRANT__SERVICE__API_KEY}"
 
 # Start Qdrant in background
 /qdrant/qdrant &
@@ -10,7 +11,7 @@ QDRANT_PID=$!
 
 # Wait for Qdrant to be ready
 echo "Waiting for Qdrant to start..."
-until wget -qO- "$QDRANT_URL/readyz" > /dev/null 2>&1; do
+until wget -qO- --header="api-key: $API_KEY" "$QDRANT_URL/readyz" > /dev/null 2>&1; do
     sleep 1
 done
 echo "Qdrant ready."
@@ -26,7 +27,6 @@ for COLLECTION_DIR in "$SNAPSHOT_DIR"/*/; do
     # Find snapshot file (not checksum)
     SNAPSHOT=""
     for f in "$COLLECTION_DIR"*.snapshot; do
-        # Skip checksum files
         case "$f" in
             *.checksum) continue ;;
         esac
@@ -40,7 +40,9 @@ for COLLECTION_DIR in "$SNAPSHOT_DIR"/*/; do
     echo "Found snapshot: $SNAPSHOT"
 
     # Check if collection already exists
-    HTTP_STATUS=$(wget -qO- "$QDRANT_URL/collections/$COLLECTION" 2>/dev/null | grep -o '"status":"[^"]*"' | head -1)
+    HTTP_STATUS=$(wget -qO- \
+        --header="api-key: $API_KEY" \
+        "$QDRANT_URL/collections/$COLLECTION" 2>/dev/null | grep -o '"status":"[^"]*"' | head -1)
     echo "Collection $COLLECTION status response: $HTTP_STATUS"
 
     if echo "$HTTP_STATUS" | grep -q '"status":"green"'; then
@@ -53,6 +55,7 @@ for COLLECTION_DIR in "$SNAPSHOT_DIR"/*/; do
     RESPONSE=$(wget -qO- \
         --method=PUT \
         --header="Content-Type: application/json" \
+        --header="api-key: $API_KEY" \
         --body-data="{\"location\":\"file://$SNAPSHOT\"}" \
         "$QDRANT_URL/collections/$COLLECTION/snapshots/recover" 2>&1)
     echo "Recovery response: $RESPONSE"
@@ -61,7 +64,9 @@ for COLLECTION_DIR in "$SNAPSHOT_DIR"/*/; do
     echo "Waiting for '$COLLECTION' to become green..."
     for i in $(seq 1 60); do
         sleep 5
-        STATUS=$(wget -qO- "$QDRANT_URL/collections/$COLLECTION" 2>/dev/null | grep -o '"status":"[^"]*"' | head -1)
+        STATUS=$(wget -qO- \
+            --header="api-key: $API_KEY" \
+            "$QDRANT_URL/collections/$COLLECTION" 2>/dev/null | grep -o '"status":"[^"]*"' | head -1)
         echo "  Attempt $i: $STATUS"
         if echo "$STATUS" | grep -q '"status":"green"'; then
             echo "  '$COLLECTION' is green!"