'use server';

import { openai } from '@ai-sdk/openai';
import { streamObject } from 'ai';
import { createStreamableValue } from '@ai-sdk/rsc';
import { PartialNotification, notificationSchema } from './schema';

export async function generateNotifications(context: string) {
  const notificationsStream = createStreamableValue<PartialNotification>();

  const result = streamObject({
    model: openai('gpt-4-turbo'),
    prompt: `Generate 3 notifications for a messages app in this context: ${context}`,
    schema: notificationSchema,
  });

  try {
    for await (const partialObject of result.partialObjectStream) {
      notificationsStream.update(partialObject);
    }
  } finally {
    notificationsStream.done();
  }

  return notificationsStream.value;
}