Fix LocalDocQA object has no attribute llm when adding docs

Signed-off-by: Benjamin Huo <benjamin@kubesphere.io>
This commit is contained in:
Benjamin Huo 2023-07-16 11:39:46 +08:00
parent 1cd4ba8a51
commit 4e6edecd9e
1 changed files with 1 additions and 1 deletions

View File

@ -150,7 +150,7 @@ def get_vector_store(local_doc_qa, vs_id, files, sentence_size, history, one_con
filelist = [] filelist = []
if not os.path.exists(os.path.join(KB_ROOT_PATH, vs_id, "content")): if not os.path.exists(os.path.join(KB_ROOT_PATH, vs_id, "content")):
os.makedirs(os.path.join(KB_ROOT_PATH, vs_id, "content")) os.makedirs(os.path.join(KB_ROOT_PATH, vs_id, "content"))
if local_doc_qa.llm and local_doc_qa.embeddings: if local_doc_qa.llm_model_chain and local_doc_qa.embeddings:
if isinstance(files, list): if isinstance(files, list):
for file in files: for file in files:
filename = os.path.split(file.name)[-1] filename = os.path.split(file.name)[-1]