RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-02 12:46:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix faiss_cache bug

Browse Source
This commit is contained in:
imClumsyPanda 2024-06-09 22:51:48 +08:00
parent 6019af4f33
commit b110fcd01b
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libs/chatchat-server/chatchat/server/knowledge_base/kb_cache/faiss_cache.py
View File

@ -110,7 +110,8 @@ class KBFaissPool(_FaissPool):
if os.path.isfile(os.path.join(vs_path, "index.faiss")): if os.path.isfile(os.path.join(vs_path, "index.faiss")):
embeddings = get_Embeddings(embed_model=embed_model) embeddings = get_Embeddings(embed_model=embed_model)
vector_store = FAISS.load_local(vs_path, embeddings, normalize_L2=True) vector_store = FAISS.load_local(vs_path, embeddings, normalize_L2=True,
allow_dangerous_deserialization=True)
elif create: elif create:
# create an empty vector store # create an empty vector store
if not os.path.exists(vs_path): if not os.path.exists(vs_path):