diff --git a/api/core/workflow/nodes/template_transform/entities.py b/api/core/workflow/nodes/template_transform/entities.py
index 2d3d35b84c..d9099a8118 100644
--- a/api/core/workflow/nodes/template_transform/entities.py
+++ b/api/core/workflow/nodes/template_transform/entities.py
@@ -1,6 +1,4 @@
-from typing import Literal, Union
 
-from pydantic import BaseModel
 
 from core.workflow.entities.base_node_data_entities import BaseNodeData
 from core.workflow.entities.variable_entities import VariableSelector
diff --git a/api/core/workflow/nodes/template_transform/template_transform_node.py b/api/core/workflow/nodes/template_transform/template_transform_node.py
index 3fb880d926..724b84495c 100644
--- a/api/core/workflow/nodes/template_transform/template_transform_node.py
+++ b/api/core/workflow/nodes/template_transform/template_transform_node.py
@@ -1,9 +1,8 @@
 from typing import Optional, cast
+
 from core.helper.code_executor.code_executor import CodeExecutionException, CodeExecutor
-from core.workflow.entities.base_node_data_entities import BaseNodeData
 from core.workflow.entities.node_entities import NodeRunResult, NodeType
 from core.workflow.entities.variable_pool import VariablePool
-
 from core.workflow.nodes.base_node import BaseNode
 from core.workflow.nodes.template_transform.entities import TemplateTransformNodeData
 from models.workflow import WorkflowNodeExecutionStatus