From 1c5d07871fab0ff9b9ea12cde72bf5982e145fc6 Mon Sep 17 00:00:00 2001
From: StyleZhang <jasonapring2015@outlook.com>
Date: Wed, 13 Mar 2024 17:06:16 +0800
Subject: [PATCH] hooks

---
 .../components/workflow/hooks/use-workflow.ts    | 16 +++++++++-------
 1 file changed, 9 insertions(+), 7 deletions(-)

diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts
index 7cb8dab283..6a4d98b070 100644
--- a/web/app/components/workflow/hooks/use-workflow.ts
+++ b/web/app/components/workflow/hooks/use-workflow.ts
@@ -93,13 +93,15 @@ export const useWorkflow = () => {
       return list
 
     const traverse = (root: Node, callback: (node: Node) => void) => {
-      const incomers = getIncomers(root, nodes, edges)
+      if (root) {
+        const incomers = getIncomers(root, nodes, edges)
 
-      if (incomers.length) {
-        incomers.forEach((node) => {
-          callback(node)
-          traverse(node, callback)
-        })
+        if (incomers.length) {
+          incomers.forEach((node) => {
+            callback(node)
+            traverse(node, callback)
+          })
+        }
       }
     }
     traverse(currentNode, (node) => {
@@ -107,7 +109,7 @@ export const useWorkflow = () => {
     })
 
     const length = list.length
-    if (length && list.some(item => item.data.type === BlockEnum.Start)) {
+    if (length) {
       return list.reverse().filter((item) => {
         return SUPPORT_OUTPUT_VARS_NODE.includes(item.data.type)
       })