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

artifacts/code/client.tsx

@@ -1,11 +1,11 @@
 import { toast } from "sonner";
-import { CodeEditor } from "@/components/code-editor";
+import { CodeEditor } from "@/components/chat/code-editor";
 import {
   Console,
   type ConsoleOutput,
   type ConsoleOutputContent,
-} from "@/components/console";
-import { Artifact } from "@/components/create-artifact";
+} from "@/components/chat/console";
+import { Artifact } from "@/components/chat/create-artifact";
 import {
   CopyIcon,
   LogsIcon,
@@ -13,7 +13,7 @@ import {
   PlayIcon,
   RedoIcon,
   UndoIcon,
-} from "@/components/icons";
+} from "@/components/chat/icons";
 import { generateUUID } from "@/lib/utils";
 
 const OUTPUT_HANDLERS = {
@@ -93,7 +93,7 @@ export const codeArtifact = new Artifact<"code", Metadata>({
   content: ({ metadata, setMetadata, ...props }) => {
     return (
       <>
-        <div className="px-1">
+        <div className="relative min-h-[200px]">
           <CodeEditor {...props} />
         </div>
 
@@ -193,14 +193,20 @@ export const codeArtifact = new Artifact<"code", Metadata>({
               },
             ],
           }));
-        } catch (error: any) {
+        } catch (error: unknown) {
           setMetadata((metadata) => ({
             ...metadata,
             outputs: [
               ...metadata.outputs.filter((output) => output.id !== runId),
               {
                 id: runId,
-                contents: [{ type: "text", value: error.message }],
+                contents: [
+                  {
+                    type: "text",
+                    value:
+                      error instanceof Error ? error.message : String(error),
+                  },
+                ],
                 status: "failed",
               },
             ],

artifacts/code/server.ts

@@ -1,75 +1,59 @@
-import { streamObject } from "ai";
-import { z } from "zod";
+import { streamText } from "ai";
 import { codePrompt, updateDocumentPrompt } from "@/lib/ai/prompts";
-import { getArtifactModel } from "@/lib/ai/providers";
+import { getLanguageModel } from "@/lib/ai/providers";
 import { createDocumentHandler } from "@/lib/artifacts/server";
 
+function stripFences(code: string): string {
+  return code
+    .replace(/^```[\w]*\n?/, "")
+    .replace(/\n?```\s*$/, "")
+    .trim();
+}
+
 export const codeDocumentHandler = createDocumentHandler<"code">({
   kind: "code",
-  onCreateDocument: async ({ title, dataStream }) => {
+  onCreateDocument: async ({ title, dataStream, modelId }) => {
     let draftContent = "";
 
-    const { fullStream } = streamObject({
-      model: getArtifactModel(),
-      system: codePrompt,
+    const { fullStream } = streamText({
+      model: getLanguageModel(modelId),
+      system: `${codePrompt}\n\nOutput ONLY the code. No explanations, no markdown fences, no wrapping.`,
       prompt: title,
-      schema: z.object({
-        code: z.string(),
-      }),
     });
 
     for await (const delta of fullStream) {
-      const { type } = delta;
-
-      if (type === "object") {
-        const { object } = delta;
-        const { code } = object;
-
-        if (code) {
-          dataStream.write({
-            type: "data-codeDelta",
-            data: code ?? "",
-            transient: true,
-          });
-
-          draftContent = code;
-        }
+      if (delta.type === "text-delta") {
+        draftContent += delta.text;
+        dataStream.write({
+          type: "data-codeDelta",
+          data: stripFences(draftContent),
+          transient: true,
+        });
       }
     }
 
-    return draftContent;
+    return stripFences(draftContent);
   },
-  onUpdateDocument: async ({ document, description, dataStream }) => {
+  onUpdateDocument: async ({ document, description, dataStream, modelId }) => {
     let draftContent = "";
 
-    const { fullStream } = streamObject({
-      model: getArtifactModel(),
-      system: updateDocumentPrompt(document.content, "code"),
+    const { fullStream } = streamText({
+      model: getLanguageModel(modelId),
+      system: `${updateDocumentPrompt(document.content, "code")}\n\nOutput ONLY the complete updated code. No explanations, no markdown fences, no wrapping.`,
       prompt: description,
-      schema: z.object({
-        code: z.string(),
-      }),
     });
 
     for await (const delta of fullStream) {
-      const { type } = delta;
-
-      if (type === "object") {
-        const { object } = delta;
-        const { code } = object;
-
-        if (code) {
-          dataStream.write({
-            type: "data-codeDelta",
-            data: code ?? "",
-            transient: true,
-          });
-
-          draftContent = code;
-        }
+      if (delta.type === "text-delta") {
+        draftContent += delta.text;
+        dataStream.write({
+          type: "data-codeDelta",
+          data: stripFences(draftContent),
+          transient: true,
+        });
       }
     }
 
-    return draftContent;
+    return stripFences(draftContent);
   },
 });