Skip to content

Commit afddc40

Browse files
authored
fix: 修复知识库检索向量模型不一致时错误提示 (#897)
1 parent 624c82a commit afddc40

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

apps/application/chat_pipeline/step/search_dataset_step/impl/base_search_dataset_step.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def get_model_by_id(_id, user_id):
3535
def get_embedding_id(dataset_id_list):
3636
dataset_list = QuerySet(DataSet).filter(id__in=dataset_id_list)
3737
if len(set([dataset.embedding_mode_id for dataset in dataset_list])) > 1:
38-
raise Exception("知识库未向量模型不一致")
38+
raise Exception("关联知识库的向量模型不一致,无法召回分段。")
3939
if len(dataset_list) == 0:
4040
raise Exception("知识库设置错误,请重新设置知识库")
4141
return dataset_list[0].embedding_mode_id

apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
def get_embedding_id(dataset_id_list):
2626
dataset_list = QuerySet(DataSet).filter(id__in=dataset_id_list)
2727
if len(set([dataset.embedding_mode_id for dataset in dataset_list])) > 1:
28-
raise Exception("知识库未向量模型不一致")
28+
raise Exception("关联知识库的向量模型不一致,无法召回分段。")
2929
if len(dataset_list) == 0:
3030
raise Exception("知识库设置错误,请重新设置知识库")
3131
return dataset_list[0].embedding_mode_id

0 commit comments

Comments
 (0)