Spaces:
Sleeping
Sleeping
File size: 1,570 Bytes
f3a9ef2 93dd66e f3a9ef2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
'use client';
import { useChat, type Message } from 'ai/react';
import { cn } from '@/lib/utils';
import { ChatList } from '@/components/chat/ChatList';
import { ChatPanel } from '@/components/chat-panel';
import { ChatScrollAnchor } from '@/components/chat-scroll-anchor';
import { toast } from 'react-hot-toast';
import { useAtom } from 'jotai';
import { datasetAtom } from '@/state';
import ImageList from './ImageList';
export interface ChatProps extends React.ComponentProps<'div'> {
initialMessages?: Message[];
id?: string;
}
export function Chat({ id, initialMessages, className }: ChatProps) {
const [dataset] = useAtom(datasetAtom);
const { messages, append, reload, stop, isLoading, input, setInput } =
useChat({
initialMessages,
id,
body: {
id,
dataset: dataset.filter(entity => entity.selected),
},
onResponse(response) {
if (response.status === 401) {
toast.error(response.statusText);
}
},
});
return (
<>
<div className={cn('pb-[150px] pt-4 md:pt-10 h-full', className)}>
<div className="flex h-full">
<div className="w-1/2 relative border-r-2 border-gray-200">
<ImageList />
</div>
<div className="w-1/2 relative overflow-auto">
<ChatList messages={messages} isLoading={isLoading} />
<ChatScrollAnchor trackVisibility={isLoading} />
</div>
</div>
</div>
<ChatPanel
id={id}
isLoading={isLoading}
stop={stop}
append={append}
reload={reload}
messages={messages}
input={input}
setInput={setInput}
/>
</>
);
}
|