diff --git a/ee/tabby-ui/components/chat/chat.tsx b/ee/tabby-ui/components/chat/chat.tsx index a762504fb..8466fdcd0 100644 --- a/ee/tabby-ui/components/chat/chat.tsx +++ b/ee/tabby-ui/components/chat/chat.tsx @@ -21,6 +21,7 @@ import { QuestionAnswerList } from './question-answer' import { useTabbyAnswer } from './use-tabby-answer' import { useLatest } from '@/lib/hooks/use-latest' import { useDebounceCallback } from '@/lib/hooks/use-debounce' +import { ListSkeleton } from '../skeleton' type ChatContextValue = { isLoading: boolean @@ -131,9 +132,10 @@ function ChatRenderer( }: ChatProps, ref: React.ForwardedRef ) { + const [initialized, setInitialzed] = React.useState(false) + const isOnLoadExecuted = React.useRef(false) const [qaPairs, setQaPairs] = React.useState(initialMessages ?? []) const [input, setInput] = React.useState('') - const loaded = React.useRef(false) const { triggerRequest, isLoading, error, answer, stop } = useTabbyAnswer({ api, @@ -334,7 +336,7 @@ function ChatRenderer( } React.useEffect(() => { - if (!loaded.current) return + if (!isOnLoadExecuted.current) return onThreadUpdates(qaPairs) }, [qaPairs]) @@ -351,12 +353,15 @@ function ChatRenderer( ) React.useEffect(() => { - if (loaded?.current) return + if (isOnLoadExecuted.current) return - loaded.current = true + isOnLoadExecuted.current = true onLoaded?.() + setInitialzed(true) }, []) + if (!initialized) return + return (