From f6c6f260dbd3a7eed47e0ad54744af3eebfce4ad Mon Sep 17 00:00:00 2001 From: -LAN- Date: Fri, 18 Jul 2025 01:31:33 +0800 Subject: [PATCH] tests: fix tests Signed-off-by: -LAN- --- .../workflow/nodes/test_parameter_extractor.py | 4 +++- .../workflow/nodes/test_template_transform.py | 1 + api/tests/integration_tests/workflow/nodes/test_tool.py | 4 +++- 3 files changed, 7 insertions(+), 2 deletions(-) diff --git a/api/tests/integration_tests/workflow/nodes/test_parameter_extractor.py b/api/tests/integration_tests/workflow/nodes/test_parameter_extractor.py index dd8466afa6..edd70193a8 100644 --- a/api/tests/integration_tests/workflow/nodes/test_parameter_extractor.py +++ b/api/tests/integration_tests/workflow/nodes/test_parameter_extractor.py @@ -74,13 +74,15 @@ def init_parameter_extractor_node(config: dict): variable_pool.add(["a", "b123", "args1"], 1) variable_pool.add(["a", "b123", "args2"], 2) - return ParameterExtractorNode( + node = ParameterExtractorNode( id=str(uuid.uuid4()), graph_init_params=init_params, graph=graph, graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()), config=config, ) + node.init_node_data(config.get("data", {})) + return node def test_function_calling_parameter_extractor(setup_model_mock): diff --git a/api/tests/integration_tests/workflow/nodes/test_template_transform.py b/api/tests/integration_tests/workflow/nodes/test_template_transform.py index 1f617fc92d..f71a5ee140 100644 --- a/api/tests/integration_tests/workflow/nodes/test_template_transform.py +++ b/api/tests/integration_tests/workflow/nodes/test_template_transform.py @@ -76,6 +76,7 @@ def test_execute_code(setup_code_executor_mock): graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()), config=config, ) + node.init_node_data(config.get("data", {})) # execute node result = node._run() diff --git a/api/tests/integration_tests/workflow/nodes/test_tool.py b/api/tests/integration_tests/workflow/nodes/test_tool.py index 6907e0163e..8476c1f874 100644 --- a/api/tests/integration_tests/workflow/nodes/test_tool.py +++ b/api/tests/integration_tests/workflow/nodes/test_tool.py @@ -50,13 +50,15 @@ def init_tool_node(config: dict): conversation_variables=[], ) - return ToolNode( + node = ToolNode( id=str(uuid.uuid4()), graph_init_params=init_params, graph=graph, graph_runtime_state=GraphRuntimeState(variable_pool=variable_pool, start_at=time.perf_counter()), config=config, ) + node.init_node_data(config.get("data", {})) + return node def test_tool_variable_invoke():