app/api/text-generation/route.ts

import { streamText } from 'ai';
import getProviderForModel from '@/lib/ai/provider-manager';

export const runtime = 'edge';

export async function POST(req: Request) {
  const { prompt } = await req.json();

  const { client, actualModel } = getProviderForModel('text');

  const result = await streamText({
    model: client(actualModel),
    messages: [
      {
        role: 'user',
        content: prompt,
      },
    ],
  });

  return result.toTextStreamResponse();
}