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

fix: workflow result generation render

Browse Source
This commit is contained in:
zxhlyh
2026-01-28 21:00:57 +08:00
parent 0c1e812d21
commit 3af927556e
5 changed files with 124 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

98
web/app/components/workflow/hooks/use-workflow-run-event/use-workflow-text-chunk.ts
View File

@ -1,13 +1,25 @@
import type { TextChunkResponse } from '@/types/workflow'
import { produce } from 'immer'
import { useCallback } from 'react'
import { useCallback, useRef } from 'react'
import { v4 as uuidV4 } from 'uuid'
import { useWorkflowStore } from '@/app/components/workflow/store'
export const useWorkflowTextChunk = () => {
const workflowStore = useWorkflowStore()
const toolCallIdRef = useRef<string | null>(null)
const handleWorkflowTextChunk = useCallback((params: TextChunkResponse) => {
const { data: { text } } = params
const { data: {
text,
chunk_type,
tool_name,
tool_arguments,
tool_icon,
tool_icon_dark,
tool_error,
tool_elapsed_time,
tool_files,
} } = params
const {
workflowRunningData,
setWorkflowRunningData,
@ -15,7 +27,87 @@ export const useWorkflowTextChunk = () => {
setWorkflowRunningData(produce(workflowRunningData!, (draft) => {
draft.resultTabActive = true
draft.resultText += text
if (chunk_type === 'text') {
draft.resultText += text
if (!draft.resultLLMGenerationItems)
draft.resultLLMGenerationItems = []
const isNotCompletedTextItemIndex = draft.resultLLMGenerationItems?.findIndex(item => item.type === 'text' && !item.textCompleted)
if (isNotCompletedTextItemIndex > -1) {
draft.resultLLMGenerationItems![isNotCompletedTextItemIndex].text += text
}
else {
draft.resultLLMGenerationItems?.push({
id: uuidV4(),
type: 'text',
text,
})
}
}
if (chunk_type === 'tool_call') {
if (!draft.resultLLMGenerationItems)
draft.resultLLMGenerationItems = []
const isNotCompletedTextItemIndex = draft.resultLLMGenerationItems?.findIndex(item => item.type === 'text' && !item.textCompleted)
if (isNotCompletedTextItemIndex > -1) {
draft.resultLLMGenerationItems![isNotCompletedTextItemIndex].textCompleted = true
}
toolCallIdRef.current = uuidV4()
draft.resultLLMGenerationItems?.push({
id: toolCallIdRef.current,
type: 'tool',
toolName: tool_name,
toolArguments: tool_arguments,
toolIcon: tool_icon,
toolIconDark: tool_icon_dark,
})
}
if (chunk_type === 'tool_result') {
const currentToolCallIndex = draft.resultLLMGenerationItems?.findIndex(item => item.id === toolCallIdRef.current) ?? -1
if (currentToolCallIndex > -1) {
draft.resultLLMGenerationItems![currentToolCallIndex].toolError = tool_error
draft.resultLLMGenerationItems![currentToolCallIndex].toolDuration = tool_elapsed_time
draft.resultLLMGenerationItems![currentToolCallIndex].toolFiles = tool_files
draft.resultLLMGenerationItems![currentToolCallIndex].toolOutput = text
}
}
if (chunk_type === 'thought_start') {
if (!draft.resultLLMGenerationItems)
draft.resultLLMGenerationItems = []
const isNotCompletedTextItemIndex = draft.resultLLMGenerationItems?.findIndex(item => item.type === 'text' && !item.textCompleted)
if (isNotCompletedTextItemIndex > -1) {
draft.resultLLMGenerationItems![isNotCompletedTextItemIndex].textCompleted = true
}
toolCallIdRef.current = uuidV4()
draft.resultLLMGenerationItems?.push({
id: toolCallIdRef.current,
type: 'thought',
thoughtOutput: '',
})
}
if (chunk_type === 'thought') {
const currentThoughtIndex = draft.resultLLMGenerationItems?.findIndex(item => item.id === toolCallIdRef.current) ?? -1
if (currentThoughtIndex > -1) {
draft.resultLLMGenerationItems![currentThoughtIndex].thoughtOutput += text
}
}
if (chunk_type === 'thought_end') {
const currentThoughtIndex = draft.resultLLMGenerationItems?.findIndex(item => item.id === toolCallIdRef.current) ?? -1
if (currentThoughtIndex > -1) {
draft.resultLLMGenerationItems![currentThoughtIndex].thoughtOutput += text
draft.resultLLMGenerationItems![currentThoughtIndex].thoughtCompleted = true
}
}
}))
}, [workflowStore])