feat: v1 — persistent shell, model gateway, artifact improvements (#1462)

artifacts/text/client.tsx

@@ -1,7 +1,7 @@
 import { toast } from "sonner";
-import { Artifact } from "@/components/create-artifact";
-import { DiffView } from "@/components/diffview";
-import { DocumentSkeleton } from "@/components/document-skeleton";
+import { Artifact } from "@/components/chat/create-artifact";
+import { DiffView } from "@/components/chat/diffview";
+import { DocumentSkeleton } from "@/components/chat/document-skeleton";
 import {
   ClockRewind,
   CopyIcon,
@@ -9,8 +9,8 @@ import {
   PenIcon,
   RedoIcon,
   UndoIcon,
-} from "@/components/icons";
-import { Editor } from "@/components/text-editor";
+} from "@/components/chat/icons";
+import { Editor } from "@/components/chat/text-editor";
 import type { Suggestion } from "@/lib/db/schema";
 import { getSuggestions } from "../actions";
 
@@ -69,21 +69,28 @@ export const textArtifact = new Artifact<"text", TextArtifactMetadata>({
     }
 
     if (mode === "diff") {
-      const oldContent = getDocumentContentById(currentVersionIndex - 1);
-      const newContent = getDocumentContentById(currentVersionIndex);
+      const selectedContent = getDocumentContentById(currentVersionIndex);
+      const prevContent =
+        currentVersionIndex > 0
+          ? getDocumentContentById(currentVersionIndex - 1)
+          : selectedContent;
 
-      return <DiffView newContent={newContent} oldContent={oldContent} />;
+      return (
+        <div className="flex flex-row px-4 py-8 md:px-16 md:py-12 lg:px-20">
+          <DiffView newContent={selectedContent} oldContent={prevContent} />
+        </div>
+      );
     }
 
     return (
-      <div className="flex flex-row px-4 py-8 md:p-20">
+      <div className="flex flex-row px-4 py-8 md:px-16 md:py-12 lg:px-20">
         <Editor
           content={content}
           currentVersionIndex={currentVersionIndex}
           isCurrentVersion={isCurrentVersion}
           onSaveContent={onSaveContent}
           status={status}
-          suggestions={metadata ? metadata.suggestions : []}
+          suggestions={isCurrentVersion && metadata ? metadata.suggestions : []}
         />
 
         {metadata?.suggestions && metadata.suggestions.length > 0 ? (

artifacts/text/server.ts

@@ -1,15 +1,15 @@
 import { smoothStream, streamText } from "ai";
 import { updateDocumentPrompt } from "@/lib/ai/prompts";
-import { getArtifactModel } from "@/lib/ai/providers";
+import { getLanguageModel } from "@/lib/ai/providers";
 import { createDocumentHandler } from "@/lib/artifacts/server";
 
 export const textDocumentHandler = createDocumentHandler<"text">({
   kind: "text",
-  onCreateDocument: async ({ title, dataStream }) => {
+  onCreateDocument: async ({ title, dataStream, modelId }) => {
     let draftContent = "";
 
     const { fullStream } = streamText({
-      model: getArtifactModel(),
+      model: getLanguageModel(modelId),
       system:
         "Write about the given topic. Markdown is supported. Use headings wherever appropriate.",
       experimental_transform: smoothStream({ chunking: "word" }),
@@ -17,16 +17,11 @@ export const textDocumentHandler = createDocumentHandler<"text">({
     });
 
     for await (const delta of fullStream) {
-      const { type } = delta;
-
-      if (type === "text-delta") {
-        const { text } = delta;
-
-        draftContent += text;
-
+      if (delta.type === "text-delta") {
+        draftContent += delta.text;
         dataStream.write({
           type: "data-textDelta",
-          data: text,
+          data: delta.text,
           transient: true,
         });
       }
@@ -34,35 +29,22 @@ export const textDocumentHandler = createDocumentHandler<"text">({
 
     return draftContent;
   },
-  onUpdateDocument: async ({ document, description, dataStream }) => {
+  onUpdateDocument: async ({ document, description, dataStream, modelId }) => {
     let draftContent = "";
 
     const { fullStream } = streamText({
-      model: getArtifactModel(),
+      model: getLanguageModel(modelId),
       system: updateDocumentPrompt(document.content, "text"),
       experimental_transform: smoothStream({ chunking: "word" }),
       prompt: description,
-      providerOptions: {
-        openai: {
-          prediction: {
-            type: "content",
-            content: document.content,
-          },
-        },
-      },
     });
 
     for await (const delta of fullStream) {
-      const { type } = delta;
-
-      if (type === "text-delta") {
-        const { text } = delta;
-
-        draftContent += text;
-
+      if (delta.type === "text-delta") {
+        draftContent += delta.text;
         dataStream.write({
           type: "data-textDelta",
-          data: text,
+          data: delta.text,
           transient: true,
         });
       }