fix: add validation to ensure model_id is not empty in various nodes

Author: wxg0103

apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py

@@ -277,6 +277,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 model_id is None or 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)

apps/application/flow/step_node/image_generate_step_node/impl/base_image_generate_node.py

@@ -4,7 +4,7 @@
 
 import requests
 from langchain_core.messages import BaseMessage, HumanMessage, AIMessage
-
+from django.utils.translation import gettext_lazy as _
 from application.flow.common import WorkflowMode
 from application.flow.i_step_node import NodeResult
 from application.flow.step_node.image_generate_step_node.i_image_generate_node import IImageGenerateNode
@@ -37,6 +37,9 @@ def execute(self, model_id, prompt, negative_prompt, dialogue_number, dialogue_t
                 model_id = reference_data.get('model_id', model_id)
                 model_params_setting = reference_data.get('model_params_setting')
 
+        if model_id is None or model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
+
         workspace_id = self.workflow_manage.get_body().get('workspace_id')
         tti_model = get_model_instance_by_model_workspace_id(model_id, workspace_id,
                                                              **(model_params_setting or {}))

apps/application/flow/step_node/image_to_video_step_node/impl/base_image_to_video_node.py

@@ -6,7 +6,7 @@
 import requests
 from django.db.models import QuerySet
 from langchain_core.messages import BaseMessage, HumanMessage, AIMessage
-
+from django.utils.translation import gettext_lazy as _
 from application.flow.common import WorkflowMode
 from application.flow.i_step_node import NodeResult
 from application.flow.step_node.image_to_video_step_node.i_image_to_video_node import IImageToVideoNode
@@ -40,7 +40,8 @@ def execute(self, model_id, prompt, negative_prompt, dialogue_number, dialogue_t
             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 model_id is None or model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
         workspace_id = self.workflow_manage.get_body().get('workspace_id')
         ttv_model = get_model_instance_by_model_workspace_id(model_id, workspace_id,
                                                              **(model_params_setting or {}))

apps/application/flow/step_node/image_understand_step_node/impl/base_image_understand_node.py

@@ -7,7 +7,7 @@
 
 from django.db.models import QuerySet
 from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage, AIMessage
-
+from django.utils.translation import gettext_lazy as _
 from application.flow.i_step_node import NodeResult, INode
 from application.flow.step_node.image_understand_step_node.i_image_understand_node import IImageUnderstandNode
 from application.flow.tools import Reasoning
@@ -153,6 +153,8 @@ def execute(self, model_id, system, prompt, dialogue_number, dialogue_type, hist
                 model_id = reference_data.get('model_id', model_id)
                 model_params_setting = reference_data.get('model_params_setting')
 
+        if model_id is None or model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
         # 处理不正确的参数
         workspace_id = self.workflow_manage.get_body().get('workspace_id')
         image_model = get_model_instance_by_model_workspace_id(model_id, workspace_id,

apps/application/flow/step_node/speech_to_text_step_node/impl/base_speech_to_text_node.py

@@ -33,6 +33,10 @@ def execute(self, stt_model_id, audio, model_params_setting=None, stt_model_id_t
                 stt_model_id = reference_data.get('stt_model_id', reference_data.get('model_id', stt_model_id))
                 model_params_setting = reference_data.get('model_params_setting')
 
+        from django.utils.translation import gettext_lazy as _
+
+        if stt_model_id is None or stt_model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
         workspace_id = self.workflow_manage.get_body().get('workspace_id')
         stt_model = get_model_instance_by_model_workspace_id(stt_model_id, workspace_id, **(model_params_setting or {}))
         audio_list = audio

apps/application/flow/step_node/text_to_speech_step_node/impl/base_text_to_speech_node.py

@@ -56,6 +56,11 @@ def execute(self, tts_model_id,
             if reference_data and isinstance(reference_data, dict):
                 tts_model_id = reference_data.get('tts_model_id', reference_data.get('model_id', tts_model_id))
                 model_params_setting = reference_data.get('model_params_setting')
+
+        from django.utils.translation import gettext_lazy as _
+
+        if tts_model_id is None or tts_model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
         # 分割文本为合理片段
         content = _remove_empty_lines(content)
         content_chunks = [content[i:i + max_length]

apps/application/flow/step_node/text_to_video_step_node/impl/base_text_to_video_node.py

@@ -38,6 +38,10 @@ def execute(self, model_id, prompt, negative_prompt, dialogue_number, dialogue_t
                 model_id = reference_data.get('model_id', model_id)
                 model_params_setting = reference_data.get('model_params_setting')
 
+        from django.utils.translation import gettext_lazy as _
+
+        if model_id is None or model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
         workspace_id = self.workflow_manage.get_body().get('workspace_id')
         ttv_model = get_model_instance_by_model_workspace_id(model_id, workspace_id,
                                                              **(model_params_setting or {}))

apps/application/flow/step_node/video_understand_step_node/impl/base_video_understand_node.py

@@ -148,6 +148,11 @@ def execute(self, model_id, system, prompt, dialogue_number, dialogue_type, hist
                 model_id = reference_data.get('model_id', model_id)
                 model_params_setting = reference_data.get('model_params_setting')
 
+        from django.utils.translation import gettext_lazy as _
+
+        if model_id is None or model_id == '':
+            raise Exception(_('Model is not allowed to be empty'))
+
         workspace_id = self.workflow_manage.get_body().get('workspace_id')
         if model_setting is None:
             model_setting = {'reasoning_content_enable': False, 'reasoning_content_end': '</think>',

apps/locales/en_US/LC_MESSAGES/django.po

@@ -9188,3 +9188,6 @@ msgstr "Batch delete tools"
 msgid "Batch move tools"
 msgstr "Batch move tools"
 
+msgid "Model is not allowed to be empty"
+msgstr ""
+

apps/locales/zh_CN/LC_MESSAGES/django.po

@@ -9311,3 +9311,6 @@ msgstr "批量删除工具"
 msgid "Batch move tools"
 msgstr "批量移动工具"
 
+msgid "Model is not allowed to be empty"
+msgstr "模型不能为空"
+