34 lines
785 B
TypeScript
34 lines
785 B
TypeScript
|
"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();
|
||
|
|
}
|