From 270fd9cb076cea3e35105c0e8ff0ae443056a92e Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Tue, 14 Oct 2025 20:13:56 +0800 Subject: [PATCH] fix: discorrect entity reference --- api/services/workflow_service.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/api/services/workflow_service.py b/api/services/workflow_service.py index 8dc1f64045..a5e3581215 100644 --- a/api/services/workflow_service.py +++ b/api/services/workflow_service.py @@ -23,6 +23,8 @@ from core.workflow.nodes import NodeType from core.workflow.nodes.base.node import Node from core.workflow.nodes.node_mapping import LATEST_VERSION, NODE_TYPE_CLASSES_MAPPING from core.workflow.nodes.start.entities import StartNodeData +from core.workflow.nodes.trigger_plugin.entities import PluginTriggerData +from core.workflow.nodes.trigger_schedule.entities import TriggerScheduleNodeData from core.workflow.nodes.trigger_webhook.entities import WebhookData from core.workflow.system_variable import SystemVariable from core.workflow.workflow_entry import WorkflowEntry @@ -38,7 +40,6 @@ from models.workflow import Workflow, WorkflowNodeExecutionModel, WorkflowNodeEx from repositories.factory import DifyAPIRepositoryFactory from services.enterprise.plugin_manager_service import PluginCredentialType from services.errors.app import IsDraftWorkflowError, WorkflowHashNotEqualError -from services.workflow.entities import PluginTriggerData, ScheduleTriggerData from services.workflow.workflow_converter import WorkflowConverter from .errors.workflow_service import DraftWorkflowDeletionError, WorkflowInUseError @@ -637,7 +638,7 @@ class WorkflowService: elif node_type == NodeType.TRIGGER_PLUGIN: start_data = PluginTriggerData.model_validate(node_data) elif node_type == NodeType.TRIGGER_SCHEDULE: - start_data = ScheduleTriggerData.model_validate(node_data) + start_data = TriggerScheduleNodeData.model_validate(node_data) else: start_data = StartNodeData.model_validate(node_data) user_inputs = _rebuild_file_for_user_inputs_in_start_node(