refact workflow run
This commit is contained in:
parent
b98dd22491
commit
754baf8d02
@ -18,7 +18,7 @@ export const useWorkflowFailed = () => {
|
|||||||
status: WorkflowRunningStatus.Failed,
|
status: WorkflowRunningStatus.Failed,
|
||||||
}
|
}
|
||||||
}))
|
}))
|
||||||
}, [])
|
}, [workflowStore])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowFailed,
|
handleWorkflowFailed,
|
||||||
|
@ -27,7 +27,7 @@ export const useWorkflowFinished = () => {
|
|||||||
draft.resultText = data.outputs[Object.keys(data.outputs)[0]]
|
draft.resultText = data.outputs[Object.keys(data.outputs)[0]]
|
||||||
}
|
}
|
||||||
}))
|
}))
|
||||||
}, [])
|
}, [workflowStore])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowFinished,
|
handleWorkflowFinished,
|
||||||
|
@ -145,7 +145,7 @@ export const useWorkflowNodeFinished = () => {
|
|||||||
})
|
})
|
||||||
setEdges(newEdges)
|
setEdges(newEdges)
|
||||||
}
|
}
|
||||||
}, [])
|
}, [workflowStore, store])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowNodeFinished,
|
handleWorkflowNodeFinished,
|
||||||
|
@ -12,7 +12,6 @@ export const useWorkflowNodeIterationFinished = () => {
|
|||||||
|
|
||||||
const handleWorkflowNodeIterationFinished = useCallback((params: IterationFinishedResponse) => {
|
const handleWorkflowNodeIterationFinished = useCallback((params: IterationFinishedResponse) => {
|
||||||
const { data } = params
|
const { data } = params
|
||||||
|
|
||||||
const {
|
const {
|
||||||
workflowRunningData,
|
workflowRunningData,
|
||||||
setWorkflowRunningData,
|
setWorkflowRunningData,
|
||||||
@ -23,6 +22,7 @@ export const useWorkflowNodeIterationFinished = () => {
|
|||||||
setNodes,
|
setNodes,
|
||||||
} = store.getState()
|
} = store.getState()
|
||||||
const nodes = getNodes()
|
const nodes = getNodes()
|
||||||
|
|
||||||
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
|
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
|
||||||
const tracing = draft.tracing!
|
const tracing = draft.tracing!
|
||||||
const currIterationNode = tracing.find(trace => trace.node_id === data.node_id)
|
const currIterationNode = tracing.find(trace => trace.node_id === data.node_id)
|
||||||
@ -40,7 +40,7 @@ export const useWorkflowNodeIterationFinished = () => {
|
|||||||
currentNode.data._runningStatus = data.status
|
currentNode.data._runningStatus = data.status
|
||||||
})
|
})
|
||||||
setNodes(newNodes)
|
setNodes(newNodes)
|
||||||
}, [])
|
}, [workflowStore, store])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowNodeIterationFinished,
|
handleWorkflowNodeIterationFinished,
|
||||||
|
@ -40,7 +40,7 @@ export const useWorkflowNodeIterationNext = () => {
|
|||||||
setIterTimes(iterTimes + 1)
|
setIterTimes(iterTimes + 1)
|
||||||
})
|
})
|
||||||
setNodes(newNodes)
|
setNodes(newNodes)
|
||||||
}, [])
|
}, [workflowStore, store])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowNodeIterationNext,
|
handleWorkflowNodeIterationNext,
|
||||||
|
@ -79,7 +79,7 @@ export const useWorkflowNodeIterationStarted = () => {
|
|||||||
})
|
})
|
||||||
})
|
})
|
||||||
setEdges(newEdges)
|
setEdges(newEdges)
|
||||||
}, [])
|
}, [workflowStore, store, reactflow])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowNodeIterationStarted,
|
handleWorkflowNodeIterationStarted,
|
||||||
|
@ -90,7 +90,7 @@ export const useWorkflowNodeRetry = () => {
|
|||||||
currentNode.data._retryIndex = data.retry_index
|
currentNode.data._retryIndex = data.retry_index
|
||||||
})
|
})
|
||||||
setNodes(newNodes)
|
setNodes(newNodes)
|
||||||
}, [])
|
}, [workflowStore, store])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowNodeRetry,
|
handleWorkflowNodeRetry,
|
||||||
|
@ -113,7 +113,7 @@ export const useWorkflowNodeStarted = () => {
|
|||||||
})
|
})
|
||||||
setEdges(newEdges)
|
setEdges(newEdges)
|
||||||
}
|
}
|
||||||
}, [])
|
}, [workflowStore, store, reactflow])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowNodeStarted,
|
handleWorkflowNodeStarted,
|
||||||
|
@ -1,13 +1,13 @@
|
|||||||
import {
|
import {
|
||||||
useWorkflowStarted,
|
|
||||||
useWorkflowFinished,
|
|
||||||
useWorkflowFailed,
|
useWorkflowFailed,
|
||||||
useWorkflowNodeStarted,
|
useWorkflowFinished,
|
||||||
useWorkflowNodeFinished,
|
useWorkflowNodeFinished,
|
||||||
useWorkflowNodeIterationStarted,
|
|
||||||
useWorkflowNodeIterationNext,
|
|
||||||
useWorkflowNodeIterationFinished,
|
useWorkflowNodeIterationFinished,
|
||||||
|
useWorkflowNodeIterationNext,
|
||||||
|
useWorkflowNodeIterationStarted,
|
||||||
useWorkflowNodeRetry,
|
useWorkflowNodeRetry,
|
||||||
|
useWorkflowNodeStarted,
|
||||||
|
useWorkflowStarted,
|
||||||
useWorkflowTextChunk,
|
useWorkflowTextChunk,
|
||||||
useWorkflowTextReplace,
|
useWorkflowTextReplace,
|
||||||
} from '.'
|
} from '.'
|
||||||
|
@ -50,7 +50,7 @@ export const useWorkflowStarted = () => {
|
|||||||
})
|
})
|
||||||
})
|
})
|
||||||
setEdges(newEdges)
|
setEdges(newEdges)
|
||||||
}, [])
|
}, [workflowStore, store])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowStarted,
|
handleWorkflowStarted,
|
||||||
|
@ -17,7 +17,7 @@ export const useWorkflowTextChunk = () => {
|
|||||||
draft.resultTabActive = true
|
draft.resultTabActive = true
|
||||||
draft.resultText += text
|
draft.resultText += text
|
||||||
}))
|
}))
|
||||||
}, [])
|
}, [workflowStore])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowTextChunk,
|
handleWorkflowTextChunk,
|
||||||
|
@ -15,7 +15,7 @@ export const useWorkflowTextReplace = () => {
|
|||||||
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
|
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
|
||||||
draft.resultText = text
|
draft.resultText = text
|
||||||
}))
|
}))
|
||||||
}, [])
|
}, [workflowStore])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
handleWorkflowTextReplace,
|
handleWorkflowTextReplace,
|
||||||
|
@ -242,19 +242,19 @@ export const useWorkflowRun = () => {
|
|||||||
onTextReplace: (params) => {
|
onTextReplace: (params) => {
|
||||||
handleWorkflowTextReplace(params)
|
handleWorkflowTextReplace(params)
|
||||||
},
|
},
|
||||||
onTTSChunk: (messageId: string, audio: string, audioType?: string) => {
|
onTTSChunk: (messageId: string, audio: string) => {
|
||||||
if (!audio || audio === '')
|
if (!audio || audio === '')
|
||||||
return
|
return
|
||||||
player.playAudioWithAudio(audio, true)
|
player.playAudioWithAudio(audio, true)
|
||||||
AudioPlayerManager.getInstance().resetMsgId(messageId)
|
AudioPlayerManager.getInstance().resetMsgId(messageId)
|
||||||
},
|
},
|
||||||
onTTSEnd: (messageId: string, audio: string, audioType?: string) => {
|
onTTSEnd: (messageId: string, audio: string) => {
|
||||||
player.playAudioWithAudio(audio, false)
|
player.playAudioWithAudio(audio, false)
|
||||||
},
|
},
|
||||||
...restCallback,
|
...restCallback,
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
}, [store, reactflow, workflowStore, doSyncWorkflowDraft])
|
}, [store, workflowStore, doSyncWorkflowDraft, handleWorkflowStarted, handleWorkflowFinished, handleWorkflowFailed, handleWorkflowNodeStarted, handleWorkflowNodeFinished, handleWorkflowNodeIterationStarted, handleWorkflowNodeIterationNext, handleWorkflowNodeIterationFinished, handleWorkflowNodeRetry, handleWorkflowTextChunk, handleWorkflowTextReplace, pathname])
|
||||||
|
|
||||||
const handleStopRun = useCallback((taskId: string) => {
|
const handleStopRun = useCallback((taskId: string) => {
|
||||||
const appId = useAppStore.getState().appDetail?.id
|
const appId = useAppStore.getState().appDetail?.id
|
||||||
|
Loading…
Reference in New Issue
Block a user