@bian-womp/spark-workbench 0.2.69 → 0.2.71

package/lib/esm/index.js

@@ -1,4 +1,5 @@
 import { generateId, GraphBuilder, createEngine, StepEngine, PullEngine, BatchedEngine, isTypedOutput, getTypedOutputValue, getTypedOutputTypeId, isInputPrivate, getInputTypeId, createSimpleGraphRegistry, createSimpleGraphDef, createAsyncGraphDef, createAsyncGraphRegistry, createProgressGraphDef, createProgressGraphRegistry, createValidationGraphDef, createValidationGraphRegistry } from '@bian-womp/spark-graph';
+import lod from 'lodash';
 import { RuntimeApiClient } from '@bian-womp/spark-remote';
 import { Position, Handle, useUpdateNodeInternals, useReactFlow, ReactFlowProvider, ReactFlow, Background, BackgroundVariant, MiniMap, Controls } from '@xyflow/react';
 import React, { useCallback, useState, useRef, useEffect, useMemo, createContext, useContext, useImperativeHandle } from 'react';
@@ -122,13 +123,15 @@ class InMemoryWorkbench extends AbstractWorkbench {
     constructor() {
         super(...arguments);
         this.def = { nodes: [], edges: [] };
-        this.positions = {};
         this.listeners = new Map();
+        this.positions = {};
         this.selection = {
             nodes: [],
             edges: [],
         };
         this.viewport = null;
+        this.runtimeState = null;
+        this.historyState = undefined;
         this.copiedData = null;
     }
     setRegistry(registry) {
@@ -208,18 +211,19 @@ class InMemoryWorkbench extends AbstractWorkbench {
                 inputs: options?.inputs,
                 copyOutputsFrom: options?.copyOutputsFrom,
             },
-            dry: options?.dry,
+            ...lod.pick(options, ["dry", "commit", "reason"]),
         });
         this.refreshValidation();
         return id;
     }
-    removeNode(nodeId) {
+    removeNode(nodeId, options) {
         this.def.nodes = this.def.nodes.filter((n) => n.nodeId !== nodeId);
         this.def.edges = this.def.edges.filter((e) => e.source.nodeId !== nodeId && e.target.nodeId !== nodeId);
         delete this.positions[nodeId];
         this.emit("graphChanged", {
             def: this.def,
             change: { type: "removeNode", nodeId },
+            ...options,
         });
         this.refreshValidation();
     }
@@ -234,16 +238,17 @@ class InMemoryWorkbench extends AbstractWorkbench {
         this.emit("graphChanged", {
             def: this.def,
             change: { type: "connect", edgeId: id },
-            dry: options?.dry,
+            ...options,
         });
         this.refreshValidation();
         return id;
     }
-    disconnect(edgeId) {
+    disconnect(edgeId, options) {
         this.def.edges = this.def.edges.filter((e) => e.id !== edgeId);
         this.emit("graphChanged", {
             def: this.def,
             change: { type: "disconnect", edgeId },
+            ...options,
         });
         this.emit("validationChanged", this.validate());
     }
@@ -272,32 +277,26 @@ class InMemoryWorkbench extends AbstractWorkbench {
         });
     }
     // Position and selection APIs for React Flow bridge
