contentEditorFlow.ts

import type { Flow } from '@genkit-ai/core';
import { ai } from '../config.js';
import { UserFacingError } from '../errors/UserFacingError.js';
import { ContentEditorInputSchema, ContentEditorOutputSchema } from '../schemas/index.js';

export const contentEditorFlow: Flow<typeof ContentEditorInputSchema, typeof ContentEditorOutputSchema> = ai.defineFlow(
  {
    name: 'contentEditorFlow',
    inputSchema: ContentEditorInputSchema,
    outputSchema: ContentEditorOutputSchema,
  },
  async (input, { context }) => {
    const prompt = ai.prompt('content_editor');
    const resp = context ? await prompt(input, { context }) : await prompt(input);
    const output = resp?.output;

    if (typeof output === 'string') {
      return { edited: output };
    }

    if (output === null || output === undefined) {
      throw new UserFacingError('Invalid model output for contentEditorFlow', {
        details: { reason: 'No output returned from prompt', input },
      });
    }

    const parsed = ContentEditorOutputSchema.safeParse(output);
    if (!parsed.success) {
      throw new UserFacingError('Schema validation failed for contentEditorFlow output', {
        details: parsed.error.flatten(),
      });
    }
    return parsed.data;
  }
);