提交 8c469fda 作者: imClumsyPanda

update webui.py

上级 5c349791
...@@ -80,7 +80,7 @@ class LocalDocQA: ...@@ -80,7 +80,7 @@ class LocalDocQA:
vector_store = FAISS.from_documents(docs, self.embeddings) 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")}""" 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) vector_store.save_local(vs_path)
return vs_path return vs_path if len(docs)>0 else None
def get_knowledge_based_answer(self, def get_knowledge_based_answer(self,
query, query,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论