RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-01-28 17:53:33 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update server.knowledge_base.utils

Browse Source
This commit is contained in:
imClumsyPanda 2023-08-04 15:32:42 +08:00
parent 7bfbe18011
commit b4f80ca370
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/knowledge_base/utils.py
View File

@ -3,6 +3,7 @@ from typing import List
from server.utils import torch_gc
from configs.model_config import KB_ROOT_PATH
from langchain.vectorstores import FAISS
from langchain.schema import Document
from langchain.embeddings.huggingface import HuggingFaceEmbeddings
from configs.model_config import (CACHED_VS_NUM, VECTOR_SEARCH_TOP_K,
embedding_model_dict, EMBEDDING_MODEL, EMBEDDING_DEVICE)