@bian-womp/spark-workbench 0.2.92 → 0.2.93

package/lib/cjs/index.cjs

@@ -3657,6 +3657,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
     const [systemErrors, setSystemErrors] = React.useState([]);
     const [registryErrors, setRegistryErrors] = React.useState([]);
     const [inputValidationErrors, setInputValidationErrors] = React.useState([]);
+    const [registryVersion, setRegistryVersion] = React.useState(0);
     const clearSystemErrors = React.useCallback(() => setSystemErrors([]), []);
     const clearRegistryErrors = React.useCallback(() => setRegistryErrors([]), []);
     const clearInputValidationErrors = React.useCallback(() => setInputValidationErrors([]), []);
@@ -4319,6 +4320,8 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             try {
                 setRegistry(newReg);
                 wb.setRegistry(newReg);
+                // Increment registry version to trigger UI updates
+                setRegistryVersion((v) => v + 1);
                 // Trigger a graph update so the UI revalidates with new types/enums/nodes
                 try {
                     await runner.update(wb.def);
@@ -4516,6 +4519,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
         updateEdgeType,
         triggerExternal,
         uiVersion,
+        registryVersion,
         overrides,
         getNodeDisplayName,
         setNodeName,
@@ -4557,6 +4561,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
         wb,
         runner,
         uiVersion,
+        registryVersion,
         overrides,
         getNodeDisplayName,
         setNodeName,
@@ -5538,10 +5543,9 @@ function useKeyboardShortcutToast() {
 }
 
 const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, getDefaultNodeSize }, ref) => {
-    const { wb, registry, inputsMap, inputDefaultsMap, outputsMap, outputTypesMap, valuesTick, nodeStatus, edgeStatus, validationByNode, validationByEdge, uiVersion, runner, engineKind, overrides, } = useWorkbenchContext();
+    const { wb, registry, inputsMap, inputDefaultsMap, outputsMap, outputTypesMap, valuesTick, nodeStatus, edgeStatus, validationByNode, validationByEdge, uiVersion, registryVersion, runner, engineKind, overrides, } = useWorkbenchContext();
     const nodeValidation = validationByNode;
     const edgeValidation = validationByEdge.errors;
-    const [registryVersion, setRegistryVersion] = React.useState(0);
     const [historyState, setHistoryState] = React.useState(wb.getHistory());
     const prevNodesRef = React.useRef([]);
     const prevEdgesRef = React.useRef([]);
@@ -5635,6 +5639,7 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
         const resolver = (nodeTypeId) => custom.has(nodeTypeId) ? `spark-${nodeTypeId}` : "spark-default";
         return { nodeTypes: types, resolveNodeType: resolver };
         // Include uiVersion to recompute when custom renderers are registered
+        // Include registryVersion to recompute when registry enums/types change
     }, [wb, registry, uiVersion, ui]);
     const edgeTypes = React.useMemo(() => {
         // Use default edge renderer override if registered, otherwise use DefaultEdge
@@ -5922,12 +5927,6 @@ const WorkbenchCanvas = React.forwardRef(({ showValues, toString, toElement, get
     const onCloseSelectionMenu = React.useCallback(() => {
         setSelectionMenuOpen(false);
     }, []);
-    React.useEffect(() => {
-        const off = runner.on("registry", () => {
-            setRegistryVersion((v) => v + 1);
-        });
-        return () => off();
-    }, [runner]);
     React.useEffect(() => {
         const off = wb.on("historyChanged", (event) => {
             setHistoryState(event.history);