//native
type Ai21 = {
apiKey: string;
};
/**
* Search
*
*/
export async function main(
auth: Ai21,
body: {
query: string;
maxSegments?: number;
path?: string;
labels?: string[];
labels_filter?: {} | {} | {};
labels_filter_mode?: "AND" | "OR";
fileIds?: string[];
retrievalStrategy?: "segments" | "default" | "add_neighbors" | "full_doc";
maxNeighbors?: number;
retrievalSimilarityThreshold?: number;
hybridSearchAlpha?: number;
},
) {
const url = new URL(`https://api.ai21.com/studio/v1/library/search`);
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