Hugging Face's logo

Spaces:
thexForce
/
guard
Sleeping

App Files Community
Junaidb commited on
Commit
76bed36
·
verified ·
1 Parent(s): 143eae9

Update databaseengine.py

Browse files
Files changed (1) hide show
  1. databaseengine.py +3 -2
databaseengine.py CHANGED
@@ -47,8 +47,9 @@ class DatabaseEngine():
47
  promptdocument = collection1.find_one({"version":promptversion})
48
 
49
 
50
- contextdocument=collection2.find_one({"version":promptversion})
51
- responsedocument=collection2.find_one({"version":promptversion})
 
52
 
53
  actual_saved_response=responsedocument["response"]
54
  actual_saved_context=contextdocument["context"]
 
47
  promptdocument = collection1.find_one({"version":promptversion})
48
 
49
 
50
+ contextdocument=collection2.find_one({"version":promptversion,"type":"context"})
51
+ responsedocument=collection2.find_one({"version":promptversion,"type":"response"})
52
+
53
 
54
  actual_saved_response=responsedocument["response"]
55
  actual_saved_context=contextdocument["context"]