From 3ba2164d6411c3fee79af435ead8f3db6953cbb3 Mon Sep 17 00:00:00 2001
From: Emir Karabeg <emirkarabeg@berkeley.edu>
Date: Mon, 10 Feb 2025 15:57:36 -0800
Subject: [PATCH] Added loops to logs

---
 stores/workflow/logger.ts   | 1 +
 stores/workflow/registry.ts | 7 ++++---
 2 files changed, 5 insertions(+), 3 deletions(-)

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<WorkflowRegistry>()(
           useWorkflowStore.setState({
             blocks,
             edges,
-            loops: loops || {},
+            loops,
             history: history || {
               past: [],
               present: {
@@ -108,14 +108,15 @@ export const useWorkflowRegistry = create<WorkflowRegistry>()(
             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',
                   },