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

fix: 修改工作流中模型参数为非必填参数

Browse Source
This commit is contained in:
shaohuzhang1 2024-08-29 15:19:29 +08:00 committed by shaohuzhang1
parent 96ac12ea31
commit 644bb61c4e
4 changed files with 4 additions and 4 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/i_chat_node.py
View File

@ -38,6 +38,6 @@ class IChatNode(INode):
def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id, def execute(self, model_id, system, prompt, dialogue_number, history_chat_record, stream, chat_id,
chat_record_id, chat_record_id,
model_params_setting, model_params_setting=None,
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
pass pass

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

@ -74,7 +74,7 @@ def get_default_model_params_setting(model_id):
class BaseChatNode(IChatNode): 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,
model_params_setting, model_params_setting=None,
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
if model_params_setting is None: if model_params_setting is None:
model_params_setting = get_default_model_params_setting(model_id) model_params_setting = get_default_model_params_setting(model_id)

2
apps/application/flow/step_node/question_node/i_question_node.py
View File

@ -36,6 +36,6 @@ class IQuestionNode(INode):
return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data) return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
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,
model_params_setting, model_params_setting=None,
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
pass pass

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

@ -74,7 +74,7 @@ def get_default_model_params_setting(model_id):
class BaseQuestionNode(IQuestionNode): 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,
model_params_setting, model_params_setting=None,
**kwargs) -> NodeResult: **kwargs) -> NodeResult:
if model_params_setting is None: if model_params_setting is None:
model_params_setting = get_default_model_params_setting(model_id) model_params_setting = get_default_model_params_setting(model_id)