update class method of KnowledgeBase and KnowledgeFile

This commit is contained in:
imClumsyPanda 2023-08-05 13:48:55 +08:00
parent e8a88f174d
commit 7beeaacea6
1 changed files with 1 additions and 5 deletions

View File

@ -1,9 +1,5 @@
import os.path
from server.knowledge_base.utils import (get_file_path, get_vs_path,
refresh_vs_cache, load_embeddings)
from configs.model_config import (embedding_model_dict, EMBEDDING_MODEL, EMBEDDING_DEVICE)
from langchain.vectorstores import FAISS
from server.utils import torch_gc
from server.knowledge_base.utils import (get_file_path)
from server.knowledge_base import KnowledgeBase