@bian-womp/spark-workbench 0.2.0 → 0.2.2

package/lib/esm/index.js

@@ -1,9 +1,10 @@
 import { GraphBuilder, StepEngine, HybridEngine, PullEngine, BatchedEngine, PushEngine, isTypedOutput, getTypedOutputValue, getTypedOutputTypeId, isInputPrivate, getInputTypeId, createSimpleGraphRegistry, createSimpleGraphDef, createAsyncGraphDef, createAsyncGraphRegistry, createProgressGraphDef, createProgressGraphRegistry, createValidationGraphDef, createValidationGraphRegistry, Registry } from '@bian-womp/spark-graph';
 import { HttpPollingTransport, WebSocketTransport, RemoteRunner } from '@bian-womp/spark-remote';
-import React, { useCallback, useState, useEffect, useMemo, createContext, useContext, useRef, useImperativeHandle } from 'react';
-import { jsx, jsxs } from 'react/jsx-runtime';
+import React, { useCallback, useState, useRef, useEffect, useMemo, createContext, useContext, useImperativeHandle } from 'react';
+import { Position, Handle, useUpdateNodeInternals, useReactFlow, ReactFlow, Background, MiniMap, Controls } from '@xyflow/react';
+import { jsx, jsxs, Fragment } from 'react/jsx-runtime';
 import { XCircleIcon, WarningCircleIcon, PlugsConnectedIcon, ClockClockwiseIcon, WifiHighIcon, WifiSlashIcon } from '@phosphor-icons/react';
-import { Handle, Position, useReactFlow, ReactFlow, Background, MiniMap, Controls } from '@xyflow/react';
+import isEqual from 'lodash/isEqual';
 import cx from 'classnames';
 
 class DefaultUIExtensionRegistry {
@@ -199,6 +200,20 @@ class InMemoryWorkbench extends AbstractWorkbench {
         });
         this.emit("validationChanged", this.validate());
     }
+    updateEdgeType(edgeId, typeId) {
+        const e = this.def.edges.find((x) => x.id === edgeId);
+        if (!e)
+            return;
+        if (!typeId)
+            delete e.typeId;
+        else
+            e.typeId = typeId;
+        this.emit("graphChanged", {
+            def: this.def,
+            change: { type: "updateEdgeType", edgeId, typeId },
+        });
+        this.refreshValidation();
+    }
     updateParams(nodeId, params) {
         const n = this.def.nodes.find((n) => n.nodeId === nodeId);
         if (!n)
@@ -230,6 +245,10 @@ class InMemoryWorkbench extends AbstractWorkbench {
     setSelection(sel) {
         this.selection = { nodes: [...sel.nodes], edges: [...sel.edges] };
         this.emit("selectionChanged", this.selection);
+        this.emit("graphUiChanged", {
+            def: this.def,
+            change: { type: "selection" },
+        });
     }
     getSelection() {
         return {
@@ -237,18 +256,6 @@ class InMemoryWorkbench extends AbstractWorkbench {
             edges: [...this.selection.edges],
         };
     }
-    toggleNodeSelection(nodeId) {
-        this.selection.nodes = this.selection.nodes.includes(nodeId)
-            ? this.selection.nodes.filter((id) => id !== nodeId)
-            : [...this.selection.nodes, nodeId];
-        this.emit("selectionChanged", this.selection);
-    }
-    toggleEdgeSelection(edgeId) {
-        this.selection.edges = this.selection.edges.includes(edgeId)
-            ? this.selection.edges.filter((id) => id !== edgeId)
-            : [...this.selection.edges, edgeId];
-        this.emit("selectionChanged", this.selection);
-    }
     on(event, handler) {
         if (!this.listeners.has(event))
             this.listeners.set(event, new Set());
@@ -758,49 +765,91 @@ function useWorkbenchBridge(wb) {
         });
     }, [wb]);
     const onNodesChange = useCallback((changes) => {
+        // Apply position updates
         changes.forEach((c) => {
-            if (c.type === "position" && c.position)
+            if (c.type === "position" && c.position) {
                 wb.setPosition(c.id, c.position);
-            if (c.type === "remove")
-                wb.removeNode(c.id);
-            if (c.type === "select")
-                wb.toggleNodeSelection(c.id);
+            }
         });
+        // Derive next node selection from change set
+        const current = wb.getSelection();
+        const nextNodeIds = new Set(current.nodes);
+        let selectionChanged = false;
+        for (const change of changes) {
+            const type = change?.type;
+            if (type === "select") {
+                const id = change.id;
+                const selected = change.selected;
+                if (typeof selected === "boolean") {
+                    if (selected) {
+                        if (!nextNodeIds.has(id)) {
+                            nextNodeIds.add(id);
+                            selectionChanged = true;
+                        }
+                    }
+                    else if (nextNodeIds.delete(id)) {
+                        selectionChanged = true;
+                    }
+                }
+            }
+            else if (type === "remove") {
+                const id = change.id;
+                if (nextNodeIds.delete(id))
+                    selectionChanged = true;
+            }
+        }
+        if (selectionChanged) {
+            wb.setSelection({
+                nodes: Array.from(nextNodeIds),
+                edges: current.edges,
+            });
+        }
     }, [wb]);
     const onEdgesDelete = useCallback((edges) => edges.forEach((e) => wb.disconnect(e.id)), [wb]);
     const onEdgesChange = useCallback((changes) => {
-        changes.forEach((c) => {
-            if (c.type === "remove")
-                wb.disconnect(c.id);
-            else if (c.type === "select")
-                wb.toggleEdgeSelection(c.id);
-        });
+        const current = wb.getSelection();
+        const nextEdgeIds = new Set(current.edges);
+        let selectionChanged = false;
+        for (const change of changes) {
+            const type = change?.type;
+            if (type === "select") {
+                const id = change.id;
+                const selected = change.selected;
+                if (typeof selected === "boolean") {
+                    if (selected) {
+                        if (!nextEdgeIds.has(id)) {
+                            nextEdgeIds.add(id);
+                            selectionChanged = true;
+                        }
+                    }
+                    else if (nextEdgeIds.delete(id)) {
+                        selectionChanged = true;
+                    }
+                }
+            }
+            else if (type === "remove") {
+                const id = change.id;
+                if (nextEdgeIds.delete(id))
+                    selectionChanged = true;
+            }
+        }
+        if (selectionChanged) {
+            wb.setSelection({
+                nodes: current.nodes,
+                edges: Array.from(nextEdgeIds),
+            });
+        }
     }, [wb]);
     const onNodesDelete = useCallback((nodes) => {
         for (const n of nodes)
             wb.removeNode(n.id);
     }, [wb]);
-    const onSelectionChange = useCallback((sel) => {
-        const next = {
-            nodes: sel.nodes.map((n) => n.id),
-            edges: sel.edges.map((e) => e.id),
-        };
-        const cur = wb.getSelection();
-        const sameLen = cur.nodes.length === next.nodes.length &&
-            cur.edges.length === next.edges.length;
-        const same = sameLen &&
-            cur.nodes.every((id, i) => id === next.nodes[i]) &&
-            cur.edges.every((id, i) => id === next.edges[i]);
-        if (!same)
-            wb.setSelection(next);
-    }, [wb]);
     return {
         onConnect,
         onNodesChange,
         onEdgesChange,
         onEdgesDelete,
         onNodesDelete,
-        onSelectionChange,
     };
 }
 function useWorkbenchGraphTick(wb) {
@@ -840,6 +889,44 @@ function useWorkbenchVersionTick(runner) {
     }, [runner]);
     return version;
 }
