132 lines
3.2 KiB
TypeScript
132 lines
3.2 KiB
TypeScript
'use client';
|
|
|
|
import type { Attachment, Message } from 'ai';
|
|
import { useChat } from 'ai/react';
|
|
import { AnimatePresence } from 'framer-motion';
|
|
import { useState } from 'react';
|
|
import useSWR, { useSWRConfig } from 'swr';
|
|
import { useWindowSize } from 'usehooks-ts';
|
|
|
|
import { ChatHeader } from '@/components/chat-header';
|
|
import type { Vote } from '@/lib/db/schema';
|
|
import { fetcher } from '@/lib/utils';
|
|
|
|
import { Block, type UIBlock } from './block';
|
|
import { BlockStreamHandler } from './block-stream-handler';
|
|
import { MultimodalInput } from './multimodal-input';
|
|
import { Messages } from './messages';
|
|
|
|
export function Chat({
|
|
id,
|
|
initialMessages,
|
|
selectedModelId,
|
|
}: {
|
|
id: string;
|
|
initialMessages: Array<Message>;
|
|
selectedModelId: string;
|
|
}) {
|
|
const { mutate } = useSWRConfig();
|
|
|
|
const {
|
|
messages,
|
|
setMessages,
|
|
handleSubmit,
|
|
input,
|
|
setInput,
|
|
append,
|
|
isLoading,
|
|
stop,
|
|
reload,
|
|
data: streamingData,
|
|
} = useChat({
|
|
id,
|
|
body: { id, modelId: selectedModelId },
|
|
initialMessages,
|
|
onFinish: () => {
|
|
mutate('/api/history');
|
|
},
|
|
});
|
|
|
|
const { width: windowWidth = 1920, height: windowHeight = 1080 } =
|
|
useWindowSize();
|
|
|
|
const [block, setBlock] = useState<UIBlock>({
|
|
documentId: 'init',
|
|
content: '',
|
|
title: '',
|
|
status: 'idle',
|
|
isVisible: false,
|
|
boundingBox: {
|
|
top: windowHeight / 4,
|
|
left: windowWidth / 4,
|
|
width: 250,
|
|
height: 50,
|
|
},
|
|
});
|
|
|
|
const { data: votes } = useSWR<Array<Vote>>(
|
|
`/api/vote?chatId=${id}`,
|
|
fetcher,
|
|
);
|
|
|
|
const [attachments, setAttachments] = useState<Array<Attachment>>([]);
|
|
|
|
return (
|
|
<>
|
|
<div className="flex flex-col min-w-0 h-dvh bg-background">
|
|
<ChatHeader selectedModelId={selectedModelId} />
|
|
|
|
<Messages
|
|
chatId={id}
|
|
block={block}
|
|
setBlock={setBlock}
|
|
isLoading={isLoading}
|
|
votes={votes}
|
|
messages={messages}
|
|
setMessages={setMessages}
|
|
reload={reload}
|
|
/>
|
|
|
|
<form className="flex mx-auto px-4 bg-background pb-4 md:pb-6 gap-2 w-full md:max-w-3xl">
|
|
<MultimodalInput
|
|
chatId={id}
|
|
input={input}
|
|
setInput={setInput}
|
|
handleSubmit={handleSubmit}
|
|
isLoading={isLoading}
|
|
stop={stop}
|
|
attachments={attachments}
|
|
setAttachments={setAttachments}
|
|
messages={messages}
|
|
setMessages={setMessages}
|
|
append={append}
|
|
/>
|
|
</form>
|
|
</div>
|
|
|
|
<AnimatePresence>
|
|
{block?.isVisible && (
|
|
<Block
|
|
chatId={id}
|
|
input={input}
|
|
setInput={setInput}
|
|
handleSubmit={handleSubmit}
|
|
isLoading={isLoading}
|
|
stop={stop}
|
|
attachments={attachments}
|
|
setAttachments={setAttachments}
|
|
append={append}
|
|
block={block}
|
|
setBlock={setBlock}
|
|
messages={messages}
|
|
setMessages={setMessages}
|
|
reload={reload}
|
|
votes={votes}
|
|
/>
|
|
)}
|
|
</AnimatePresence>
|
|
|
|
<BlockStreamHandler streamingData={streamingData} setBlock={setBlock} />
|
|
</>
|
|
);
|
|
}
|