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.document import Document
import numpy as np
import copy
class MyFAISS(FAISS, VectorStore):
@ -95,7 +96,8 @@ class MyFAISS(FAISS, VectorStore):
for id in id_seq:
if id == id_seq[0]:
_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:
_id0 = self.index_to_docstore_id[id]
doc0 = self.docstore.search(_id0)