//native

type Ai21 = {

  apiKey: string;

};

/**

 * Modify Assistant

 *

 */

export async function main(

  auth: Ai21,

  assistant_id: string,

  body: {

    name?: string;

    description?: string;

    optimization?: "cost" | "latency";

    avatar?: string;

    is_archived?: false | true;

    models?: string[];

    tools?: "file_search" | "web_search" | "plan_approval"[];

    tool_resources?: { file_search?: {}; web_search?: {}; plan_approval?: {} };

  },

) {

  const url = new URL(

    `https://api.ai21.com/studio/v1/assistants/${assistant_id}`,

  );

  const response = await fetch(url, {

    method: "PATCH",

    headers: {

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

      Authorization: "Bearer " + auth.apiKey,

    },

    body: JSON.stringify(body),

  });

  if (!response.ok) {

    const text = await response.text();

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

  }

  return await response.json();

}