知识库工具的下拉列表改为动态获取，不必重启服务。 (#4126)

* 1、知识库工具的下拉列表改为动态获取，不必重启服务。
chatchat-space · Jun 3, 2024 · a84221c · a84221c
1 parent e941310
commit a84221c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/libs/chatchat-server/chatchat/server/utils.py b/libs/chatchat-server/chatchat/server/utils.py
@@ -724,6 +724,7 @@ def update_search_local_knowledgebase_tool():
     search_local_knowledgebase_tool=tools_registry._TOOLS_REGISTRY.get("search_local_knowledgebase")
     if search_local_knowledgebase_tool:
         search_local_knowledgebase_tool.description = " ".join(re.split(r"\n+\s*", template_knowledge))
+        search_local_knowledgebase_tool.args["database"]["choices"]=[kb.kb_name for kb in kbs]
 
 
 def get_tool(name: str = None) -> Union[BaseTool, Dict[str, BaseTool]]:

diff --git a/libs/chatchat-server/chatchat/webui_pages/dialogue/dialogue.py b/libs/chatchat-server/chatchat/webui_pages/dialogue/dialogue.py
@@ -106,7 +106,7 @@ def clear_conv(name: str = None):
     chat_box.reset_history(name=name or None)
 
 
-@st.cache_data
+# @st.cache_data
 def list_tools(_api: ApiRequest):
     return _api.list_tools()