fix: code node dose not work as expected

This commit is contained in:
Yeuoly
2024-03-10 17:55:24 +08:00
committed by takatost
parent 2d8497f79b
commit 1e6feadc7e
3 changed files with 16 additions and 18 deletions

View File

@ -101,7 +101,6 @@ class CodeNode(BaseNode):
)
variables[variable] = value
# Run code
try:
result = CodeExecutor.execute_code(
@ -109,15 +108,16 @@ class CodeNode(BaseNode):
code=code,
inputs=variables
)
except CodeExecutionException as e:
# Transform result
result = self._transform_result(result, node_data.outputs)
except (CodeExecutionException, ValueError) as e:
return NodeRunResult(
status=WorkflowNodeExecutionStatus.FAILED,
inputs=variables,
error=str(e)
)
# Transform result
result = self._transform_result(result, node_data.outputs)
return NodeRunResult(
status=WorkflowNodeExecutionStatus.SUCCEEDED,
inputs=variables,