From 6d128f24f0a6e794bc5640cafa4fb63c80f55cf1 Mon Sep 17 00:00:00 2001 From: allseeeeeee <130981946+allseeeeeee@users.noreply.github.com> Date: Mon, 15 Apr 2024 21:29:40 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BB=A5lite=E6=A8=A1=E5=BC=8F=E5=90=AF?= =?UTF-8?q?=E5=8A=A8=E6=97=B6=E4=BB=85=E5=8A=A0=E8=BD=BD=E4=BA=86=E5=9C=A8?= =?UTF-8?q?=E7=BA=BF=E5=B5=8C=E5=85=A5=E6=A8=A1=E5=9E=8B,=E6=9C=AA?= =?UTF-8?q?=E5=8A=A0=E8=BD=BD=E9=85=8D=E7=BD=AE=E7=9A=84=E9=BB=98=E8=AE=A4?= =?UTF-8?q?=E5=B5=8C=E5=85=A5=E6=A8=A1=E5=9E=8B,=E5=AF=BC=E8=87=B4?= =?UTF-8?q?=E5=88=9B=E5=BB=BA=E7=9F=A5=E8=AF=86=E5=BA=93=E6=97=B6=E6=8A=A5?= =?UTF-8?q?=E9=94=99=20'bge-large-zh-v1.5'=20is=20not=20in=20list.=20(#367?= =?UTF-8?q?0)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 修复方式: 如果lite模式启动, 配置的默认嵌入模型也添加到列表 Co-authored-by: LuoJingtian --- webui_pages/knowledge_base/knowledge_base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/webui_pages/knowledge_base/knowledge_base.py b/webui_pages/knowledge_base/knowledge_base.py index 76a97377..44e8a852 100644 --- a/webui_pages/knowledge_base/knowledge_base.py +++ b/webui_pages/knowledge_base/knowledge_base.py @@ -108,6 +108,8 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): if is_lite: embed_models = list_online_embed_models() + if EMBEDDING_MODEL not in embed_models: + embed_models.append(EMBEDDING_MODEL) else: embed_models = list_embed_models() + list_online_embed_models()