-    setPosition(nodeId, pos, opts) {
-        this.positions[nodeId] = pos;
-        this.emit("graphUiChanged", {
-            def: this.def,
-            change: { type: "moveNode", nodeId, pos },
-            commit: !!opts?.commit === true,
-        });
-    }
-    setPositions(map, opts) {
+    setPositions(map, options) {
         this.positions = { ...map };
         this.emit("graphUiChanged", {
             def: this.def,
             change: { type: "moveNodes" },
-            commit: opts?.commit,
+            ...options,
         });
     }
     getPositions() {
         return { ...this.positions };
     }
-    setSelection(sel, opts) {
+    setSelection(sel, options) {
+        if (lod.isEqual(this.selection, sel))
+            return;
         this.selection = { nodes: [...sel.nodes], edges: [...sel.edges] };
         this.emit("selectionChanged", this.selection);
         this.emit("graphUiChanged", {
             def: this.def,
             change: { type: "selection" },
-            commit: opts?.commit,
+            ...options,
         });
     }
     getSelection() {
@@ -309,7 +308,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
     /**
      * Delete all selected nodes and edges.
      */
-    deleteSelection() {
+    deleteSelection(options) {
         const selection = this.getSelection();
         // Delete all selected nodes (this will also remove connected edges)
         for (const nodeId of selection.nodes) {
@@ -320,14 +319,14 @@ class InMemoryWorkbench extends AbstractWorkbench {
             this.disconnect(edgeId);
         }
         // Clear selection
-        this.setSelection({ nodes: [], edges: [] });
+        this.setSelection({ nodes: [], edges: [] }, options);
     }
-    setViewport(viewport, opts) {
+    setViewport(viewport, options) {
         this.viewport = { ...viewport };
         this.emit("graphUiChanged", {
             def: this.def,
             change: { type: "viewport" },
-            commit: opts?.commit,
+            ...options,
         });
     }
     getViewport() {
@@ -369,6 +368,27 @@ class InMemoryWorkbench extends AbstractWorkbench {
             this.viewport = { ...ui.viewport };
         }
     }
+    getRuntimeState() {
+        return this.runtimeState ? { ...this.runtimeState } : null;
+    }
+    setRuntimeState(runtime) {
+        this.runtimeState = runtime ? { ...runtime } : null;
+    }
+    getHistory() {
+        return this.historyState;
+    }
+    setHistory(history) {
+        this.historyState = history;
+    }
+    getNodeRuntimeMetadata(nodeId) {
+        return this.runtimeState?.nodes[nodeId];
+    }
+    updateNodeRuntimeMetadata(nodeId, updater) {
+        const current = this.runtimeState ?? { nodes: {} };
+        const nodeMeta = current.nodes[nodeId] ?? {};
+        const updated = updater({ ...nodeMeta });
+        this.runtimeState = { nodes: { ...current.nodes, [nodeId]: updated } };
+    }
     on(event, handler) {
         if (!this.listeners.has(event))
             this.listeners.set(event, new Set());
@@ -472,7 +492,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
      * Returns the mapping from original node IDs to new node IDs.
      * Uses copyOutputsFrom to copy outputs from original nodes (like duplicate does).
      */
-    pasteCopiedData(data, center) {
+    pasteCopiedData(data, center, options) {
         const nodeIdMap = new Map();
         const edgeIds = [];
         // Add nodes
@@ -512,10 +532,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
             }
         }
         // Select the newly pasted nodes
-        this.setSelection({
-            nodes: Array.from(nodeIdMap.values()),
-            edges: edgeIds,
-        });
+        this.setSelection({ nodes: Array.from(nodeIdMap.values()), edges: edgeIds }, options);
         return { nodeIdMap, edgeIds };
     }
     /**
@@ -677,14 +694,10 @@ class AbstractGraphRunner {
     async redo() {
         return false;
     }
-    async canUndo() {
-        return false;
-    }
-    async canRedo() {
-        return false;
-    }
     // Optional commit support
-    async commit() { }
+    async commit(_reason) {
+        return undefined;
+    }
 }
 
 // Counter for generating readable runner IDs
@@ -1431,10 +1444,11 @@ class RemoteGraphRunner extends AbstractGraphRunner {
         const client = await this.ensureClient();
         await client.setExtData(data);
     }
-    async commit() {
+    async commit(reason) {
         const client = await this.ensureClient();
         try {
-            await client.commit();
+            const history = await client.commit(reason);
+            return history;
         }
         catch (err) {
             console.error("[RemoteGraphRunner] Error committing:", err);
@@ -1459,24 +1473,6 @@ class RemoteGraphRunner extends AbstractGraphRunner {
             return false;
         }
     }
-    async canUndo() {
-        const client = await this.ensureClient();
-        try {
-            return await client.canUndo();
-        }
-        catch {
-            return false;
-        }
-    }
-    async canRedo() {
-        const client = await this.ensureClient();
-        try {
-            return await client.canRedo();
-        }
-        catch {
-            return false;
-        }
-    }
     async snapshotFull() {
         const client = await this.ensureClient();
         try {
@@ -1914,13 +1910,17 @@ function useWorkbenchBridge(wb) {
         wb.connect({
             source: { nodeId: params.source, handle: params.sourceHandle },
             target: { nodeId: params.target, handle: params.targetHandle },
-        });
+        }, { commit: true });
     }, [wb]);
     const onNodesChange = useCallback((changes) => {
         // Apply position updates continuously, but mark commit only on drag end
+        const positions = {};
+        let commit = false;
         changes.forEach((c) => {
             if (c.type === "position" && c.position) {
-                wb.setPosition(c.id, c.position, { commit: !c.dragging });
+                positions[c.id] = c.position;
+                if (!c.dragging)
+                    commit = true;
             }
         });
         // Derive next node selection from change set
@@ -1951,13 +1951,13 @@ function useWorkbenchBridge(wb) {
             }
         }
         if (selectionChanged) {
-            wb.setSelection({
-                nodes: Array.from(nextNodeIds),
-                edges: current.edges,
-            });
+            wb.setSelection({ nodes: Array.from(nextNodeIds), edges: current.edges }, { commit: !(Object.keys(positions).length && commit) });
+        }
+        if (Object.keys(positions).length > 0) {
+            wb.setPositions(positions, { commit });
         }
     }, [wb]);
-    const onEdgesDelete = useCallback((edges) => edges.forEach((e) => wb.disconnect(e.id)), [wb]);
+    const onEdgesDelete = useCallback((edges) => edges.forEach((e, idx) => wb.disconnect(e.id, { commit: idx === edges.length - 1 })), [wb]);
     const onEdgesChange = useCallback((changes) => {
         const current = wb.getSelection();
         const nextEdgeIds = new Set(current.edges);
@@ -1986,15 +1986,11 @@ function useWorkbenchBridge(wb) {
             }
         }
         if (selectionChanged) {
-            wb.setSelection({
-                nodes: current.nodes,
-                edges: Array.from(nextEdgeIds),
-            });
+            wb.setSelection({ nodes: current.nodes, edges: Array.from(nextEdgeIds) }, { commit: true });
         }
     }, [wb]);
     const onNodesDelete = useCallback((nodes) => {
-        for (const n of nodes)
-            wb.removeNode(n.id);
+        nodes.forEach((n, idx) => wb.removeNode(n.id, { commit: idx === nodes.length - 1 }));
     }, [wb]);
     return {
         onConnect,
@@ -2438,6 +2434,7 @@ async function download(wb, runner) {
     try {
         const def = wb.export();
         const uiState = wb.getUIState();
+        const runtimeState = wb.getRuntimeState();
         let snapshot;
         if (runner.isRunning()) {
             const fullSnapshot = await runner.snapshotFull();
@@ -2447,6 +2444,7 @@ async function download(wb, runner) {
                 extData: {
                     ...(fullSnapshot.extData || {}),
                     ui: uiState,
+                    runtime: runtimeState || undefined,
                 },
             };
         }
@@ -2457,7 +2455,7 @@ async function download(wb, runner) {
                 inputs,
                 outputs: {},
                 environment: {},
-                extData: { ui: uiState },
+                extData: { ui: uiState, runtime: runtimeState || undefined },
             };
         }
         downloadJSON(snapshot, `spark-snapshot-${generateTimestamp()}.json`);
@@ -2483,6 +2481,9 @@ async function upload(parsed, wb, runner) {
     if (extData.ui && typeof extData.ui === "object") {
         wb.setUIState(extData.ui);
     }
+    if (extData.runtime && typeof extData.runtime === "object") {
+        wb.setRuntimeState(extData.runtime);
+    }
     if (runner.isRunning()) {
         await runner.applySnapshotFull({
             def,
@@ -2510,6 +2511,18 @@ function useWorkbenchContext() {
     return ctx;
 }
 
+// Helper to compute invalidated status from runtime metadata
+function computeInvalidatedFromMetadata(metadata) {
+    if (!metadata)
+        return true;
+    const { lastSuccessAt, lastInputAt, lastRunAt } = metadata;
+    if (!lastSuccessAt && !lastRunAt)
+        return true;
+    if (!lastInputAt || Object.keys(lastInputAt).length === 0)
+        return false;
+    const maxInputTime = Math.max(...Object.values(lastInputAt));
+    return maxInputTime > (lastSuccessAt ?? lastRunAt ?? 0);
+}
 function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVersion, children, }) {
     const [nodeStatus, setNodeStatus] = useState({});
     const [edgeStatus, setEdgeStatus] = useState({});
@@ -2600,27 +2613,35 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
         }
         return out;
     }, [def, outputsMap, registry]);
-    // Initialize nodes as invalidated by default until first successful run
+    // Initialize nodes and derive invalidated status from persisted metadata
     useEffect(() => {
+        const workbenchRuntimeState = wb.getRuntimeState() ?? { nodes: {} };
         setNodeStatus((prev) => {
             const next = { ...prev };
+            const metadata = workbenchRuntimeState;
             for (const n of def.nodes) {
                 const cur = next[n.nodeId] ?? (next[n.nodeId] = {});
+                const nodeMeta = metadata.nodes[n.nodeId];
                 const updates = {};
                 if (cur.invalidated === undefined) {
-                    updates.invalidated = true;
+                    updates.invalidated = computeInvalidatedFromMetadata(nodeMeta);
                 }
-                // Ensure activeRunIds is always initialized as an array
                 if (cur.activeRunIds === undefined) {
                     updates.activeRunIds = [];
                 }
+                if (cur.activeRuns === undefined) {
+                    updates.activeRuns = 0;
+                }
+                if (nodeMeta?.lastErrorSummary && cur.lastError === undefined) {
+                    updates.lastError = nodeMeta.lastErrorSummary;
+                }
                 if (Object.keys(updates).length > 0) {
                     next[n.nodeId] = { ...cur, ...updates };
                 }
             }
             return next;
         });
-    }, [def]);
+    }, [def, wb]);
     // Auto layout (simple layered layout)
     const runAutoLayout = useCallback(() => {
         const cur = wb.export();
@@ -2687,10 +2708,31 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             }
             curX += maxWidth + H_GAP;
         }
-        wb.setPositions(pos, { commit: true });
+        wb.setPositions(pos, { commit: true, reason: "auto-layout" });
     }, [wb, registry, overrides?.getDefaultNodeSize]);
     const updateEdgeType = useCallback((edgeId, typeId) => wb.updateEdgeType(edgeId, typeId), [wb]);
     const triggerExternal = useCallback((nodeId, event) => runner.triggerExternal(nodeId, event), [runner]);
+    // Helper to save runtime metadata to extData.runtime and workbench state
+    const saveRuntimeMetadata = useCallback(async () => {
+        try {
+            const current = wb.getRuntimeState() ?? { nodes: {} };
+            const metadata = { nodes: { ...current.nodes } };
+            // Clean up metadata for nodes that no longer exist
+            const nodeIds = new Set(def.nodes.map((n) => n.nodeId));
+            for (const nodeId of Object.keys(metadata.nodes)) {
+                if (!nodeIds.has(nodeId)) {
+                    delete metadata.nodes[nodeId];
+                }
+            }
+            // Save cleaned metadata to workbench state
+            wb.setRuntimeState(metadata);
+            // Save to extData.runtime via runner (no snapshotFull)
+            await runner.setExtData?.({ runtime: metadata });
+        }
+        catch (err) {
+            console.warn("[WorkbenchContext] Failed to save runtime metadata:", err);
+        }
+    }, [wb, def, runner]);
     // Subscribe to runner/workbench events
     useEffect(() => {
         const add = (source, type) => (payload) => setEvents((prev) => {
@@ -2733,9 +2775,18 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                 wb.refreshValidation();
         };
         const offRunnerValue = runner.on("value", (e) => {
+            const now = Date.now();
             if (e?.io === "input") {
-                const nodeId = e?.nodeId;
-                const handle = e?.handle;
+                const nodeId = e.nodeId;
+                const handle = e.handle;
+                // Track input timestamp in workbench runtime state
+                wb.updateNodeRuntimeMetadata(nodeId, (nodeMeta) => ({
+                    ...nodeMeta,
+                    lastInputAt: {
+                        ...(nodeMeta.lastInputAt ?? {}),
+                        [handle]: now,
+                    },
+                }));
                 setNodeStatus((s) => ({
                     ...s,
                     [nodeId]: { ...s[nodeId], invalidated: true },
@@ -2743,6 +2794,18 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                 // Clear validation errors for this input when a valid value is set
                 setInputValidationErrors((prev) => prev.filter((err) => !(err.nodeId === nodeId && err.handle === handle)));
             }
+            else if (e?.io === "output") {
+                const nodeId = e.nodeId;
+                const handle = e.handle;
+                // Track output timestamp in workbench runtime state
+                wb.updateNodeRuntimeMetadata(nodeId, (nodeMeta) => ({
+                    ...nodeMeta,
+                    lastOutputAt: {
+                        ...(nodeMeta.lastOutputAt ?? {}),
+                        [handle]: now,
+                    },
+                }));
+            }
             return add("runner", "value")(e);
         });
         const offRunnerError = runner.on("error", (e) => {
@@ -2761,6 +2824,24 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             else if (nodeError.kind === "node-run" && nodeError.nodeId) {
                 const nodeId = nodeError.nodeId;
                 const runId = nodeError.runId;
+                const now = Date.now();
+                // Track error timestamp and summary in workbench runtime state
+                const err = nodeError.err;
+                let errorSummary;
+                if (err && typeof err === "object") {
+                    const message = err.message || String(err);
+                    const code = err.code || err.statusCode;
+                    errorSummary = {
+                        message: typeof message === "string" ? message : String(message),
+                        code: typeof code === "number" ? code : undefined,
+                    };
+                }
+                wb.updateNodeRuntimeMetadata(nodeId, (nodeMeta) => ({
+                    ...nodeMeta,
+                    lastErrorAt: now,
+                    lastRunAt: now,
+                    ...(errorSummary ? { lastErrorSummary: errorSummary } : {}),
+                }));
                 setNodeStatus((s) => ({
                     ...s,
                     [nodeId]: {
@@ -2815,6 +2896,30 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             // If resolvedHandles are included in the event, use them directly (more efficient)
             if (e?.resolvedHandles && Object.keys(e.resolvedHandles).length > 0) {
                 applyResolvedHandles(e.resolvedHandles);
+                // Mark nodes whose handles changed as invalid
+                const affectedNodeIds = Object.keys(e.resolvedHandles);
+                if (affectedNodeIds.length > 0) {
+                    setNodeStatus((prev) => {
+                        const next = { ...prev };
+                        for (const id of affectedNodeIds) {
+                            const cur = next[id] ?? (next[id] = { activeRuns: 0, activeRunIds: [] });
+                            next[id] = { ...cur, invalidated: true };
+                        }
+                        return next;
+                    });
+                }
+            }
+            // For broader invalidations (e.g. registry-changed, graph-updated), mark all nodes invalid
+            if (e?.reason === "registry-changed" || e?.reason === "graph-updated") {
+                setNodeStatus((prev) => {
+                    const next = { ...prev };
+                    for (const n of def.nodes) {
+                        const cur = next[n.nodeId] ??
+                            (next[n.nodeId] = { activeRuns: 0, activeRunIds: [] });
+                        next[n.nodeId] = { ...cur, invalidated: true };
+                    }
+                    return next;
+                });
             }
             return add("runner", "invalidate")(e);
         });
@@ -2824,6 +2929,12 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             if (s.kind === "node-start") {
                 const id = s.nodeId;
                 const runId = s.runId;
+                const now = Date.now();
+                // Track run timestamp in workbench runtime state
+                wb.updateNodeRuntimeMetadata(id, (nodeMeta) => ({
+                    ...nodeMeta,
+                    lastRunAt: now,
+                }));
                 // Validate runId is a non-empty string
                 const isValidRunId = runId && typeof runId === "string" && runId.length > 0;
                 if (!isValidRunId) {
@@ -2846,7 +2957,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                     };
                 });
                 // Start fallback animation window
-                setFallbackStarts((prev) => ({ ...prev, [id]: Date.now() }));
+                setFallbackStarts((prev) => ({ ...prev, [id]: now }));
             }
             else if (s.kind === "node-progress") {
                 const id = s.nodeId;
@@ -2861,8 +2972,21 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             else if (s.kind === "node-done") {
                 const id = s.nodeId;
                 const runId = s.runId;
+                const now = Date.now();
                 // Validate runId is a non-empty string
                 const isValidRunId = runId && typeof runId === "string" && runId.length > 0;
+                const hadError = !!(isValidRunId && errorRunsRef.current[id]?.[runId]);
+                // Track success timestamp if no error in workbench runtime state
+                if (!hadError) {
+                    wb.updateNodeRuntimeMetadata(id, (nodeMeta) => {
+                        const updated = { ...nodeMeta, lastSuccessAt: now };
+                        // Clear error summary on success
+                        if (updated.lastErrorSummary) {
+                            delete updated.lastErrorSummary;
+                        }
+                        return updated;
+                    });
+                }
                 setNodeStatus((prev) => {
                     const current = prev[id]?.activeRuns ?? 0;
                     const currentRunIds = prev[id]?.activeRunIds ?? [];
@@ -2874,7 +2998,6 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                     const nextRunIds = isValidRunId
                         ? currentRunIds.filter((rid) => rid !== runId)
                         : currentRunIds;
-                    const hadError = !!(isValidRunId && errorRunsRef.current[id]?.[runId]);
                     const keepProgress = hadError || nextActive > 0;
                     // Clear error flag for this runId
                     if (isValidRunId && errorRunsRef.current[id]?.[runId]) {
@@ -2943,11 +3066,40 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             }
         });
         const offWbGraphChangedForUpdate = wb.on("graphChanged", async (event) => {
+            // Build detailed reason from change type
+            let reason = "graph-changed";
+            if (event.change) {
+                const changeType = event.change.type;
+                if (changeType === "addNode") {
+                    reason = "add-node";
+                }
+                else if (changeType === "removeNode") {
+                    reason = "remove-node";
+                }
+                else if (changeType === "connect") {
+                    reason = "connect-edge";
+                }
+                else if (changeType === "disconnect") {
+                    reason = "disconnect-edge";
+                }
+                else if (changeType === "updateParams") {
+                    reason = "update-node-params";
+                }
+                else if (changeType === "updateEdgeType") {
+                    reason = "update-edge-type";
+                }
+            }
             if (!runner.isRunning()) {
-                // If runner not running, commit immediately (no update needed)
-                await runner.commit().catch((err) => {
-                    console.error("[WorkbenchContext] Error committing:", err);
-                });
+                if (event.commit) {
+                    await saveRuntimeMetadata();
+                    const history = await runner.commit(reason).catch((err) => {
+                        console.error("[WorkbenchContext] Error committing:", err);
+                        return undefined;
+                    });
+                    if (history) {
+                        wb.setHistory(history);
+                    }
+                }
                 return;
             }
             try {
@@ -2972,10 +3124,18 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                 else {
                     await runner.update(event.def, { dry: event.dry });
                 }
-                // Wait for update to complete, then commit
-                await runner.commit().catch((err) => {
-                    console.error("[WorkbenchContext] Error committing after update:", err);
-                });
+                if (event.commit) {
+                    await saveRuntimeMetadata();
+                    const history = await runner
+                        .commit(event.reason ?? reason)
+                        .catch((err) => {
+                        console.error("[WorkbenchContext] Error committing after update:", err);
+                        return undefined;
+                    });
+                    if (history) {
+                        wb.setHistory(history);
+                    }
+                }
             }
             catch (err) {
                 console.error("[WorkbenchContext] Error updating graph:", err);
@@ -2985,17 +3145,49 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
         const offWbSelectionChanged = wb.on("selectionChanged", async (sel) => {
             setSelectedNodeId(sel.nodes?.[0]);
             setSelectedEdgeId(sel.edges?.[0]);
-            // Commit on selection change
-            await runner.commit().catch((err) => {
-                console.error("[WorkbenchContext] Error committing selection change:", err);
-            });
+            if (sel.commit) {
+                await saveRuntimeMetadata();
+                const history = await runner
+                    .commit(sel.reason ?? "selection")
+                    .catch((err) => {
+                    console.error("[WorkbenchContext] Error committing selection change:", err);
+                    return undefined;
+                });
+                if (history) {
+                    wb.setHistory(history);
+                }
+            }
         });
         const offWbGraphUiChanged = wb.on("graphUiChanged", async (event) => {
             // Only commit if commit flag is true (e.g., drag end, not during dragging)
             if (event.commit) {
-                await runner.commit().catch((err) => {
+                // Build detailed reason from change type
+                let reason = "ui-changed";
+                if (event.change) {
+                    const changeType = event.change.type;
+                    if (changeType === "moveNode") {
+                        reason = "move-node";
+                    }
+                    else if (changeType === "moveNodes") {
+                        reason = "move-nodes";
+                    }
+                    else if (changeType === "selection") {
+                        reason = "selection";
+                    }
+                    else if (changeType === "viewport") {
+                        reason = "viewport";
+                    }
+                }
+                await saveRuntimeMetadata();
+                const history = await runner
+                    .commit(event.reason ?? reason)
+                    .catch((err) => {
                     console.error("[WorkbenchContext] Error committing UI changes:", err);
+                    return undefined;
                 });
+                if (history) {
+                    wb.setHistory(history);
+                }
             }
         });
         const offWbError = wb.on("error", add("workbench", "error"));
@@ -3016,11 +3208,24 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                 console.error("Failed to handle registry changed event");
             }
         });
-        // Handle transport disconnect: reset runtime status when connection is lost
+        // Handle transport changes: reset runtime status when connection is lost
         const offRunnerTransport = runner.on("transport", (t) => {
             if (t.state === "disconnected") {
                 console.info("[WorkbenchContext] Transport disconnected, resetting node status");
-                setNodeStatus({});
+                // Reinitialize node status with invalidated=true for all nodes
+                setNodeStatus(() => {
+                    const next = {};
+                    const metadata = wb.getRuntimeState() ?? { nodes: {} };
+                    for (const n of def.nodes) {
+                        const nodeMeta = metadata.nodes[n.nodeId];
+                        next[n.nodeId] = {
+                            activeRuns: 0,
+                            activeRunIds: [],
+                            invalidated: computeInvalidatedFromMetadata(nodeMeta),
+                        };
+                    }
+                    return next;
+                });
                 setEdgeStatus({});
                 setFallbackStarts({});
                 errorRunsRef.current = {};
@@ -3209,6 +3414,7 @@ function createNodeContextMenuHandlers(nodeId, wb, runner, registry, outputsMap,
         try {
             const typeId = outputTypesMap?.[nodeId]?.[handleId];
             const raw = outputsMap?.[nodeId]?.[handleId];
+            let newNodeId;
             if (!typeId || raw === undefined)
                 return;
             const unwrap = (v) => isTypedOutput(v) ? getTypedOutputValue(v) : v;
@@ -3234,23 +3440,21 @@ function createNodeContextMenuHandlers(nodeId, wb, runner, registry, outputsMap,
                 const nodeDesc = registry.nodes.get(singleTarget.nodeTypeId);
                 const inType = getInputTypeId(nodeDesc?.inputs, singleTarget.inputHandle);
                 const coerced = await coerceIfNeeded(typeId, inType, unwrap(raw));
-                wb.addNode({
+                newNodeId = wb.addNode({
                     typeId: singleTarget.nodeTypeId,
                     position: { x: pos.x + 180, y: pos.y },
                 }, { inputs: { [singleTarget.inputHandle]: coerced } });
-                return;
             }
-            if (isArray && arrTarget) {
+            else if (isArray && arrTarget) {
                 const nodeDesc = registry.nodes.get(arrTarget.nodeTypeId);
                 const inType = getInputTypeId(nodeDesc?.inputs, arrTarget.inputHandle);
                 const coerced = await coerceIfNeeded(typeId, inType, unwrap(raw));
-                wb.addNode({
+                newNodeId = wb.addNode({
                     typeId: arrTarget.nodeTypeId,
                     position: { x: pos.x + 180, y: pos.y },
                 }, { inputs: { [arrTarget.inputHandle]: coerced } });
-                return;
             }
-            if (isArray && elemTarget) {
+            else if (isArray && elemTarget) {
                 const nodeDesc = registry.nodes.get(elemTarget.nodeTypeId);
                 const inType = getInputTypeId(nodeDesc?.inputs, elemTarget.inputHandle);
                 const src = unwrap(raw);
@@ -3262,19 +3466,21 @@ function createNodeContextMenuHandlers(nodeId, wb, runner, registry, outputsMap,
                 for (let idx = 0; idx < coercedItems.length; idx++) {
                     const col = idx % COLS;
                     const row = Math.floor(idx / COLS);
-                    wb.addNode({
+                    newNodeId = wb.addNode({
                         typeId: elemTarget.nodeTypeId,
                         position: { x: pos.x + (col + 1) * DX, y: pos.y + row * DY },
                     }, { inputs: { [elemTarget.inputHandle]: coercedItems[idx] } });
                 }
-                return;
+            }
+            if (newNodeId) {
+                wb.setSelection({ nodes: [newNodeId], edges: [] }, { commit: true, reason: "bake" });
             }
         }
         catch { }
     };
     return {
         onDelete: () => {
-            wb.removeNode(nodeId);
+            wb.removeNode(nodeId, { commit: true });
             onClose();
         },
         onDuplicate: async () => {
@@ -3299,10 +3505,7 @@ function createNodeContextMenuHandlers(nodeId, wb, runner, registry, outputsMap,
                 dry: true,
             });
             // Select the newly duplicated node
-            wb.setSelection({
-                nodes: [newNodeId],
-                edges: [],
-            });
+            wb.setSelection({ nodes: [newNodeId], edges: [] }, { commit: true, reason: "duplicate" });
             onClose();
         },
         onDuplicateWithEdges: async () => {
@@ -3335,10 +3538,9 @@ function createNodeContextMenuHandlers(nodeId, wb, runner, registry, outputsMap,
                 }, { dry: true });
             }
             // Select the newly duplicated node and edges
-            wb.setSelection({
-                nodes: [newNodeId],
-                edges: [],
-            });
+            if (newNodeId) {
+                wb.setSelection({ nodes: [newNodeId], edges: [] }, { commit: true, reason: "duplicate-with-edges" });
+            }
             onClose();
         },
         onRunPull: async () => {
@@ -3412,7 +3614,7 @@ function createSelectionContextMenuHandlers(wb, onClose, getDefaultNodeSize, onC
                 onClose();
             },
         onDelete: () => {
-            wb.deleteSelection();
+            wb.deleteSelection({ commit: true, reason: "delete-selection" });
             onClose();
         },
         onClose,
@@ -3421,7 +3623,7 @@ function createSelectionContextMenuHandlers(wb, onClose, getDefaultNodeSize, onC
 /**
  * Creates base default context menu handlers.
  */
-function createDefaultContextMenuHandlers(onAddNode, onClose, onPaste, runner, getCopiedData, clearCopiedData) {
+function createDefaultContextMenuHandlers(onAddNode, onClose, onPaste, runner, getCopiedData, clearCopiedData, history) {
     // Wrap paste handler to clear storage after paste
     const wrappedOnPaste = onPaste && getCopiedData && clearCopiedData
         ? (position) => {
@@ -3429,16 +3631,17 @@ function createDefaultContextMenuHandlers(onAddNode, onClose, onPaste, runner, g
             clearCopiedData();
         }
         : onPaste;
-    // Function to check if paste data exists (called dynamically when menu opens)
     const hasPasteData = getCopiedData ? () => !!getCopiedData() : undefined;
+    const canUndo = history ? history.undoCount > 0 : undefined;
+    const canRedo = history ? history.redoCount > 0 : undefined;
     return {
         onAddNode,
         onPaste: wrappedOnPaste,
         hasPasteData,
         onUndo: runner ? () => runner.undo().then(() => onClose()) : undefined,
         onRedo: runner ? () => runner.redo().then(() => onClose()) : undefined,
-        canUndo: runner ? () => runner.canUndo().catch(() => false) : undefined,
-        canRedo: runner ? () => runner.canRedo().catch(() => false) : undefined,
+        canUndo,
+        canRedo,
         onClose,
     };
 }
@@ -3962,6 +4165,16 @@ function DefaultNodeContent({ data, isConnectable, }) {
                 } })] }));
 }
 
+// Helper to format shortcut for current platform
+function formatShortcut(shortcut) {
+    const isMac = typeof navigator !== "undefined" &&
+        navigator.userAgent.toLowerCase().includes("mac");
+    return shortcut.replace(/⌘\/Ctrl/g, isMac ? "⌘" : "Ctrl");
+}
+function ContextMenuButton({ label, onClick, disabled = false, shortcut, enableKeyboardShortcuts = true, }) {
+    return (jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100 disabled:opacity-50 disabled:cursor-not-allowed flex items-center justify-between", onClick: onClick, disabled: disabled, children: [jsx("span", { children: label }), enableKeyboardShortcuts && shortcut && (jsx("span", { className: "text-gray-400 text-xs ml-4", children: formatShortcut(shortcut) }))] }));
+}
+
 function DefaultContextMenu({ open, clientPos, handlers, registry, nodeIds, enableKeyboardShortcuts = true, keyboardShortcuts = {
     undo: "⌘/Ctrl + Z",
     redo: "⌘/Ctrl + Shift + Z",
@@ -3969,41 +4182,24 @@ function DefaultContextMenu({ open, clientPos, handlers, registry, nodeIds, enab
 }, }) {
     const rf = useReactFlow();
     const [query, setQuery] = useState("");
-    const [canUndo, setCanUndo] = useState(false);
-    const [canRedo, setCanRedo] = useState(false);
     const [hasPasteData, setHasPasteData] = useState(false);
     const q = query.trim().toLowerCase();
     const filteredIds = q
         ? nodeIds.filter((id) => id.toLowerCase().includes(q))
         : nodeIds;
-    // Check undo/redo availability and paste data when menu opens
+    const canUndo = handlers.canUndo ?? false;
+    const canRedo = handlers.canRedo ?? false;
     useEffect(() => {
         if (!open)
             return;
-        let cancelled = false;
-        const checkAvailability = async () => {
-            if (handlers.canUndo) {
-                const result = await handlers.canUndo();
-                if (!cancelled)
-                    setCanUndo(result);
-            }
-            if (handlers.canRedo) {
-                const result = await handlers.canRedo();
-                if (!cancelled)
-                    setCanRedo(result);
-            }
-            // Check paste data dynamically
-            if (handlers.hasPasteData) {
-                const result = handlers.hasPasteData();
-                if (!cancelled)
-                    setHasPasteData(result);
-            }
-        };
-        checkAvailability();
-        return () => {
-            cancelled = true;
-        };
-    }, [open, handlers.canUndo, handlers.canRedo, handlers.hasPasteData]);
+        if (handlers.hasPasteData) {
+            const result = handlers.hasPasteData();
+            setHasPasteData(result);
+        }
+        else {
+            setHasPasteData(false);
+        }
+    }, [open, handlers.hasPasteData]);
     const root = { __children: {} };
     for (const id of filteredIds) {
         const parts = id.split(".");
@@ -4068,12 +4264,6 @@ function DefaultContextMenu({ open, clientPos, handlers, registry, nodeIds, enab
         handlers.onPaste(p);
         handlers.onClose();
     };
-    // Helper to format shortcut for current platform
-    const formatShortcut = (shortcut) => {
-        const isMac = typeof navigator !== "undefined" &&
-            navigator.userAgent.toLowerCase().includes("mac");
-        return shortcut.replace(/⌘\/Ctrl/g, isMac ? "⌘" : "Ctrl");
-    };
     const renderTree = (tree, path = []) => {
         const entries = Object.entries(tree?.__children ?? {}).sort((a, b) => a[0].localeCompare(b[0]));
         return (jsx("div", { children: entries.map(([key, child]) => {
@@ -4091,13 +4281,17 @@ function DefaultContextMenu({ open, clientPos, handlers, registry, nodeIds, enab
     return (jsxs("div", { ref: ref, tabIndex: -1, className: "fixed z-[1000] bg-white border border-gray-300 rounded-lg shadow-lg p-1 min-w-[180px] text-sm text-gray-700 select-none", style: { left: x, top: y }, onClick: (e) => e.stopPropagation(), onMouseDown: (e) => e.stopPropagation(), onWheel: (e) => e.stopPropagation(), onContextMenu: (e) => {
             e.preventDefault();
             e.stopPropagation();
-        }, children: [hasPasteData && handlers.onPaste && (jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100 disabled:opacity-50 disabled:cursor-not-allowed flex items-center justify-between", onClick: handlePaste, children: [jsx("span", { children: "Paste" }), enableKeyboardShortcuts && keyboardShortcuts.paste && (jsx("span", { className: "text-gray-400 text-xs ml-4", children: formatShortcut(keyboardShortcuts.paste) }))] })), (handlers.onUndo || handlers.onRedo) && (jsxs(Fragment, { children: [hasPasteData && handlers.onPaste && (jsx("div", { className: "h-px bg-gray-200 my-1" })), handlers.onUndo && (jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100 disabled:opacity-50 disabled:cursor-not-allowed flex items-center justify-between", onClick: handlers.onUndo, disabled: !canUndo, children: [jsx("span", { children: "Undo" }), enableKeyboardShortcuts && keyboardShortcuts.undo && (jsx("span", { className: "text-gray-400 text-xs ml-4", children: formatShortcut(keyboardShortcuts.undo) }))] })), handlers.onRedo && (jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100 disabled:opacity-50 disabled:cursor-not-allowed flex items-center justify-between", onClick: handlers.onRedo, disabled: !canRedo, children: [jsx("span", { children: "Redo" }), enableKeyboardShortcuts && keyboardShortcuts.redo && (jsx("span", { className: "text-gray-400 text-xs ml-4", children: formatShortcut(keyboardShortcuts.redo) }))] }))] })), hasPasteData &&
+        }, children: [hasPasteData && handlers.onPaste && (jsx(ContextMenuButton, { label: "Paste", onClick: handlePaste, shortcut: keyboardShortcuts.paste, enableKeyboardShortcuts: enableKeyboardShortcuts })), (handlers.onUndo || handlers.onRedo) && (jsxs(Fragment, { children: [hasPasteData && handlers.onPaste && (jsx("div", { className: "h-px bg-gray-200 my-1" })), handlers.onUndo && (jsx(ContextMenuButton, { label: "Undo", onClick: handlers.onUndo, disabled: !canUndo, shortcut: keyboardShortcuts.undo, enableKeyboardShortcuts: enableKeyboardShortcuts })), handlers.onRedo && (jsx(ContextMenuButton, { label: "Redo", onClick: handlers.onRedo, disabled: !canRedo, shortcut: keyboardShortcuts.redo, enableKeyboardShortcuts: enableKeyboardShortcuts }))] })), hasPasteData &&
                 handlers.onPaste &&
                 !handlers.onUndo &&
-                !handlers.onRedo && jsx("div", { className: "h-px bg-gray-200 my-1" }), jsxs("div", { className: "px-2 py-1 font-semibold text-gray-700", children: ["Add Node", " ", jsxs("span", { className: "text-gray-500 font-normal", children: ["(", totalCount, ")"] })] }), jsx("div", { className: "px-2 pb-1", children: jsx("input", { ref: inputRef, type: "text", value: query, onChange: (e) => setQuery(e.target.value), placeholder: "Filter nodes...", className: "w-full border border-gray-300 px-2 py-1 text-sm outline-none focus:border-gray-400 select-text", onClick: (e) => e.stopPropagation(), onMouseDown: (e) => e.stopPropagation(), onWheel: (e) => e.stopPropagation() }) }), jsx("div", { className: "max-h-60 overflow-auto", children: totalCount > 0 ? (renderTree(root)) : (jsx("div", { className: "px-3 py-2 text-gray-400", children: "No matches" })) })] }));
+                !handlers.onRedo && jsx("div", { className: "h-px bg-gray-200 my-1" }), jsxs("div", { className: "px-2 py-1 font-semibold text-gray-700", children: ["Add Node", " ", jsxs("span", { className: "text-gray-500 font-normal", children: ["(", totalCount, ")"] })] }), jsx("div", { className: "px-2 pb-1", children: jsx("input", { ref: inputRef, type: "text", value: query, onChange: (e) => setQuery(e.target.value), placeholder: "Filter nodes...", className: "w-full border border-gray-300 rounded px-2 py-1 text-sm outline-none focus:border-gray-400 select-text", onClick: (e) => e.stopPropagation(), onMouseDown: (e) => e.stopPropagation(), onWheel: (e) => e.stopPropagation() }) }), jsx("div", { className: "max-h-60 overflow-auto", children: totalCount > 0 ? (renderTree(root)) : (jsx("div", { className: "px-3 py-2 text-gray-400", children: "No matches" })) })] }));
 }
 
-function NodeContextMenu({ open, clientPos, nodeId, handlers, canRunPull, bakeableOutputs, }) {
+function NodeContextMenu({ open, clientPos, nodeId, handlers, canRunPull, bakeableOutputs, enableKeyboardShortcuts = true, keyboardShortcuts = {
+    copy: "⌘/Ctrl + C",
+    duplicate: "⌘/Ctrl + D",
+    delete: "Delete",
+}, }) {
     const ref = useRef(null);
     // outside click + ESC
     useEffect(() => {
@@ -4135,7 +4329,7 @@ function NodeContextMenu({ open, clientPos, nodeId, handlers, canRunPull, bakeab
     return (jsxs("div", { ref: ref, tabIndex: -1, className: "fixed z-[1000] bg-white border border-gray-300 rounded-lg shadow-lg p-1 min-w-[180px] text-sm text-gray-700 select-none", style: { left: x, top: y }, onClick: (e) => e.stopPropagation(), onMouseDown: (e) => e.stopPropagation(), onWheel: (e) => e.stopPropagation(), onContextMenu: (e) => {
             e.preventDefault();
             e.stopPropagation();
-        }, children: [jsxs("div", { className: "px-2 py-1 font-semibold text-gray-700", children: ["Node (", nodeId, ")"] }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onDelete, children: "Delete" }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onDuplicate, children: "Duplicate" }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onDuplicateWithEdges, children: "Duplicate with edges" }), canRunPull && (jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onRunPull, children: "Run (pull)" })), jsx("div", { className: "h-px bg-gray-200 my-1" }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onCopy, children: "Copy" }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onCopyId, children: "Copy Node ID" }), bakeableOutputs.length > 0 && (jsxs(Fragment, { children: [jsx("div", { className: "h-px bg-gray-200 my-1" }), jsx("div", { className: "px-2 py-1 font-semibold text-gray-700", children: "Bake" }), bakeableOutputs.map((h) => (jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: () => handlers.onBake(h), children: ["Bake: ", h] }, h)))] }))] }));
+        }, children: [jsxs("div", { className: "px-2 py-1 font-semibold text-gray-700", children: ["Node (", nodeId, ")"] }), jsx(ContextMenuButton, { label: "Delete", onClick: handlers.onDelete, shortcut: keyboardShortcuts.delete, enableKeyboardShortcuts: enableKeyboardShortcuts }), jsx(ContextMenuButton, { label: "Duplicate", onClick: handlers.onDuplicate, shortcut: keyboardShortcuts.duplicate, enableKeyboardShortcuts: enableKeyboardShortcuts }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onDuplicateWithEdges, children: "Duplicate with edges" }), canRunPull && (jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onRunPull, children: "Run (pull)" })), jsx("div", { className: "h-px bg-gray-200 my-1" }), jsx(ContextMenuButton, { label: "Copy", onClick: handlers.onCopy, shortcut: keyboardShortcuts.copy, enableKeyboardShortcuts: enableKeyboardShortcuts }), jsx("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: handlers.onCopyId, children: "Copy Node ID" }), bakeableOutputs.length > 0 && (jsxs(Fragment, { children: [jsx("div", { className: "h-px bg-gray-200 my-1" }), jsx("div", { className: "px-2 py-1 font-semibold text-gray-700", children: "Bake" }), bakeableOutputs.map((h) => (jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100", onClick: () => handlers.onBake(h), children: ["Bake: ", h] }, h)))] }))] }));
 }
 
 function SelectionContextMenu({ open, clientPos, handlers, enableKeyboardShortcuts = true, keyboardShortcuts = {
@@ -4168,12 +4362,6 @@ function SelectionContextMenu({ open, clientPos, handlers, enableKeyboardShortcu
         if (open)
             ref.current?.focus();
     }, [open]);
-    // Helper to format shortcut for current platform
-    const formatShortcut = (shortcut) => {
-        const isMac = typeof navigator !== "undefined" &&
-            navigator.userAgent.toLowerCase().includes("mac");
-        return shortcut.replace(/⌘\/Ctrl/g, isMac ? "⌘" : "Ctrl");
-    };
     if (!open || !clientPos)
         return null;
     // Clamp menu position to viewport
@@ -4185,7 +4373,7 @@ function SelectionContextMenu({ open, clientPos, handlers, enableKeyboardShortcu
     return (jsxs("div", { ref: ref, tabIndex: -1, className: "fixed z-[1000] bg-white border border-gray-300 rounded-lg shadow-lg p-1 min-w-[180px] text-sm text-gray-700 select-none", style: { left: x, top: y }, onClick: (e) => e.stopPropagation(), onMouseDown: (e) => e.stopPropagation(), onWheel: (e) => e.stopPropagation(), onContextMenu: (e) => {
             e.preventDefault();
             e.stopPropagation();
-        }, children: [jsx("div", { className: "px-2 py-1 font-semibold text-gray-700", children: "Selection" }), jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100 flex items-center justify-between", onClick: handlers.onCopy, children: [jsx("span", { children: "Copy" }), enableKeyboardShortcuts && keyboardShortcuts.copy && (jsx("span", { className: "text-gray-400 text-xs ml-4", children: formatShortcut(keyboardShortcuts.copy) }))] }), jsxs("button", { className: "block w-full text-left px-2 py-1 hover:bg-gray-100 flex items-center justify-between", onClick: handlers.onDelete, children: [jsx("span", { children: "Delete" }), enableKeyboardShortcuts && keyboardShortcuts.delete && (jsx("span", { className: "text-gray-400 text-xs ml-4", children: formatShortcut(keyboardShortcuts.delete) }))] })] }));
+        }, children: [jsx("div", { className: "px-2 py-1 font-semibold text-gray-700", children: "Selection" }), jsx(ContextMenuButton, { label: "Copy", onClick: handlers.onCopy, shortcut: keyboardShortcuts.copy, enableKeyboardShortcuts: enableKeyboardShortcuts }), jsx(ContextMenuButton, { label: "Delete", onClick: handlers.onDelete, shortcut: keyboardShortcuts.delete, enableKeyboardShortcuts: enableKeyboardShortcuts })] }));
 }
 
 const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, getDefaultNodeSize }, ref) => {
@@ -4193,7 +4381,6 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
     const nodeValidation = validationByNode;
     const edgeValidation = validationByEdge.errors;
     const [registryVersion, setRegistryVersion] = useState(0);
-    // Keep stable references for nodes/edges to avoid unnecessary updates
     const prevNodesRef = useRef([]);
     const prevEdgesRef = useRef([]);
     function retainStabilityById(prev, next, isSame) {
@@ -4523,7 +4710,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
         setNodeMenuOpen(false);
         setSelectionMenuOpen(false);
     };
-    const addNodeAt = useCallback(async (typeId, opts) => wb.addNode({ typeId, position: opts.position }, { inputs: opts.inputs }), [wb]);
+    const addNodeAt = useCallback(async (typeId, opts) => wb.addNode({ typeId, position: opts.position }, { inputs: opts.inputs, commit: true }), [wb]);
     const onCloseMenu = useCallback(() => {
         setMenuOpen(false);
     }, []);
@@ -4548,16 +4735,13 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
                 get: () => wb.getCopiedData(),
                 set: (data) => wb.setCopiedData(data),
             };
-        const baseHandlers = createDefaultContextMenuHandlers(addNodeAt, onCloseMenu, 
-        // Paste handler - checks storage dynamically when called
-        // Only provide handler if storage has data or might have data (for dynamic checking)
-        (position) => {
+        const baseHandlers = createDefaultContextMenuHandlers(addNodeAt, onCloseMenu, (position) => {
             const data = storage.get();
             if (!data)
                 return;
-            wb.pasteCopiedData(data, position);
+            wb.pasteCopiedData(data, position, { commit: true, reason: "paste" });
             onCloseMenu();
-        }, runner, () => storage.get(), () => storage.set(null));
+        }, runner, () => storage.get(), () => storage.set(null), wb.getHistory());
         if (overrides?.getDefaultContextMenuHandlers) {
             return overrides.getDefaultContextMenuHandlers(wb, baseHandlers);
         }
@@ -4576,9 +4760,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
         }, runner);
         if (overrides?.getSelectionContextMenuHandlers) {
             const selection = wb.getSelection();
-            return overrides.getSelectionContextMenuHandlers(wb, selection, baseHandlers, {
-                getDefaultNodeSize: overrides.getDefaultNodeSize,
-            });
+            return overrides.getSelectionContextMenuHandlers(wb, selection, baseHandlers, { getDefaultNodeSize: overrides.getDefaultNodeSize });
         }
         return baseHandlers;
     }, [wb, runner, overrides, onCloseSelectionMenu]);
