fix: 以lite模式启动时仅加载了在线嵌入模型,未加载配置的默认嵌入模型,导致创建知识库时报错 'bge-large-zh-v1.5' is not in list. (#3670)

修复方式: 如果lite模式启动, 配置的默认嵌入模型也添加到列表

Co-authored-by: LuoJingtian <luojingtian@siiture.com>
This commit is contained in:
allseeeeeee 2024-04-15 21:29:40 +08:00 committed by GitHub
parent 8721aad18d
commit 6d128f24f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -108,6 +108,8 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None):
if is_lite: if is_lite:
embed_models = list_online_embed_models() embed_models = list_online_embed_models()
if EMBEDDING_MODEL not in embed_models:
embed_models.append(EMBEDDING_MODEL)
else: else:
embed_models = list_embed_models() + list_online_embed_models() embed_models = list_embed_models() + list_online_embed_models()