Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make document call component clickable #558

Merged
merged 4 commits into from
Nov 19, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion app/(chat)/chat/[id]/page.tsx
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@

import { cookies } from 'next/headers';
import { notFound } from 'next/navigation';

Expand Down
52 changes: 40 additions & 12 deletions components/document.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -3,14 +3,19 @@ import type { SetStateAction } from 'react';
import type { UIBlock } from './block';
import { FileIcon, LoaderIcon, MessageIcon, PencilEditIcon } from './icons';

const getActionText = (type: 'create' | 'update' | 'request-suggestions') => {
const getActionText = (
type: 'create' | 'update' | 'request-suggestions',
tense: 'present' | 'past',
) => {
switch (type) {
case 'create':
return 'Creating';
return tense === 'present' ? 'Creating' : 'Created';
case 'update':
return 'Updating';
return tense === 'present' ? 'Updating' : 'Updated';
case 'request-suggestions':
return 'Adding suggestions';
return tense === 'present'
? 'Adding suggestions'
: 'Added suggestions to';
default:
return null;
}
Expand All @@ -26,7 +31,6 @@ interface DocumentToolResultProps {
export function DocumentToolResult({
type,
result,
block,
setBlock,
}: DocumentToolResultProps) {
return (
Expand Down Expand Up @@ -62,8 +66,8 @@ export function DocumentToolResult({
<MessageIcon />
) : null}
</div>
<div className="">
{getActionText(type)} {result.title}
<div className="text-left">
{`${getActionText(type, 'past')} "${result.title}"`}
</div>
</button>
);
Expand All @@ -72,11 +76,35 @@ export function DocumentToolResult({
interface DocumentToolCallProps {
type: 'create' | 'update' | 'request-suggestions';
args: { title: string };
setBlock: (value: SetStateAction<UIBlock>) => void;
}

export function DocumentToolCall({ type, args }: DocumentToolCallProps) {
export function DocumentToolCall({
type,
args,
setBlock,
}: DocumentToolCallProps) {
return (
<div className="w-fit border py-2 px-3 rounded-xl flex flex-row items-start justify-between gap-3">
<button
type="button"
className="cursor pointer w-fit border py-2 px-3 rounded-xl flex flex-row items-start justify-between gap-3"
onClick={(event) => {
const rect = event.currentTarget.getBoundingClientRect();

const boundingBox = {
top: rect.top,
left: rect.left,
width: rect.width,
height: rect.height,
};

setBlock((currentBlock) => ({
...currentBlock,
isVisible: true,
boundingBox,
}));
}}
>
<div className="flex flex-row gap-3 items-start">
<div className="text-zinc-500 mt-1">
{type === 'create' ? (
Expand All @@ -88,12 +116,12 @@ export function DocumentToolCall({ type, args }: DocumentToolCallProps) {
) : null}
</div>

<div className="">
{getActionText(type)} {args.title}
<div className="text-left">
{`${getActionText(type, 'present')} ${args.title ? `"${args.title}"` : ''}`}
</div>
</div>

<div className="animate-spin mt-1">{<LoaderIcon />}</div>
</div>
</button>
);
}
13 changes: 11 additions & 2 deletions components/message.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -104,13 +104,22 @@ export const PreviewMessage = ({
{toolName === 'getWeather' ? (
<Weather />
) : toolName === 'createDocument' ? (
<DocumentToolCall type="create" args={args} />
<DocumentToolCall
type="create"
args={args}
setBlock={setBlock}
/>
) : toolName === 'updateDocument' ? (
<DocumentToolCall type="update" args={args} />
<DocumentToolCall
type="update"
args={args}
setBlock={setBlock}
/>
) : toolName === 'requestSuggestions' ? (
<DocumentToolCall
type="request-suggestions"
args={args}
setBlock={setBlock}
/>
) : null}
</div>
Expand Down