From 9add46ebbddd8f0e9f052c6b1901a2043ca79e97 Mon Sep 17 00:00:00 2001 From: liunux4odoo Date: Fri, 11 Aug 2023 14:11:34 +0800 Subject: [PATCH] add recreate vector store to knowledge base mamagement ui --- webui_pages/knowledge_base/knowledge_base.py | 21 +++++++++++--------- 1 file changed, 12 insertions(+), 9 deletions(-) diff --git a/webui_pages/knowledge_base/knowledge_base.py b/webui_pages/knowledge_base/knowledge_base.py index 53a86a3..0a5e532 100644 --- a/webui_pages/knowledge_base/knowledge_base.py +++ b/webui_pages/knowledge_base/knowledge_base.py @@ -181,12 +181,15 @@ def knowledge_base_page(api: ApiRequest): st.session_state.files = [] # todo: freezed - # if cols[1].button( - # "重建知识库", - # help="无需上传文件，通过其它方式将文档拷贝到对应知识库content目录下，点击本按钮即可重建知识库。", - # use_container_width=True, - # type="primary", - # ): - # progress = st.progress(0.0, "") - # for d in api.recreate_vector_store(kb): - # progress.progress(d["finished"] / d["total"], f"正在处理： {d['doc']}") + if cols[1].button( + "重建知识库", + help="无需上传文件，通过其它方式将文档拷贝到对应知识库content目录下，点击本按钮即可重建知识库。", + use_container_width=True, + type="primary", + ): + empty = st.empty() + empty.progress(0.0, "") + for d in api.recreate_vector_store(kb): + print(d) + empty.progress(d["finished"] / d["total"], f"正在处理： {d['doc']}") + empty.write("重建完毕")