Update databaseengine.py

databaseengine.py

@@ -25,7 +25,7 @@ class DatabaseEngine():
         
         collection=db1.get_collection(coll1)
         collection.update_one(
-         {"_id": data["prompt"]},
+         {"version": data["promptversion"]},
          {"$set": {"biological_context_alignment":data["biological_context_alignment"]}},
         )
 
@@ -35,11 +35,11 @@ class DatabaseEngine():
         collection1=db1.get_collection(coll1)
         collection2=db2.get_collection(coll2)
         
-        promptdocument = collection1.find_one({"_id":promptversion})
+        promptdocument = collection1.find_one({"version":promptversion})
         
 
-        contextdocument=collection2.find_one({"_id":promptversion})
-        responsedocument=collection2.find_one({"_id":promptversion})
+        contextdocument=collection2.find_one({"version":promptversion})
+        responsedocument=collection2.find_one({"version":promptversion})
 
         
         return {