diff --git a/ui/src/workflow/common/data.ts b/ui/src/workflow/common/data.ts
index 013945ffb..1059fc6ea 100644
--- a/ui/src/workflow/common/data.ts
+++ b/ui/src/workflow/common/data.ts
@@ -847,6 +847,10 @@ export const applicationLoopMenuNodes = [
   },
 ]
 export const knowledgeLoopMenuNodes = [
+  {
+    label: t('views.tool.dataSource.title'),
+    list: [dataSourceLocalNode, dataSourceWebNode],
+  },
   {
     label: t('workflow.nodes.classify.aiCapability'),
     list: [
@@ -864,7 +868,7 @@ export const knowledgeLoopMenuNodes = [
   },
   {
     label: t('views.knowledge.title'),
-    list: [rerankerNode, documentExtractNode],
+    list: [documentSplitNode, knowledgeWriteNode, documentExtractNode],
   },
   {
     label: t('workflow.nodes.classify.businessLogic'),