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

fix another regression

Browse files

(cherry picked from commit 9b528b77741b898b52560d2564fff0d1d6fada1c)

Files changed (1) hide show
  1. document_qa/document_qa_engine.py +3 -3
document_qa/document_qa_engine.py CHANGED
@@ -253,14 +253,14 @@ class DocumentQAEngine:
253
  else:
254
  hash = metadata[0]['hash']
255
 
256
- if hash not in self.embeddings_dict.keys() and (
257
- 'documents' in self.embeddings_dict[hash].get() and len(self.embeddings_dict[hash].get()['documents']) == 0):
258
  self.embeddings_dict[hash] = Chroma.from_texts(texts,
259
  embedding=self.embedding_function,
260
  metadatas=metadata,
261
  collection_name=hash)
262
  else:
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,
 
253
  else:
254
  hash = metadata[0]['hash']
255
 
256
+ if hash not in self.embeddings_dict.keys():
 
257
  self.embeddings_dict[hash] = Chroma.from_texts(texts,
258
  embedding=self.embedding_function,
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,