From b275abe5cda9dcdea985e8f4f05f6d6b1e42d5cd Mon Sep 17 00:00:00 2001 From: shaohuzhang1 Date: Mon, 24 Jun 2024 15:55:55 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=B7=BB=E5=8A=A0=E8=8A=82=E7=82=B9?= =?UTF-8?q?=E5=90=8D=E7=A7=B0?= 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 | 1 + .../flow/step_node/condition_node/impl/base_condition_node.py | 1 + .../flow/step_node/direct_reply_node/impl/base_reply_node.py | 1 + .../flow/step_node/question_node/impl/base_question_node.py | 1 + .../search_dataset_node/impl/base_search_dataset_node.py | 1 + .../flow/step_node/start_node/impl/base_start_node.py | 1 + 6 files changed, 6 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 766c31c80..b33736aa7 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 @@ -170,6 +170,7 @@ class BaseChatNode(IChatNode): def get_details(self, index: int, **kwargs): return { + 'name':self.node.properties.get('stepName'), "index": index, 'run_time': self.context.get('run_time'), 'system': self.node_params.get('system'), diff --git a/apps/application/flow/step_node/condition_node/impl/base_condition_node.py b/apps/application/flow/step_node/condition_node/impl/base_condition_node.py index 94f004b19..3164bb9fe 100644 --- a/apps/application/flow/step_node/condition_node/impl/base_condition_node.py +++ b/apps/application/flow/step_node/condition_node/impl/base_condition_node.py @@ -39,6 +39,7 @@ class BaseConditionNode(IConditionNode): def get_details(self, index: int, **kwargs): return { + 'name': self.node.properties.get('stepName'), "index": index, 'run_time': self.context.get('run_time'), 'branch_id': self.context.get('branch_id'), diff --git a/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py b/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py index b7e77dde0..8b18c7bde 100644 --- a/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py +++ b/apps/application/flow/step_node/direct_reply_node/impl/base_reply_node.py @@ -80,6 +80,7 @@ class BaseReplyNode(IReplyNode): def get_details(self, index: int, **kwargs): return { + 'name': self.node.properties.get('stepName'), "index": index, 'run_time': self.context.get('run_time'), 'type': self.node.type, 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 4163e79b4..f75bf914d 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 @@ -167,6 +167,7 @@ class BaseQuestionNode(IQuestionNode): def get_details(self, index: int, **kwargs): return { + 'name': self.node.properties.get('stepName'), "index": index, 'run_time': self.context.get('run_time'), 'system': self.node_params.get('system'), diff --git a/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py b/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py index 2b233ecb2..0adb03203 100644 --- a/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py +++ b/apps/application/flow/step_node/search_dataset_node/impl/base_search_dataset_node.py @@ -80,6 +80,7 @@ class BaseSearchDatasetNode(ISearchDatasetStepNode): def get_details(self, index: int, **kwargs): return { + 'name': self.node.properties.get('stepName'), "index": index, 'run_time': self.context.get('run_time'), 'paragraph_list': self.context.get('paragraph_list'), diff --git a/apps/application/flow/step_node/start_node/impl/base_start_node.py b/apps/application/flow/step_node/start_node/impl/base_start_node.py index 8ed8a67fd..aedc4f984 100644 --- a/apps/application/flow/step_node/start_node/impl/base_start_node.py +++ b/apps/application/flow/step_node/start_node/impl/base_start_node.py @@ -21,6 +21,7 @@ class BaseStartStepNode(IStarNode): def get_details(self, index: int, **kwargs): return { + 'name': self.node.properties.get('stepName'), "index": index, "question": self.context.get('question'), 'run_time': self.context.get('run_time'),