lfoppiano commited on
Commit
ba399dc
1 Parent(s): 153c004

fix condition to establish when we need to cleanup the storage

Browse files
Files changed (1) hide show
  1. document_qa/document_qa_engine.py +2 -1
document_qa/document_qa_engine.py CHANGED
@@ -251,7 +251,8 @@ class DocumentQAEngine:
251
  else:
252
  hash = metadata[0]['hash']
253
 
254
- if hash not in self.embeddings_dict.keys():
 
255
  self.embeddings_dict[hash] = Chroma.from_texts(texts,
256
  embedding=self.embedding_function,
257
  metadatas=metadata,
 
251
  else:
252
  hash = metadata[0]['hash']
253
 
254
+ if hash not in self.embeddings_dict.keys() and (
255
+ 'documents' in self.embeddings_dict.get() and len(self.embeddings_dict.get()['documents']) == 0):
256
  self.embeddings_dict[hash] = Chroma.from_texts(texts,
257
  embedding=self.embedding_function,
258
  metadatas=metadata,