weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: delete knowledge base error

This commit is contained in:
liunux4odoo 2023-08-11 23:35:27 +08:00
parent 802ddb7b55
commit 2bb9a7949f
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
webui_pages/knowledge_base/knowledge_base.py
View File

@ -32,6 +32,7 @@ def config_aggrid(
def knowledge_base_page(api: ApiRequest): def knowledge_base_page(api: ApiRequest):
# api = ApiRequest(base_url="http://127.0.0.1:7861", no_remote_api=True) # api = ApiRequest(base_url="http://127.0.0.1:7861", no_remote_api=True)
kb_list = get_kb_details() kb_list = get_kb_details()
kb_names = [x["kb_name"] for x in kb_list]
cols = st.columns([3, 1, 1, 3]) cols = st.columns([3, 1, 1, 3])
new_kb_name = cols[0].text_input( new_kb_name = cols[0].text_input(
@ -46,7 +47,7 @@ def knowledge_base_page(api: ApiRequest):
disabled=not bool(new_kb_name), disabled=not bool(new_kb_name),
use_container_width=True, use_container_width=True,
) and new_kb_name: ) and new_kb_name:
if new_kb_name in kb_list: if new_kb_name in kb_names:
st.error(f"名为 {new_kb_name} 的知识库已经存在！") st.error(f"名为 {new_kb_name} 的知识库已经存在！")
else: else:
ret = api.create_knowledge_base(new_kb_name) ret = api.create_knowledge_base(new_kb_name)
@ -58,7 +59,7 @@ def knowledge_base_page(api: ApiRequest):
disabled=not bool(new_kb_name), disabled=not bool(new_kb_name),
use_container_width=True, use_container_width=True,
) and new_kb_name: ) and new_kb_name:
if new_kb_name in kb_list: if new_kb_name in kb_names:
ret = api.delete_knowledge_base(new_kb_name) ret = api.delete_knowledge_base(new_kb_name)
st.toast(ret["msg"]) st.toast(ret["msg"])
st.experimental_rerun() st.experimental_rerun()

2
webui_pages/utils.py
View File

@ -409,7 +409,7 @@ class ApiRequest:
else: else:
response = self.post( response = self.post(
"/knowledge_base/delete_knowledge_base", "/knowledge_base/delete_knowledge_base",
json={"knowledge_base_name": knowledge_base_name}, json=f"{knowledge_base_name}",
) )
return response.json() return response.json()