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

update MyFAISS with method mentioned in https://github.com/imClumsyPanda/langchain-ChatGLM/issues/613#issuecomment-1590460465

This commit is contained in:
imClumsyPanda 2023-06-14 21:20:52 +08:00
parent a55aab498f
commit 58d6a9a098
3 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
img/qr_code_30.jpg
View File

Binary file not shown.
Side by Side

Before

Width:  |  Height:  |  Size: 291 KiB

BIN
img/qr_code_31.jpg
View File

Binary file not shown.
Side by Side

Before

Width:  |  Height:  |  Size: 265 KiB

4
vectorstores/MyFAISS.py
View File

@ -5,6 +5,7 @@ from typing import Any, Callable, List, Tuple, Dict
from langchain.docstore.base import Docstore from langchain.docstore.base import Docstore
from langchain.docstore.document import Document from langchain.docstore.document import Document
import numpy as np import numpy as np
import copy
class MyFAISS(FAISS, VectorStore): class MyFAISS(FAISS, VectorStore):
@ -95,7 +96,8 @@ class MyFAISS(FAISS, VectorStore):
for id in id_seq: for id in id_seq:
if id == id_seq[0]: if id == id_seq[0]:
_id = self.index_to_docstore_id[id] _id = self.index_to_docstore_id[id]
doc = self.docstore.search(_id) # doc = self.docstore.search(_id)
doc = copy.deepcopy(self.docstore.search(_id))
else: else:
_id0 = self.index_to_docstore_id[id] _id0 = self.index_to_docstore_id[id]
doc0 = self.docstore.search(_id0) doc0 = self.docstore.search(_id0)