Introduce Plugins (#13836)

Signed-off-by: yihong0618 <zouzou0208@gmail.com>
Signed-off-by: -LAN- <laipz8200@outlook.com>
Signed-off-by: xhe <xw897002528@gmail.com>
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: takatost <takatost@gmail.com>
Co-authored-by: kurokobo <kuro664@gmail.com>
Co-authored-by: Novice Lee <novicelee@NoviPro.local>
Co-authored-by: zxhlyh <jasonapring2015@outlook.com>
Co-authored-by: AkaraChen <akarachen@outlook.com>
Co-authored-by: Yi <yxiaoisme@gmail.com>
Co-authored-by: Joel <iamjoel007@gmail.com>
Co-authored-by: JzoNg <jzongcode@gmail.com>
Co-authored-by: twwu <twwu@dify.ai>
Co-authored-by: Hiroshi Fujita <fujita-h@users.noreply.github.com>
Co-authored-by: AkaraChen <85140972+AkaraChen@users.noreply.github.com>
Co-authored-by: NFish <douxc512@gmail.com>
Co-authored-by: Wu Tianwei <30284043+WTW0313@users.noreply.github.com>
Co-authored-by: 非法操作 <hjlarry@163.com>
Co-authored-by: Novice <857526207@qq.com>
Co-authored-by: Hiroki Nagai <82458324+nagaihiroki-git@users.noreply.github.com>
Co-authored-by: Gen Sato <52241300+halogen22@users.noreply.github.com>
Co-authored-by: eux <euxuuu@gmail.com>
Co-authored-by: huangzhuo1949 <167434202+huangzhuo1949@users.noreply.github.com>
Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
Co-authored-by: lotsik <lotsik@mail.ru>
Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
Co-authored-by: nite-knite <nkCoding@gmail.com>
Co-authored-by: Jyong <76649700+JohnJyong@users.noreply.github.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: gakkiyomi <gakkiyomi@aliyun.com>
Co-authored-by: CN-P5 <heibai2006@gmail.com>
Co-authored-by: CN-P5 <heibai2006@qq.com>
Co-authored-by: Chuehnone <1897025+chuehnone@users.noreply.github.com>
Co-authored-by: yihong <zouzou0208@gmail.com>
Co-authored-by: Kevin9703 <51311316+Kevin9703@users.noreply.github.com>
Co-authored-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: Boris Feld <lothiraldan@gmail.com>
Co-authored-by: mbo <himabo@gmail.com>
Co-authored-by: mabo <mabo@aeyes.ai>
Co-authored-by: Warren Chen <warren.chen830@gmail.com>
Co-authored-by: JzoNgKVO <27049666+JzoNgKVO@users.noreply.github.com>
Co-authored-by: jiandanfeng <chenjh3@wangsu.com>
Co-authored-by: zhu-an <70234959+xhdd123321@users.noreply.github.com>
Co-authored-by: zhaoqingyu.1075 <zhaoqingyu.1075@bytedance.com>
Co-authored-by: 海狸大師 <86974027+yenslife@users.noreply.github.com>
Co-authored-by: Xu Song <xusong.vip@gmail.com>
Co-authored-by: rayshaw001 <396301947@163.com>
Co-authored-by: Ding Jiatong <dingjiatong@gmail.com>
Co-authored-by: Bowen Liang <liangbowen@gf.com.cn>
Co-authored-by: JasonVV <jasonwangiii@outlook.com>
Co-authored-by: le0zh <newlight@qq.com>
Co-authored-by: zhuxinliang <zhuxinliang@didiglobal.com>
Co-authored-by: k-zaku <zaku99@outlook.jp>
Co-authored-by: luckylhb90 <luckylhb90@gmail.com>
Co-authored-by: hobo.l <hobo.l@binance.com>
Co-authored-by: jiangbo721 <365065261@qq.com>
Co-authored-by: 刘江波 <jiangbo721@163.com>
Co-authored-by: Shun Miyazawa <34241526+miya@users.noreply.github.com>
Co-authored-by: EricPan <30651140+Egfly@users.noreply.github.com>
Co-authored-by: crazywoola <427733928@qq.com>
Co-authored-by: sino <sino2322@gmail.com>
Co-authored-by: Jhvcc <37662342+Jhvcc@users.noreply.github.com>
Co-authored-by: lowell <lowell.hu@zkteco.in>
Co-authored-by: Boris Polonsky <BorisPolonsky@users.noreply.github.com>
Co-authored-by: Ademílson Tonato <ademilsonft@outlook.com>
Co-authored-by: Ademílson Tonato <ademilson.tonato@refurbed.com>
Co-authored-by: IWAI, Masaharu <iwaim.sub@gmail.com>
Co-authored-by: Yueh-Po Peng (Yabi) <94939112+y10ab1@users.noreply.github.com>
Co-authored-by: Jason <ggbbddjm@gmail.com>
Co-authored-by: Xin Zhang <sjhpzx@gmail.com>
Co-authored-by: yjc980121 <3898524+yjc980121@users.noreply.github.com>
Co-authored-by: heyszt <36215648+hieheihei@users.noreply.github.com>
Co-authored-by: Abdullah AlOsaimi <osaimiacc@gmail.com>
Co-authored-by: Abdullah AlOsaimi <189027247+osaimi@users.noreply.github.com>
Co-authored-by: Yingchun Lai <laiyingchun@apache.org>
Co-authored-by: Hash Brown <hi@xzd.me>
Co-authored-by: zuodongxu <192560071+zuodongxu@users.noreply.github.com>
Co-authored-by: Masashi Tomooka <tmokmss@users.noreply.github.com>
Co-authored-by: aplio <ryo.091219@gmail.com>
Co-authored-by: Obada Khalili <54270856+obadakhalili@users.noreply.github.com>
Co-authored-by: Nam Vu <zuzoovn@gmail.com>
Co-authored-by: Kei YAMAZAKI <1715090+kei-yamazaki@users.noreply.github.com>
Co-authored-by: TechnoHouse <13776377+deephbz@users.noreply.github.com>
Co-authored-by: Riddhimaan-Senapati <114703025+Riddhimaan-Senapati@users.noreply.github.com>
Co-authored-by: MaFee921 <31881301+2284730142@users.noreply.github.com>
Co-authored-by: te-chan <t-nakanome@sakura-is.co.jp>
Co-authored-by: HQidea <HQidea@users.noreply.github.com>
Co-authored-by: Joshbly <36315710+Joshbly@users.noreply.github.com>
Co-authored-by: xhe <xw897002528@gmail.com>
Co-authored-by: weiwenyan-dev <154779315+weiwenyan-dev@users.noreply.github.com>
Co-authored-by: ex_wenyan.wei <ex_wenyan.wei@tcl.com>
Co-authored-by: engchina <12236799+engchina@users.noreply.github.com>
Co-authored-by: engchina <atjapan2015@gmail.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: 呆萌闷油瓶 <253605712@qq.com>
Co-authored-by: Kemal <kemalmeler@outlook.com>
Co-authored-by: Lazy_Frog <4590648+lazyFrogLOL@users.noreply.github.com>
Co-authored-by: Yi Xiao <54782454+YIXIAO0@users.noreply.github.com>
Co-authored-by: Steven sun <98230804+Tuyohai@users.noreply.github.com>
Co-authored-by: steven <sunzwj@digitalchina.com>
Co-authored-by: Kalo Chin <91766386+fdb02983rhy@users.noreply.github.com>
Co-authored-by: Katy Tao <34019945+KatyTao@users.noreply.github.com>
Co-authored-by: depy <42985524+h4ckdepy@users.noreply.github.com>
Co-authored-by: 胡春东 <gycm520@gmail.com>
Co-authored-by: Junjie.M <118170653@qq.com>
Co-authored-by: MuYu <mr.muzea@gmail.com>
Co-authored-by: Naoki Takashima <39912547+takatea@users.noreply.github.com>
Co-authored-by: Summer-Gu <37869445+gubinjie@users.noreply.github.com>
Co-authored-by: Fei He <droxer.he@gmail.com>
Co-authored-by: ybalbert001 <120714773+ybalbert001@users.noreply.github.com>
Co-authored-by: Yuanbo Li <ybalbert@amazon.com>
Co-authored-by: douxc <7553076+douxc@users.noreply.github.com>
Co-authored-by: liuzhenghua <1090179900@qq.com>
Co-authored-by: Wu Jiayang <62842862+Wu-Jiayang@users.noreply.github.com>
Co-authored-by: Your Name <you@example.com>
Co-authored-by: kimjion <45935338+kimjion@users.noreply.github.com>
Co-authored-by: AugNSo <song.tiankai@icloud.com>
Co-authored-by: llinvokerl <38915183+llinvokerl@users.noreply.github.com>
Co-authored-by: liusurong.lsr <liusurong.lsr@alibaba-inc.com>
Co-authored-by: Vasu Negi <vasu-negi@users.noreply.github.com>
Co-authored-by: Hundredwz <1808096180@qq.com>
Co-authored-by: Xiyuan Chen <52963600+GareArc@users.noreply.github.com>

web/app/components/workflow/hooks/use-workflow-run.ts

@@ -8,24 +8,16 @@ import { v4 as uuidV4 } from 'uuid'
 import { usePathname } from 'next/navigation'
 import { useWorkflowStore } from '../store'
 import { useNodesSyncDraft } from '../hooks'
-import {
-  BlockEnum,
-  NodeRunningStatus,
-  WorkflowRunningStatus,
-} from '../types'
-import { DEFAULT_ITER_TIMES } from '../constants'
+import { WorkflowRunningStatus } from '../types'
 import { useWorkflowUpdate } from './use-workflow-interactions'
+import { useWorkflowRunEvent } from './use-workflow-run-event/use-workflow-run-event'
 import { useStore as useAppStore } from '@/app/components/app/store'
 import type { IOtherOptions } from '@/service/base'
 import { ssePost } from '@/service/base'
 import { stopWorkflowRun } from '@/service/workflow'
 import { useFeaturesStore } from '@/app/components/base/features/hooks'
 import { AudioPlayerManager } from '@/app/components/base/audio-btn/audio.player.manager'
-import {
-  getFilesInLogs,
-} from '@/app/components/base/file-uploader/utils'
-import { ErrorHandleTypeEnum } from '@/app/components/workflow/nodes/_base/components/error-handle/types'
-import type { NodeTracing, VersionHistory } from '@/types/workflow'
+import type { VersionHistory } from '@/types/workflow'
 
 export const useWorkflowRun = () => {
   const store = useStoreApi()
@@ -35,6 +27,20 @@ export const useWorkflowRun = () => {
   const { doSyncWorkflowDraft } = useNodesSyncDraft()
   const { handleUpdateWorkflowCanvas } = useWorkflowUpdate()
   const pathname = usePathname()
+  const {
+    handleWorkflowStarted,
+    handleWorkflowFinished,
+    handleWorkflowFailed,
+    handleWorkflowNodeStarted,
+    handleWorkflowNodeFinished,
+    handleWorkflowNodeIterationStarted,
+    handleWorkflowNodeIterationNext,
+    handleWorkflowNodeIterationFinished,
+    handleWorkflowNodeRetry,
+    handleWorkflowAgentLog,
+    handleWorkflowTextChunk,
+    handleWorkflowTextReplace,
+  } = useWorkflowRunEvent()
 
   const handleBackupDraft = useCallback(() => {
     const {
@@ -113,6 +119,7 @@ export const useWorkflowRun = () => {
       onIterationNext,
       onIterationFinish,
       onNodeRetry,
+      onAgentLog,
       onError,
       ...restCallback
     } = callback || {}
@@ -132,8 +139,6 @@ export const useWorkflowRun = () => {
     if (appDetail?.mode === 'workflow')
       url = `/apps/${appDetail.id}/workflows/draft/run`
 
-    let prevNodeId = ''
-
     const {
       setWorkflowRunningData,
     } = workflowStore.getState()
@@ -166,584 +171,96 @@ export const useWorkflowRun = () => {
       },
       {
         onWorkflowStarted: (params) => {
-          const { task_id, data } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            setIterParallelLogMap,
-          } = workflowStore.getState()
-          const {
-            getNodes,
-            setNodes,
-            edges,
-            setEdges,
-          } = store.getState()
-          setIterParallelLogMap(new Map())
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            draft.task_id = task_id
-            draft.result = {
-              ...draft?.result,
-              ...data,
-              status: WorkflowRunningStatus.Running,
-            }
-          }))
-          const nodes = getNodes()
-          const newNodes = produce(nodes, (draft) => {
-            draft.forEach((node) => {
-              node.data._waitingRun = true
-              node.data._runningBranchId = undefined
-            })
-          })
-          setNodes(newNodes)
-          const newEdges = produce(edges, (draft) => {
-            draft.forEach((edge) => {
-              edge.data = {
-                ...edge.data,
-                _sourceRunningStatus: undefined,
-                _targetRunningStatus: undefined,
-                _waitingRun: true,
-              }
-            })
-          })
-          setEdges(newEdges)
+          handleWorkflowStarted(params)
 
           if (onWorkflowStarted)
             onWorkflowStarted(params)
         },
         onWorkflowFinished: (params) => {
-          const { data } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-          } = workflowStore.getState()
-
-          const isStringOutput = data.outputs && Object.keys(data.outputs).length === 1 && typeof data.outputs[Object.keys(data.outputs)[0]] === 'string'
-
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            draft.result = {
-              ...draft.result,
-              ...data,
-              files: getFilesInLogs(data.outputs),
-            } as any
-            if (isStringOutput) {
-              draft.resultTabActive = true
-              draft.resultText = data.outputs[Object.keys(data.outputs)[0]]
-            }
-          }))
-
-          prevNodeId = ''
+          handleWorkflowFinished(params)
 
           if (onWorkflowFinished)
             onWorkflowFinished(params)
         },
         onError: (params) => {
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-          } = workflowStore.getState()
-
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            draft.result = {
-              ...draft.result,
-              status: WorkflowRunningStatus.Failed,
-            }
-          }))
+          handleWorkflowFailed()
 
           if (onError)
             onError(params)
         },
         onNodeStarted: (params) => {
-          const { data } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            iterParallelLogMap,
-            setIterParallelLogMap,
-          } = workflowStore.getState()
-          const {
-            getNodes,
-            setNodes,
-            edges,
-            setEdges,
-            transform,
-          } = store.getState()
-          const nodes = getNodes()
-          const node = nodes.find(node => node.id === data.node_id)
-          if (node?.parentId) {
-            setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-              const tracing = draft.tracing!
-              const iterations = tracing.find(trace => trace.node_id === node?.parentId)
-              const currIteration = iterations?.details![node.data.iteration_index] || iterations?.details![iterations.details!.length - 1]
-              if (!data.parallel_run_id) {
-                currIteration?.push({
-                  ...data,
-                  status: NodeRunningStatus.Running,
-                } as any)
-              }
-              else {
-                const nodeId = iterations?.node_id as string
-                if (!iterParallelLogMap.has(nodeId as string))
-                  iterParallelLogMap.set(iterations?.node_id as string, new Map())
+          handleWorkflowNodeStarted(
+            params,
+            {
+              clientWidth,
+              clientHeight,
+            },
+          )
 
-                const currentIterLogMap = iterParallelLogMap.get(nodeId)!
-                if (!currentIterLogMap.has(data.parallel_run_id))
-                  currentIterLogMap.set(data.parallel_run_id, [{ ...data, status: NodeRunningStatus.Running } as any])
-                else
-                  currentIterLogMap.get(data.parallel_run_id)!.push({ ...data, status: NodeRunningStatus.Running } as any)
-                setIterParallelLogMap(iterParallelLogMap)
-                if (iterations)
-                  iterations.details = Array.from(currentIterLogMap.values())
-              }
-            }))
-          }
-          else {
-            setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-              draft.tracing!.push({
-                ...data,
-                status: NodeRunningStatus.Running,
-              } as any)
-            }))
-
-            const {
-              setViewport,
-            } = reactflow
-            const currentNodeIndex = nodes.findIndex(node => node.id === data.node_id)
-            const currentNode = nodes[currentNodeIndex]
-            const position = currentNode.position
-            const zoom = transform[2]
-
-            if (!currentNode.parentId) {
-              setViewport({
-                x: (clientWidth - 400 - currentNode.width! * zoom) / 2 - position.x * zoom,
-                y: (clientHeight - currentNode.height! * zoom) / 2 - position.y * zoom,
-                zoom: transform[2],
-              })
-            }
-            const newNodes = produce(nodes, (draft) => {
-              draft[currentNodeIndex].data._runningStatus = NodeRunningStatus.Running
-              draft[currentNodeIndex].data._waitingRun = false
-            })
-            setNodes(newNodes)
-            const newEdges = produce(edges, (draft) => {
-              const incomeEdges = draft.filter((edge) => {
-                return edge.target === data.node_id
-              })
-
-              incomeEdges.forEach((edge) => {
-                const incomeNode = nodes.find(node => node.id === edge.source)!
-                if (
-                  (!incomeNode.data._runningBranchId && edge.sourceHandle === 'source')
-                  || (incomeNode.data._runningBranchId && edge.sourceHandle === incomeNode.data._runningBranchId)
-                ) {
-                  edge.data = {
-                    ...edge.data,
-                    _sourceRunningStatus: incomeNode.data._runningStatus,
-                    _targetRunningStatus: NodeRunningStatus.Running,
-                    _waitingRun: false,
-                  }
-                }
-              })
-            })
-            setEdges(newEdges)
-          }
           if (onNodeStarted)
             onNodeStarted(params)
         },
         onNodeFinished: (params) => {
-          const { data } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            iterParallelLogMap,
-            setIterParallelLogMap,
-          } = workflowStore.getState()
-          const {
-            getNodes,
-            setNodes,
-            edges,
-            setEdges,
-          } = store.getState()
-          const nodes = getNodes()
-          const nodeParentId = nodes.find(node => node.id === data.node_id)!.parentId
-          if (nodeParentId) {
-            if (!data.execution_metadata.parallel_mode_run_id) {
-              setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-                const tracing = draft.tracing!
-                const iterations = tracing.find(trace => trace.node_id === nodeParentId) // the iteration node
-
-                if (iterations && iterations.details) {
-                  const iterationIndex = data.execution_metadata?.iteration_index || 0
-                  if (!iterations.details[iterationIndex])
-                    iterations.details[iterationIndex] = []
-
-                  const currIteration = iterations.details[iterationIndex]
-                  const nodeIndex = currIteration.findIndex(node =>
-                    node.node_id === data.node_id && (
-                      node.execution_metadata?.parallel_id === data.execution_metadata?.parallel_id || node.parallel_id === data.execution_metadata?.parallel_id),
-                  )
-                  if (nodeIndex !== -1) {
-                    currIteration[nodeIndex] = {
-                      ...currIteration[nodeIndex],
-                      ...(currIteration[nodeIndex].retryDetail
-                        ? { retryDetail: currIteration[nodeIndex].retryDetail }
-                        : {}),
-                      ...data,
-                    } as any
-                  }
-                  else {
-                    currIteration.push({
-                      ...data,
-                    } as any)
-                  }
-                }
-              }))
-            }
-            else {
-              // open parallel mode
-              setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-                const tracing = draft.tracing!
-                const iterations = tracing.find(trace => trace.node_id === nodeParentId) // the iteration node
-
-                if (iterations && iterations.details) {
-                  const iterRunID = data.execution_metadata?.parallel_mode_run_id
-
-                  const currIteration = iterParallelLogMap.get(iterations.node_id)?.get(iterRunID)
-                  const nodeIndex = currIteration?.findIndex(node =>
-                    node.node_id === data.node_id && (
-                      node?.parallel_run_id === data.execution_metadata?.parallel_mode_run_id),
-                  )
-                  if (currIteration) {
-                    if (nodeIndex !== undefined && nodeIndex !== -1) {
-                      currIteration[nodeIndex] = {
-                        ...currIteration[nodeIndex],
-                        ...data,
-                      } as any
-                    }
-                    else {
-                      currIteration.push({
-                        ...data,
-                      } as any)
-                    }
-                  }
-                  setIterParallelLogMap(iterParallelLogMap)
-                  const iterLogMap = iterParallelLogMap.get(iterations.node_id)
-                  if (iterLogMap)
-                    iterations.details = Array.from(iterLogMap.values())
-                }
-              }))
-            }
-          }
-          else {
-            setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-              const currentIndex = draft.tracing!.findIndex((trace) => {
-                if (!trace.execution_metadata?.parallel_id)
-                  return trace.node_id === data.node_id
-                return trace.node_id === data.node_id && trace.execution_metadata?.parallel_id === data.execution_metadata?.parallel_id
-              })
-              if (currentIndex > -1 && draft.tracing) {
-                draft.tracing[currentIndex] = {
-                  ...data,
-                  ...(draft.tracing[currentIndex].extras
-                    ? { extras: draft.tracing[currentIndex].extras }
-                    : {}),
-                  ...(draft.tracing[currentIndex].retryDetail
-                    ? { retryDetail: draft.tracing[currentIndex].retryDetail }
-                    : {}),
-                } as any
-              }
-            }))
-            const newNodes = produce(nodes, (draft) => {
-              const currentNode = draft.find(node => node.id === data.node_id)!
-              currentNode.data._runningStatus = data.status as any
-              if (data.status === NodeRunningStatus.Exception) {
-                if (data.execution_metadata.error_strategy === ErrorHandleTypeEnum.failBranch)
-                  currentNode.data._runningBranchId = ErrorHandleTypeEnum.failBranch
-              }
-              else {
-                if (data.node_type === BlockEnum.IfElse)
-                  currentNode.data._runningBranchId = data?.outputs?.selected_case_id
-
-                if (data.node_type === BlockEnum.QuestionClassifier)
-                  currentNode.data._runningBranchId = data?.outputs?.class_id
-              }
-            })
-            setNodes(newNodes)
-            const newEdges = produce(edges, (draft) => {
-              const incomeEdges = draft.filter((edge) => {
-                return edge.target === data.node_id
-              })
-              incomeEdges.forEach((edge) => {
-                edge.data = {
-                  ...edge.data,
-                  _targetRunningStatus: data.status as any,
-                }
-              })
-            })
-            setEdges(newEdges)
-            prevNodeId = data.node_id
-          }
+          handleWorkflowNodeFinished(params)
 
           if (onNodeFinished)
             onNodeFinished(params)
         },
         onIterationStart: (params) => {
-          const { data } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            setIterTimes,
-          } = workflowStore.getState()
-          const {
-            getNodes,
-            setNodes,
-            edges,
-            setEdges,
-            transform,
-          } = store.getState()
-          const nodes = getNodes()
-          setIterTimes(DEFAULT_ITER_TIMES)
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            draft.tracing!.push({
-              ...data,
-              status: NodeRunningStatus.Running,
-              details: [],
-              iterDurationMap: {},
-            } as any)
-          }))
-
-          const {
-            setViewport,
-          } = reactflow
-          const currentNodeIndex = nodes.findIndex(node => node.id === data.node_id)
-          const currentNode = nodes[currentNodeIndex]
-          const position = currentNode.position
-          const zoom = transform[2]
-
-          if (!currentNode.parentId) {
-            setViewport({
-              x: (clientWidth - 400 - currentNode.width! * zoom) / 2 - position.x * zoom,
-              y: (clientHeight - currentNode.height! * zoom) / 2 - position.y * zoom,
-              zoom: transform[2],
-            })
-          }
-          const newNodes = produce(nodes, (draft) => {
-            draft[currentNodeIndex].data._runningStatus = NodeRunningStatus.Running
-            draft[currentNodeIndex].data._iterationLength = data.metadata.iterator_length
-            draft[currentNodeIndex].data._waitingRun = false
-          })
-          setNodes(newNodes)
-          const newEdges = produce(edges, (draft) => {
-            const incomeEdges = draft.filter(edge => edge.target === data.node_id)
-
-            incomeEdges.forEach((edge) => {
-              edge.data = {
-                ...edge.data,
-                _sourceRunningStatus: nodes.find(node => node.id === edge.source)!.data._runningStatus,
-                _targetRunningStatus: NodeRunningStatus.Running,
-                _waitingRun: false,
-              }
-            })
-          })
-          setEdges(newEdges)
+          handleWorkflowNodeIterationStarted(
+            params,
+            {
+              clientWidth,
+              clientHeight,
+            },
+          )
 
           if (onIterationStart)
             onIterationStart(params)
         },
         onIterationNext: (params) => {
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            iterTimes,
-            setIterTimes,
-          } = workflowStore.getState()
-
-          const { data } = params
-          const {
-            getNodes,
-            setNodes,
-          } = store.getState()
-
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            const iteration = draft.tracing!.find(trace => trace.node_id === data.node_id)
-            if (iteration) {
-              if (iteration.iterDurationMap && data.duration)
-                iteration.iterDurationMap[data.parallel_mode_run_id ?? `${data.index - 1}`] = data.duration
-              if (iteration.details!.length >= iteration.metadata.iterator_length!)
-                return
-            }
-            if (!data.parallel_mode_run_id)
-              iteration?.details!.push([])
-          }))
-          const nodes = getNodes()
-          const newNodes = produce(nodes, (draft) => {
-            const currentNode = draft.find(node => node.id === data.node_id)!
-            currentNode.data._iterationIndex = iterTimes
-            setIterTimes(iterTimes + 1)
-          })
-          setNodes(newNodes)
+          handleWorkflowNodeIterationNext(params)
 
           if (onIterationNext)
             onIterationNext(params)
         },
         onIterationFinish: (params) => {
-          const { data } = params
-
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            setIterTimes,
-          } = workflowStore.getState()
-          const {
-            getNodes,
-            setNodes,
-          } = store.getState()
-          const nodes = getNodes()
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            const tracing = draft.tracing!
-            const currIterationNode = tracing.find(trace => trace.node_id === data.node_id)
-            if (currIterationNode) {
-              Object.assign(currIterationNode, {
-                ...data,
-                status: NodeRunningStatus.Succeeded,
-              })
-            }
-          }))
-          setIterTimes(DEFAULT_ITER_TIMES)
-          const newNodes = produce(nodes, (draft) => {
-            const currentNode = draft.find(node => node.id === data.node_id)!
-
-            currentNode.data._runningStatus = data.status
-          })
-          setNodes(newNodes)
-
-          prevNodeId = data.node_id
+          handleWorkflowNodeIterationFinished(params)
 
           if (onIterationFinish)
             onIterationFinish(params)
         },
         onNodeRetry: (params) => {
-          const { data } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-            iterParallelLogMap,
-            setIterParallelLogMap,
-          } = workflowStore.getState()
-          const {
-            getNodes,
-            setNodes,
-          } = store.getState()
-
-          const nodes = getNodes()
-          const currentNode = nodes.find(node => node.id === data.node_id)!
-          const nodeParent = nodes.find(node => node.id === currentNode.parentId)
-          if (nodeParent) {
-            if (!data.execution_metadata.parallel_mode_run_id) {
-              setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-                const tracing = draft.tracing!
-                const iteration = tracing.find(trace => trace.node_id === nodeParent.id)
-
-                if (iteration && iteration.details?.length) {
-                  const currentNodeRetry = iteration.details[nodeParent.data._iterationIndex - 1]?.find(item => item.node_id === data.node_id)
-
-                  if (currentNodeRetry) {
-                    if (currentNodeRetry?.retryDetail)
-                      currentNodeRetry?.retryDetail.push(data as NodeTracing)
-                    else
-                      currentNodeRetry.retryDetail = [data as NodeTracing]
-                  }
-                }
-              }))
-            }
-            else {
-              setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-                const tracing = draft.tracing!
-                const iteration = tracing.find(trace => trace.node_id === nodeParent.id)
-
-                if (iteration && iteration.details?.length) {
-                  const iterRunID = data.execution_metadata?.parallel_mode_run_id
-
-                  const currIteration = iterParallelLogMap.get(iteration.node_id)?.get(iterRunID)
-                  const currentNodeRetry = currIteration?.find(item => item.node_id === data.node_id)
-
-                  if (currentNodeRetry) {
-                    if (currentNodeRetry?.retryDetail)
-                      currentNodeRetry?.retryDetail.push(data as NodeTracing)
-                    else
-                      currentNodeRetry.retryDetail = [data as NodeTracing]
-                  }
-                  setIterParallelLogMap(iterParallelLogMap)
-                  const iterLogMap = iterParallelLogMap.get(iteration.node_id)
-                  if (iterLogMap)
-                    iteration.details = Array.from(iterLogMap.values())
-                }
-              }))
-            }
-          }
-          else {
-            setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-              const tracing = draft.tracing!
-              const currentRetryNodeIndex = tracing.findIndex(trace => trace.node_id === data.node_id)
-
-              if (currentRetryNodeIndex > -1) {
-                const currentRetryNode = tracing[currentRetryNodeIndex]
-                if (currentRetryNode.retryDetail)
-                  draft.tracing![currentRetryNodeIndex].retryDetail!.push(data as NodeTracing)
-                else
-                  draft.tracing![currentRetryNodeIndex].retryDetail = [data as NodeTracing]
-              }
-            }))
-          }
-          const newNodes = produce(nodes, (draft) => {
-            const currentNode = draft.find(node => node.id === data.node_id)!
-
-            currentNode.data._retryIndex = data.retry_index
-          })
-          setNodes(newNodes)
+          handleWorkflowNodeRetry(params)
 
           if (onNodeRetry)
             onNodeRetry(params)
         },
