Sfoglia il codice sorgente

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

tags/1.9.0
17hz 1 mese fa
parent
commit
928bef9d82
Nessun account collegato all'indirizzo email del committer
1 ha cambiato i file con 4 aggiunte e 2 eliminazioni
  1. 4
    2
      web/app/components/base/markdown-blocks/think-block.tsx

+ 4
- 2
web/app/components/base/markdown-blocks/think-block.tsx Vedi File

@@ -1,5 +1,6 @@
import React, { useEffect, useRef, useState } from 'react'
import { useTranslation } from 'react-i18next'
import { useChatContext } from '../chat/chat/context'

const hasEndThink = (children: any): boolean => {
if (typeof children === 'string')
@@ -35,6 +36,7 @@ const removeEndThink = (children: any): any => {
}

const useThinkTimer = (children: any) => {
const { isResponding } = useChatContext()
const [startTime] = useState(Date.now())
const [elapsedTime, setElapsedTime] = useState(0)
const [isComplete, setIsComplete] = useState(false)
@@ -54,9 +56,9 @@ const useThinkTimer = (children: any) => {
}, [startTime, isComplete])

useEffect(() => {
if (hasEndThink(children))
if (hasEndThink(children) || !isResponding)
setIsComplete(true)
}, [children])
}, [children, isResponding])

return { elapsedTime, isComplete }
}

Loading…
Annulla
Salva