From 28fb1db71596487ddbc71bc6e17826355223ca90 Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Wed, 3 Jul 2024 17:49:32 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E8=A2=AB=E5=88=A0=E9=99=A4=E6=97=B6,=E8=8A=82=E7=82=B9?= =?UTF-8?q?=E6=8A=A5=E9=94=99=E6=8F=90=E7=A4=BA=20(#701)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../flow/step_node/ai_chat_step_node/impl/base_chat_node.py | 2 ++ .../flow/step_node/question_node/impl/base_question_node.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py index 48df8a463..4443d851b 100644 --- a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py +++ b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py @@ -126,6 +126,8 @@ class BaseChatNode(IChatNode): def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id, chat_record_id, **kwargs) -> NodeResult: model = QuerySet(Model).filter(id=model_id).first() + if model is None: + raise Exception("模型不存在") chat_model = ModelProvideConstants[model.provider].value.get_model(model.model_type, model.model_name, json.loads( rsa_long_decrypt(model.credential)), 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 65fc52c32..d8257b138 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 @@ -126,6 +126,8 @@ class BaseQuestionNode(IQuestionNode): def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id, chat_record_id, **kwargs) -> NodeResult: model = QuerySet(Model).filter(id=model_id).first() + if model is None: + raise Exception("模型不存在") chat_model = ModelProvideConstants[model.provider].value.get_model(model.model_type, model.model_name, json.loads( rsa_long_decrypt(model.credential)),