More information about HFInference #87

class { constructor(apiKey = "", defaultOptions = {}) { __publicField(this, "apiKey"); __publicField(this, "defaultOptions"); this.apiKey = apiKey; this.defaultOptions = defaultOptions; } async fillMask(args, options) { return this.request(args, options); } async summarization(args, options) { var _a; return (_a = await this.request(args, options)) == null ? void 0 : _a[0]; } async questionAnswer(args, options) { return await this.request(args, options); } async tableQuestionAnswer(args, options) { return await this.request(args, options); } async textClassification(args, options) { var _a; return (_a = await this.request(args, options)) == null ? void 0 : _a[0]; } async textGeneration(args, options) { var _a; return (_a = await this.request(args, options)) == null ? void 0 : _a[0]; } async tokenClassification(args, options) { return HfInference.toArray(await this.request(args, options)); } async translation(args, options) { var _a; return (_a = await this.request(args, options)) == null ? void 0 : _a[0]; } async zeroShotClassification(args, options) { return HfInference.toArray(await this.request(args, options)); } async conversational(args, options) { return await this.request(args, options); } async featureExtraction(args, options) { return await this.request(args, options); } async automaticSpeechRecognition(args, options) { return await this.request(args, { ...options, binary: true }); } async audioClassification(args, options) { return await this.request(args, { ...options, binary: true }); } async imageClassification(args, options) { return await this.request(args, { ...options, binary: true }); } async objectDetection(args, options) { return await this.request(args, { ...options, binary: true }); } async imageSegmentation(args, options) { return await this.request(args, { ...options, binary: true }); } async textToImage(args, options) { return await this.request(args, { ...options, blob: true }); } async request(args, options) { const mergedOptions = {...this.defaultOptions, ...options}; const {model, ...otherArgs} = args; const headers = {}; if (this.apiKey) { headers["Authorization"] = `Bearer ${this.apiKey}`; } if (!(options == null ? void 0 : options.binary)) { headers["Content-Type"] = "application/json"; } if ((options == null ? void 0 : options.binary) && mergedOptions.wait_for_model) { headers["X-Wait-For-Model"] = "true"; } const response = await fetch(`https://api-inference.huggingface.co/models/${model}`, { headers, method: "POST", body: (options == null ? void 0 : options.binary) ? args.data : JSON.stringify({ ...otherArgs, options: mergedOptions }) }); if (mergedOptions.retry_on_error !== false && response.status === 503 && !mergedOptions.wait_for_model) { return this.request(args, { ...mergedOptions, wait_for_model: true }); } let output; if (options == null ? void 0 : options.blob) { if (!response.ok) { throw new Error("An error occurred while fetching the blob"); } return await response.arrayBuffer(); } else { output = await response.json(); if (output.error) { throw new Error(output.error); } } return output; } static toArray(obj) { if (Array.isArray(obj)) { return obj; } return [obj]; } }