youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 02:18:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: running status

Browse Source
This commit is contained in:
StyleZhang
2024-03-21 15:03:49 +08:00
parent 8bdaab96b1
commit 267d9568c6
7 changed files with 82 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
web/app/components/workflow/hooks/use-workflow-run.ts
View File

@ -130,6 +130,7 @@ export const useWorkflowRun = () => {
params: any,
callback?: IOtherOptions,
) => {
handleLoadBackupDraft()
const {
onWorkflowStarted,
onWorkflowFinished,
@ -138,15 +139,6 @@ export const useWorkflowRun = () => {
...restCallback
} = callback || {}
workflowStore.setState({ historyWorkflowData: undefined })
const { backupDraft } = workflowStore.getState()
const {
setNodes,
setEdges,
} = store.getState()
const {
nodes,
edges,
} = backupDraft!
const appDetail = useAppStore.getState().appDetail
const workflowContainer = document.getElementById('workflow-container')
@ -174,6 +166,10 @@ export const useWorkflowRun = () => {
workflowRunningData,
setWorkflowRunningData,
} = workflowStore.getState()
const {
getNodes,
setNodes,
} = store.getState()
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
draft.task_id = task_id
draft.result = {
@ -183,7 +179,7 @@ export const useWorkflowRun = () => {
}
}))
const newNodes = produce(nodes, (draft) => {
const newNodes = produce(getNodes(), (draft) => {
draft.forEach((node) => {
node.data._runningStatus = NodeRunningStatus.Waiting
})
@ -216,6 +212,13 @@ export const useWorkflowRun = () => {
workflowRunningData,
setWorkflowRunningData,
} = workflowStore.getState()
const {
getNodes,
setNodes,
edges,
setEdges,
} = store.getState()
const nodes = getNodes()
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
draft.tracing!.push({
...data,
@ -257,6 +260,11 @@ export const useWorkflowRun = () => {
workflowRunningData,
setWorkflowRunningData,
} = workflowStore.getState()
const {
getNodes,
setNodes,
} = store.getState()
const nodes = getNodes()
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
const currentIndex = draft.tracing!.findIndex(trace => trace.node_id === data.node_id)