From 9e8a4c8a71d89b34e50ecaf7d8b15675210014fb Mon Sep 17 00:00:00 2001 From: -LAN- Date: Sat, 14 Mar 2026 18:16:51 +0800 Subject: [PATCH] Keep dify_graph node base generic --- api/dify_graph/nodes/base/node.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/api/dify_graph/nodes/base/node.py b/api/dify_graph/nodes/base/node.py index eb00f48d6c..e944851d21 100644 --- a/api/dify_graph/nodes/base/node.py +++ b/api/dify_graph/nodes/base/node.py @@ -179,8 +179,7 @@ class Node(Generic[NodeDataT]): # Skip base class itself if cls is Node: return - # Only register production node implementations defined under - # dify_graph.nodes.* or core.workflow.nodes.*. + # Only register production node implementations defined under dify_graph.nodes.*. # This prevents test helper subclasses from polluting the global registry and # accidentally overriding real node types (e.g., a test Answer node). module_name = getattr(cls, "__module__", "") @@ -188,7 +187,7 @@ class Node(Generic[NodeDataT]): node_type = cls.node_type version = cls.version() bucket = Node._registry.setdefault(node_type, {}) - if module_name.startswith(("dify_graph.nodes.", "core.workflow.nodes.")): + if module_name.startswith("dify_graph.nodes."): # Production node definitions take precedence and may override bucket[version] = cls # type: ignore[index] else: @@ -380,10 +379,6 @@ class Node(Generic[NodeDataT]): start_event.provider_id = f"{plugin_id}/{provider_name}" start_event.provider_type = getattr(self.node_data, "provider_type", "") - if self.node_type == NodeType.TRIGGER_PLUGIN: - start_event.provider_id = getattr(self.node_data, "provider_id", "") - start_event.provider_type = getattr(self.node_data, "provider_type", "") - from dify_graph.nodes.agent.agent_node import AgentNode from dify_graph.nodes.agent.entities import AgentNodeData