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

fix: imporve the condition for stopping the think timer. (#25365)

Browse source
This commit is contained in:
17hz 2025-09-10 08:45:00 +08:00 • committed by GitHub
commit 928bef9d82
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
web/app/components/base/markdown-blocks/think-block.tsx
View file

@ -1,5 +1,6 @@
import React, { useEffect, useRef, useState } from 'react' import React, { useEffect, useRef, useState } from 'react'
import { useTranslation } from 'react-i18next' import { useTranslation } from 'react-i18next'
import { useChatContext } from '../chat/chat/context'
const hasEndThink = (children: any): boolean => { const hasEndThink = (children: any): boolean => {
if (typeof children === 'string') if (typeof children === 'string')
@ -35,6 +36,7 @@ const removeEndThink = (children: any): any => {
} }
const useThinkTimer = (children: any) => { const useThinkTimer = (children: any) => {
const { isResponding } = useChatContext()
const [startTime] = useState(Date.now()) const [startTime] = useState(Date.now())
const [elapsedTime, setElapsedTime] = useState(0) const [elapsedTime, setElapsedTime] = useState(0)
const [isComplete, setIsComplete] = useState(false) const [isComplete, setIsComplete] = useState(false)
@ -54,9 +56,9 @@ const useThinkTimer = (children: any) => {
}, [startTime, isComplete]) }, [startTime, isComplete])
useEffect(() => { useEffect(() => {
if (hasEndThink(children)) if (hasEndThink(children) || !isResponding)
setIsComplete(true) setIsComplete(true)
}, [children]) }, [children, isResponding])
return { elapsedTime, isComplete } return { elapsedTime, isComplete }
} }