AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-05-13 15:18:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: status query not stop when leaving document embedding detail page (#4754)

Browse Source
This commit is contained in:
doufa 2024-05-31 08:07:48 +00:00 committed by GitHub
parent 7aaa1ff270
commit b1c9671a60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
web/app/components/datasets/documents/detail/embedding/index.tsx
View File

@ -126,17 +126,13 @@ const EmbeddingDetail: FC<Props> = ({ detail, stopPosition = 'top', datasetId: d
return status
}
const [isStopQuery, setIsStopQuery] = useState(false)
const isStopQueryRef = useRef(isStopQuery)
useEffect(() => {
isStopQueryRef.current = isStopQuery
}, [isStopQuery])
const stopQueryStatus = () => {
setIsStopQuery(true)
}
const isStopQuery = useRef(false)
const stopQueryStatus = useCallback(() => {
isStopQuery.current = true
}, [])
const startQueryStatus = async () => {
if (isStopQueryRef.current)
const startQueryStatus = useCallback(async () => {
if (isStopQuery.current)
return
try {
@ -146,6 +142,7 @@ const EmbeddingDetail: FC<Props> = ({ detail, stopPosition = 'top', datasetId: d
detailUpdate()
return
}
await sleep(2500)
await startQueryStatus()
}
@ -153,16 +150,15 @@ const EmbeddingDetail: FC<Props> = ({ detail, stopPosition = 'top', datasetId: d
await sleep(2500)
await startQueryStatus()
}
}
}, [stopQueryStatus])
useEffect(() => {
setIsStopQuery(false)
isStopQuery.current = false
startQueryStatus()
return () => {
stopQueryStatus()
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [])
}, [startQueryStatus, stopQueryStatus])
const { data: indexingEstimateDetail, error: indexingEstimateErr } = useSWR({
action: 'fetchIndexingEstimate',