initial nodes

web/app/components/rag-pipeline/hooks/use-pipeline-refresh-draft.ts

@@ -19,9 +19,14 @@ export const usePipelineRefreshDraft = () => {
     } = workflowStore.getState()
     setIsSyncingWorkflowDraft(true)
     fetchWorkflowDraft(`/rag/pipelines/${pipelineId}/workflows/draft`).then((response) => {
+      const {
+        nodes: processedNodes,
+        viewport,
+      } = processNodesWithoutDataSource(response.graph.nodes, response.graph.viewport)
       handleUpdateWorkflowCanvas({
         ...response.graph,
-        nodes: processNodesWithoutDataSource(response.graph.nodes),
+        nodes: processedNodes,
+        viewport,
       } as WorkflowDataUpdater)
       setSyncWorkflowDraftHash(response.hash)
       setEnvSecrets((response.environment_variables || []).filter(env => env.value_type === 'secret').reduce((acc, env) => {

web/app/components/rag-pipeline/index.tsx

@@ -23,7 +23,7 @@ const RagPipeline = () => {
   } = usePipelineInit()
   const nodesData = useMemo(() => {
     if (data)
-      return processNodesWithoutDataSource(initialNodes(data.graph.nodes, data.graph.edges))
+      return initialNodes(data.graph.nodes, data.graph.edges)
 
     return []
   }, [data])
@@ -41,15 +41,20 @@ const RagPipeline = () => {
       </div>
     )
   }
+
+  const {
+    nodes: processedNodes,
+    viewport,
+  } = processNodesWithoutDataSource(nodesData, data.graph.viewport)
   return (
     <WorkflowWithDefaultContext
       edges={edgesData}
-      nodes={nodesData}
+      nodes={processedNodes}
     >
       <RagPipelineMain
         edges={edgesData}
-        nodes={nodesData}
+        nodes={processedNodes}
-        viewport={data.graph.viewport}
+        viewport={viewport}
       />
     </WorkflowWithDefaultContext>
   )

web/app/components/rag-pipeline/utils/nodes.ts

@@ -1,3 +1,4 @@
+import type { Viewport } from 'reactflow'
 import type { Node } from '@/app/components/workflow/types'
 import { BlockEnum } from '@/app/components/workflow/types'
 import { generateNewNode } from '@/app/components/workflow/utils'
@@ -5,15 +6,23 @@ import { CUSTOM_DATA_SOURCE_EMPTY_NODE } from '@/app/components/workflow/nodes/d
 import { CUSTOM_NOTE_NODE } from '@/app/components/workflow/note-node/constants'
 import { NoteTheme } from '@/app/components/workflow/note-node/types'
 import type { NoteNodeType } from '@/app/components/workflow/note-node/types'
-import { CUSTOM_NODE } from '@/app/components/workflow/constants'
+import {
+  CUSTOM_NODE,
+  NODE_WIDTH_X_OFFSET,
+  START_INITIAL_POSITION,
+} from '@/app/components/workflow/constants'
 
-export const processNodesWithoutDataSource = (nodes: Node[]) => {
+export const processNodesWithoutDataSource = (nodes: Node[], viewport?: Viewport) => {
   let leftNode
   for (let i = 0; i < nodes.length; i++) {
     const node = nodes[i]
 
-    if (node.data.type === BlockEnum.DataSource)
+    if (node.data.type === BlockEnum.DataSource) {
-      return nodes
+      return {
+        nodes,
+        viewport,
+      }
+    }
 
     if (node.type === CUSTOM_NODE && !leftNode)
       leftNode = node
@@ -23,6 +32,8 @@ export const processNodesWithoutDataSource = (nodes: Node[]) => {
   }
 
   if (leftNode) {
+    const startX = leftNode.position.x - NODE_WIDTH_X_OFFSET
+    const startY = leftNode.position.y
     const { newNode } = generateNewNode({
       id: 'data-source-empty',
       type: CUSTOM_DATA_SOURCE_EMPTY_NODE,
@@ -34,8 +45,8 @@ export const processNodesWithoutDataSource = (nodes: Node[]) => {
         _isTempNode: true,
       },
       position: {
-        x: leftNode.position.x - 500,
+        x: startX,
-        y: leftNode.position.y,
+        y: startY,
       },
     })
     const newNoteNode = generateNewNode({
@@ -54,16 +65,26 @@ export const processNodesWithoutDataSource = (nodes: Node[]) => {
         _isTempNode: true,
       } as NoteNodeType,
       position: {
-        x: leftNode.position.x - 500,
+        x: startX,
-        y: leftNode.position.y + 100,
+        y: startY + 100,
       },
     }).newNode
-    return [
+    return {
-      newNode,
+      nodes: [
-      newNoteNode,
+        newNode,
-      ...nodes,
+        newNoteNode,
-    ]
+        ...nodes,
+      ],
+      viewport: {
+        x: (START_INITIAL_POSITION.x - startX) * (viewport?.zoom || 1),
+        y: (START_INITIAL_POSITION.y - startY) * (viewport?.zoom || 1),
+        zoom: viewport?.zoom || 1,
+      },
+    }
   }
 
-  return nodes
+  return {
+    nodes,
+    viewport,
+  }
 }