fix chroma embeddings when an already seen file is uplaoded

Author: lfoppiano

document_qa/document_qa_engine.py

@@ -205,11 +205,16 @@ def create_memory_embeddings(self, pdf_path, doc_id=None, chunk_size=500, perc_o
         if doc_id:
             hash = doc_id
         else:
-
             hash = metadata[0]['hash']
 
         if hash not in self.embeddings_dict.keys():
-            self.embeddings_dict[hash] = Chroma.from_texts(texts, embedding=self.embedding_function, metadatas=metadata, collection_name=hash)
+            self.embeddings_dict[hash] = Chroma.from_texts(texts, embedding=self.embedding_function, metadatas=metadata,
+                                                           collection_name=hash)
+        else:
+            self.embeddings_dict[hash].delete(ids=self.embeddings_dict[hash].get()['ids'])
+            self.embeddings_dict[hash] = Chroma.from_texts(texts, embedding=self.embedding_function, metadatas=metadata,
+                                                           collection_name=hash)
+
 
         self.embeddings_root_path = None