Explorar el Código

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

Sidharth Mohanty hace 1 año
padre
commit
d065cbf934
Se han modificado 1 ficheros con 4 adiciones y 0 borrados
  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