fix: workflow run edge status (#3236)

2 years ago · 1384a6d0fd
parent 28089c98c1
commit 1384a6d0fd
1 changed files with 11 additions and 0 deletions
--- a/web/app/components/workflow/hooks/use-workflow-run.ts
+++ b/web/app/components/workflow/hooks/use-workflow-run.ts
@ -176,6 +176,8 @@ export const useWorkflowRun = () => {
          const {
            getNodes,
            setNodes,
+            edges,
+            setEdges,
          } = store.getState()
          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
            draft.task_id = task_id
@ -192,6 +194,15 @@ export const useWorkflowRun = () => {
            })
          })
          setNodes(newNodes)
+          const newEdges = produce(edges, (draft) => {
+            draft.forEach((edge) => {
+              edge.data = {
+                ...edge.data,
+                _runned: false,
+              }
+            })
+          })
+          setEdges(newEdges)

          if (onWorkflowStarted)
            onWorkflowStarted(params)