RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-08 16:10:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update streamlit-aggrid; fix knowledge file status error (#4179)

Browse Source
This commit is contained in:
liunux4odoo 2024-06-13 11:00:44 +08:00 committed by GitHub
parent 7bec36e5b6
commit af536a3309
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libs/chatchat-server/chatchat/server/knowledge_base/kb_cache/faiss_cache.py
View File

@ -128,7 +128,7 @@ class KBFaissPool(_FaissPool):
except Exception as e: except Exception as e:
if locked: # we don't know exception raised before or after atomic.release if locked: # we don't know exception raised before or after atomic.release
self.atomic.release() self.atomic.release()
logger.error(e) logger.error(e, exc_info=True)
raise RuntimeError(f"向量库 {kb_name} 加载失败。") raise RuntimeError(f"向量库 {kb_name} 加载失败。")
return self.get((kb_name, vector_name)) return self.get((kb_name, vector_name))

8
libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py
View File

@ -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_folder"] = doc_details["in_folder"].replace(True, "✓").replace(False, "×")
doc_details["in_db"] = doc_details["in_db"].replace(True, "✓").replace(False, "×") doc_details["in_db"] = doc_details["in_db"].replace(True, "✓").replace(False, "×")
print(doc_details)
gb = config_aggrid( gb = config_aggrid(
doc_details, doc_details,
{ {
@ -217,8 +218,8 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None):
("docs_count", "文档数量"): {}, ("docs_count", "文档数量"): {},
("text_splitter", "分词器"): {}, ("text_splitter", "分词器"): {},
# ("create_time", "创建时间"): {}, # ("create_time", "创建时间"): {},
("in_folder", "源文件"): {"cellRenderer": cell_renderer}, ("in_folder", "源文件"): {},
("in_db", "向量库"): {"cellRenderer": cell_renderer}, ("in_db", "向量库"): {},
}, },
"multiple", "multiple",
) )
@ -236,7 +237,7 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None):
) )
selected_rows = doc_grid.get("selected_rows", []) selected_rows = doc_grid.get("selected_rows", [])
selected_rows = selected_rows.to_dict("records")
cols = st.columns(4) cols = st.columns(4)
file_name, file_path = file_exists(kb, selected_rows) file_name, file_path = file_exists(kb, selected_rows)
if file_path: if file_path:
@ -330,6 +331,7 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None):
if selected_rows: if selected_rows:
file_name = selected_rows[0]["file_name"] file_name = selected_rows[0]["file_name"]
docs = api.search_kb_docs(knowledge_base_name=selected_kb, file_name=file_name) docs = api.search_kb_docs(knowledge_base_name=selected_kb, file_name=file_name)
data = [ data = [
{"seq": i + 1, "id": x["id"], "page_content": x["page_content"], "source": x["metadata"].get("source"), {"seq": i + 1, "id": x["id"], "page_content": x["page_content"], "source": x["metadata"].get("source"),
"type": x["type"], "type": x["type"],

2
libs/chatchat-server/pyproject.toml
View File

@ -57,7 +57,7 @@ streamlit-option-menu = "0.3.12"
streamlit-antd-components = "0.3.1" streamlit-antd-components = "0.3.1"
streamlit-chatbox = "1.1.12.post2" streamlit-chatbox = "1.1.12.post2"
streamlit-modal = "0.1.0" streamlit-modal = "0.1.0"
streamlit-aggrid = "0.3.4.post3" streamlit-aggrid = "1.0.5"
streamlit-extras = "0.4.2" streamlit-extras = "0.4.2"
xinference_client = { version = "^0.11.1", optional = true } xinference_client = { version = "^0.11.1", optional = true }
zhipuai = { version = "^2.1.0", optional = true } zhipuai = { version = "^2.1.0", optional = true }