-        onParallelBranchStarted: (params) => {
-          // console.log(params, 'parallel start')
-        },
-        onParallelBranchFinished: (params) => {
-          // console.log(params, 'finished')
+        onAgentLog: (params) => {
+          handleWorkflowAgentLog(params)
+
+          if (onAgentLog)
+            onAgentLog(params)
         },
         onTextChunk: (params) => {
-          const { data: { text } } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-          } = workflowStore.getState()
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            draft.resultTabActive = true
-            draft.resultText += text
-          }))
+          handleWorkflowTextChunk(params)
         },
         onTextReplace: (params) => {
-          const { data: { text } } = params
-          const {
-            workflowRunningData,
-            setWorkflowRunningData,
-          } = workflowStore.getState()
-          setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
-            draft.resultText = text
-          }))
+          handleWorkflowTextReplace(params)
         },
-        onTTSChunk: (messageId: string, audio: string, audioType?: string) => {
+        onTTSChunk: (messageId: string, audio: string) => {
           if (!audio || audio === '')
             return
           player.playAudioWithAudio(audio, true)
           AudioPlayerManager.getInstance().resetMsgId(messageId)
         },
-        onTTSEnd: (messageId: string, audio: string, audioType?: string) => {
+        onTTSEnd: (messageId: string, audio: string) => {
           player.playAudioWithAudio(audio, false)
         },
         ...restCallback,
       },
     )
-  }, [store, reactflow, workflowStore, doSyncWorkflowDraft])
+  }, [store, workflowStore, doSyncWorkflowDraft, handleWorkflowStarted, handleWorkflowFinished, handleWorkflowFailed, handleWorkflowNodeStarted, handleWorkflowNodeFinished, handleWorkflowNodeIterationStarted, handleWorkflowNodeIterationNext, handleWorkflowNodeIterationFinished, handleWorkflowNodeRetry, handleWorkflowTextChunk, handleWorkflowTextReplace, handleWorkflowAgentLog, pathname])
 
   const handleStopRun = useCallback((taskId: string) => {
     const appId = useAppStore.getState().appDetail?.id