dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add redirect

Browse source
This commit is contained in:
Gary Meehan 2023-10-04 22:27:23 +01:00
parent 0a201f63c3
commit 5c3001e29a
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
components/chat.tsx
View file

@ -20,6 +20,7 @@ import { useState } from 'react'
import { Button } from './ui/button' import { Button } from './ui/button'
import { Input } from './ui/input' import { Input } from './ui/input'
import { toast } from 'react-hot-toast' import { toast } from 'react-hot-toast'
import { usePathname, useRouter } from 'next/navigation'
const IS_PREVIEW = process.env.VERCEL_ENV === 'preview' const IS_PREVIEW = process.env.VERCEL_ENV === 'preview'
export interface ChatProps extends React.ComponentProps<'div'> { export interface ChatProps extends React.ComponentProps<'div'> {
@ -28,6 +29,8 @@ export interface ChatProps extends React.ComponentProps<'div'> {
} }
export function Chat({ id, initialMessages, className }: ChatProps) { export function Chat({ id, initialMessages, className }: ChatProps) {
const router = useRouter()
const path = usePathname()
const [previewToken, setPreviewToken] = useLocalStorage<string | null>( const [previewToken, setPreviewToken] = useLocalStorage<string | null>(
'ai-token', 'ai-token',
null null
@ -46,6 +49,12 @@ export function Chat({ id, initialMessages, className }: ChatProps) {
if (response.status === 401) { if (response.status === 401) {
toast.error(response.statusText) toast.error(response.statusText)
} }
},
onFinish() {
if (!path.includes('chat')) {
router.push(`/chat/${id}`, { shallow: true })
router.refresh()
}
} }
}) })
return ( return (