Spaces:
Sleeping
Sleeping
File size: 1,783 Bytes
68af3c5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | """Migrate local Qdrant collection to Qdrant Cloud. Run once before deploying."""
import asyncio, os, sys
COLLECTION = "knowledge_base"
CLOUD_URL = os.getenv("QDRANT_CLOUD_URL", "")
CLOUD_KEY = os.getenv("QDRANT_CLOUD_API_KEY", "")
if not CLOUD_URL or not CLOUD_KEY:
print("Set QDRANT_CLOUD_URL and QDRANT_CLOUD_API_KEY env vars first")
sys.exit(1)
async def migrate():
from qdrant_client import AsyncQdrantClient
from qdrant_client.models import PointStruct
src = AsyncQdrantClient(host="localhost", port=6333)
dst = AsyncQdrantClient(url=CLOUD_URL, api_key=CLOUD_KEY)
info = await src.get_collection(COLLECTION)
print(f"Source collection: {info.points_count} points")
try:
await dst.delete_collection(COLLECTION)
except Exception:
pass
# Create with both dense and sparse vector configs
await dst.create_collection(
COLLECTION,
vectors_config=info.config.params.vectors,
sparse_vectors_config=info.config.params.sparse_vectors,
)
print("Created cloud collection (dense + sparse)")
offset = None
total = 0
while True:
records, offset = await src.scroll(
COLLECTION, offset=offset, limit=50,
with_payload=True, with_vectors=True
)
if not records:
break
structs = [
PointStruct(id=r.id, vector=r.vector, payload=r.payload or {})
for r in records
if r.vector is not None
]
if structs:
await dst.upsert(COLLECTION, points=structs)
total += len(structs)
print(f"Migrated {total} points...")
if offset is None:
break
print(f"Done — {total} points migrated to {CLOUD_URL}")
asyncio.run(migrate())
|