From aaa98c76d5baae3674484da8d6ef49b580292652 Mon Sep 17 00:00:00 2001 From: takatost Date: Wed, 26 Jun 2024 23:56:30 +0800 Subject: [PATCH] optimize --- .../workflow/entities/workflow_runtime_state_entities.py | 5 +---- api/core/workflow/graph.py | 3 --- 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/api/core/workflow/entities/workflow_runtime_state_entities.py b/api/core/workflow/entities/workflow_runtime_state_entities.py index ffe4e8f109..39141c52e7 100644 --- a/api/core/workflow/entities/workflow_runtime_state_entities.py +++ b/api/core/workflow/entities/workflow_runtime_state_entities.py @@ -8,7 +8,7 @@ from core.app.entities.app_invoke_entities import InvokeFrom from core.workflow.entities.node_entities import NodeRunResult from core.workflow.entities.variable_pool import VariablePool from core.workflow.graph import Graph, GraphNode -from core.workflow.nodes.base_node import BaseNode, UserFrom +from core.workflow.nodes.base_node import UserFrom from models.workflow import WorkflowNodeExecutionStatus, WorkflowType @@ -26,9 +26,6 @@ class RuntimeNode(BaseModel): graph_node: GraphNode """graph node""" - node_instance: BaseNode - """node instance""" - node_run_result: Optional[NodeRunResult] = None """node run result""" diff --git a/api/core/workflow/graph.py b/api/core/workflow/graph.py index 25482e10c7..df49199d15 100644 --- a/api/core/workflow/graph.py +++ b/api/core/workflow/graph.py @@ -14,9 +14,6 @@ class GraphNode(BaseModel): children_node_ids: list[str] = [] """children node ids""" - source_handle: Optional[str] = None - """current node source handle from the previous node result""" - run_condition_callback: Optional[Callable] = None """condition function check if the node can be executed"""