Execute AI model

type Cloudflare = {

  token: string;

  email: string;

  key: string;

};

/**

 * Execute AI model

 * This endpoint provides users with the capability to run specific AI models on-demand. 

By submitting the required input data, users can receive real-time predictions or results generated by the chosen AI 

model. The endpoint supports various AI model types, ensuring flexibility and adaptability for diverse use cases.

 */

export async function main(

  auth: Cloudflare,

  account_identifier: string,

  model_name: string,

  body: { [k: string]: unknown }

) {

  const url = new URL(

    `https://api.cloudflare.com/client/v4/accounts/${account_identifier}/ai/run/${model_name}`

  );

  const response = await fetch(url, {

    method: "POST",

    headers: {

      "X-AUTH-EMAIL": auth.email,

      "X-AUTH-KEY": auth.key,

      "Content-Type": "application/json",

      Authorization: "Bearer " + auth.token,

    },

    body: JSON.stringify(body),

  });

  if (!response.ok) {

    const text = await response.text();

    throw new Error(`${response.status} ${text}`);

  }

  return await response.json();

}