youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: template transform

Browse Source
This commit is contained in:
Yeuoly
2024-03-11 21:52:49 +08:00
parent 8dc4d122b9
commit a5394fa2ce
4 changed files with 58 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/workflow/nodes/template_transform/template_transform_node.py
View File

@ -7,6 +7,7 @@ from core.workflow.nodes.base_node import BaseNode
from core.workflow.nodes.template_transform.entities import TemplateTransformNodeData
from models.workflow import WorkflowNodeExecutionStatus
MAX_TEMPLATE_TRANSFORM_OUTPUT_LENGTH = 1000
class TemplateTransformNode(BaseNode):
_node_data_cls = TemplateTransformNodeData
@ -48,7 +49,6 @@ class TemplateTransformNode(BaseNode):
)
variables[variable] = value
# Run code
try:
result = CodeExecutor.execute_code(
@ -62,6 +62,13 @@ class TemplateTransformNode(BaseNode):
status=WorkflowNodeExecutionStatus.FAILED,
error=str(e)
)
if len(result['result']) > MAX_TEMPLATE_TRANSFORM_OUTPUT_LENGTH:
return NodeRunResult(
inputs=variables,
status=WorkflowNodeExecutionStatus.FAILED,
error=f"Output length exceeds {MAX_TEMPLATE_TRANSFORM_OUTPUT_LENGTH} characters"
)
return NodeRunResult(
status=WorkflowNodeExecutionStatus.SUCCEEDED,