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

fix workflow outputs

Browse Source
This commit is contained in:
JzoNg
2024-03-19 12:33:42 +08:00
parent 978ee93df7
commit cf0c96e0d1
3 changed files with 33 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
web/app/components/share/text-generation/result/index.tsx
View File

@ -14,6 +14,7 @@ import type { PromptConfig } from '@/models/debug'
import type { InstalledApp } from '@/models/explore'
import type { ModerationService } from '@/models/common'
import { TransferMethod, type VisionFile, type VisionSettings } from '@/types/app'
import { BlockEnum } from '@/app/components/workflow/types'
export type IResultProps = {
isWorkflow: boolean
@ -181,35 +182,44 @@ const Result: FC<IResultProps> = ({
if (isWorkflow) {
// ###TODO###
let outputsExisted = false
sendWorkflowMessage(
data,
{
onWorkflowStarted: ({ workflow_run_id }) => {
tempMessageId = workflow_run_id
// console.log('onWorkflowStarted runID: ', workflow_run_id)
},
onTextChunk: ({ data }) => {
res.push(data.text)
setCompletionRes(res.join(''))
onNodeStarted: ({ data }) => {
// console.log('onNodeStarted: ', data.node_type)
},
onWorkflowFinished: () => {
onNodeFinished: ({ data }) => {
// console.log('onNodeFinished: ', data.node_type, data)
if (data.node_type === BlockEnum.LLM && data.outputs.text)
setCompletionRes(data.outputs.text)
if (data.node_type === BlockEnum.End && data.outputs)
outputsExisted = true
},
onWorkflowFinished: ({ data }) => {
// console.log('onWorkflowFinished: ', data)
if (isTimeout)
return
if (data.error) {
notify({ type: 'error', message: data.error })
setRespondingFalse()
onCompleted(getCompletionRes(), taskId, false)
clearInterval(runId)
return
}
if (!outputsExisted) {
notify({ type: 'info', message: 'Outputs not existed.' })
setCompletionRes('')
}
setRespondingFalse()
setMessageId(tempMessageId)
onCompleted(getCompletionRes(), taskId, true)
clearInterval(runId)
},
onTextReplace: ({ data }) => {
res = [data.text]
setCompletionRes(res.join(''))
},
onError() {
if (isTimeout)
return
setRespondingFalse()
onCompleted(getCompletionRes(), taskId, false)
clearInterval(runId)
},
},
isInstalledApp,
installedAppInfo?.id,