Merge branch 'feat/workflow-continue-on-error' into deploy/dev

langgenius · Dec 5, 2024 · 0527005 · 0527005
2 parents d2c88fa + 31128ef
commit 0527005
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/web/app/components/workflow/nodes/_base/components/error-handle/error-handle-on-node.tsx b/web/app/components/workflow/nodes/_base/components/error-handle/error-handle-on-node.tsx
@@ -1,4 +1,6 @@
+import { useEffect } from 'react'
 import { useTranslation } from 'react-i18next'
+import { useUpdateNodeInternals } from 'reactflow'
 import { NodeSourceHandle } from '../node-handle'
 import { ErrorHandleTypeEnum } from './types'
 import type { Node } from '@/app/components/workflow/types'
@@ -12,6 +14,12 @@ const ErrorHandleOnNode = ({
 }: ErrorHandleOnNodeProps) => {
   const { t } = useTranslation()
   const { error_strategy } = data
+  const updateNodeInternals = useUpdateNodeInternals()
+
+  useEffect(() => {
+    if (error_strategy === ErrorHandleTypeEnum.failBranch)
+      updateNodeInternals(id)
+  }, [error_strategy, id, updateNodeInternals])
 
   if (!error_strategy)
     return null