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

update MyFAISS.py and cli_demo.py

This commit is contained in:
imClumsyPanda 2023-08-04 23:26:42 +08:00
parent 00cb7bf598
commit 2eb35291b2
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vectorstores/MyFAISS.py
View File

@ -129,12 +129,12 @@ class MyFAISS(FAISS, VectorStore):
if len(ids) == 0: if len(ids) == 0:
return f"docs delete fail" return f"docs delete fail"
else: else:
# 从 self.index 中删除对应id
self.index.remove_ids(np.array(ids, 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)
self.docstore._dict.pop(id) self.docstore._dict.pop(id)
# TODO: 从 self.index 中删除对应id
# self.index.reset()
self.save_local(vs_path) self.save_local(vs_path)
return f"docs delete success" return f"docs delete success"
except Exception as e: except Exception as e: