diff --git a/libs/chatchat-server/chatchat/server/knowledge_base/kb_cache/faiss_cache.py b/libs/chatchat-server/chatchat/server/knowledge_base/kb_cache/faiss_cache.py index 36d21162..477eb1b7 100644 --- a/libs/chatchat-server/chatchat/server/knowledge_base/kb_cache/faiss_cache.py +++ b/libs/chatchat-server/chatchat/server/knowledge_base/kb_cache/faiss_cache.py @@ -128,7 +128,7 @@ class KBFaissPool(_FaissPool): except Exception as e: if locked: # we don't know exception raised before or after atomic.release self.atomic.release() - logger.error(e) + logger.error(e, exc_info=True) raise RuntimeError(f"向量库 {kb_name} 加载失败。") return self.get((kb_name, vector_name)) diff --git a/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py b/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py index 03995573..66654e9b 100644 --- a/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py +++ b/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py @@ -206,6 +206,7 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): ]] doc_details["in_folder"] = doc_details["in_folder"].replace(True, "✓").replace(False, "×") doc_details["in_db"] = doc_details["in_db"].replace(True, "✓").replace(False, "×") + print(doc_details) gb = config_aggrid( doc_details, { @@ -217,8 +218,8 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): ("docs_count", "文档数量"): {}, ("text_splitter", "分词器"): {}, # ("create_time", "创建时间"): {}, - ("in_folder", "源文件"): {"cellRenderer": cell_renderer}, - ("in_db", "向量库"): {"cellRenderer": cell_renderer}, + ("in_folder", "源文件"): {}, + ("in_db", "向量库"): {}, }, "multiple", ) @@ -236,7 +237,7 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): ) selected_rows = doc_grid.get("selected_rows", []) - + selected_rows = selected_rows.to_dict("records") cols = st.columns(4) file_name, file_path = file_exists(kb, selected_rows) if file_path: @@ -330,6 +331,7 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): if selected_rows: file_name = selected_rows[0]["file_name"] docs = api.search_kb_docs(knowledge_base_name=selected_kb, file_name=file_name) + data = [ {"seq": i + 1, "id": x["id"], "page_content": x["page_content"], "source": x["metadata"].get("source"), "type": x["type"], diff --git a/libs/chatchat-server/pyproject.toml b/libs/chatchat-server/pyproject.toml index 30d98a96..087276d4 100644 --- a/libs/chatchat-server/pyproject.toml +++ b/libs/chatchat-server/pyproject.toml @@ -57,7 +57,7 @@ streamlit-option-menu = "0.3.12" streamlit-antd-components = "0.3.1" streamlit-chatbox = "1.1.12.post2" streamlit-modal = "0.1.0" -streamlit-aggrid = "0.3.4.post3" +streamlit-aggrid = "1.0.5" streamlit-extras = "0.4.2" xinference_client = { version = "^0.11.1", optional = true } zhipuai = { version = "^2.1.0", optional = true }