import { cohere } from '@ai-sdk/cohere';
import { convertToModelMessages, streamText, UIMessage } from 'ai';

export const maxDuration = 30;

export async function POST(req: Request) {
  const { messages }: { messages: UIMessage[] } = await req.json();

  const prompt = convertToModelMessages(messages);

  const result = streamText({
    model: cohere('command-r-plus'),
    prompt,
  });

  return result.toUIMessageStreamResponse();
}