DH-4778 fix migration script (#211)

Dataherald · May 7, 2024 · c2a2cb0 · c2a2cb0
1 parent 6ec3d97
commit c2a2cb0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/apps/ai/server/database/migrations/sprint_60/DH-4778_version_history_refactor.py b/apps/ai/server/database/migrations/sprint_60/DH-4778_version_history_refactor.py
@@ -36,8 +36,10 @@
 
             if query_response:
                 question_id = ObjectId(query_response["question_id"])
-                updated_by = ObjectId(doc["updated_by"])
-
+                if "updated_by" in doc:
+                    updated_by = ObjectId(doc["updated_by"])
+                else:
+                    updated_by = None
                 # add new field question id, change updated_by to ObjectId
                 data_store[query_collection].update_one(
                     {"_id": doc["_id"]},