Create a training

//native

type Replicate = {

  token: string;

};

/**

 * Create a training

 * Start a new training of the model version you specify.

 */

export async function main(

  auth: Replicate,

  model_owner: string,

  model_name: string,

  version_id: string,

  body: {

    destination: string;

    input: {};

    webhook?: string;

    webhook_events_filter?: "start" | "output" | "logs" | "completed"[];

  },

) {

  const url = new URL(

    `https://api.replicate.com/v1/models/${model_owner}/${model_name}/versions/${version_id}/trainings`,

  );

  const response = await fetch(url, {

    method: "POST",

    headers: {

      "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();

}