Преглед на файлове

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

Sidharth Mohanty преди 1 година
родител
ревизия
d065cbf934
променени са 1 файла, в които са добавени 4 реда и са изтрити 0 реда
  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