From 7ef48c32e05a5805be037f45122f9181f014c7bf Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Mon, 8 Dec 2025 11:35:54 -0800
Subject: [PATCH] fix workflow graph building logic for loop blocks (#4228)

---
 skyvern/forge/sdk/workflow/service.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/skyvern/forge/sdk/workflow/service.py b/skyvern/forge/sdk/workflow/service.py
index c03c973f..18810bf5 100644
--- a/skyvern/forge/sdk/workflow/service.py
+++ b/skyvern/forge/sdk/workflow/service.py
@@ -1326,7 +1326,8 @@ class WorkflowService:
         self,
         blocks: list[BlockTypeVar],
     ) -> tuple[str, dict[str, BlockTypeVar], dict[str, str | None]]:
-        all_blocks = get_all_blocks(blocks)
+        # all_blocks = get_all_blocks(blocks)
+        all_blocks = blocks
         label_to_block: dict[str, BlockTypeVar] = {}
         default_next_map: dict[str, str | None] = {}