mirror of
https://github.com/langgenius/dify.git
synced 2026-03-25 00:07:56 +08:00
772dbe620d
Simpler approach: disable the view picker toggle when preview is running, preventing users from switching views during active runs. This replaces the previous mounted ref guard approach (commitsa0188bd9b5,b7f1eb9b7b,8332f0de2b) which added complexity to handle post-unmount operations. Disabling the toggle is more direct and follows KISS principle. Changes: - Add disabled prop to ViewPicker based on isResponding state - Revert mounted ref guards in use-chat-flow-control.ts - Revert isMountedRef parameter in use-nodes/edges-interactions-without-sync.ts - Revert defensive type check in markdown-utils.ts (no longer needed)
91 lines
3.0 KiB
TypeScript
91 lines
3.0 KiB
TypeScript
import { useCallback } from 'react'
|
|
import { DEFAULT_ITER_TIMES, DEFAULT_LOOP_TIMES } from '../../../constants'
|
|
import { useEdgesInteractionsWithoutSync } from '../../../hooks/use-edges-interactions-without-sync'
|
|
import { useNodesInteractionsWithoutSync } from '../../../hooks/use-nodes-interactions-without-sync'
|
|
import { useStore, useWorkflowStore } from '../../../store'
|
|
import { WorkflowRunningStatus } from '../../../types'
|
|
|
|
type UseChatFlowControlParams = {
|
|
stopChat?: (taskId: string) => void
|
|
}
|
|
|
|
export function useChatFlowControl({
|
|
stopChat,
|
|
}: UseChatFlowControlParams) {
|
|
const workflowStore = useWorkflowStore()
|
|
const setIsResponding = useStore(s => s.setIsResponding)
|
|
const resetChatPreview = useStore(s => s.resetChatPreview)
|
|
const setActiveTaskId = useStore(s => s.setActiveTaskId)
|
|
const setHasStopResponded = useStore(s => s.setHasStopResponded)
|
|
const setSuggestedQuestionsAbortController = useStore(s => s.setSuggestedQuestionsAbortController)
|
|
const invalidateRun = useStore(s => s.invalidateRun)
|
|
const { handleNodeCancelRunningStatus } = useNodesInteractionsWithoutSync()
|
|
const { handleEdgeCancelRunningStatus } = useEdgesInteractionsWithoutSync()
|
|
|
|
const { setIterTimes, setLoopTimes } = workflowStore.getState()
|
|
|
|
const handleResponding = useCallback((responding: boolean) => {
|
|
setIsResponding(responding)
|
|
}, [setIsResponding])
|
|
|
|
const handleStop = useCallback(() => {
|
|
const {
|
|
activeTaskId,
|
|
suggestedQuestionsAbortController,
|
|
workflowRunningData,
|
|
setWorkflowRunningData,
|
|
} = workflowStore.getState()
|
|
const runningStatus = workflowRunningData?.result?.status
|
|
const isActiveRun = runningStatus === WorkflowRunningStatus.Running || runningStatus === WorkflowRunningStatus.Waiting
|
|
setHasStopResponded(true)
|
|
handleResponding(false)
|
|
if (stopChat && activeTaskId)
|
|
stopChat(activeTaskId)
|
|
setIterTimes(DEFAULT_ITER_TIMES)
|
|
setLoopTimes(DEFAULT_LOOP_TIMES)
|
|
if (suggestedQuestionsAbortController)
|
|
suggestedQuestionsAbortController.abort()
|
|
setSuggestedQuestionsAbortController(null)
|
|
setActiveTaskId('')
|
|
invalidateRun()
|
|
if (isActiveRun && workflowRunningData) {
|
|
setWorkflowRunningData({
|
|
...workflowRunningData,
|
|
result: {
|
|
...workflowRunningData.result,
|
|
status: WorkflowRunningStatus.Stopped,
|
|
},
|
|
})
|
|
}
|
|
if (isActiveRun) {
|
|
handleNodeCancelRunningStatus()
|
|
handleEdgeCancelRunningStatus()
|
|
}
|
|
}, [
|
|
handleResponding,
|
|
setIterTimes,
|
|
setLoopTimes,
|
|
stopChat,
|
|
workflowStore,
|
|
setHasStopResponded,
|
|
setSuggestedQuestionsAbortController,
|
|
setActiveTaskId,
|
|
invalidateRun,
|
|
handleNodeCancelRunningStatus,
|
|
handleEdgeCancelRunningStatus,
|
|
])
|
|
|
|
const handleRestart = useCallback(() => {
|
|
handleStop()
|
|
resetChatPreview()
|
|
setIterTimes(DEFAULT_ITER_TIMES)
|
|
setLoopTimes(DEFAULT_LOOP_TIMES)
|
|
}, [handleStop, setIterTimes, setLoopTimes, resetChatPreview])
|
|
|
|
return {
|
|
handleResponding,
|
|
handleStop,
|
|
handleRestart,
|
|
}
|
|
}
|