diff --git a/libs/chatchat-server/chatchat/server/utils.py b/libs/chatchat-server/chatchat/server/utils.py index a810c26f..314ef0cc 100644 --- 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 index d211097e..cdd40aa6 100644 --- 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()