Add message actions (#482)

This commit is contained in:
Jeremy 2024-11-05 17:15:51 +03:00 committed by GitHub
parent 94f563f179
commit 171914941e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
20 changed files with 1011 additions and 150 deletions

View file

@ -1,8 +1,29 @@
'use server';
import { CoreMessage, CoreUserMessage, generateText } from 'ai';
import { cookies } from 'next/headers';
import { customModel } from '@/ai';
export async function saveModelId(model: string) {
const cookieStore = await cookies();
cookieStore.set('model-id', model);
}
export async function generateTitleFromUserMessage({
message,
}: {
message: CoreUserMessage;
}) {
const { text: title } = await generateText({
model: customModel('gpt-3.5-turbo'),
system: `\n
- you will generate a short title based on the first message a user begins a conversation with
- ensure it is not more than 80 characters long
- the title should be a summary of the user's message
- do not use quotes or colons`,
prompt: JSON.stringify(message),
});
return title;
}