Restore Ultracite + fix sidebar (#1233)
This commit is contained in:
Hayden Bleasel
GitHub
parent
8fbfc253fa
commit
947ed094a6
177 changed files with 6908 additions and 8306 deletions
|
|
@ -1,7 +1,7 @@
|
|||
import { Artifact } from '@/components/create-artifact';
|
||||
import { DiffView } from '@/components/diffview';
|
||||
import { DocumentSkeleton } from '@/components/document-skeleton';
|
||||
import { Editor } from '@/components/text-editor';
|
||||
import { toast } from "sonner";
|
||||
import { Artifact } from "@/components/create-artifact";
|
||||
import { DiffView } from "@/components/diffview";
|
||||
import { DocumentSkeleton } from "@/components/document-skeleton";
|
||||
import {
|
||||
ClockRewind,
|
||||
CopyIcon,
|
||||
|
|
@ -9,18 +9,18 @@ import {
|
|||
PenIcon,
|
||||
RedoIcon,
|
||||
UndoIcon,
|
||||
} from '@/components/icons';
|
||||
import type { Suggestion } from '@/lib/db/schema';
|
||||
import { toast } from 'sonner';
|
||||
import { getSuggestions } from '../actions';
|
||||
} from "@/components/icons";
|
||||
import { Editor } from "@/components/text-editor";
|
||||
import type { Suggestion } from "@/lib/db/schema";
|
||||
import { getSuggestions } from "../actions";
|
||||
|
||||
interface TextArtifactMetadata {
|
||||
suggestions: Array<Suggestion>;
|
||||
}
|
||||
type TextArtifactMetadata = {
|
||||
suggestions: Suggestion[];
|
||||
};
|
||||
|
||||
export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
||||
kind: 'text',
|
||||
description: 'Useful for text content, like drafting essays and emails.',
|
||||
export const textArtifact = new Artifact<"text", TextArtifactMetadata>({
|
||||
kind: "text",
|
||||
description: "Useful for text content, like drafting essays and emails.",
|
||||
initialize: async ({ documentId, setMetadata }) => {
|
||||
const suggestions = await getSuggestions({ documentId });
|
||||
|
||||
|
|
@ -29,7 +29,7 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
});
|
||||
},
|
||||
onStreamPart: ({ streamPart, setMetadata, setArtifact }) => {
|
||||
if (streamPart.type === 'data-suggestion') {
|
||||
if (streamPart.type === "data-suggestion") {
|
||||
setMetadata((metadata) => {
|
||||
return {
|
||||
suggestions: [...metadata.suggestions, streamPart.data],
|
||||
|
|
@ -37,18 +37,18 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
});
|
||||
}
|
||||
|
||||
if (streamPart.type === 'data-textDelta') {
|
||||
if (streamPart.type === "data-textDelta") {
|
||||
setArtifact((draftArtifact) => {
|
||||
return {
|
||||
...draftArtifact,
|
||||
content: draftArtifact.content + streamPart.data,
|
||||
isVisible:
|
||||
draftArtifact.status === 'streaming' &&
|
||||
draftArtifact.status === "streaming" &&
|
||||
draftArtifact.content.length > 400 &&
|
||||
draftArtifact.content.length < 450
|
||||
? true
|
||||
: draftArtifact.isVisible,
|
||||
status: 'streaming',
|
||||
status: "streaming",
|
||||
};
|
||||
});
|
||||
}
|
||||
|
|
@ -68,40 +68,38 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
return <DocumentSkeleton artifactKind="text" />;
|
||||
}
|
||||
|
||||
if (mode === 'diff') {
|
||||
if (mode === "diff") {
|
||||
const oldContent = getDocumentContentById(currentVersionIndex - 1);
|
||||
const newContent = getDocumentContentById(currentVersionIndex);
|
||||
|
||||
return <DiffView oldContent={oldContent} newContent={newContent} />;
|
||||
return <DiffView newContent={newContent} oldContent={oldContent} />;
|
||||
}
|
||||
|
||||
return (
|
||||
<>
|
||||
<div className="flex flex-row px-4 py-8 md:p-20">
|
||||
<Editor
|
||||
content={content}
|
||||
suggestions={metadata ? metadata.suggestions : []}
|
||||
isCurrentVersion={isCurrentVersion}
|
||||
currentVersionIndex={currentVersionIndex}
|
||||
status={status}
|
||||
onSaveContent={onSaveContent}
|
||||
/>
|
||||
<div className="flex flex-row px-4 py-8 md:p-20">
|
||||
<Editor
|
||||
content={content}
|
||||
currentVersionIndex={currentVersionIndex}
|
||||
isCurrentVersion={isCurrentVersion}
|
||||
onSaveContent={onSaveContent}
|
||||
status={status}
|
||||
suggestions={metadata ? metadata.suggestions : []}
|
||||
/>
|
||||
|
||||
{metadata?.suggestions && metadata.suggestions.length > 0 ? (
|
||||
<div className="h-dvh w-12 shrink-0 md:hidden" />
|
||||
) : null}
|
||||
</div>
|
||||
</>
|
||||
{metadata?.suggestions && metadata.suggestions.length > 0 ? (
|
||||
<div className="h-dvh w-12 shrink-0 md:hidden" />
|
||||
) : null}
|
||||
</div>
|
||||
);
|
||||
},
|
||||
actions: [
|
||||
{
|
||||
icon: <ClockRewind size={18} />,
|
||||
description: 'View changes',
|
||||
description: "View changes",
|
||||
onClick: ({ handleVersionChange }) => {
|
||||
handleVersionChange('toggle');
|
||||
handleVersionChange("toggle");
|
||||
},
|
||||
isDisabled: ({ currentVersionIndex, setMetadata }) => {
|
||||
isDisabled: ({ currentVersionIndex }) => {
|
||||
if (currentVersionIndex === 0) {
|
||||
return true;
|
||||
}
|
||||
|
|
@ -111,9 +109,9 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
},
|
||||
{
|
||||
icon: <UndoIcon size={18} />,
|
||||
description: 'View Previous version',
|
||||
description: "View Previous version",
|
||||
onClick: ({ handleVersionChange }) => {
|
||||
handleVersionChange('prev');
|
||||
handleVersionChange("prev");
|
||||
},
|
||||
isDisabled: ({ currentVersionIndex }) => {
|
||||
if (currentVersionIndex === 0) {
|
||||
|
|
@ -125,9 +123,9 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
},
|
||||
{
|
||||
icon: <RedoIcon size={18} />,
|
||||
description: 'View Next version',
|
||||
description: "View Next version",
|
||||
onClick: ({ handleVersionChange }) => {
|
||||
handleVersionChange('next');
|
||||
handleVersionChange("next");
|
||||
},
|
||||
isDisabled: ({ isCurrentVersion }) => {
|
||||
if (isCurrentVersion) {
|
||||
|
|
@ -139,24 +137,24 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
},
|
||||
{
|
||||
icon: <CopyIcon size={18} />,
|
||||
description: 'Copy to clipboard',
|
||||
description: "Copy to clipboard",
|
||||
onClick: ({ content }) => {
|
||||
navigator.clipboard.writeText(content);
|
||||
toast.success('Copied to clipboard!');
|
||||
toast.success("Copied to clipboard!");
|
||||
},
|
||||
},
|
||||
],
|
||||
toolbar: [
|
||||
{
|
||||
icon: <PenIcon />,
|
||||
description: 'Add final polish',
|
||||
description: "Add final polish",
|
||||
onClick: ({ sendMessage }) => {
|
||||
sendMessage({
|
||||
role: 'user',
|
||||
role: "user",
|
||||
parts: [
|
||||
{
|
||||
type: 'text',
|
||||
text: 'Please add final polish and check for grammar, add section titles for better structure, and ensure everything reads smoothly.',
|
||||
type: "text",
|
||||
text: "Please add final polish and check for grammar, add section titles for better structure, and ensure everything reads smoothly.",
|
||||
},
|
||||
],
|
||||
});
|
||||
|
|
@ -164,14 +162,14 @@ export const textArtifact = new Artifact<'text', TextArtifactMetadata>({
|
|||
},
|
||||
{
|
||||
icon: <MessageIcon />,
|
||||
description: 'Request suggestions',
|
||||
description: "Request suggestions",
|
||||
onClick: ({ sendMessage }) => {
|
||||
sendMessage({
|
||||
role: 'user',
|
||||
role: "user",
|
||||
parts: [
|
||||
{
|
||||
type: 'text',
|
||||
text: 'Please add suggestions you have that could improve the writing.',
|
||||
type: "text",
|
||||
text: "Please add suggestions you have that could improve the writing.",
|
||||
},
|
||||
],
|
||||
});
|
||||
|
|
|
|||
|
|
@ -1,31 +1,31 @@
|
|||
import { smoothStream, streamText } from 'ai';
|
||||
import { myProvider } from '@/lib/ai/providers';
|
||||
import { createDocumentHandler } from '@/lib/artifacts/server';
|
||||
import { updateDocumentPrompt } from '@/lib/ai/prompts';
|
||||
import { smoothStream, streamText } from "ai";
|
||||
import { updateDocumentPrompt } from "@/lib/ai/prompts";
|
||||
import { myProvider } from "@/lib/ai/providers";
|
||||
import { createDocumentHandler } from "@/lib/artifacts/server";
|
||||
|
||||
export const textDocumentHandler = createDocumentHandler<'text'>({
|
||||
kind: 'text',
|
||||
export const textDocumentHandler = createDocumentHandler<"text">({
|
||||
kind: "text",
|
||||
onCreateDocument: async ({ title, dataStream }) => {
|
||||
let draftContent = '';
|
||||
let draftContent = "";
|
||||
|
||||
const { fullStream } = streamText({
|
||||
model: myProvider.languageModel('artifact-model'),
|
||||
model: myProvider.languageModel("artifact-model"),
|
||||
system:
|
||||
'Write about the given topic. Markdown is supported. Use headings wherever appropriate.',
|
||||
experimental_transform: smoothStream({ chunking: 'word' }),
|
||||
"Write about the given topic. Markdown is supported. Use headings wherever appropriate.",
|
||||
experimental_transform: smoothStream({ chunking: "word" }),
|
||||
prompt: title,
|
||||
});
|
||||
|
||||
for await (const delta of fullStream) {
|
||||
const { type } = delta;
|
||||
|
||||
if (type === 'text-delta') {
|
||||
if (type === "text-delta") {
|
||||
const { text } = delta;
|
||||
|
||||
draftContent += text;
|
||||
|
||||
dataStream.write({
|
||||
type: 'data-textDelta',
|
||||
type: "data-textDelta",
|
||||
data: text,
|
||||
transient: true,
|
||||
});
|
||||
|
|
@ -35,17 +35,17 @@ export const textDocumentHandler = createDocumentHandler<'text'>({
|
|||
return draftContent;
|
||||
},
|
||||
onUpdateDocument: async ({ document, description, dataStream }) => {
|
||||
let draftContent = '';
|
||||
let draftContent = "";
|
||||
|
||||
const { fullStream } = streamText({
|
||||
model: myProvider.languageModel('artifact-model'),
|
||||
system: updateDocumentPrompt(document.content, 'text'),
|
||||
experimental_transform: smoothStream({ chunking: 'word' }),
|
||||
model: myProvider.languageModel("artifact-model"),
|
||||
system: updateDocumentPrompt(document.content, "text"),
|
||||
experimental_transform: smoothStream({ chunking: "word" }),
|
||||
prompt: description,
|
||||
providerOptions: {
|
||||
openai: {
|
||||
prediction: {
|
||||
type: 'content',
|
||||
type: "content",
|
||||
content: document.content,
|
||||
},
|
||||
},
|
||||
|
|
@ -55,13 +55,13 @@ export const textDocumentHandler = createDocumentHandler<'text'>({
|
|||
for await (const delta of fullStream) {
|
||||
const { type } = delta;
|
||||
|
||||
if (type === 'text-delta') {
|
||||
if (type === "text-delta") {
|
||||
const { text } = delta;
|
||||
|
||||
draftContent += text;
|
||||
|
||||
dataStream.write({
|
||||
type: 'data-textDelta',
|
||||
type: "data-textDelta",
|
||||
data: text,
|
||||
transient: true,
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue