diff --git a/api/core/workflow/nodes/code/code_executor.py b/api/core/workflow/nodes/code/code_executor.py index 3ecd7cfd89..058ee83d46 100644 --- a/api/core/workflow/nodes/code/code_executor.py +++ b/api/core/workflow/nodes/code/code_executor.py @@ -1,8 +1,8 @@ from os import environ from httpx import post -from yarl import URL from pydantic import BaseModel +from yarl import URL from core.workflow.nodes.code.python_template import PythonTemplateTransformer diff --git a/api/core/workflow/nodes/code/code_node.py b/api/core/workflow/nodes/code/code_node.py index dc69fdc84a..32f6776850 100644 --- a/api/core/workflow/nodes/code/code_node.py +++ b/api/core/workflow/nodes/code/code_node.py @@ -1,10 +1,10 @@ -from typing import Optional, cast, Union +from typing import Optional, Union, cast + 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.code.code_executor import CodeExecutionException, CodeExecutor from core.workflow.nodes.code.entities import CodeNodeData -from core.workflow.nodes.code.code_executor import CodeExecutor, CodeExecutionException from models.workflow import WorkflowNodeExecutionStatus MAX_NUMBER = 2 ** 63 - 1 @@ -151,7 +151,7 @@ class CodeNode(BaseNode): :param variable: variable :return: """ - if not isinstance(value, (int, float)): + if not isinstance(value, int | float): raise ValueError(f"{variable} in input form must be a number") if value > MAX_NUMBER or value < MIN_NUMBER: diff --git a/api/core/workflow/nodes/code/entities.py b/api/core/workflow/nodes/code/entities.py index 731b00f8c8..2212d77e2d 100644 --- a/api/core/workflow/nodes/code/entities.py +++ b/api/core/workflow/nodes/code/entities.py @@ -1,8 +1,10 @@ +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 -from pydantic import BaseModel -from typing import Literal, Union class CodeNodeData(BaseNodeData): """