+function useThrottledValue(value, intervalMs) {
+    const [throttled, setThrottled] = useState(value);
+    const lastSetAtRef = useRef(0);
+    const timeoutRef = useRef(null);
+    useEffect(() => {
+        const now = typeof performance !== "undefined" && performance.now
+            ? performance.now()
+            : Date.now();
+        const elapsed = now - lastSetAtRef.current;
+        if (elapsed >= intervalMs) {
+            lastSetAtRef.current = now;
+            setThrottled(value);
+        }
+        else {
+            if (timeoutRef.current !== null) {
+                window.clearTimeout(timeoutRef.current);
+            }
+            timeoutRef.current = window.setTimeout(() => {
+                lastSetAtRef.current =
+                    typeof performance !== "undefined" && performance.now
+                        ? performance.now()
+                        : Date.now();
+                setThrottled(value);
+                timeoutRef.current = null;
+            }, Math.max(0, intervalMs - elapsed));
+        }
+        return () => { };
+    }, [value, intervalMs]);
+    useEffect(() => {
+        return () => {
+            if (timeoutRef.current !== null) {
+                window.clearTimeout(timeoutRef.current);
+                timeoutRef.current = null;
+            }
+        };
+    }, []);
+    return throttled;
+}
 // Query param helpers
 function setSearchParam(key, val) {
     if (typeof window === "undefined")
@@ -1000,7 +1087,13 @@ function summarizeDeep(value) {
     return value;
 }
 
+// Shared UI constants for node layout to keep mapping and rendering in sync
+const NODE_HEADER_HEIGHT_PX = 24;
+const NODE_ROW_HEIGHT_PX = 22;
+
 function toReactFlow(def, positions, registry, opts) {
+    const EDGE_STYLE_ERROR = { stroke: "#ef4444", strokeWidth: 2 };
+    const EDGE_STYLE_RUNNING = { stroke: "#3b82f6" };
     const nodeHandleMap = {};
     const nodes = def.nodes.map((n) => {
         const desc = registry.nodes.get(n.typeId);
@@ -1012,6 +1105,35 @@ function toReactFlow(def, positions, registry, opts) {
             inputs: new Set(inputHandles.map((h) => h.id)),
             outputs: new Set(outputHandles.map((h) => h.id)),
         };
+        // Match DefaultNode sizing heuristics to avoid hidden nodes during re-measure
+        const HEADER_SIZE = NODE_HEADER_HEIGHT_PX;
+        const ROW_SIZE = NODE_ROW_HEIGHT_PX;
+        const maxRows = Math.max(inputHandles.length, outputHandles.length);
+        const initialWidth = opts.showValues ? 320 : 240;
+        const initialHeight = HEADER_SIZE + maxRows * ROW_SIZE;
+        // Precompute handle bounds so edges can render immediately without waiting for measurement
+        const handles = [
+            // Inputs on the left as targets
+            ...inputHandles.map((h, i) => ({
+                id: h.id,
+                type: "target",
+                position: Position.Left,
+                x: 0,
+                y: HEADER_SIZE + i * ROW_SIZE,
+                width: 1,
+                height: ROW_SIZE + 2,
+            })),
+            // Outputs on the right as sources
+            ...outputHandles.map((h, i) => ({
+                id: h.id,
+                type: "source",
+                position: Position.Right,
+                x: initialWidth - 1,
+                y: HEADER_SIZE + i * ROW_SIZE,
+                width: 1,
+                height: ROW_SIZE + 2,
+            })),
+        ];
         return {
             id: n.nodeId,
             data: {
@@ -1019,7 +1141,23 @@ function toReactFlow(def, positions, registry, opts) {
                 params: n.params,
                 inputHandles,
                 outputHandles,
+                handleLayout: [
+                    ...inputHandles.map((h, i) => ({
+                        id: h.id,
+                        type: "target",
+                        position: Position.Left,
+                        y: HEADER_SIZE + i * ROW_SIZE + ROW_SIZE / 2,
+                    })),
+                    ...outputHandles.map((h, i) => ({
+                        id: h.id,
+                        type: "source",
+                        position: Position.Right,
+                        y: HEADER_SIZE + i * ROW_SIZE + ROW_SIZE / 2,
+                    })),
+                ],
                 showValues: opts.showValues,
+                renderWidth: initialWidth,
+                renderHeight: initialHeight,
                 inputValues: opts.inputs?.[n.nodeId],
                 outputValues: opts.outputs?.[n.nodeId],
                 status: opts.nodeStatus?.[n.nodeId],
@@ -1036,6 +1174,11 @@ function toReactFlow(def, positions, registry, opts) {
             selected: opts.selectedNodeIds
                 ? opts.selectedNodeIds.has(n.nodeId)
                 : undefined,
+            initialWidth,
+            initialHeight,
+            handles,
+            width: initialWidth,
+            height: initialHeight,
         };
     });
     const edges = def.edges
@@ -1052,9 +1195,9 @@ function toReactFlow(def, positions, registry, opts) {
         const hasError = !!st?.lastError;
         const isInvalidEdge = !!opts.edgeValidation?.[e.id];
         const style = hasError || isInvalidEdge
-            ? { stroke: "#ef4444", strokeWidth: 2 }
+            ? EDGE_STYLE_ERROR
             : isRunning
-                ? { stroke: "#3b82f6" }
+                ? EDGE_STYLE_RUNNING
                 : undefined;
         return {
             id: e.id,
@@ -1082,17 +1225,35 @@ function getNodeBorderClassNames(args) {
     const hasValidationWarning = !hasValidationError && issues.length > 0;
     const isRunning = !!status.activeRuns;
     const isInvalid = !!status.invalidated && !isRunning && !hasError;
-    const borderWidth = selected ? "border-2" : "border";
+    // Keep border width constant to avoid layout reflow on selection toggles
+    const borderWidth = "border";
     const borderStyle = isInvalid ? "border-dashed" : "border-solid";
-    const borderColor = hasError || hasValidationError
-        ? "border-red-500"
+    const severity = hasError || hasValidationError
+        ? "red"
         : hasValidationWarning
-            ? "border-amber-500"
+            ? "amber"
             : isRunning
-                ? "border-blue-500"
-                : "border-gray-500 dark:border-gray-400";
-    const ring = isRunning ? " ring-2 ring-blue-200 dark:ring-blue-900" : "";
-    return `${borderWidth} ${borderStyle} ${borderColor}${ring}`.trim();
+                ? "blue"
+                : "gray";
+    const borderBySeverity = {
+        red: "border-red-500",
+        amber: "border-amber-500",
+        blue: "border-blue-500",
+        gray: "border-gray-500 dark:border-gray-400",
+    };
+    const ringBySeverity = {
+        red: "ring-2 ring-red-300 dark:ring-red-900",
+        amber: "ring-2 ring-amber-300 dark:ring-amber-900",
+        blue: "ring-2 ring-blue-200 dark:ring-blue-900",
+        gray: "ring-2 ring-gray-300 dark:ring-gray-500",
+    };
+    const borderColor = borderBySeverity[severity];
+    const ring = isRunning
+        ? ringBySeverity.blue
+        : selected
+            ? ringBySeverity[severity === "blue" ? "gray" : severity]
+            : "";
+    return [borderWidth, borderStyle, borderColor, ring].join(" ").trim();
 }
 
 const WorkbenchContext = createContext(null);
@@ -1228,6 +1389,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, children, }) {
         });
         wb.setPositions(pos);
     }, [wb]);
+    const updateEdgeType = useCallback((edgeId, typeId) => wb.updateEdgeType(edgeId, typeId), [wb]);
     // Subscribe to runner/workbench events
     useEffect(() => {
         const add = (source, type) => (payload) => setEvents((prev) => {
@@ -1238,8 +1400,15 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, children, }) {
                 if (changeType === "moveNode" || changeType === "moveNodes")
                     return prev;
             }
+            const nextNo = prev.length > 0 ? (prev[0]?.no ?? 0) + 1 : 1;
             const next = [
-                { at: Date.now(), source, type, payload: structuredClone(payload) },
+                {
+                    no: nextNo,
+                    at: Date.now(),
+                    source,
+                    type,
+                    payload: structuredClone(payload),
+                },
                 ...prev,
             ];
             return next.length > 200 ? next.slice(0, 200) : next;
@@ -1554,6 +1723,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, children, }) {
         step,
         flush,
         runAutoLayout,
+        updateEdgeType,
     }), [
         wb,
         runner,
@@ -1580,6 +1750,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, children, }) {
         step,
         flush,
         runAutoLayout,
+        wb,
     ]);
     return (jsx(WorkbenchContext.Provider, { value: value, children: children }));
 }
