//native

type Ai21 = {

  apiKey: string;

};

/**

 * Conversational Rag

 *

 */

export async function main(

  auth: Ai21,

  body: {

    messages: { role: "assistant" | "user"; content: string }[];

    path?: string;

    labels?: string[];

    file_ids?: string[];

    max_segments?: number;

    retrieval_strategy?: "default" | "segments" | "add_neighbors" | "full_doc";

    retrieval_similarity_threshold?: number;

    max_neighbors?: number;

    hybrid_search_alpha?: number;

    response_language?:

      | "dutch"

      | "english"

      | "french"

      | "german"

      | "hebrew"

      | "italian"

      | "portuguese"

      | "spanish";

  },

) {

  const url = new URL(`https://api.ai21.com/studio/v1/conversational-rag`);

  const response = await fetch(url, {

    method: "POST",

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

}