diff --git a/stores/workflow/logger.ts b/stores/workflow/logger.ts index 032b8e6b31..f7ee585cc1 100644 --- a/stores/workflow/logger.ts +++ b/stores/workflow/logger.ts @@ -6,6 +6,7 @@ export function initializeStateLogger() { current: { blocks: state.blocks, edges: state.edges, + loops: state.loops, }, history: { past: state.history.past, diff --git a/stores/workflow/registry.ts b/stores/workflow/registry.ts index 4024ce5b65..cf199aa6ee 100644 --- a/stores/workflow/registry.ts +++ b/stores/workflow/registry.ts @@ -40,7 +40,7 @@ export const useWorkflowRegistry = create()( useWorkflowStore.setState({ blocks, edges, - loops: loops || {}, + loops, history: history || { past: [], present: { @@ -108,14 +108,15 @@ export const useWorkflowRegistry = create()( newActiveWorkflowId = remainingIds[0] const savedState = localStorage.getItem(`workflow-${newActiveWorkflowId}`) if (savedState) { - const { blocks, edges, history } = JSON.parse(savedState) + const { blocks, edges, history, loops } = JSON.parse(savedState) useWorkflowStore.setState({ blocks, edges, + loops, history: history || { past: [], present: { - state: { blocks, edges }, + state: { blocks, edges, loops }, timestamp: Date.now(), action: 'Initial state', },