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

components/chat/message-editor.tsx

@@ -0,0 +1,33 @@
+"use client";
+
+import type { UseChatHelpers } from "@ai-sdk/react";
+import { deleteTrailingMessages } from "@/app/(chat)/actions";
+import type { ChatMessage } from "@/lib/types";
+
+export async function submitEditedMessage({
+  message,
+  text,
+  setMessages,
+  regenerate,
+}: {
+  message: ChatMessage;
+  text: string;
+  setMessages: UseChatHelpers<ChatMessage>["setMessages"];
+  regenerate: UseChatHelpers<ChatMessage>["regenerate"];
+}) {
+  await deleteTrailingMessages({ id: message.id });
+
+  setMessages((messages) => {
+    const index = messages.findIndex((m) => m.id === message.id);
+    if (index === -1) {
+      return messages;
+    }
+
+    return [
+      ...messages.slice(0, index),
+      { ...message, parts: [{ type: "text" as const, text }] },
+    ];
+  });
+
+  regenerate();
+}