refact workflow run

This commit is contained in:
zxhlyh 2024-12-24 15:38:54 +08:00
parent b98dd22491
commit 754baf8d02
14 changed files with 36 additions and 36 deletions

View File

@ -18,7 +18,7 @@ export const useWorkflowFailed = () => {
status: WorkflowRunningStatus.Failed, status: WorkflowRunningStatus.Failed,
} }
})) }))
}, []) }, [workflowStore])
return { return {
handleWorkflowFailed, handleWorkflowFailed,

View File

@ -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,

View File

@ -145,7 +145,7 @@ export const useWorkflowNodeFinished = () => {
}) })
setEdges(newEdges) setEdges(newEdges)
} }
}, []) }, [workflowStore, store])
return { return {
handleWorkflowNodeFinished, handleWorkflowNodeFinished,

View File

@ -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,

View File

@ -40,7 +40,7 @@ export const useWorkflowNodeIterationNext = () => {
setIterTimes(iterTimes + 1) setIterTimes(iterTimes + 1)
}) })
setNodes(newNodes) setNodes(newNodes)
}, []) }, [workflowStore, store])
return { return {
handleWorkflowNodeIterationNext, handleWorkflowNodeIterationNext,

View File

@ -79,7 +79,7 @@ export const useWorkflowNodeIterationStarted = () => {
}) })
}) })
setEdges(newEdges) setEdges(newEdges)
}, []) }, [workflowStore, store, reactflow])
return { return {
handleWorkflowNodeIterationStarted, handleWorkflowNodeIterationStarted,

View File

@ -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,

View File

@ -113,7 +113,7 @@ export const useWorkflowNodeStarted = () => {
}) })
setEdges(newEdges) setEdges(newEdges)
} }
}, []) }, [workflowStore, store, reactflow])
return { return {
handleWorkflowNodeStarted, handleWorkflowNodeStarted,

View File

@ -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 '.'

View File

@ -50,7 +50,7 @@ export const useWorkflowStarted = () => {
}) })
}) })
setEdges(newEdges) setEdges(newEdges)
}, []) }, [workflowStore, store])
return { return {
handleWorkflowStarted, handleWorkflowStarted,

View File

@ -17,7 +17,7 @@ export const useWorkflowTextChunk = () => {
draft.resultTabActive = true draft.resultTabActive = true
draft.resultText += text draft.resultText += text
})) }))
}, []) }, [workflowStore])
return { return {
handleWorkflowTextChunk, handleWorkflowTextChunk,

View File

@ -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,

View File

@ -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