lfoppiano commited on
Commit
cef567f
1 Parent(s): 6c39cdf

fix dimensionality error when switching from one model to the other and using the same document

Browse files
Files changed (1) hide show
  1. document_qa/document_qa_engine.py +3 -2
document_qa/document_qa_engine.py CHANGED
@@ -259,8 +259,9 @@ class DocumentQAEngine:
259
  metadatas=metadata,
260
  collection_name=hash)
261
  else:
262
- if 'documents' in self.embeddings_dict[hash].get() and len(self.embeddings_dict[hash].get()['documents']) == 0:
263
- self.embeddings_dict[hash].delete(ids=self.embeddings_dict[hash].get()['ids'])
 
264
  self.embeddings_dict[hash] = Chroma.from_texts(texts,
265
  embedding=self.embedding_function,
266
  metadatas=metadata,
 
259
  metadatas=metadata,
260
  collection_name=hash)
261
  else:
262
+ # if 'documents' in self.embeddings_dict[hash].get() and len(self.embeddings_dict[hash].get()['documents']) == 0:
263
+ # self.embeddings_dict[hash].delete(ids=self.embeddings_dict[hash].get()['ids'])
264
+ self.embeddings_dict[hash].delete_collection()
265
  self.embeddings_dict[hash] = Chroma.from_texts(texts,
266
  embedding=self.embedding_function,
267
  metadatas=metadata,