@@ -2,8 +2,10 @@ import AlertCircle from "@/assets/icons/alert-circle.svg?react";
22import { ArtifactNode } from "@/components/dag-visualizer/ArtifactNode" ;
33import { DagControls } from "@/components/dag-visualizer/Controls" ;
44import { ElkEdge } from "@/components/dag-visualizer/elk-edge" ;
5+ import { GlobalSheets } from "@/components/dag-visualizer/global-sheets" ;
56import { PreviewArtifactNode } from "@/components/dag-visualizer/PreviewArtifact" ;
67import { PreviewStepNode } from "@/components/dag-visualizer/PreviewStep" ;
8+ import { SheetProvider } from "@/components/dag-visualizer/sheet-context" ;
79import { StepNode } from "@/components/dag-visualizer/StepNode" ;
810import { EmptyState } from "@/components/EmptyState" ;
911import { Spinner } from "@zenml-io/react-component-library/components/server" ;
@@ -58,33 +60,36 @@ export function DAG() {
5860 ) ;
5961
6062 return (
61- < ReactFlow
62- minZoom = { - 2 }
63- connectOnClick = { false }
64- nodesDraggable = { false }
65- nodesConnectable = { false }
66- edgeTypes = { customEdge }
67- nodeTypes = { customNodes }
68- nodes = { nodes }
69- edges = { edges }
70- edgesFocusable = { false }
71- onlyRenderVisibleElements
72- multiSelectionKeyCode = { null }
73- onNodesChange = { onNodesChange }
74- onEdgesChange = { onEdgesChange }
75- fitView
76- fitViewOptions = {
77- shouldFitView
78- ? undefined
79- : topMostNode
80- ? {
81- nodes : [ topMostNode ] ,
82- maxZoom : 1.2
83- }
84- : undefined
85- }
86- >
87- < DagControls refetch = { handleRefetch } />
88- </ ReactFlow >
63+ < SheetProvider >
64+ < ReactFlow
65+ minZoom = { - 2 }
66+ connectOnClick = { false }
67+ nodesDraggable = { false }
68+ nodesConnectable = { false }
69+ edgeTypes = { customEdge }
70+ nodeTypes = { customNodes }
71+ nodes = { nodes }
72+ edges = { edges }
73+ edgesFocusable = { false }
74+ onlyRenderVisibleElements
75+ multiSelectionKeyCode = { null }
76+ onNodesChange = { onNodesChange }
77+ onEdgesChange = { onEdgesChange }
78+ fitView
79+ fitViewOptions = {
80+ shouldFitView
81+ ? undefined
82+ : topMostNode
83+ ? {
84+ nodes : [ topMostNode ] ,
85+ maxZoom : 1.2
86+ }
87+ : undefined
88+ }
89+ >
90+ < DagControls refetch = { handleRefetch } />
91+ </ ReactFlow >
92+ < GlobalSheets />
93+ </ SheetProvider >
8994 ) ;
9095}
0 commit comments