Skip to content

Commit 5fc868a

Browse files
committed
chore: disable generate button when no sub
1 parent d7a49d1 commit 5fc868a

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

  • apps/desktop/src/routes/_protected/database/$id/sql/-components/chat

apps/desktop/src/routes/_protected/database/$id/sql/-components/chat/chat-messages.tsx

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import { useStickToBottom } from 'use-stick-to-bottom'
2121
import { Markdown } from '~/components/markdown'
2222
import { connectionStore } from '~/entities/connection/store'
2323
import { UserAvatar } from '~/entities/user/components'
24+
import { useSubscription } from '~/entities/user/hooks'
2425
import { Route } from '../..'
2526
import { chatHooks, runnerHooks } from '../../-page'
2627
import { ChatImages } from './chat-images'
@@ -349,6 +350,7 @@ function AssistantMessage({ message, isLast, status, className, ...props }: { me
349350
const { chat } = Route.useLoaderData()
350351
const ref = useRef<HTMLDivElement>(null)
351352
const { height } = useElementSize(ref)
353+
const { subscription } = useSubscription()
352354

353355
const isLoading = isLast ? status === 'streaming' || status === 'submitted' : false
354356

@@ -387,7 +389,7 @@ function AssistantMessage({ message, isLast, status, className, ...props }: { me
387389
<ChatMessageFooterButton
388390
icon={<RiRestartLine className="size-4 text-muted-foreground" />}
389391
tooltip="Regenerate message"
390-
disabled={status === 'streaming' || status === 'submitted'}
392+
disabled={status === 'streaming' || status === 'submitted' || !subscription}
391393
onClick={() => chat.regenerate({ messageId: message.id })}
392394
/>
393395
)}

0 commit comments

Comments
 (0)