add workflow run events

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

@@ -13,3 +13,5 @@ export * from './use-workflow-node-retry'
 export * from './use-workflow-text-chunk'
 export * from './use-workflow-text-replace'
 export * from './use-workflow-agent-log'
+export * from './use-workflow-suspended'
+export * from './use-workflow-node-human-input-required'

web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-node-human-input-required.ts

@@ -0,0 +1,44 @@
+import { useCallback } from 'react'
+import {
+  useStoreApi,
+} from 'reactflow'
+import produce from 'immer'
+import { useWorkflowStore } from '@/app/components/workflow/store'
+import type { HumanInputRequiredResponse } from '@/types/workflow'
+import { NodeRunningStatus } from '@/app/components/workflow/types'
+import { WorkflowRunningStatus } from '@/app/components/workflow/types'
+
+export const useWorkflowNodeHumanInputRequired = () => {
+  const workflowStore = useWorkflowStore()
+  const store = useStoreApi()
+
+  const handleWorkflowNodeHumanInputRequired = useCallback((params: HumanInputRequiredResponse) => {
+    const { data } = params
+    const {
+      workflowRunningData,
+      setWorkflowRunningData,
+    } = workflowStore.getState()
+    const {
+      getNodes,
+      setNodes,
+    } = store.getState()
+    const nodes = getNodes()
+    const currentNodeIndex = nodes.findIndex(node => node.id === data.node_id)
+    const newNodes = produce(nodes, (draft) => {
+      draft[currentNodeIndex].data._runningStatus = NodeRunningStatus.Suspended
+      // draft[currentNodeIndex].data._waitingRun = false
+    })
+    setNodes(newNodes)
+
+    setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
+      draft.result = {
+        ...draft.result,
+        status: WorkflowRunningStatus.Suspended,
+      }
+    }))
+  }, [workflowStore])
+
+  return {
+    handleWorkflowNodeHumanInputRequired,
+  }
+}

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

@@ -3,6 +3,7 @@ import {
   useWorkflowFailed,
   useWorkflowFinished,
   useWorkflowNodeFinished,
+  useWorkflowNodeHumanInputRequired,
   useWorkflowNodeIterationFinished,
   useWorkflowNodeIterationNext,
   useWorkflowNodeIterationStarted,
@@ -12,6 +13,7 @@ import {
   useWorkflowNodeRetry,
   useWorkflowNodeStarted,
   useWorkflowStarted,
+  useWorkflowSuspended,
   useWorkflowTextChunk,
   useWorkflowTextReplace,
 } from '.'
@@ -32,6 +34,8 @@ export const useWorkflowRunEvent = () => {
   const { handleWorkflowTextChunk } = useWorkflowTextChunk()
   const { handleWorkflowTextReplace } = useWorkflowTextReplace()
   const { handleWorkflowAgentLog } = useWorkflowAgentLog()
+  const { handleWorkflowSuspended } = useWorkflowSuspended()
+  const { handleWorkflowNodeHumanInputRequired } = useWorkflowNodeHumanInputRequired()
 
   return {
     handleWorkflowStarted,
@@ -49,5 +53,7 @@ export const useWorkflowRunEvent = () => {
     handleWorkflowTextChunk,
     handleWorkflowTextReplace,
     handleWorkflowAgentLog,
+    handleWorkflowSuspended,
+    handleWorkflowNodeHumanInputRequired,
   }
 }

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

@@ -0,0 +1,26 @@
+import { useCallback } from 'react'
+import produce from 'immer'
+import { useWorkflowStore } from '@/app/components/workflow/store'
+import { WorkflowRunningStatus } from '@/app/components/workflow/types'
+
+export const useWorkflowSuspended = () => {
+  const workflowStore = useWorkflowStore()
+
+  const handleWorkflowSuspended = useCallback(() => {
+    const {
+      workflowRunningData,
+      setWorkflowRunningData,
+    } = workflowStore.getState()
+
+    setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
+      draft.result = {
+        ...draft.result,
+        status: WorkflowRunningStatus.Suspended,
+      }
+    }))
+  }, [workflowStore])
+
+  return {
+    handleWorkflowSuspended,
+  }
+}