From 5a7973ea4202cffaee2789dee528aab016b74020 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Sat, 22 Apr 2023 16:53:04 +0800 Subject: [PATCH] update webui.py --- webui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/webui.py b/webui.py index bf613c9..04ed490 100644 --- a/webui.py +++ b/webui.py @@ -17,10 +17,10 @@ LLM_HISTORY_LEN = 3 def get_vs_list(): if not os.path.exists(VS_ROOT_PATH): return [] - return ["新建知识库"] + os.listdir(VS_ROOT_PATH) + return os.listdir(VS_ROOT_PATH) -vs_list = get_vs_list() +vs_list = ["新建知识库"] + get_vs_list() embedding_model_dict_list = list(embedding_model_dict.keys())