Create a rerank request

//native

type Togetherai = {

  api_key: string;

};

/**

 * Create a rerank request

 * Query a reranker model

 */

export async function main(

  auth: Togetherai,

  body: {

    model: string;

    query: string;

    documents: {}[] | string[];

    top_n?: number;

    return_documents?: false | true;

    rank_fields?: string[];

  },

) {

  const url = new URL(`https://api.together.xyz/v1/rerank`);

  const response = await fetch(url, {

    method: "POST",

    headers: {

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

      Authorization: "Bearer " + auth.api_key,

    },

    body: JSON.stringify(body),

  });

  if (!response.ok) {

    const text = await response.text();

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

  }

  return await response.json();

}