@@ -4624,6 +4806,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
         redo: "⌘/Ctrl + Shift + Z",
         copy: "⌘/Ctrl + C",
         paste: "⌘/Ctrl + V",
+        duplicate: "⌘/Ctrl + D",
         delete: "Delete",
     };
     // Keyboard shortcut handler
@@ -4648,9 +4831,9 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
                 e.preventDefault();
                 if (runner &&
                     "onUndo" in defaultContextMenuHandlers &&
-                    defaultContextMenuHandlers.onUndo) {
-                    const canUndo = await runner.canUndo().catch(() => false);
-                    if (canUndo) {
+                    defaultContextMenuHandlers.onUndo &&
+                    defaultContextMenuHandlers.canUndo) {
+                    if (defaultContextMenuHandlers.canUndo) {
                         defaultContextMenuHandlers.onUndo();
                     }
                 }
@@ -4661,9 +4844,9 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
                 e.preventDefault();
                 if (runner &&
                     "onRedo" in defaultContextMenuHandlers &&
-                    defaultContextMenuHandlers.onRedo) {
-                    const canRedo = await runner.canRedo().catch(() => false);
-                    if (canRedo) {
+                    defaultContextMenuHandlers.onRedo &&
+                    defaultContextMenuHandlers.canRedo) {
+                    if (defaultContextMenuHandlers.canRedo) {
                         defaultContextMenuHandlers.onRedo();
                     }
                 }
@@ -4674,12 +4857,26 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
                 const selection = wb.getSelection();
                 if (selection.nodes.length > 0 || selection.edges.length > 0) {
                     e.preventDefault();
-                    if (selectionContextMenuHandlers.onCopy) {
+                    // If single node selected, use node context menu handler; otherwise use selection handler
+                    if (selection.nodes.length === 1 && nodeContextMenuHandlers?.onCopy) {
+                        nodeContextMenuHandlers.onCopy();
+                    }
+                    else if (selectionContextMenuHandlers.onCopy) {
                         selectionContextMenuHandlers.onCopy();
                     }
                 }
                 return;
             }
+            // Duplicate: Cmd/Ctrl + D
+            if (modKey && key === "d" && !e.shiftKey && !e.altKey) {
+                const selection = wb.getSelection();
+                if (selection.nodes.length === 1 &&
+                    nodeContextMenuHandlers?.onDuplicate) {
+                    e.preventDefault();
+                    nodeContextMenuHandlers.onDuplicate();
+                }
+                return;
+            }
             // Paste: Cmd/Ctrl + V
             if (modKey && key === "v" && !e.shiftKey && !e.altKey) {
                 e.preventDefault();
@@ -4709,6 +4906,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
         runner,
         defaultContextMenuHandlers,
         selectionContextMenuHandlers,
+        nodeContextMenuHandlers,
         rfInstanceRef,
     ]);
     // Get custom renderers from UI extension registry (reactive to uiVersion changes)
@@ -4725,7 +4923,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
     const onMoveEnd = useCallback(() => {
         if (rfInstanceRef.current) {
             const viewport = rfInstanceRef.current.getViewport();
-            wb.setViewport({ x: viewport.x, y: viewport.y, zoom: viewport.zoom }, { commit: true });
+            wb.setViewport({ x: viewport.x, y: viewport.y, zoom: viewport.zoom });
         }
     }, [wb]);
     const viewportRef = useRef(null);
@@ -4761,7 +4959,9 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
                             ? { enableKeyboardShortcuts, keyboardShortcuts }
                             : {}) })) : (jsx(DefaultContextMenu, { open: menuOpen, clientPos: menuPos, handlers: defaultContextMenuHandlers, registry: registry, nodeIds: nodeIds, enableKeyboardShortcuts: enableKeyboardShortcuts, keyboardShortcuts: keyboardShortcuts })), !!nodeAtMenu &&
                         nodeContextMenuHandlers &&
