weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update MyFAISS.py

This commit is contained in:
imClumsyPanda 2023-08-06 16:49:52 +08:00
parent 0ad63aee63
commit d7039d05a1
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vectorstores/MyFAISS.py
View File

@ -129,8 +129,10 @@ class MyFAISS(FAISS, VectorStore):
if len(ids) == 0: if len(ids) == 0:
return f"docs delete fail" return f"docs delete fail"
else: else:
_reversed_index = {v: k for k, v in self.index_to_docstore_id.items()}
index_to_delete = [_reversed_index[i] for i in ids]
# 从 self.index 中删除对应id # 从 self.index 中删除对应id
self.index.remove_ids(np.array(ids, dtype=np.int64)) self.index.remove_ids(np.array(index_to_delete, dtype=np.int64))
for id in ids: for id in ids:
index = list(self.index_to_docstore_id.keys())[list(self.index_to_docstore_id.values()).index(id)] index = list(self.index_to_docstore_id.keys())[list(self.index_to_docstore_id.values()).index(id)]
self.index_to_docstore_id.pop(index) self.index_to_docstore_id.pop(index)