Parcourir la source

[Bug fix] Qna pair not loading properly (#817)

Sidharth Mohanty il y a 1 an
Parent
commit
d065cbf934
1 fichiers modifiés avec 4 ajouts et 0 suppressions
  1. 4 0
      embedchain/embedchain.py

+ 4 - 0
embedchain/embedchain.py

@@ -350,6 +350,10 @@ class EmbedChain(JSONSerializable):
 
         # get existing ids, and discard doc if any common id exist.
         where = {"url": src}
+        # if data type is qna_pair, we check for question
+        if chunker.data_type == DataType.QNA_PAIR:
+            where = {"question": src[0]}
+
         if self.config.id is not None:
             where["app_id"] = self.config.id