From e99b382b16a7f547f8a696a6131712b8cc3c4306 Mon Sep 17 00:00:00 2001
From: Pratik Narola <pratik.narola@tether.io>
Date: Sat, 23 May 2026 19:57:41 +0530
Subject: [PATCH] fix: restore_test.sh uses upload endpoint and tightens
 source-file glob

file:// snapshot recovery is disabled by default in recent Qdrant
(returns 403 on /collections/.../snapshots/recover with a file:// URL).
Switched to POST /snapshots/upload with multipart form-data which
doesn't need an allowlist.

Also tightened the find -name glob from "${SOURCE_COLLECTION}_*" to
"${SOURCE_COLLECTION}_[0-9]*" so a source named "mem0_v3" does not
accidentally match "mem0_v3_entities_*" files in the same dir.
---
 scripts/restore_test.sh | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

diff --git a/scripts/restore_test.sh b/scripts/restore_test.sh
index a2c7d02..4aaa183 100755
--- a/scripts/restore_test.sh
+++ b/scripts/restore_test.sh
@@ -26,11 +26,13 @@ TEST_COLLECTION="${TEST_COLLECTION:-mem0_v3_restore_test}"
 
 log() { printf '[%s] %s\n' "$(date -u +%FT%TZ)" "$*"; }
 
-# Pick the most recently-modified backup file matching SOURCE_COLLECTION_*.
-latest="$(find "$BACKUP_DIR" -type f -name "${SOURCE_COLLECTION}_*" -printf '%T@ %p\n' 2>/dev/null \
+# Pick the most recent backup for SOURCE_COLLECTION. The "_[0-9]*" suffix
+# anchors on the timestamp digit so a collection named "mem0_v3" does NOT
+# also match "mem0_v3_entities_*" files in the same directory.
+latest="$(find "$BACKUP_DIR" -type f -name "${SOURCE_COLLECTION}_[0-9]*" -printf '%T@ %p\n' 2>/dev/null \
   | sort -nr | head -1 | cut -d' ' -f2-)"
 if [ -z "$latest" ]; then
-  log "ERROR: no backup found under $BACKUP_DIR matching ${SOURCE_COLLECTION}_*"
+  log "ERROR: no backup found under $BACKUP_DIR matching ${SOURCE_COLLECTION}_[0-9]*"
   exit 1
 fi
 log "latest backup: $latest"
@@ -50,11 +52,14 @@ snap_basename="$(basename "$latest")"
 log "copying snapshot into container: /tmp/$snap_basename"
 docker cp "$latest" "$QDRANT_CONTAINER:/tmp/$snap_basename"
 
-log "restoring into $TEST_COLLECTION"
-docker exec "$QDRANT_CONTAINER" curl -fsS -X PUT \
-  "http://localhost:6333/collections/$TEST_COLLECTION/snapshots/recover" \
-  -H "Content-Type: application/json" \
-  -d "{\"location\":\"file:///tmp/$snap_basename\",\"priority\":\"snapshot\"}" \
+# Use the upload endpoint (multipart) rather than recover-from-URL —
+# file:// recovery is disabled by default in recent Qdrant (returns 403),
+# and upload doesn't need an allowlist.
+log "restoring into $TEST_COLLECTION via /snapshots/upload"
+docker exec "$QDRANT_CONTAINER" curl -fsS -X POST \
+  "http://localhost:6333/collections/$TEST_COLLECTION/snapshots/upload?priority=snapshot" \
+  -H "Content-Type: multipart/form-data" \
+  -F "snapshot=@/tmp/$snap_basename" \
   >/dev/null
 
 restored_count=$(docker exec "$QDRANT_CONTAINER" curl -fsS -X POST \