update: init

TabbyML · Dec 18, 2024 · 2d844a2 · 2d844a2
1 parent c87eaf6
commit 2d844a2
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 11 deletions.
diff --git a/ee/tabby-ui/components/chat/chat-panel.tsx b/ee/tabby-ui/components/chat/chat-panel.tsx
@@ -323,6 +323,7 @@ function ChatPanelRenderer(
             setInput={setInput}
             isLoading={isLoading}
             chatInputRef={chatInputRef}
+            isInitializing={!initialized}
           />
           <FooterText className="hidden sm:block" />
         </div>

diff --git a/ee/tabby-ui/components/chat/chat.tsx b/ee/tabby-ui/components/chat/chat.tsx
@@ -38,7 +38,6 @@ import {
 } from '@/lib/types/chat'
 import { cn, nanoid } from '@/lib/utils'
 
-import { ListSkeleton } from '../skeleton'
 import { ChatPanel, ChatPanelRef } from './chat-panel'
 import { ChatScrollAnchor } from './chat-scroll-anchor'
 import { EmptyScreen } from './empty-screen'
@@ -142,7 +141,7 @@ function ChatRenderer(
   }: ChatProps,
   ref: React.ForwardedRef<ChatRef>
 ) {
-  const [initialized, setInitialzed] = React.useState(false)
+  const [initialized, setInitialized] = React.useState(false)
   const [threadId, setThreadId] = React.useState<string | undefined>()
   const isOnLoadExecuted = React.useRef(false)
   const [qaPairs, setQaPairs] = React.useState(initialMessages ?? [])
@@ -548,7 +547,7 @@ function ChatRenderer(
         }
       }
 
-      setInitialzed(true)
+      setInitialized(true)
     }
 
     if (!fetchingSources && !initialized) {
@@ -578,11 +577,6 @@ function ChatRenderer(
   )
 
   const chatMaxWidthClass = maxWidth ? `max-w-${maxWidth}` : 'max-w-2xl'
-  if (!initialized) {
-    return (
-      <ListSkeleton className={`${chatMaxWidthClass} mx-auto pt-4 md:pt-10`} />
-    )
-  }
 
   return (
     <ChatContext.Provider

diff --git a/ee/tabby-ui/components/chat/prompt-form.tsx b/ee/tabby-ui/components/chat/prompt-form.tsx
@@ -32,14 +32,22 @@ export interface PromptProps
   onSubmit: (value: string) => Promise<void>
   isLoading: boolean
   chatInputRef: React.RefObject<HTMLTextAreaElement>
+  isInitializing?: boolean
 }
 
 export interface PromptFormRef {
   focus: () => void
 }
 
 function PromptFormRenderer(
-  { onSubmit, input, setInput, isLoading, chatInputRef }: PromptProps,
+  {
+    onSubmit,
+    input,
+    setInput,
+    isLoading,
+    chatInputRef,
+    isInitializing
+  }: PromptProps,
   ref: React.ForwardedRef<PromptFormRef>
 ) {
   const { formRef, onKeyDown } = useEnterSubmit()
@@ -137,7 +145,7 @@ function PromptFormRenderer(
     HTMLFormElement
   > = async e => {
     e.preventDefault()
-    if (!input?.trim() || isLoading) {
+    if (!input?.trim() || isLoading || isInitializing) {
       return
     }
 
@@ -233,7 +241,7 @@ function PromptFormRenderer(
                         <Button
                           type="submit"
                           size="icon"
-                          disabled={isLoading || input === ''}
+                          disabled={isInitializing || isLoading || input === ''}
                         >
                           <IconArrowElbow />
                           <span className="sr-only">Send message</span>