zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复模型被删除时,节点报错提示 (#701)

Browse Source
This commit is contained in:
shaohuzhang1 2024-07-03 17:49:32 +08:00 committed by GitHub
parent 327b0190d0
commit 28fb1db715
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -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, def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id, chat_record_id,
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
model = QuerySet(Model).filter(id=model_id).first() 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, chat_model = ModelProvideConstants[model.provider].value.get_model(model.model_type, model.model_name,
json.loads( json.loads(
rsa_long_decrypt(model.credential)), rsa_long_decrypt(model.credential)),

2
apps/application/flow/step_node/question_node/impl/base_question_node.py
View File

@ -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, def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id, chat_record_id,
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
model = QuerySet(Model).filter(id=model_id).first() 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, chat_model = ModelProvideConstants[model.provider].value.get_model(model.model_type, model.model_name,
json.loads( json.loads(
rsa_long_decrypt(model.credential)), rsa_long_decrypt(model.credential)),