refactor(workflow): remove code node helper imports (#32759)

Co-authored-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
99
2026-03-01 16:31:45 +08:00
committed by GitHub
parent 36fad7256d
commit 00e52796e6
8 changed files with 59 additions and 51 deletions

View File

@ -112,7 +112,6 @@ class MockNodeFactory(DifyNodeFactory):
graph_runtime_state=self.graph_runtime_state,
mock_config=self.mock_config,
code_executor=self._code_executor,
code_providers=self._code_providers,
code_limits=self._code_limits,
)
elif node_type == NodeType.HTTP_REQUEST:

View File

@ -1,7 +1,6 @@
from configs import dify_config
from core.helper.code_executor.code_executor import CodeLanguage
from core.workflow.nodes.code.code_node import CodeNode
from core.workflow.nodes.code.entities import CodeNodeData
from core.workflow.nodes.code.entities import CodeLanguage, CodeNodeData
from core.workflow.nodes.code.exc import (
CodeNodeError,
DepthLimitError,
@ -438,7 +437,7 @@ class TestCodeNodeInitialization:
"outputs": {"x": {"type": "number"}},
}
node.init_node_data(data)
node._node_data = node._hydrate_node_data(data)
assert node._node_data.title == "Test Node"
assert node._node_data.code_language == CodeLanguage.PYTHON3
@ -454,7 +453,7 @@ class TestCodeNodeInitialization:
"outputs": {"x": {"type": "number"}},
}
node.init_node_data(data)
node._node_data = node._hydrate_node_data(data)
assert node._node_data.code_language == CodeLanguage.JAVASCRIPT

View File

@ -1,8 +1,7 @@
import pytest
from pydantic import ValidationError
from core.helper.code_executor.code_executor import CodeLanguage
from core.workflow.nodes.code.entities import CodeNodeData
from core.workflow.nodes.code.entities import CodeLanguage, CodeNodeData
from core.workflow.variables.types import SegmentType