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

update server.knowledge_base.utils

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 server.utils import torch_gc
from configs.model_config import KB_ROOT_PATH from configs.model_config import KB_ROOT_PATH
from langchain.vectorstores import FAISS from langchain.vectorstores import FAISS
from langchain.schema import Document
from langchain.embeddings.huggingface import HuggingFaceEmbeddings from langchain.embeddings.huggingface import HuggingFaceEmbeddings
from configs.model_config import (CACHED_VS_NUM, VECTOR_SEARCH_TOP_K, from configs.model_config import (CACHED_VS_NUM, VECTOR_SEARCH_TOP_K,
embedding_model_dict, EMBEDDING_MODEL, EMBEDDING_DEVICE) embedding_model_dict, EMBEDDING_MODEL, EMBEDDING_DEVICE)