jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: llm text stream support for workflow app (#3798)

Browse source 
Co-authored-by: JzoNg <jzongcode@gmail.com>
This commit is contained in:
takatost 2024-04-28 17:37:00 +08:00 • committed by GitHub
commit ff67a6d338
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
27 changed files with 549 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

22
web/app/components/workflow/hooks/use-workflow-run.ts
View file

@ -124,6 +124,7 @@ export const useWorkflowRun = () => {
status: WorkflowRunningStatus.Running,
},
tracing: [],
resultText: '',
})
ssePost(
@ -284,6 +285,27 @@ export const useWorkflowRun = () => {
if (onNodeFinished)
onNodeFinished(params)
},
onTextChunk: (params) => {
const { data: { text } } = params
const {
workflowRunningData,
setWorkflowRunningData,
} = workflowStore.getState()
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
draft.resultTabActive = true
draft.resultText += text
}))
},
onTextReplace: (params) => {
const { data: { text } } = params
const {
workflowRunningData,
setWorkflowRunningData,
} = workflowStore.getState()
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
draft.resultText = text
}))
},
...restCallback,
},
)