@@ -1615,7 +1786,7 @@ function DebugEvents({ autoScroll, onAutoScrollChange, hideWorkbench, onHideWork
             return String(v);
         }
     };
-    return (jsxs("div", { className: "flex flex-col h-full min-h-0", children: [jsxs("div", { className: "flex items-center justify-between mb-1", children: [jsx("div", { className: "font-semibold", children: "Events" }), jsxs("div", { className: "flex items-center gap-2", children: [jsxs("label", { className: "flex items-center gap-1 text-xs text-gray-700", children: [jsx("input", { type: "checkbox", checked: hideWorkbench, onChange: (e) => onHideWorkbenchChange?.(e.target.checked) }), jsx("span", { children: "Hide workbench" })] }), jsxs("label", { className: "flex items-center gap-1 text-xs text-gray-700", children: [jsx("input", { type: "checkbox", checked: autoScroll, onChange: (e) => onAutoScrollChange?.(e.target.checked) }), jsx("span", { children: "Auto scroll" })] }), jsx("button", { onClick: clearEvents, className: "text-xs px-2 py-0.5 border border-gray-300 rounded", children: "Clear" })] })] }), jsx("div", { ref: scrollRef, className: "flex-1 overflow-auto text-[11px] leading-4 divide-y divide-gray-200", children: rows.map((ev, idx) => (jsxs("div", { className: "opacity-85 odd:bg-gray-50 px-2 py-1", children: [jsxs("div", { className: "flex items-baseline gap-2", children: [jsx("span", { className: "w-8 shrink-0 text-right text-gray-500 select-none", children: idx + 1 }), jsxs("span", { className: "text-gray-500", children: [new Date(ev.at).toLocaleTimeString(), " \u00B7 ", ev.source, ":", ev.type] })] }), jsx("pre", { className: "m-0 whitespace-pre-wrap ml-10", children: renderPayload(ev.payload) })] }, `${ev.at}:${idx}`))) })] }));
+    return (jsxs("div", { className: "flex flex-col h-full min-h-0", children: [jsxs("div", { className: "flex items-center justify-between mb-1", children: [jsx("div", { className: "font-semibold", children: "Events" }), jsxs("div", { className: "flex items-center gap-2", children: [jsxs("label", { className: "flex items-center gap-1 text-xs text-gray-700", children: [jsx("input", { type: "checkbox", checked: hideWorkbench, onChange: (e) => onHideWorkbenchChange?.(e.target.checked) }), jsx("span", { children: "Hide workbench" })] }), jsxs("label", { className: "flex items-center gap-1 text-xs text-gray-700", children: [jsx("input", { type: "checkbox", checked: autoScroll, onChange: (e) => onAutoScrollChange?.(e.target.checked) }), jsx("span", { children: "Auto scroll" })] }), jsx("button", { onClick: clearEvents, className: "text-xs px-2 py-0.5 border border-gray-300 rounded", children: "Clear" })] })] }), jsx("div", { ref: scrollRef, className: "flex-1 overflow-auto text-[11px] leading-4 divide-y divide-gray-200", children: rows.map((ev) => (jsxs("div", { className: "opacity-85 odd:bg-gray-50 px-2 py-1", children: [jsxs("div", { className: "flex items-baseline gap-2", children: [jsx("span", { className: "w-12 shrink-0 text-right text-gray-500 select-none", children: ev.no }), jsxs("span", { className: "text-gray-500", children: [new Date(ev.at).toLocaleTimeString(), " \u00B7 ", ev.source, ":", ev.type] })] }), jsx("pre", { className: "m-0 whitespace-pre-wrap ml-12", children: renderPayload(ev.payload) })] }, `${ev.at}:${ev.no}`))) })] }));
 }
 
 function Inspector({ debug, autoScroll, hideWorkbench, onAutoScrollChange, onHideWorkbenchChange, toString, toElement, contextPanel, setInput, }) {
@@ -1630,7 +1801,7 @@ function Inspector({ debug, autoScroll, hideWorkbench, onAutoScrollChange, onHid
             return String(value ?? "");
         }
     };
-    const { registry, def, selectedNodeId, selectedEdgeId, inputsMap, outputsMap, outputTypesMap, nodeStatus, validationByNode, validationByEdge, validationGlobal, valuesTick, } = useWorkbenchContext();
+    const { registry, def, selectedNodeId, selectedEdgeId, inputsMap, outputsMap, outputTypesMap, nodeStatus, validationByNode, validationByEdge, validationGlobal, valuesTick, updateEdgeType, } = useWorkbenchContext();
     const nodeValidationIssues = validationByNode.issues;
     const edgeValidationIssues = validationByEdge.issues;
     const nodeValidationHandles = validationByNode;
@@ -1707,7 +1878,11 @@ function Inspector({ debug, autoScroll, hideWorkbench, onAutoScrollChange, onHid
             setOriginals(nextOriginals);
     }, [selectedNodeId, selectedDesc, valuesTick]);
     const widthClass = debug ? "w-[480px]" : "w-[320px]";
-    return (jsxs("div", { className: `${widthClass} border-l border-gray-300 p-3 flex flex-col h-full min-h-0 overflow-hidden`, children: [contextPanel && (jsx("div", { className: "mb-2", children: contextPanel })), jsx("div", { className: "font-semibold mb-2", children: "Inspector" }), jsx("div", { className: "flex-1 overflow-auto", children: !selectedNode && !selectedEdge ? (jsxs("div", { children: [jsx("div", { className: "text-gray-500", children: "Select a node or edge." }), globalValidationIssues && globalValidationIssues.length > 0 && (jsxs("div", { className: "mt-2 text-xs bg-red-50 border border-red-200 rounded px-2 py-1", children: [jsx("div", { className: "font-semibold mb-1", children: "Validation" }), jsx("ul", { className: "list-disc ml-4", children: globalValidationIssues.map((m, i) => (jsxs("li", { className: "flex items-center gap-1", children: [jsx(IssueBadge, { level: m.level, size: 24, className: "w-6 h-6" }), jsx("span", { children: `${m.code}: ${m.message}` })] }, i))) })] }))] })) : selectedEdge ? (jsxs("div", { children: [jsxs("div", { className: "mb-2", children: [jsxs("div", { children: ["Edge: ", selectedEdge.id] }), jsxs("div", { children: [selectedEdge.source.nodeId, ".", selectedEdge.source.handle, " \u2192", " ", selectedEdge.target.nodeId, ".", selectedEdge.target.handle] }), jsxs("div", { children: ["Type: ", selectedEdge.typeId] })] }), selectedEdgeValidation.length > 0 && (jsxs("div", { className: "mt-2 text-xs bg-red-50 border border-red-200 rounded px-2 py-1", children: [jsx("div", { className: "font-semibold mb-1", children: "Validation" }), jsx("ul", { className: "list-disc ml-4", children: selectedEdgeValidation.map((m, i) => (jsxs("li", { className: "flex items-center gap-1", children: [jsx(IssueBadge, { level: m.level, size: 24, className: "w-6 h-6" }), jsx("span", { children: `${m.code}: ${m.message}` })] }, i))) })] }))] })) : (jsxs("div", { children: [selectedNode && (jsxs("div", { className: "mb-2", children: [jsxs("div", { children: ["Node: ", selectedNode.nodeId] }), jsxs("div", { children: ["Type: ", selectedNode.typeId] }), !!selectedNodeStatus?.lastError && (jsx("div", { className: "mt-2 text-sm text-red-700 bg-red-50 border border-red-200 rounded px-2 py-1 break-words", children: String(selectedNodeStatus.lastError?.message ??
+    return (jsxs("div", { className: `${widthClass} border-l border-gray-300 p-3 flex flex-col h-full min-h-0 overflow-hidden`, children: [contextPanel && jsx("div", { className: "mb-2", children: contextPanel }), jsx("div", { className: "font-semibold mb-2", children: "Inspector" }), jsxs("div", { className: "text-xs text-gray-500 mb-2", children: ["valuesTick: ", valuesTick] }), jsx("div", { className: "flex-1 overflow-auto", children: !selectedNode && !selectedEdge ? (jsxs("div", { children: [jsx("div", { className: "text-gray-500", children: "Select a node or edge." }), globalValidationIssues && globalValidationIssues.length > 0 && (jsxs("div", { className: "mt-2 text-xs bg-red-50 border border-red-200 rounded px-2 py-1", children: [jsx("div", { className: "font-semibold mb-1", children: "Validation" }), jsx("ul", { className: "list-disc ml-4", children: globalValidationIssues.map((m, i) => (jsxs("li", { className: "flex items-center gap-1", children: [jsx(IssueBadge, { level: m.level, size: 24, className: "w-6 h-6" }), jsx("span", { children: `${m.code}: ${m.message}` })] }, i))) })] }))] })) : selectedEdge ? (jsxs("div", { children: [jsxs("div", { className: "mb-2", children: [jsxs("div", { children: ["Edge: ", selectedEdge.id] }), jsxs("div", { children: [selectedEdge.source.nodeId, ".", selectedEdge.source.handle, " \u2192", " ", selectedEdge.target.nodeId, ".", selectedEdge.target.handle] }), jsxs("div", { className: "flex items-center gap-2 mt-1", children: [jsxs("label", { className: "w-20 flex flex-col", children: [jsx("span", { children: "Type" }), jsx("span", { className: "text-gray-500 text-[11px]", children: "DataTypeId" })] }), jsxs("select", { className: "border border-gray-300 rounded px-2 py-1 focus:outline-none focus:ring-2 focus:ring-blue-500 w-full", value: selectedEdge.typeId ?? "", onChange: (e) => {
+                                                const v = e.target.value;
+                                                const next = v === "" ? undefined : v;
+                                                updateEdgeType(selectedEdge.id, next);
+                                            }, children: [jsx("option", { value: "", children: "(infer from source)" }), Array.from(registry.types.keys()).map((tid) => (jsx("option", { value: tid, children: tid }, tid)))] })] })] }), selectedEdgeValidation.length > 0 && (jsxs("div", { className: "mt-2 text-xs bg-red-50 border border-red-200 rounded px-2 py-1", children: [jsx("div", { className: "font-semibold mb-1", children: "Validation" }), jsx("ul", { className: "list-disc ml-4", children: selectedEdgeValidation.map((m, i) => (jsxs("li", { className: "flex items-center gap-1", children: [jsx(IssueBadge, { level: m.level, size: 24, className: "w-6 h-6" }), jsx("span", { children: `${m.code}: ${m.message}` })] }, i))) })] }))] })) : (jsxs("div", { children: [selectedNode && (jsxs("div", { className: "mb-2", children: [jsxs("div", { children: ["Node: ", selectedNode.nodeId] }), jsxs("div", { children: ["Type: ", selectedNode.typeId] }), !!selectedNodeStatus?.lastError && (jsx("div", { className: "mt-2 text-sm text-red-700 bg-red-50 border border-red-200 rounded px-2 py-1 break-words", children: String(selectedNodeStatus.lastError?.message ??
                                         selectedNodeStatus.lastError) }))] })), jsxs("div", { className: "mb-2", children: [jsx("div", { className: "font-semibold mb-1", children: "Inputs" }), inputHandles.length === 0 ? (jsx("div", { className: "text-gray-500", children: "No inputs" })) : (inputHandles.map((h) => {
                                     const typeId = getInputTypeId(selectedDesc?.inputs, h);
                                     const isLinked = def.edges.some((e) => e.target.nodeId === selectedNodeId &&
@@ -1768,73 +1943,114 @@ function Inspector({ debug, autoScroll, hideWorkbench, onAutoScrollChange, onHid
                                         })()] }, h))))] }), selectedNodeValidation.length > 0 && (jsxs("div", { className: "mt-2 text-xs bg-red-50 border border-red-200 rounded px-2 py-1", children: [jsx("div", { className: "font-semibold mb-1", children: "Validation" }), jsx("ul", { className: "list-disc ml-4", children: selectedNodeValidation.map((m, i) => (jsxs("li", { className: "flex items-center gap-1", children: [jsx(IssueBadge, { level: m.level, size: 24, className: "w-6 h-6" }), jsx("span", { children: `${m.code}: ${m.message}` })] }, i))) })] }))] })) }), debug && (jsx("div", { className: "mt-3 flex-none min-h-0 h-[50%]", children: jsx(DebugEvents, { autoScroll: !!autoScroll, hideWorkbench: !!hideWorkbench, onAutoScrollChange: onAutoScrollChange, onHideWorkbenchChange: onHideWorkbenchChange }) }))] }));
 }
 
+function NodeHandles({ data, isConnectable, inputClassName = "!w-2 !h-2 !bg-gray-600", outputClassName = "!w-2 !h-2 !bg-gray-600", getClassName, renderLabel, labelClassName = "absolute text-[11px] text-gray-700 dark:text-gray-300 pointer-events-none", }) {
+    const layout = data.handleLayout ?? [];
+    const byId = React.useMemo(() => {
+        const m = new Map();
+        for (const h of layout) {
+            m.set(h.id, { position: h.position, y: h.y, type: h.type });
+        }
+        return m;
+    }, [layout]);
+    return (jsxs(Fragment, { children: [(data.inputHandles ?? []).map((h) => {
+                const placed = byId.get(h.id);
+                const position = placed?.position ?? Position.Left;
+                const y = placed?.y;
+                const cls = getClassName?.({ kind: "input", id: h.id, type: "target" }) ??
+                    inputClassName;
+                return (jsxs(React.Fragment, { children: [jsx(Handle, { id: h.id, type: "target", position: position, isConnectable: isConnectable, className: cls, style: y !== undefined ? { top: y } : undefined }), renderLabel && (jsx("div", { className: labelClassName + " left-2", style: {
+                                top: (y ?? 0) - 8,
+                                right: "50%",
+                                whiteSpace: "nowrap",
+                                overflow: "hidden",
+                                textOverflow: "ellipsis",
+                            }, children: renderLabel({ kind: "input", id: h.id }) }))] }, h.id));
+            }), (data.outputHandles ?? []).map((h) => {
+                const placed = byId.get(h.id);
+                const position = placed?.position ?? Position.Right;
+                const y = placed?.y;
+                const cls = getClassName?.({ kind: "output", id: h.id, type: "source" }) ??
+                    outputClassName;
+                return (jsxs(React.Fragment, { children: [jsx(Handle, { id: h.id, type: "source", position: position, isConnectable: isConnectable, className: cls, style: y !== undefined ? { top: y } : undefined }), renderLabel && (jsx("div", { className: labelClassName + " right-2", style: {
+                                top: (y ?? 0) - 8,
+                                left: "50%",
+                                textAlign: "right",
+                                whiteSpace: "nowrap",
+                                overflow: "hidden",
+                                textOverflow: "ellipsis",
+                            }, children: renderLabel({ kind: "output", id: h.id }) }))] }, h.id));
+            })] }));
+}
+
 const DefaultNode = React.memo(function DefaultNode({ id, data, selected, isConnectable, }) {
+    const updateNodeInternals = useUpdateNodeInternals();
     const { typeId, showValues, inputValues, outputValues, toString } = data;
     const inputEntries = data.inputHandles ?? [];
     const outputEntries = data.outputHandles ?? [];
+    React.useEffect(() => {
+        updateNodeInternals(id);
+    }, [
+        id,
+        inputEntries.length,
+        outputEntries.length,
+        showValues,
+        updateNodeInternals,
+    ]);
     const status = data.status ?? { activeRuns: 0 };
     const validation = data.validation ?? {
         inputs: [],
         outputs: [],
         issues: [],
     };
-    const HEADER_SIZE = 24;
-    const ROW_SIZE = 22;
-    const maxRows = Math.max(inputEntries.length, outputEntries.length);
-    const minHeight = HEADER_SIZE + maxRows * ROW_SIZE;
-    const minWidth = data.showValues ? 320 : 240;
-    const topFor = (i) => HEADER_SIZE + i * ROW_SIZE + ROW_SIZE / 2;
     const hasError = !!status.lastError;
-    const hasValidationError = validation.issues.some((i) => i.level === "error");
-    const hasValidationWarning = !hasValidationError && validation.issues.length > 0;
     const isRunning = !!status.activeRuns;
-    const isInvalid = !!status.invalidated && !isRunning && !hasError;
-    // Border color encodes severity; thickness encodes selection; style (dashed) encodes invalidated
-    const borderWidth = selected ? "border-2" : "border";
-    const borderStyle = isInvalid ? "border-dashed" : "border-solid";
-    const borderColor = hasError || hasValidationError
-        ? "border-red-500"
-        : hasValidationWarning
-            ? "border-amber-500"
-            : isRunning
-                ? "border-blue-500"
-                : "border-gray-500 dark:border-gray-400";
-    const ringClasses = isRunning
-        ? "ring-2 ring-blue-200 dark:ring-blue-900"
-        : undefined;
-    const borderClasses = cx(borderWidth, borderStyle, borderColor, ringClasses);
+    const containerBorder = getNodeBorderClassNames({
+        selected,
+        status,
+        validation,
+    });
     const pct = Math.round(Math.max(0, Math.min(1, Number(status.progress) || 0)) * 100);
-    return (jsxs("div", { className: cx("rounded-lg bg-white/70 !dark:bg-stone-900", borderClasses), style: { position: "relative", minHeight: minHeight, minWidth }, children: [jsxs("div", { className: "flex h-6 items-center justify-center px-2 border-b border-solid border-gray-500 dark:border-gray-400 text-gray-600 dark:text-gray-300", children: [jsx("strong", { className: "flex-1 h-full leading-6 text-xs", children: typeId }), jsxs("div", { className: "flex items-center gap-1", children: [hasError && (jsx("span", { title: String(status.lastError?.message ?? status.lastError), children: jsx(XCircleIcon, { size: 12, weight: "fill", className: "text-red-500" }) })), validation.issues && validation.issues.length > 0 && (jsx(IssueBadge, { level: validation.issues.some((i) => i.level === "error")
+    return (jsxs("div", { className: cx("rounded-lg bg-white/70 !dark:bg-stone-900", containerBorder), style: {
+            position: "relative",
+            minWidth: typeof data.renderWidth === "number" ? data.renderWidth : undefined,
+            minHeight: typeof data.renderHeight === "number" ? data.renderHeight : undefined,
+        }, children: [jsxs("div", { className: "flex items-center justify-center px-2 border-b border-solid border-gray-500 dark:border-gray-400 text-gray-600 dark:text-gray-300", style: {
+                    maxHeight: NODE_HEADER_HEIGHT_PX,
+                    minHeight: NODE_HEADER_HEIGHT_PX,
+                }, children: [jsx("strong", { className: "flex-1 h-full text-sm", style: { lineHeight: `${NODE_HEADER_HEIGHT_PX}px` }, children: typeId }), jsxs("div", { className: "flex items-center gap-1", children: [hasError && (jsx("span", { title: String(status.lastError?.message ?? status.lastError), children: jsx(XCircleIcon, { size: 12, weight: "fill", className: "text-red-500" }) })), validation.issues && validation.issues.length > 0 && (jsx(IssueBadge, { level: validation.issues.some((i) => i.level === "error")
                                     ? "error"
                                     : "warning", size: 12, className: "w-3 h-3", title: validation.issues
                                     .map((v) => `${v.code}: ${v.message}`)
-                                    .join("; ") })), jsxs("span", { className: "text-[10px] opacity-70", children: ["(", id, ")"] })] })] }), (isRunning || pct > 0) && (jsx("div", { className: "h-1 bg-blue-200 dark:bg-blue-900", children: jsx("div", { className: "h-1 bg-blue-500 transition-all", style: { width: `${pct}%` } }) })), inputEntries.map((entry, i) => {
-                const vIssues = validation.inputs.filter((v) => v.handle === entry.id);
-                const hasAny = vIssues.length > 0;
-                const hasErr = vIssues.some((v) => v.level === "error");
-                const title = vIssues.map((v) => `${v.code}: ${v.message}`).join("; ");
-                return (jsxs(React.Fragment, { children: [jsx(Handle, { id: entry.id, type: "target", position: Position.Left, isConnectable: isConnectable, className: cx("!w-3 !h-3 !bg-white !dark:bg-stone-900 !border-gray-500 dark:!border-gray-400", hasAny && (hasErr ? "!border-red-500" : "!border-amber-500")), style: { left: -5, top: topFor(i) } }), jsxs("div", { className: "absolute left-2 text-[11px] text-gray-700 dark:text-gray-300 pointer-events-none", style: {
-                                top: topFor(i) - 8,
-                                right: "50%",
-                                whiteSpace: "nowrap",
-                                overflow: "hidden",
-                                textOverflow: "ellipsis",
-                            }, title: `${entry.id}: ${entry.typeId}`, children: [entry.id, hasAny && (jsx(IssueBadge, { level: hasErr ? "error" : "warning", size: 12, className: "ml-1", title: title })), showValues && (jsx("span", { className: "ml-1 opacity-60", children: toString(entry.typeId, inputValues?.[entry.id]) }))] })] }, `in-${entry.id}`));
-            }), outputEntries.map((entry, i) => {
-                const vIssues = validation.outputs.filter((v) => v.handle === entry.id);
-                const hasAny = vIssues.length > 0;
-                const hasErr = vIssues.some((v) => v.level === "error");
-                const title = vIssues.map((v) => `${v.code}: ${v.message}`).join("; ");
-                const resolved = resolveOutputDisplay(outputValues?.[entry.id], entry.typeId);
-                return (jsxs(React.Fragment, { children: [jsx(Handle, { id: entry.id, type: "source", position: Position.Right, isConnectable: isConnectable, className: cx("!w-3 !h-3 !bg-white !dark:bg-stone-900 !border-gray-500 dark:!border-gray-400 !rounded-none", hasAny && (hasErr ? "!border-red-500" : "!border-amber-500")), style: { right: -5, top: topFor(i) } }), jsxs("div", { className: "absolute right-2 text-[11px] text-gray-700 dark:text-gray-300 pointer-events-none", style: {
-                                top: topFor(i) - 8,
-                                textAlign: "right",
-                                left: "50%",
-                                whiteSpace: "nowrap",
-                                overflow: "hidden",
-                                textOverflow: "ellipsis",
-                            }, title: `${entry.id}: ${entry.typeId}`, children: [entry.id, hasAny && (jsx(IssueBadge, { level: hasErr ? "error" : "warning", size: 12, className: "ml-1", title: title })), showValues && (jsx("span", { className: "ml-1 opacity-60", children: toString(resolved.typeId, resolved.value) }))] })] }, `out-${entry.id}`));
-            })] }));
+                                    .join("; ") })), jsxs("span", { className: "text-[10px] opacity-70", children: ["(", id, ")"] })] })] }), jsx("div", { className: cx("h-px", (isRunning || pct > 0) && "bg-blue-200 dark:bg-blue-900"), children: jsx("div", { className: cx("h-px transition-all", (isRunning || pct > 0) && "bg-blue-500"), style: { width: isRunning || pct > 0 ? `${pct}%` : 0 } }) }), jsx(NodeHandles, { data: data, isConnectable: isConnectable, getClassName: ({ kind, id }) => {
+                    const vIssues = (kind === "input" ? validation.inputs : validation.outputs).filter((v) => v.handle === id);
+                    const hasAny = vIssues.length > 0;
+                    const hasErr = vIssues.some((v) => v.level === "error");
+                    return cx("!w-3 !h-3 !bg-white !dark:bg-stone-900 !border-gray-500 dark:!border-gray-400", kind === "output" && "!rounded-none", hasAny && (hasErr ? "!border-red-500" : "!border-amber-500"));
+                }, renderLabel: ({ kind, id }) => {
+                    const entries = kind === "input" ? inputEntries : outputEntries;
+                    const entry = entries.find((e) => e.id === id);
+                    if (!entry)
+                        return id;
+                    const vIssues = (kind === "input" ? validation.inputs : validation.outputs).filter((v) => v.handle === id);
+                    const hasAny = vIssues.length > 0;
+                    const hasErr = vIssues.some((v) => v.level === "error");
+                    const title = vIssues
+                        .map((v) => `${v.code}: ${v.message}`)
+                        .join("; ");
+                    // Compose label with truncated value to prevent layout growth
+                    const valueText = (() => {
+                        if (!showValues)
+                            return undefined;
+                        if (kind === "input") {
+                            const txt = toString(entry.typeId, inputValues?.[entry.id]);
+                            return typeof txt === "string" ? txt : String(txt);
+                        }
+                        const resolved = resolveOutputDisplay(outputValues?.[entry.id], entry.typeId);
+                        const txt = toString(resolved.typeId, resolved.value);
+                        return typeof txt === "string" ? txt : String(txt);
+                    })();
+                    return (jsxs("span", { className: "flex items-center gap-1 w-full", children: [kind === "output" ? (jsxs(Fragment, { children: [valueText !== undefined && (jsx("span", { className: "opacity-60 truncate pl-1", style: { flex: 1, minWidth: 0, maxWidth: "100%" }, children: valueText })), jsx("span", { className: "truncate shrink-0", style: { maxWidth: "40%" }, children: id })] })) : (jsxs(Fragment, { children: [jsx("span", { className: "truncate shrink-0", style: { maxWidth: "40%" }, children: id }), valueText !== undefined && (jsx("span", { className: "opacity-60 truncate pr-1", style: { flex: 1, minWidth: 0, maxWidth: "100%" }, children: valueText }))] })), hasAny && (jsx(IssueBadge, { level: hasErr ? "error" : "warning", size: 12, className: "shrink-0", title: title }))] }));
+                } })] }));
 });
 DefaultNode.displayName = "DefaultNode";
 
@@ -1999,6 +2215,58 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement }, r
     const { wb, registry, inputsMap, outputsMap, valuesTick, nodeStatus, edgeStatus, validationByNode, validationByEdge, } = useWorkbenchContext();
     const nodeValidation = validationByNode;
     const edgeValidation = validationByEdge.errors;
+    // Keep stable references for nodes/edges to avoid unnecessary updates
+    const prevNodesRef = useRef([]);
+    const prevEdgesRef = useRef([]);
+    function retainStabilityById(prev, next, isSame) {
+        if (prev.length === 0)
+            return next;
+        const map = new Map();
+        for (const p of prev)
+            map.set(p.id, p);
+        const out = new Array(next.length);
+        for (let i = 0; i < next.length; i++) {
+            const n = next[i];
+            const p = map.get(n.id);
+            out[i] = p && isSame(p, n) ? p : n;
+        }
+        return out;
+    }
+    const isSameNode = (a, b) => {
+        // Compare the parts that affect rendering
+        const pick = (n) => ({
+            position: n.position,
+            type: n.type,
+            selected: n.selected,
+            initialWidth: n.initialWidth,
+            initialHeight: n.initialHeight,
+            data: n.data && {
+                typeId: n.data.typeId,
+                inputHandles: n.data.inputHandles,
+                outputHandles: n.data.outputHandles,
+                showValues: n.data.showValues,
+                inputValues: n.data.inputValues,
+                outputValues: n.data.outputValues,
+                status: n.data.status,
+                validation: n.data.validation,
+            },
+        });
+        return isEqual(pick(a), pick(b));
+    };
+    const isSameEdge = (a, b) => {
+        const pick = (e) => ({
+            source: e.source,
+            target: e.target,
+            sourceHandle: e.sourceHandle,
+            targetHandle: e.targetHandle,
+            selected: e.selected,
+            animated: e.animated,
+            style: e.style,
+            label: e.label,
+            type: e.type,
+        });
+        return isEqual(pick(a), pick(b));
+    };
     // Expose imperative API
     const rfInstanceRef = useRef(null);
     useImperativeHandle(ref, () => ({
@@ -2009,7 +2277,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement }, r
             catch { }
         },
     }));
-    const { onConnect, onNodesChange, onEdgesChange, onEdgesDelete, onNodesDelete, onSelectionChange, } = useWorkbenchBridge(wb);
+    const { onConnect, onNodesChange, onEdgesChange, onEdgesDelete, onNodesDelete, } = useWorkbenchBridge(wb);
     const { nodeTypes, resolveNodeType } = useMemo(() => {
         // Build nodeTypes map using UI extension registry
         const ui = wb.getUI();
@@ -2047,8 +2315,91 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement }, r
             selectedNodeIds: new Set(sel.nodes),
             selectedEdgeIds: new Set(sel.edges),
         });
-        // console.info(nodeTypes, out);
-        return out;
+        // Retain references for unchanged items
+        const stableNodes = retainStabilityById(prevNodesRef.current, out.nodes, isSameNode);
+        const stableEdges = retainStabilityById(prevEdgesRef.current, out.edges, isSameEdge);
+        // Debug: log updates/additions/removals (use value equality, not reference)
+        try {
+            const prevNodeIds = new Set(prevNodesRef.current.map((n) => n.id));
+            const nextNodeIds = new Set(out.nodes.map((n) => n.id));
+            const addedNodeIds = out.nodes
+                .filter((n) => !prevNodeIds.has(n.id))
+                .map((n) => n.id);
+            const removedNodeIds = prevNodesRef.current
+                .filter((n) => !nextNodeIds.has(n.id))
+                .map((n) => n.id);
+            const prevNodeMap = new Map(prevNodesRef.current.map((n) => [n.id, n]));
+            const changedNodeIds = out.nodes
+                .filter((n) => {
+                const p = prevNodeMap.get(n.id);
+                return p ? !isSameNode(p, n) : false;
+            })
+                .map((n) => n.id);
+            // Detect handle updates (ids/length changes) for targeted debug
+            const toIds = (arr) => Array.isArray(arr) ? arr.map((h) => h?.id) : [];
+            const handlesEqual = (a, b) => {
+                const aIds = toIds(a);
+                const bIds = toIds(b);
+                if (aIds.length !== bIds.length)
+                    return false;
+                for (let i = 0; i < aIds.length; i++) {
+                    if (aIds[i] !== bIds[i])
+                        return false;
+                }
+                return true;
+            };
+            const handleChanged = out.nodes
+                .filter((n) => {
+                const p = prevNodeMap.get(n.id);
+                if (!p)
+                    return false;
+                const inChanged = !handlesEqual(p.data?.inputHandles, n.data?.inputHandles);
+                const outChanged = !handlesEqual(p.data?.outputHandles, n.data?.outputHandles);
+                return inChanged || outChanged;
+            })
+                .map((n) => n.id);
+            const prevEdgeIds = new Set(prevEdgesRef.current.map((e) => e.id));
+            const nextEdgeIds = new Set(out.edges.map((e) => e.id));
+            const addedEdgeIds = out.edges
+                .filter((e) => !prevEdgeIds.has(e.id))
+                .map((e) => e.id);
+            const removedEdgeIds = prevEdgesRef.current
+                .filter((e) => !nextEdgeIds.has(e.id))
+                .map((e) => e.id);
+            const prevEdgeMap = new Map(prevEdgesRef.current.map((e) => [e.id, e]));
+            const changedEdgeIds = out.edges
+                .filter((e) => {
+                const p = prevEdgeMap.get(e.id);
+                return p ? !isSameEdge(p, e) : false;
+            })
+                .map((e) => e.id);
+            if (addedNodeIds.length ||
+                removedNodeIds.length ||
+                changedNodeIds.length ||
+                handleChanged.length) {
+                // eslint-disable-next-line no-console
+                console.debug("[WorkbenchCanvas] node updates", {
+                    added: addedNodeIds,
+                    removed: removedNodeIds,
+                    changed: changedNodeIds,
+                    handleChanged,
+                });
+            }
+            if (addedEdgeIds.length ||
+                removedEdgeIds.length ||
+                changedEdgeIds.length) {
+                // eslint-disable-next-line no-console
+                console.debug("[WorkbenchCanvas] edge updates", {
+                    added: addedEdgeIds,
+                    removed: removedEdgeIds,
+                    changed: changedEdgeIds,
+                });
+            }
+        }
+        catch { }
+        prevNodesRef.current = stableNodes;
+        prevEdgesRef.current = stableEdges;
+        return { nodes: stableNodes, edges: stableEdges };
     }, [
         showValues,
         inputsMap,
@@ -2060,9 +2411,9 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement }, r
         edgeStatus,
         nodeValidation,
         edgeValidation,
-        nodeTypes,
         resolveNodeType,
     ]);
+    const throttled = useThrottledValue({ nodes, edges }, 100);
     const [menuOpen, setMenuOpen] = useState(false);
     const [menuPos, setMenuPos] = useState(null);
     const [nodeMenuOpen, setNodeMenuOpen] = useState(false);
@@ -2089,7 +2440,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement }, r
     const addNodeAt = (typeId, pos) => {
         wb.addNode({ typeId, position: pos });
     };
-    return (jsx("div", { className: "w-full h-full", onContextMenu: onContextMenu, children: jsxs(ReactFlow, { nodes: nodes, edges: edges, nodeTypes: nodeTypes, selectionOnDrag: true, onConnect: onConnect, onEdgesChange: onEdgesChange, onEdgesDelete: onEdgesDelete, onNodesDelete: onNodesDelete, onNodesChange: onNodesChange, onSelectionChange: onSelectionChange, deleteKeyCode: ["Backspace", "Delete"], fitView: true, onInit: (inst) => (rfInstanceRef.current = inst), children: [jsx(Background, {}), jsx(MiniMap, {}), jsx(Controls, {}), jsx(DefaultContextMenu, { open: menuOpen, clientPos: menuPos, onAdd: addNodeAt, onClose: () => setMenuOpen(false) }), jsx(NodeContextMenu, { open: nodeMenuOpen, clientPos: nodeMenuPos, nodeId: nodeAtMenu, onClose: () => setNodeMenuOpen(false) })] }) }));
+    return (jsx("div", { className: "w-full h-full", onContextMenu: onContextMenu, children: jsxs(ReactFlow, { nodes: throttled.nodes, edges: throttled.edges, nodeTypes: nodeTypes, onlyRenderVisibleElements: true, selectionOnDrag: true, onConnect: onConnect, onEdgesChange: onEdgesChange, onEdgesDelete: onEdgesDelete, onNodesDelete: onNodesDelete, onNodesChange: onNodesChange, deleteKeyCode: ["Backspace", "Delete"], fitView: true, onInit: (inst) => (rfInstanceRef.current = inst), children: [jsx(Background, {}), jsx(MiniMap, {}), jsx(Controls, {}), jsx(DefaultContextMenu, { open: menuOpen, clientPos: menuPos, onAdd: addNodeAt, onClose: () => setMenuOpen(false) }), jsx(NodeContextMenu, { open: nodeMenuOpen, clientPos: nodeMenuPos, nodeId: nodeAtMenu, onClose: () => setNodeMenuOpen(false) })] }) }));
 });
 
 function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, example, onExampleChange, engine, onEngineChange, backendKind, onBackendKindChange, httpBaseUrl, onHttpBaseUrlChange, wsUrl, onWsUrlChange, debug, onDebugChange, showValues, onShowValuesChange, hideWorkbench, onHideWorkbenchChange, overrides, onInit, onChange, }) {
@@ -2201,9 +2552,14 @@ function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, ex
         if (!ex)
             return;
         const { registry: r, def } = await ex.load();
-        if (r) {
-            setRegistry(r);
-            wb.setRegistry(r);
+        // Keep registry consistent with backend:
+        // - For local backend, allow example to provide its own registry
+        // - For remote backend, NEVER overwrite the hydrated remote registry
+        if (backendKind === "local") {
+            if (r) {
+                setRegistry(r);
+                wb.setRegistry(r);
+            }
         }
         await wb.load(def);
         // Build a local runtime so seeded defaults are visible pre-run
@@ -2211,7 +2567,15 @@ function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, ex
         runAutoLayout();
         setExampleState(key);
         onExampleChange?.(key);
-    }, [runner, wb, onExampleChange, runAutoLayout, examples, setRegistry]);
+    }, [
+        runner,
+        wb,
+        onExampleChange,
+        runAutoLayout,
+        examples,
+        setRegistry,
+        backendKind,
+    ]);
     const downloadGraph = useCallback(() => {
         try {
             const def = wb.export();
@@ -2517,7 +2881,7 @@ function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, ex
                             ? "Stop engine before switching example"
                             : undefined, children: [jsx("option", { value: "", children: "Select Example\u2026" }), examples.map((ex) => (jsx("option", { value: ex.id, children: ex.label }, ex.id)))] }), jsx("label", { className: "ml-2 text-sm", children: "Backend:" }), jsxs("select", { className: "border border-gray-300 rounded px-2 py-1", value: backendKind, onChange: (e) => onBackendKindChange(e.target.value), disabled: runner.isRunning(), title: runner.isRunning()
                             ? "Stop engine before switching backend"
-                            : undefined, children: [jsx("option", { value: "local", children: "Local" }), jsx("option", { value: "remote-http", children: "Remote (HTTP)" }), jsx("option", { value: "remote-ws", children: "Remote (WebSocket)" })] }), backendKind === "remote-http" && (jsx("input", { className: "ml-2 border border-gray-300 rounded px-2 py-1 w-72", placeholder: "http://127.0.0.1:18080", value: httpBaseUrl, onChange: (e) => onHttpBaseUrlChange(e.target.value) })), backendKind === "remote-ws" && (jsx("input", { className: "ml-2 border border-gray-300 rounded px-2 py-1 w-72", placeholder: "ws://127.0.0.1:18081", value: wsUrl, onChange: (e) => onWsUrlChange(e.target.value) })), jsxs("select", { className: "border border-gray-300 rounded px-2 py-1", value: runner.getRunningEngine() ?? engine ?? "", onChange: (e) => {
+                            : undefined, children: [jsx("option", { value: "local", children: "Local" }), jsx("option", { value: "remote-http", children: "Remote (HTTP)" }), jsx("option", { value: "remote-ws", children: "Remote (WebSocket)" })] }), backendKind === "remote-http" && !!onHttpBaseUrlChange && (jsx("input", { className: "ml-2 border border-gray-300 rounded px-2 py-1 w-72", placeholder: "http://127.0.0.1:18080", value: httpBaseUrl, onChange: (e) => onHttpBaseUrlChange(e.target.value) })), backendKind === "remote-ws" && !!onWsUrlChange && (jsx("input", { className: "ml-2 border border-gray-300 rounded px-2 py-1 w-72", placeholder: "ws://127.0.0.1:18081", value: wsUrl, onChange: (e) => onWsUrlChange(e.target.value) })), jsxs("select", { className: "border border-gray-300 rounded px-2 py-1", value: runner.getRunningEngine() ?? engine ?? "", onChange: (e) => {
                             const kind = e.target.value || undefined;
                             onEngineChange?.(kind);
                         }, children: [jsx("option", { value: "", children: "Select Engine\u2026" }), jsx("option", { value: "push", children: "Push" }), jsx("option", { value: "batched", children: "Batched" }), jsx("option", { value: "pull", children: "Pull" }), jsx("option", { value: "hybrid", children: "Hybrid" }), jsx("option", { value: "step", children: "Step" })] }), runner.getRunningEngine() === "step" && (jsx("button", { className: "ml-2 border border-gray-300 rounded px-2 py-1.5", onClick: () => runner.step(), disabled: !runner.isRunning(), children: "Step" })), runner.getRunningEngine() === "batched" && (jsx("button", { className: "ml-2 border border-gray-300 rounded px-2 py-1.5", onClick: () => runner.flush(), disabled: !runner.isRunning(), children: "Flush" })), runner.isRunning() ? (jsx("button", { className: "border border-gray-300 rounded px-2 py-1.5", onClick: () => runner.dispose(), disabled: !runner.isRunning(), children: "Stop" })) : (jsx("button", { className: "border border-gray-300 rounded px-2 py-1.5", onClick: () => {
@@ -2560,5 +2924,5 @@ function WorkbenchStudio({ engine, onEngineChange, example, onExampleChange, bac
             }, httpBaseUrl: httpBaseUrl, onHttpBaseUrlChange: onHttpBaseUrlChange, wsUrl: wsUrl, onWsUrlChange: onWsUrlChange, debug: debug, onDebugChange: onDebugChange, showValues: showValues, onShowValuesChange: onShowValuesChange, hideWorkbench: hideWorkbench, onHideWorkbenchChange: onHideWorkbenchChange, overrides: overrides, onInit: onInit, onChange: onChange }) }));
 }
 
-export { AbstractWorkbench, CLIWorkbench, DefaultUIExtensionRegistry, InMemoryWorkbench, Inspector, LocalGraphRunner, RemoteGraphRunner, WorkbenchCanvas, WorkbenchContext, WorkbenchProvider, WorkbenchStudio, formatDataUrlAsLabel, formatDeclaredTypeSignature, getNodeBorderClassNames, preformatValueForDisplay, resolveOutputDisplay, summarizeDeep, toReactFlow, useQueryParamBoolean, useQueryParamString, useWorkbenchBridge, useWorkbenchContext, useWorkbenchGraphTick, useWorkbenchGraphUiTick, useWorkbenchVersionTick };
+export { AbstractWorkbench, CLIWorkbench, DefaultUIExtensionRegistry, InMemoryWorkbench, Inspector, LocalGraphRunner, NodeHandles, RemoteGraphRunner, WorkbenchCanvas, WorkbenchContext, WorkbenchProvider, WorkbenchStudio, formatDataUrlAsLabel, formatDeclaredTypeSignature, getNodeBorderClassNames, preformatValueForDisplay, resolveOutputDisplay, summarizeDeep, toReactFlow, useQueryParamBoolean, useQueryParamString, useThrottledValue, useWorkbenchBridge, useWorkbenchContext, useWorkbenchGraphTick, useWorkbenchGraphUiTick, useWorkbenchVersionTick };
 //# sourceMappingURL=index.js.map