From 888e13e198ae7c1058bb3ddce197d3adee36ba91 Mon Sep 17 00:00:00 2001 From: Alex Date: Mon, 9 Sep 2024 13:01:58 +0100 Subject: [PATCH] feat: mongo vector migrate script --- scripts/migrate_to_v1_vectorstore.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/scripts/migrate_to_v1_vectorstore.py b/scripts/migrate_to_v1_vectorstore.py index 5255d222..3a5a82f0 100644 --- a/scripts/migrate_to_v1_vectorstore.py +++ b/scripts/migrate_to_v1_vectorstore.py @@ -31,5 +31,20 @@ def migrate_faiss_to_v1_vectorstore(): client.close() +def migrate_mongo_atlas_vector_to_v1_vectorstore(): + client = pymongo.MongoClient("mongodb+srv://:@/?retryWrites=true&w=majority") + db = client["docsgpt"] + vectors_collection = db["vectors"] + + for vector in vectors_collection.find(): + if "location" in vector: + del vector["location"] + if "retriever" not in vector: + vector["retriever"] = "classic" + vector["remote_data"] = None + vectors_collection.update_one({"_id": vector["_id"]}, {"$set": vector}) + + client.close() + migrate_faiss_to_v1_vectorstore() migrate_to_v1_vectorstore_mongo() \ No newline at end of file