import { Configuration, OpenAIApi } from "npm:openai@3.1.0";

/**

 * @param prompt The maximum length is 1000 characters.

 */

type Openai = {

  api_key: string;

  organization_id: string;

};

export async function main(

  auth: Openai,

  prompt: string,

  number_of_images:

    | "1"

    | "2"

    | "3"

    | "4"

    | "5"

    | "6"

    | "7"

    | "8"

    | "9"

    | "10" = "1",

  size: "256x256" | "512x512" | "1024x1024" = "1024x1024",

  response_format: "url" | "b64_json" = "url"

) {

  const configuration = new Configuration({

    apiKey: auth.api_key,

    organization: auth.organization_id,

  });

  const openai = new OpenAIApi(configuration);

  const response = await openai.createImage({

    prompt,

    n: +number_of_images,

    size,

    response_format,

  });

  return response.data.data;

}