@bian-womp/spark-workbench 0.3.26 → 0.3.28

package/lib/cjs/index.cjs

@@ -2846,6 +2846,10 @@ function toReactFlow(def, positions, sizes, registry, opts) {
         }));
         const handleLayout = geom.handleLayout;
         const handles = geom.handles;
+        const handleMetadata = {};
+        for (const handleId of Object.keys(inputSource)) {
+            handleMetadata[handleId] = sparkGraph.getInputHandleMetadata(inputSource, handleId);
+        }
         const baseData = {
             typeId: n.typeId,
             params: n.params,
@@ -2864,6 +2868,7 @@ function toReactFlow(def, positions, sizes, registry, opts) {
             inputValues: opts.inputs?.[n.nodeId],
             inputDefaults: opts.inputDefaults?.[n.nodeId],
             outputValues: opts.outputs?.[n.nodeId],
+            handleMetadata,
             status: opts.nodeStatus?.[n.nodeId],
             validation: {
                 inputs: opts.nodeValidation?.inputs?.[n.nodeId] ?? [],
@@ -5987,8 +5992,6 @@ const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
         });
         return () => off();
     }, [wb]);
-    // Memoize customData to prevent unnecessary recomputations
-    const customData = React.useMemo(() => wb.getCustomData(), [wb]);
     const rfData = React.useMemo(() => {
         const out = toReactFlow(wb.def, wb.getPositions(), wb.getSizes(), wb.registry, {
             showValues,
@@ -6006,7 +6009,7 @@ const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
             selectedEdgeIds: new Set(selection.edges),
             getDefaultNodeSize,
             ui,
-            customData,
+            customData: wb.getCustomData(),
         });
         // Retain references for unchanged items
         const stableNodes = retainStabilityById(prevNodesRef.current, out.nodes, isSameNode);
@@ -6112,13 +6115,11 @@ const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
         edgeValidation,
         resolveNodeType,
         selection,
-        customData,
         ui,
         getDefaultNodeSize,
         wb,
         uiVersion,
     ]);
-    const throttled = useThrottledValue(rfData, 100);
     const [menuState, setMenuState] = React.useState(null);
     // Compute the rectangular screen-space bounds of the current selection
     const getSelectionScreenBounds = () => {
@@ -6521,7 +6522,7 @@ const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
         return () => off();
     }, [wb]);
     const { onInit: userOnInit, ...restReactFlowProps } = reactFlowProps || {};
-    return (jsxRuntime.jsxs("div", { className: "w-full h-full relative overflow-hidden", onContextMenu: onContextMenu, children: [jsxRuntime.jsxs(react.ReactFlowProvider, { children: [jsxRuntime.jsxs(react.ReactFlow, { ...restReactFlowProps, nodes: throttled.nodes, edges: throttled.edges, nodeTypes: nodeTypes, edgeTypes: edgeTypes, connectionLineComponent: connectionLineRenderer, selectionOnDrag: true, onInit: (inst) => {
+    return (jsxRuntime.jsxs("div", { className: "w-full h-full relative overflow-hidden", onContextMenu: onContextMenu, children: [jsxRuntime.jsxs(react.ReactFlowProvider, { children: [jsxRuntime.jsxs(react.ReactFlow, { ...restReactFlowProps, nodes: rfData.nodes, edges: rfData.edges, nodeTypes: nodeTypes, edgeTypes: edgeTypes, connectionLineComponent: connectionLineRenderer, selectionOnDrag: true, onInit: (inst) => {
                             rfInstanceRef.current = inst;
                             const savedViewport = wb.getViewport();
                             if (savedViewport) {