//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();
}
Submitted by hugo697 32 days ago