diff --git a/apps/application/flow/step_node/__init__.py b/apps/application/flow/step_node/__init__.py
index b089f15ed..c323a46c9 100644
--- a/apps/application/flow/step_node/__init__.py
+++ b/apps/application/flow/step_node/__init__.py
@@ -50,13 +50,9 @@ node_list = [BaseStartStepNode, BaseChatNode, BaseSearchKnowledgeNode, BaseSearc
              BaseIntentNode, BaseLoopNode, BaseLoopStartStepNode,
              BaseLoopContinueNode,
              BaseLoopBreakNode, BaseVariableSplittingNode, BaseParameterExtractionNode, BaseVariableAggregationNode,
-<<<<<<< Updated upstream
-             BaseDataSourceLocalNode,BaseDataSourceWebNode,BaseKnowledgeWriteNode]
-=======
-             BaseDataSourceLocalNode, BaseDataSourceWebNode]
+             BaseDataSourceLocalNode, BaseDataSourceWebNode, BaseKnowledgeWriteNode]
 
 node_map = {n.type: {w: n for w in n.support} for n in node_list}
->>>>>>> Stashed changes
 
 
 def get_node(node_type, workflow_model):
diff --git a/apps/application/flow/step_node/knowledge_write_node/i_knowledge_write_node.py b/apps/application/flow/step_node/knowledge_write_node/i_knowledge_write_node.py
index eadb0becb..fec3cebad 100644
--- a/apps/application/flow/step_node/knowledge_write_node/i_knowledge_write_node.py
+++ b/apps/application/flow/step_node/knowledge_write_node/i_knowledge_write_node.py
@@ -11,23 +11,20 @@ from typing import Type
 from django.utils.translation import gettext_lazy as _
 from rest_framework import serializers
 
+from application.flow.common import WorkflowMode
 from application.flow.i_step_node import INode, NodeResult
 
 
-
 class KnowledgeWriteNodeParamSerializer(serializers.Serializer):
     paragraph_list = serializers.ListField(required=True, label=_("Paragraph list"))
     chunk_length = serializers.CharField(required=True, label=_("Child chunk length"))
 
 
-
-
 class IKnowledgeWriteNode(INode):
 
     def get_node_params_serializer_class(self) -> Type[serializers.Serializer]:
         return KnowledgeWriteNodeParamSerializer
 
-
     def _run(self):
         return self.execute(**self.node_params_serializer.data, **self.flow_params_serializer.data)
 
@@ -35,4 +32,4 @@ class IKnowledgeWriteNode(INode):
         pass
 
     type = 'knowledge-write-node'
-
+    support = [WorkflowMode.KNOWLEDGE]