fix: running line

This commit is contained in:
StyleZhang 2024-03-21 16:12:38 +08:00
parent 8e0d8fdb3f
commit e366e12be0
3 changed files with 21 additions and 13 deletions

View File

@ -130,7 +130,6 @@ export const useWorkflowRun = () => {
params: any,
callback?: IOtherOptions,
) => {
handleLoadBackupDraft()
const {
onWorkflowStarted,
onWorkflowFinished,
@ -154,6 +153,8 @@ export const useWorkflowRun = () => {
if (appDetail?.mode === 'workflow')
url = `/apps/${appDetail.id}/workflows/draft/run`
let prevNodeId = ''
ssePost(
url,
{
@ -203,6 +204,8 @@ export const useWorkflowRun = () => {
}
}))
prevNodeId = ''
if (onWorkflowFinished)
onWorkflowFinished(params)
},
@ -244,7 +247,7 @@ export const useWorkflowRun = () => {
})
setNodes(newNodes)
const newEdges = produce(edges, (draft) => {
const edge = draft.find(edge => edge.target === data.node_id)
const edge = draft.find(edge => edge.target === data.node_id && edge.source === prevNodeId)
if (edge)
edge.data = { ...edge.data, _runned: true } as any
@ -279,13 +282,15 @@ export const useWorkflowRun = () => {
})
setNodes(newNodes)
prevNodeId = data.node_id
if (onNodeFinished)
onNodeFinished(params)
},
...restCallback,
},
)
}, [store, reactflow, workflowStore, handleLoadBackupDraft])
}, [store, reactflow, workflowStore])
const handleStopRun = useCallback((taskId: string) => {
const appId = useAppStore.getState().appDetail?.id

View File

@ -11,7 +11,7 @@ const Node: FC<NodeProps<AnswerNodeType>> = ({
const { t } = useTranslation()
return (
<div className='px-3'>
<div className='mb-1 px-3 py-1'>
<InfoPanel title={t('workflow.nodes.answer.answer')} content={data.answer} />
</div>
)

View File

@ -1,23 +1,26 @@
import { memo, useCallback } from 'react'
import {
memo,
// useCallback,
} from 'react'
import {
useIsChatMode,
useWorkflow,
// useWorkflow,
} from '../hooks'
import Run from '../run'
import { useStore } from '../store'
import ChatRecord from './chat-record'
import type { WorkflowRunDetailResponse } from '@/models/log'
// import type { WorkflowRunDetailResponse } from '@/models/log'
const Record = () => {
const isChatMode = useIsChatMode()
const { renderTreeFromRecord } = useWorkflow()
// const { renderTreeFromRecord } = useWorkflow()
const historyWorkflowData = useStore(s => s.historyWorkflowData)
const getResultCallback = useCallback((res: WorkflowRunDetailResponse) => {
const { graph } = res
// const getResultCallback = useCallback((res: WorkflowRunDetailResponse) => {
// const { graph } = res
renderTreeFromRecord(graph.nodes, graph.edges, graph.viewport)
}, [renderTreeFromRecord])
// renderTreeFromRecord(graph.nodes, graph.edges, graph.viewport)
// }, [renderTreeFromRecord])
return (
<div className={`
@ -33,7 +36,7 @@ const Record = () => {
: (
<Run
runID={historyWorkflowData?.id || ''}
getResultCallback={getResultCallback}
// getResultCallback={getResultCallback}
/>
)
}