diff --git a/apps/application/flow/step_node/intent_node/impl/base_intent_node.py b/apps/application/flow/step_node/intent_node/impl/base_intent_node.py index cdf8a769abf..64822aa9a20 100644 --- a/apps/application/flow/step_node/intent_node/impl/base_intent_node.py +++ b/apps/application/flow/step_node/intent_node/impl/base_intent_node.py @@ -63,6 +63,8 @@ def execute(self, model_id, dialogue_number, history_chat_record, user_input, br if reference_data and isinstance(reference_data, dict): model_id = reference_data.get('model_id', model_id) model_params_setting = reference_data.get('model_params_setting') + if not model_id: + raise Exception(_('Model is not allowed to be empty')) # 设置默认模型参数 if model_params_setting is None and model_id: diff --git a/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py b/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py index c23e0512e8b..ae13f7c6c77 100644 --- a/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py +++ b/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py @@ -94,6 +94,10 @@ def save_context(self, details, workflow_manage): def execute(self, input_variable, variable_list, model_params_setting, model_id, **kwargs) -> NodeResult: input_variable = str(input_variable) self.context['request'] = input_variable + + if not model_id: + raise Exception(_('Model is not allowed to be empty')) + if model_params_setting is None and model_id: model_params_setting = get_default_model_params_setting(model_id) workspace_id = self.workflow_manage.get_body().get('workspace_id') diff --git a/apps/application/flow/step_node/question_node/impl/base_question_node.py b/apps/application/flow/step_node/question_node/impl/base_question_node.py index 2a83493a534..3f622c5f2dc 100644 --- a/apps/application/flow/step_node/question_node/impl/base_question_node.py +++ b/apps/application/flow/step_node/question_node/impl/base_question_node.py @@ -94,6 +94,8 @@ def execute(self, model_id, system, prompt, dialogue_number, history_chat_record if reference_data and isinstance(reference_data, dict): model_id = reference_data.get('model_id', model_id) model_params_setting = reference_data.get('model_params_setting') + if not model_id: + raise Exception(_('Model is not allowed to be empty')) if model_params_setting is None and model_id: model_params_setting = get_default_model_params_setting(model_id)