Skip to content

Commit c59c110

Browse files
committed
fix: Add model id check
1 parent 0ba4db2 commit c59c110

File tree

3 files changed

+8
-0
lines changed

3 files changed

+8
-0
lines changed

apps/application/flow/step_node/intent_node/impl/base_intent_node.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,8 @@ def execute(self, model_id, dialogue_number, history_chat_record, user_input, br
6363
if reference_data and isinstance(reference_data, dict):
6464
model_id = reference_data.get('model_id', model_id)
6565
model_params_setting = reference_data.get('model_params_setting')
66+
if not model_id:
67+
raise Exception(_('Model is not allowed to be empty'))
6668

6769
# 设置默认模型参数
6870
if model_params_setting is None and model_id:

apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,10 @@ def save_context(self, details, workflow_manage):
9494
def execute(self, input_variable, variable_list, model_params_setting, model_id, **kwargs) -> NodeResult:
9595
input_variable = str(input_variable)
9696
self.context['request'] = input_variable
97+
98+
if not model_id:
99+
raise Exception(_('Model is not allowed to be empty'))
100+
97101
if model_params_setting is None and model_id:
98102
model_params_setting = get_default_model_params_setting(model_id)
99103
workspace_id = self.workflow_manage.get_body().get('workspace_id')

apps/application/flow/step_node/question_node/impl/base_question_node.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,8 @@ def execute(self, model_id, system, prompt, dialogue_number, history_chat_record
9494
if reference_data and isinstance(reference_data, dict):
9595
model_id = reference_data.get('model_id', model_id)
9696
model_params_setting = reference_data.get('model_params_setting')
97+
if not model_id:
98+
raise Exception(_('Model is not allowed to be empty'))
9799

98100
if model_params_setting is None and model_id:
99101
model_params_setting = get_default_model_params_setting(model_id)

0 commit comments

Comments
 (0)