import { useCallback } from 'react' import produce from 'immer' import type { TextChunkResponse } from '@/types/workflow' import { useWorkflowStore } from '@/app/components/workflow/store' export const useWorkflowTextChunk = () => { const workflowStore = useWorkflowStore() const handleWorkflowTextChunk = useCallback((params: TextChunkResponse) => { const { data: { text } } = params const { workflowRunningData, setWorkflowRunningData, } = workflowStore.getState() setWorkflowRunningData(produce(workflowRunningData!, (draft) => { draft.resultTabActive = true draft.resultText += text })) }, [workflowStore]) return { handleWorkflowTextChunk, } }