-                        (NodeContextMenuRenderer ? (jsx(NodeContextMenuRenderer, { open: nodeMenuOpen, clientPos: nodeMenuPos, nodeId: nodeAtMenu, handlers: nodeContextMenuHandlers, canRunPull: canRunPull, bakeableOutputs: bakeableOutputs })) : (jsx(NodeContextMenu, { open: nodeMenuOpen, clientPos: nodeMenuPos, nodeId: nodeAtMenu, handlers: nodeContextMenuHandlers, canRunPull: canRunPull, bakeableOutputs: bakeableOutputs }))), selectionMenuOpen && selectionMenuPos && (jsx(SelectionContextMenu, { open: selectionMenuOpen, clientPos: selectionMenuPos, handlers: selectionContextMenuHandlers, enableKeyboardShortcuts: enableKeyboardShortcuts, keyboardShortcuts: keyboardShortcuts }))] }) }) }));
+                        (NodeContextMenuRenderer ? (jsx(NodeContextMenuRenderer, { open: nodeMenuOpen, clientPos: nodeMenuPos, nodeId: nodeAtMenu, handlers: nodeContextMenuHandlers, canRunPull: canRunPull, bakeableOutputs: bakeableOutputs, ...(enableKeyboardShortcuts !== false
+                                ? { enableKeyboardShortcuts, keyboardShortcuts }
+                                : {}) })) : (jsx(NodeContextMenu, { open: nodeMenuOpen, clientPos: nodeMenuPos, nodeId: nodeAtMenu, handlers: nodeContextMenuHandlers, canRunPull: canRunPull, bakeableOutputs: bakeableOutputs, enableKeyboardShortcuts: enableKeyboardShortcuts, keyboardShortcuts: keyboardShortcuts }))), selectionMenuOpen && selectionMenuPos && (jsx(SelectionContextMenu, { open: selectionMenuOpen, clientPos: selectionMenuPos, handlers: selectionContextMenuHandlers, enableKeyboardShortcuts: enableKeyboardShortcuts, keyboardShortcuts: keyboardShortcuts }))] }) }) }));
 });
 
 function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, example, onExampleChange, engine, onEngineChange, backendKind, onBackendKindChange, httpBaseUrl, onHttpBaseUrlChange, wsUrl, onWsUrlChange, debug, onDebugChange, showValues, onShowValuesChange, hideWorkbench, onHideWorkbenchChange, overrides, onInit, onChange, }) {