From d7039d05a1bbc300911271cd9e166d63588bae44 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Sun, 6 Aug 2023 16:49:52 +0800 Subject: [PATCH] update MyFAISS.py --- vectorstores/MyFAISS.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/vectorstores/MyFAISS.py b/vectorstores/MyFAISS.py index 2e575d5..3853857 100644 --- a/vectorstores/MyFAISS.py +++ b/vectorstores/MyFAISS.py @@ -129,8 +129,10 @@ class MyFAISS(FAISS, VectorStore): if len(ids) == 0: return f"docs delete fail" 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.remove_ids(np.array(ids, dtype=np.int64)) + self.index.remove_ids(np.array(index_to_delete, dtype=np.int64)) for id in ids: index = list(self.index_to_docstore_id.keys())[list(self.index_to_docstore_id.values()).index(id)] self.index_to_docstore_id.pop(index)