| @@ -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 } | |||
| } | |||