diff --git a/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py b/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py index aff5100286a..04c6b0ca167 100644 --- a/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py +++ b/apps/application/flow/step_node/parameter_extraction_node/impl/base_parameter_extraction_node.py @@ -88,8 +88,10 @@ def save_context(self, details, workflow_manage): for key, value in details.get('result').items(): self.context[key] = value self.context['result'] = details.get('result') + self.context['request'] = details.get('request') def execute(self, input_variable, variable_list, model_params_setting, model_id, **kwargs) -> NodeResult: + self.context['request'] = input_variable if model_params_setting is None: model_params_setting = get_default_model_params_setting(model_id) workspace_id = self.workflow_manage.get_body().get('workspace_id') @@ -106,6 +108,7 @@ def get_details(self, index: int, **kwargs): "index": index, 'run_time': self.context.get('run_time'), 'type': self.node.type, + 'request': self.context.get('request'), 'result': self.context.get('result'), 'status': self.status, 'err_message': self.err_message diff --git a/apps/application/flow/step_node/variable_splitting_node/impl/base_variable_splitting_node.py b/apps/application/flow/step_node/variable_splitting_node/impl/base_variable_splitting_node.py index a1376446787..981e161ed71 100644 --- a/apps/application/flow/step_node/variable_splitting_node/impl/base_variable_splitting_node.py +++ b/apps/application/flow/step_node/variable_splitting_node/impl/base_variable_splitting_node.py @@ -36,8 +36,10 @@ def save_context(self, details, workflow_manage): for key, value in details.get('result').items(): self.context[key] = value self.context['result'] = details.get('result') + self.context['request'] = details.get('request') def execute(self, input_variable, variable_list, **kwargs) -> NodeResult: + self.context['request'] = input_variable response = {v['field']: smart_jsonpath_search(input_variable, v['expression']) for v in variable_list} return NodeResult({'result': response, **response}, {}) @@ -47,6 +49,7 @@ def get_details(self, index: int, **kwargs): "index": index, 'run_time': self.context.get('run_time'), 'type': self.node.type, + 'request': self.context.get('request'), 'result': self.context.get('result'), 'status': self.status, 'err_message': self.err_message