refactor(core): update node lookup on change

bcakmakoglu · May 31, 2024 · 9f103fc · 9f103fc
1 parent 77aa0bc
commit 9f103fc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packages/core/src/store/actions.ts b/packages/core/src/store/actions.ts
@@ -557,7 +557,11 @@ export function useActions(state: State): Actions {
     updateEdgeAction(oldEdge, newConnection, state.edges, findEdge, shouldReplaceId, state.hooks.error.trigger)
 
   const applyNodeChanges: Actions['applyNodeChanges'] = (changes) => {
-    return applyChanges(changes, state.nodes)
+    const changedNodes = applyChanges(changes, state.nodes)
+
+    updateNodeLookup(state.nodeLookup, changedNodes)
+
+    return changedNodes
   }
 
   const applyEdgeChanges: Actions['applyEdgeChanges'] = (changes) => {