From 8c469fda311a5a49c190e0b84368e3e8c29710c8 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Fri, 14 Apr 2023 22:57:07 +0800 Subject: [PATCH] update webui.py --- chains/local_doc_qa.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chains/local_doc_qa.py b/chains/local_doc_qa.py index bad5e55..1248942 100644 --- a/chains/local_doc_qa.py +++ b/chains/local_doc_qa.py @@ -80,7 +80,7 @@ class LocalDocQA: vector_store = FAISS.from_documents(docs, self.embeddings) vs_path = f"""./vector_store/{os.path.splitext(file)[0]}_FAISS_{datetime.datetime.now().strftime("%Y%m%d_%H%M%S")}""" vector_store.save_local(vs_path) - return vs_path + return vs_path if len(docs)>0 else None def get_knowledge_based_answer(self, query,