@huggingface/inference 3.5.2 → 3.6.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/browser/index.cjs +1652 -0
- package/dist/browser/index.js +1652 -0
- package/dist/index.cjs +277 -971
- package/dist/index.js +268 -982
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/lib/makeRequestOptions.d.ts +16 -1
- package/dist/src/lib/makeRequestOptions.d.ts.map +1 -1
- package/dist/src/providers/novita.d.ts.map +1 -1
- package/dist/src/snippets/getInferenceSnippets.d.ts +4 -0
- package/dist/src/snippets/getInferenceSnippets.d.ts.map +1 -0
- package/dist/src/snippets/index.d.ts +1 -4
- package/dist/src/snippets/index.d.ts.map +1 -1
- package/dist/src/tasks/cv/textToVideo.d.ts.map +1 -1
- package/package.json +15 -6
- package/src/index.ts +1 -1
- package/src/lib/makeRequestOptions.ts +37 -10
- package/src/providers/fireworks-ai.ts +1 -1
- package/src/providers/hf-inference.ts +1 -1
- package/src/providers/nebius.ts +3 -3
- package/src/providers/novita.ts +7 -6
- package/src/providers/sambanova.ts +1 -1
- package/src/providers/together.ts +3 -3
- package/src/snippets/getInferenceSnippets.ts +398 -0
- package/src/snippets/index.ts +1 -5
- package/src/snippets/templates/js/fetch/basic.jinja +19 -0
- package/src/snippets/templates/js/fetch/basicAudio.jinja +19 -0
- package/src/snippets/templates/js/fetch/basicImage.jinja +19 -0
- package/src/snippets/templates/js/fetch/textToAudio.jinja +41 -0
- package/src/snippets/templates/js/fetch/textToImage.jinja +19 -0
- package/src/snippets/templates/js/fetch/zeroShotClassification.jinja +22 -0
- package/src/snippets/templates/js/huggingface.js/basic.jinja +11 -0
- package/src/snippets/templates/js/huggingface.js/basicAudio.jinja +13 -0
- package/src/snippets/templates/js/huggingface.js/basicImage.jinja +13 -0
- package/src/snippets/templates/js/huggingface.js/conversational.jinja +11 -0
- package/src/snippets/templates/js/huggingface.js/conversationalStream.jinja +19 -0
- package/src/snippets/templates/js/huggingface.js/textToImage.jinja +11 -0
- package/src/snippets/templates/js/huggingface.js/textToVideo.jinja +10 -0
- package/src/snippets/templates/js/openai/conversational.jinja +13 -0
- package/src/snippets/templates/js/openai/conversationalStream.jinja +22 -0
- package/src/snippets/templates/python/fal_client/textToImage.jinja +11 -0
- package/src/snippets/templates/python/huggingface_hub/basic.jinja +4 -0
- package/src/snippets/templates/python/huggingface_hub/basicAudio.jinja +1 -0
- package/src/snippets/templates/python/huggingface_hub/basicImage.jinja +1 -0
- package/src/snippets/templates/python/huggingface_hub/conversational.jinja +6 -0
- package/src/snippets/templates/python/huggingface_hub/conversationalStream.jinja +8 -0
- package/src/snippets/templates/python/huggingface_hub/documentQuestionAnswering.jinja +5 -0
- package/src/snippets/templates/python/huggingface_hub/imageToImage.jinja +6 -0
- package/src/snippets/templates/python/huggingface_hub/importInferenceClient.jinja +6 -0
- package/src/snippets/templates/python/huggingface_hub/textToImage.jinja +5 -0
- package/src/snippets/templates/python/huggingface_hub/textToVideo.jinja +4 -0
- package/src/snippets/templates/python/openai/conversational.jinja +13 -0
- package/src/snippets/templates/python/openai/conversationalStream.jinja +15 -0
- package/src/snippets/templates/python/requests/basic.jinja +7 -0
- package/src/snippets/templates/python/requests/basicAudio.jinja +7 -0
- package/src/snippets/templates/python/requests/basicImage.jinja +7 -0
- package/src/snippets/templates/python/requests/conversational.jinja +9 -0
- package/src/snippets/templates/python/requests/conversationalStream.jinja +16 -0
- package/src/snippets/templates/python/requests/documentQuestionAnswering.jinja +13 -0
- package/src/snippets/templates/python/requests/imageToImage.jinja +15 -0
- package/src/snippets/templates/python/requests/importRequests.jinja +10 -0
- package/src/snippets/templates/python/requests/tabular.jinja +9 -0
- package/src/snippets/templates/python/requests/textToAudio.jinja +23 -0
- package/src/snippets/templates/python/requests/textToImage.jinja +14 -0
- package/src/snippets/templates/python/requests/zeroShotClassification.jinja +8 -0
- package/src/snippets/templates/python/requests/zeroShotImageClassification.jinja +14 -0
- package/src/snippets/templates/sh/curl/basic.jinja +7 -0
- package/src/snippets/templates/sh/curl/basicAudio.jinja +5 -0
- package/src/snippets/templates/sh/curl/basicImage.jinja +5 -0
- package/src/snippets/templates/sh/curl/conversational.jinja +7 -0
- package/src/snippets/templates/sh/curl/conversationalStream.jinja +7 -0
- package/src/snippets/templates/sh/curl/zeroShotClassification.jinja +5 -0
- package/src/tasks/cv/textToVideo.ts +25 -5
- package/src/vendor/fetch-event-source/LICENSE +21 -0
- package/dist/src/snippets/curl.d.ts +0 -17
- package/dist/src/snippets/curl.d.ts.map +0 -1
- package/dist/src/snippets/js.d.ts +0 -21
- package/dist/src/snippets/js.d.ts.map +0 -1
- package/dist/src/snippets/python.d.ts +0 -4
- package/dist/src/snippets/python.d.ts.map +0 -1
- package/src/snippets/curl.ts +0 -177
- package/src/snippets/js.ts +0 -475
- package/src/snippets/python.ts +0 -563
package/dist/index.js
CHANGED
|
@@ -141,7 +141,7 @@ var makeHeaders5 = (params) => {
|
|
|
141
141
|
return { Authorization: `Bearer ${params.accessToken}` };
|
|
142
142
|
};
|
|
143
143
|
var makeUrl5 = (params) => {
|
|
144
|
-
if (params.
|
|
144
|
+
if (params.chatCompletion) {
|
|
145
145
|
return `${params.baseUrl}/inference/v1/chat/completions`;
|
|
146
146
|
}
|
|
147
147
|
return `${params.baseUrl}/inference`;
|
|
@@ -167,7 +167,7 @@ var makeUrl6 = (params) => {
|
|
|
167
167
|
if (params.task && ["feature-extraction", "sentence-similarity"].includes(params.task)) {
|
|
168
168
|
return `${params.baseUrl}/pipeline/${params.task}/${params.model}`;
|
|
169
169
|
}
|
|
170
|
-
if (params.
|
|
170
|
+
if (params.chatCompletion) {
|
|
171
171
|
return `${params.baseUrl}/models/${params.model}/v1/chat/completions`;
|
|
172
172
|
}
|
|
173
173
|
return `${params.baseUrl}/models/${params.model}`;
|
|
@@ -218,10 +218,10 @@ var makeUrl8 = (params) => {
|
|
|
218
218
|
if (params.task === "text-to-image") {
|
|
219
219
|
return `${params.baseUrl}/v1/images/generations`;
|
|
220
220
|
}
|
|
221
|
+
if (params.chatCompletion) {
|
|
222
|
+
return `${params.baseUrl}/v1/chat/completions`;
|
|
223
|
+
}
|
|
221
224
|
if (params.task === "text-generation") {
|
|
222
|
-
if (params.chatCompletion) {
|
|
223
|
-
return `${params.baseUrl}/v1/chat/completions`;
|
|
224
|
-
}
|
|
225
225
|
return `${params.baseUrl}/v1/completions`;
|
|
226
226
|
}
|
|
227
227
|
return params.baseUrl;
|
|
@@ -234,7 +234,7 @@ var NEBIUS_CONFIG = {
|
|
|
234
234
|
};
|
|
235
235
|
|
|
236
236
|
// src/providers/novita.ts
|
|
237
|
-
var NOVITA_API_BASE_URL = "https://api.novita.ai
|
|
237
|
+
var NOVITA_API_BASE_URL = "https://api.novita.ai";
|
|
238
238
|
var makeBody9 = (params) => {
|
|
239
239
|
return {
|
|
240
240
|
...params.args,
|
|
@@ -245,11 +245,12 @@ var makeHeaders9 = (params) => {
|
|
|
245
245
|
return { Authorization: `Bearer ${params.accessToken}` };
|
|
246
246
|
};
|
|
247
247
|
var makeUrl9 = (params) => {
|
|
248
|
-
if (params.
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
}
|
|
252
|
-
|
|
248
|
+
if (params.chatCompletion) {
|
|
249
|
+
return `${params.baseUrl}/v3/openai/chat/completions`;
|
|
250
|
+
} else if (params.task === "text-generation") {
|
|
251
|
+
return `${params.baseUrl}/v3/openai/completions`;
|
|
252
|
+
} else if (params.task === "text-to-video") {
|
|
253
|
+
return `${params.baseUrl}/v3/hf/${params.model}`;
|
|
253
254
|
}
|
|
254
255
|
return params.baseUrl;
|
|
255
256
|
};
|
|
@@ -296,7 +297,7 @@ var makeHeaders11 = (params) => {
|
|
|
296
297
|
return { Authorization: `Bearer ${params.accessToken}` };
|
|
297
298
|
};
|
|
298
299
|
var makeUrl11 = (params) => {
|
|
299
|
-
if (params.
|
|
300
|
+
if (params.chatCompletion) {
|
|
300
301
|
return `${params.baseUrl}/v1/chat/completions`;
|
|
301
302
|
}
|
|
302
303
|
return params.baseUrl;
|
|
@@ -323,10 +324,10 @@ var makeUrl12 = (params) => {
|
|
|
323
324
|
if (params.task === "text-to-image") {
|
|
324
325
|
return `${params.baseUrl}/v1/images/generations`;
|
|
325
326
|
}
|
|
327
|
+
if (params.chatCompletion) {
|
|
328
|
+
return `${params.baseUrl}/v1/chat/completions`;
|
|
329
|
+
}
|
|
326
330
|
if (params.task === "text-generation") {
|
|
327
|
-
if (params.chatCompletion) {
|
|
328
|
-
return `${params.baseUrl}/v1/chat/completions`;
|
|
329
|
-
}
|
|
330
331
|
return `${params.baseUrl}/v1/completions`;
|
|
331
332
|
}
|
|
332
333
|
return params.baseUrl;
|
|
@@ -373,7 +374,7 @@ function isUrl(modelOrUrl) {
|
|
|
373
374
|
|
|
374
375
|
// package.json
|
|
375
376
|
var name = "@huggingface/inference";
|
|
376
|
-
var version = "3.
|
|
377
|
+
var version = "3.6.0";
|
|
377
378
|
|
|
378
379
|
// src/providers/consts.ts
|
|
379
380
|
var HARDCODED_MODEL_ID_MAPPING = {
|
|
@@ -461,11 +462,11 @@ var providerConfigs = {
|
|
|
461
462
|
together: TOGETHER_CONFIG
|
|
462
463
|
};
|
|
463
464
|
async function makeRequestOptions(args, options) {
|
|
464
|
-
const {
|
|
465
|
+
const { provider: maybeProvider, model: maybeModel } = args;
|
|
465
466
|
const provider = maybeProvider ?? "hf-inference";
|
|
466
467
|
const providerConfig = providerConfigs[provider];
|
|
467
|
-
const {
|
|
468
|
-
if (endpointUrl && provider !== "hf-inference") {
|
|
468
|
+
const { task, chatCompletion: chatCompletion2 } = options ?? {};
|
|
469
|
+
if (args.endpointUrl && provider !== "hf-inference") {
|
|
469
470
|
throw new Error(`Cannot use endpointUrl with a third-party provider.`);
|
|
470
471
|
}
|
|
471
472
|
if (maybeModel && isUrl(maybeModel)) {
|
|
@@ -481,17 +482,21 @@ async function makeRequestOptions(args, options) {
|
|
|
481
482
|
throw new Error(`Provider ${provider} requires a model ID to be passed directly.`);
|
|
482
483
|
}
|
|
483
484
|
const hfModel = maybeModel ?? await loadDefaultModel(task);
|
|
484
|
-
const
|
|
485
|
+
const resolvedModel = providerConfig.clientSideRoutingOnly ? (
|
|
485
486
|
// eslint-disable-next-line @typescript-eslint/no-non-null-assertion
|
|
486
487
|
removeProviderPrefix(maybeModel, provider)
|
|
487
|
-
) : (
|
|
488
|
-
|
|
489
|
-
|
|
490
|
-
|
|
491
|
-
|
|
492
|
-
|
|
493
|
-
|
|
494
|
-
|
|
488
|
+
) : await getProviderModelId({ model: hfModel, provider }, args, {
|
|
489
|
+
task,
|
|
490
|
+
chatCompletion: chatCompletion2,
|
|
491
|
+
fetch: options?.fetch
|
|
492
|
+
});
|
|
493
|
+
return makeRequestOptionsFromResolvedModel(resolvedModel, args, options);
|
|
494
|
+
}
|
|
495
|
+
function makeRequestOptionsFromResolvedModel(resolvedModel, args, options) {
|
|
496
|
+
const { accessToken, endpointUrl, provider: maybeProvider, model, ...remainingArgs } = args;
|
|
497
|
+
const provider = maybeProvider ?? "hf-inference";
|
|
498
|
+
const providerConfig = providerConfigs[provider];
|
|
499
|
+
const { includeCredentials, task, chatCompletion: chatCompletion2, signal } = options ?? {};
|
|
495
500
|
const authMethod = (() => {
|
|
496
501
|
if (providerConfig.clientSideRoutingOnly) {
|
|
497
502
|
if (accessToken && accessToken.startsWith("hf_")) {
|
|
@@ -509,7 +514,7 @@ async function makeRequestOptions(args, options) {
|
|
|
509
514
|
})();
|
|
510
515
|
const url = endpointUrl ? chatCompletion2 ? endpointUrl + `/v1/chat/completions` : endpointUrl : providerConfig.makeUrl({
|
|
511
516
|
baseUrl: authMethod !== "provider-key" ? HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider) : providerConfig.baseUrl,
|
|
512
|
-
model,
|
|
517
|
+
model: resolvedModel,
|
|
513
518
|
chatCompletion: chatCompletion2,
|
|
514
519
|
task
|
|
515
520
|
});
|
|
@@ -527,7 +532,7 @@ async function makeRequestOptions(args, options) {
|
|
|
527
532
|
const body = binary ? args.data : JSON.stringify(
|
|
528
533
|
providerConfig.makeBody({
|
|
529
534
|
args: remainingArgs,
|
|
530
|
-
model,
|
|
535
|
+
model: resolvedModel,
|
|
531
536
|
task,
|
|
532
537
|
chatCompletion: chatCompletion2
|
|
533
538
|
})
|
|
@@ -1161,14 +1166,14 @@ async function zeroShotImageClassification(args, options) {
|
|
|
1161
1166
|
}
|
|
1162
1167
|
|
|
1163
1168
|
// src/tasks/cv/textToVideo.ts
|
|
1164
|
-
var SUPPORTED_PROVIDERS = ["fal-ai", "replicate"];
|
|
1169
|
+
var SUPPORTED_PROVIDERS = ["fal-ai", "novita", "replicate"];
|
|
1165
1170
|
async function textToVideo(args, options) {
|
|
1166
1171
|
if (!args.provider || !typedInclude(SUPPORTED_PROVIDERS, args.provider)) {
|
|
1167
1172
|
throw new Error(
|
|
1168
1173
|
`textToVideo inference is only supported for the following providers: ${SUPPORTED_PROVIDERS.join(", ")}`
|
|
1169
1174
|
);
|
|
1170
1175
|
}
|
|
1171
|
-
const payload = args.provider === "fal-ai" || args.provider === "replicate" ? { ...omit(args, ["inputs", "parameters"]), ...args.parameters, prompt: args.inputs } : args;
|
|
1176
|
+
const payload = args.provider === "fal-ai" || args.provider === "replicate" || args.provider === "novita" ? { ...omit(args, ["inputs", "parameters"]), ...args.parameters, prompt: args.inputs } : args;
|
|
1172
1177
|
const res = await request(payload, {
|
|
1173
1178
|
...options,
|
|
1174
1179
|
task: "text-to-video"
|
|
@@ -1180,6 +1185,13 @@ async function textToVideo(args, options) {
|
|
|
1180
1185
|
}
|
|
1181
1186
|
const urlResponse = await fetch(res.video.url);
|
|
1182
1187
|
return await urlResponse.blob();
|
|
1188
|
+
} else if (args.provider === "novita") {
|
|
1189
|
+
const isValidOutput = typeof res === "object" && !!res && "video" in res && typeof res.video === "object" && !!res.video && "video_url" in res.video && typeof res.video.video_url === "string" && isUrl(res.video.video_url);
|
|
1190
|
+
if (!isValidOutput) {
|
|
1191
|
+
throw new InferenceOutputError("Expected { video: { video_url: string } }");
|
|
1192
|
+
}
|
|
1193
|
+
const urlResponse = await fetch(res.video.video_url);
|
|
1194
|
+
return await urlResponse.blob();
|
|
1183
1195
|
} else {
|
|
1184
1196
|
const isValidOutput = typeof res === "object" && !!res && "output" in res && typeof res.output === "string" && isUrl(res.output);
|
|
1185
1197
|
if (!isValidOutput) {
|
|
@@ -1588,997 +1600,271 @@ var INFERENCE_PROVIDERS = [
|
|
|
1588
1600
|
// src/snippets/index.ts
|
|
1589
1601
|
var snippets_exports = {};
|
|
1590
1602
|
__export(snippets_exports, {
|
|
1591
|
-
|
|
1592
|
-
js: () => js_exports,
|
|
1593
|
-
python: () => python_exports
|
|
1603
|
+
getInferenceSnippets: () => getInferenceSnippets
|
|
1594
1604
|
});
|
|
1595
1605
|
|
|
1596
|
-
// src/snippets/
|
|
1597
|
-
var curl_exports = {};
|
|
1598
|
-
__export(curl_exports, {
|
|
1599
|
-
curlSnippets: () => curlSnippets,
|
|
1600
|
-
getCurlInferenceSnippet: () => getCurlInferenceSnippet,
|
|
1601
|
-
snippetBasic: () => snippetBasic,
|
|
1602
|
-
snippetFile: () => snippetFile,
|
|
1603
|
-
snippetTextGeneration: () => snippetTextGeneration,
|
|
1604
|
-
snippetZeroShotClassification: () => snippetZeroShotClassification
|
|
1605
|
-
});
|
|
1606
|
-
import { HF_HUB_INFERENCE_PROXY_TEMPLATE as HF_HUB_INFERENCE_PROXY_TEMPLATE2 } from "@huggingface/tasks";
|
|
1606
|
+
// src/snippets/getInferenceSnippets.ts
|
|
1607
1607
|
import {
|
|
1608
|
-
|
|
1609
|
-
|
|
1610
|
-
stringifyMessages
|
|
1608
|
+
inferenceSnippetLanguages,
|
|
1609
|
+
getModelInputSnippet
|
|
1611
1610
|
} from "@huggingface/tasks";
|
|
1612
|
-
|
|
1613
|
-
|
|
1614
|
-
|
|
1615
|
-
|
|
1616
|
-
|
|
1617
|
-
|
|
1618
|
-
|
|
1619
|
-
|
|
1620
|
-
|
|
1621
|
-
|
|
1622
|
-
|
|
1623
|
-
|
|
1611
|
+
import { Template } from "@huggingface/jinja";
|
|
1612
|
+
import fs from "fs";
|
|
1613
|
+
import path from "path";
|
|
1614
|
+
import { existsSync as pathExists } from "node:fs";
|
|
1615
|
+
var PYTHON_CLIENTS = ["huggingface_hub", "fal_client", "requests", "openai"];
|
|
1616
|
+
var JS_CLIENTS = ["fetch", "huggingface.js", "openai"];
|
|
1617
|
+
var SH_CLIENTS = ["curl"];
|
|
1618
|
+
var CLIENTS = {
|
|
1619
|
+
js: [...JS_CLIENTS],
|
|
1620
|
+
python: [...PYTHON_CLIENTS],
|
|
1621
|
+
sh: [...SH_CLIENTS]
|
|
1622
|
+
};
|
|
1623
|
+
var rootDirFinder = () => {
|
|
1624
|
+
let currentPath = typeof import.meta !== "undefined" && import.meta.url ? path.normalize(new URL(import.meta.url).pathname) : __dirname;
|
|
1625
|
+
while (currentPath !== "/") {
|
|
1626
|
+
if (pathExists(path.join(currentPath, "package.json"))) {
|
|
1627
|
+
return currentPath;
|
|
1624
1628
|
}
|
|
1625
|
-
|
|
1626
|
-
};
|
|
1627
|
-
var snippetTextGeneration = (model, accessToken, provider, providerModelId, opts) => {
|
|
1628
|
-
if (model.tags.includes("conversational")) {
|
|
1629
|
-
const baseUrl = provider === "hf-inference" ? `https://router.huggingface.co/hf-inference/models/${model.id}/v1/chat/completions` : HF_HUB_INFERENCE_PROXY_TEMPLATE2.replace("{{PROVIDER}}", provider) + "/v1/chat/completions";
|
|
1630
|
-
const modelId = providerModelId ?? model.id;
|
|
1631
|
-
const streaming = opts?.streaming ?? true;
|
|
1632
|
-
const exampleMessages = getModelInputSnippet(model);
|
|
1633
|
-
const messages = opts?.messages ?? exampleMessages;
|
|
1634
|
-
const config = {
|
|
1635
|
-
...opts?.temperature ? { temperature: opts.temperature } : void 0,
|
|
1636
|
-
max_tokens: opts?.max_tokens ?? 500,
|
|
1637
|
-
...opts?.top_p ? { top_p: opts.top_p } : void 0
|
|
1638
|
-
};
|
|
1639
|
-
return [
|
|
1640
|
-
{
|
|
1641
|
-
client: "curl",
|
|
1642
|
-
content: `curl '${baseUrl}' \\
|
|
1643
|
-
-H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}' \\
|
|
1644
|
-
-H 'Content-Type: application/json' \\
|
|
1645
|
-
--data '{
|
|
1646
|
-
"model": "${modelId}",
|
|
1647
|
-
"messages": ${stringifyMessages(messages, {
|
|
1648
|
-
indent: " ",
|
|
1649
|
-
attributeKeyQuotes: true,
|
|
1650
|
-
customContentEscaper: (str) => str.replace(/'/g, "'\\''")
|
|
1651
|
-
})},
|
|
1652
|
-
${stringifyGenerationConfig(config, {
|
|
1653
|
-
indent: "\n ",
|
|
1654
|
-
attributeKeyQuotes: true,
|
|
1655
|
-
attributeValueConnector: ": "
|
|
1656
|
-
})}
|
|
1657
|
-
"stream": ${!!streaming}
|
|
1658
|
-
}'`
|
|
1659
|
-
}
|
|
1660
|
-
];
|
|
1661
|
-
} else {
|
|
1662
|
-
return snippetBasic(model, accessToken, provider);
|
|
1629
|
+
currentPath = path.normalize(path.join(currentPath, ".."));
|
|
1663
1630
|
}
|
|
1631
|
+
return "/";
|
|
1664
1632
|
};
|
|
1665
|
-
var
|
|
1666
|
-
|
|
1667
|
-
|
|
1668
|
-
|
|
1669
|
-
return
|
|
1670
|
-
|
|
1671
|
-
|
|
1672
|
-
|
|
1673
|
-
|
|
1674
|
-
-d '{"inputs": ${getModelInputSnippet(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
|
|
1675
|
-
-H 'Content-Type: application/json' \\
|
|
1676
|
-
-H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`
|
|
1677
|
-
}
|
|
1678
|
-
];
|
|
1679
|
-
};
|
|
1680
|
-
var snippetFile = (model, accessToken, provider) => {
|
|
1681
|
-
if (provider !== "hf-inference") {
|
|
1682
|
-
return [];
|
|
1683
|
-
}
|
|
1684
|
-
return [
|
|
1685
|
-
{
|
|
1686
|
-
client: "curl",
|
|
1687
|
-
content: `curl https://router.huggingface.co/hf-inference/models/${model.id} \\
|
|
1688
|
-
-X POST \\
|
|
1689
|
-
--data-binary '@${getModelInputSnippet(model, true, true)}' \\
|
|
1690
|
-
-H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`
|
|
1691
|
-
}
|
|
1692
|
-
];
|
|
1693
|
-
};
|
|
1694
|
-
var curlSnippets = {
|
|
1695
|
-
// Same order as in tasks/src/pipelines.ts
|
|
1696
|
-
"text-classification": snippetBasic,
|
|
1697
|
-
"token-classification": snippetBasic,
|
|
1698
|
-
"table-question-answering": snippetBasic,
|
|
1699
|
-
"question-answering": snippetBasic,
|
|
1700
|
-
"zero-shot-classification": snippetZeroShotClassification,
|
|
1701
|
-
translation: snippetBasic,
|
|
1702
|
-
summarization: snippetBasic,
|
|
1703
|
-
"feature-extraction": snippetBasic,
|
|
1704
|
-
"text-generation": snippetTextGeneration,
|
|
1705
|
-
"image-text-to-text": snippetTextGeneration,
|
|
1706
|
-
"text2text-generation": snippetBasic,
|
|
1707
|
-
"fill-mask": snippetBasic,
|
|
1708
|
-
"sentence-similarity": snippetBasic,
|
|
1709
|
-
"automatic-speech-recognition": snippetFile,
|
|
1710
|
-
"text-to-image": snippetBasic,
|
|
1711
|
-
"text-to-speech": snippetBasic,
|
|
1712
|
-
"text-to-audio": snippetBasic,
|
|
1713
|
-
"audio-to-audio": snippetFile,
|
|
1714
|
-
"audio-classification": snippetFile,
|
|
1715
|
-
"image-classification": snippetFile,
|
|
1716
|
-
"image-to-text": snippetFile,
|
|
1717
|
-
"object-detection": snippetFile,
|
|
1718
|
-
"image-segmentation": snippetFile
|
|
1719
|
-
};
|
|
1720
|
-
function getCurlInferenceSnippet(model, accessToken, provider, providerModelId, opts) {
|
|
1721
|
-
return model.pipeline_tag && model.pipeline_tag in curlSnippets ? curlSnippets[model.pipeline_tag]?.(model, accessToken, provider, providerModelId, opts) ?? [] : [];
|
|
1722
|
-
}
|
|
1723
|
-
|
|
1724
|
-
// src/snippets/python.ts
|
|
1725
|
-
var python_exports = {};
|
|
1726
|
-
__export(python_exports, {
|
|
1727
|
-
getPythonInferenceSnippet: () => getPythonInferenceSnippet
|
|
1728
|
-
});
|
|
1729
|
-
import { openAIbaseUrl } from "@huggingface/tasks";
|
|
1730
|
-
import {
|
|
1731
|
-
getModelInputSnippet as getModelInputSnippet2,
|
|
1732
|
-
stringifyGenerationConfig as stringifyGenerationConfig2,
|
|
1733
|
-
stringifyMessages as stringifyMessages2
|
|
1734
|
-
} from "@huggingface/tasks";
|
|
1735
|
-
var HFH_INFERENCE_CLIENT_METHODS = {
|
|
1633
|
+
var templatePath = (language, client, templateName) => path.join(rootDirFinder(), "src", "snippets", "templates", language, client, `${templateName}.jinja`);
|
|
1634
|
+
var hasTemplate = (language, client, templateName) => pathExists(templatePath(language, client, templateName));
|
|
1635
|
+
var loadTemplate = (language, client, templateName) => {
|
|
1636
|
+
const template = fs.readFileSync(templatePath(language, client, templateName), "utf8");
|
|
1637
|
+
return (data) => new Template(template).render({ ...data });
|
|
1638
|
+
};
|
|
1639
|
+
var snippetImportPythonInferenceClient = loadTemplate("python", "huggingface_hub", "importInferenceClient");
|
|
1640
|
+
var snippetImportRequests = loadTemplate("python", "requests", "importRequests");
|
|
1641
|
+
var HF_PYTHON_METHODS = {
|
|
1736
1642
|
"audio-classification": "audio_classification",
|
|
1737
1643
|
"audio-to-audio": "audio_to_audio",
|
|
1738
1644
|
"automatic-speech-recognition": "automatic_speech_recognition",
|
|
1739
|
-
"
|
|
1645
|
+
"document-question-answering": "document_question_answering",
|
|
1646
|
+
"feature-extraction": "feature_extraction",
|
|
1647
|
+
"fill-mask": "fill_mask",
|
|
1740
1648
|
"image-classification": "image_classification",
|
|
1741
1649
|
"image-segmentation": "image_segmentation",
|
|
1742
1650
|
"image-to-image": "image_to_image",
|
|
1743
1651
|
"image-to-text": "image_to_text",
|
|
1744
1652
|
"object-detection": "object_detection",
|
|
1745
|
-
"text-to-image": "text_to_image",
|
|
1746
|
-
"text-to-video": "text_to_video",
|
|
1747
|
-
"zero-shot-image-classification": "zero_shot_image_classification",
|
|
1748
|
-
"document-question-answering": "document_question_answering",
|
|
1749
|
-
"visual-question-answering": "visual_question_answering",
|
|
1750
|
-
"feature-extraction": "feature_extraction",
|
|
1751
|
-
"fill-mask": "fill_mask",
|
|
1752
1653
|
"question-answering": "question_answering",
|
|
1753
1654
|
"sentence-similarity": "sentence_similarity",
|
|
1754
1655
|
summarization: "summarization",
|
|
1755
1656
|
"table-question-answering": "table_question_answering",
|
|
1657
|
+
"tabular-classification": "tabular_classification",
|
|
1658
|
+
"tabular-regression": "tabular_regression",
|
|
1756
1659
|
"text-classification": "text_classification",
|
|
1757
1660
|
"text-generation": "text_generation",
|
|
1661
|
+
"text-to-image": "text_to_image",
|
|
1662
|
+
"text-to-speech": "text_to_speech",
|
|
1663
|
+
"text-to-video": "text_to_video",
|
|
1758
1664
|
"token-classification": "token_classification",
|
|
1759
1665
|
translation: "translation",
|
|
1666
|
+
"visual-question-answering": "visual_question_answering",
|
|
1760
1667
|
"zero-shot-classification": "zero_shot_classification",
|
|
1761
|
-
"
|
|
1762
|
-
"tabular-regression": "tabular_regression"
|
|
1763
|
-
};
|
|
1764
|
-
var snippetImportInferenceClient = (accessToken, provider) => `from huggingface_hub import InferenceClient
|
|
1765
|
-
|
|
1766
|
-
client = InferenceClient(
|
|
1767
|
-
provider="${provider}",
|
|
1768
|
-
api_key="${accessToken || "{API_TOKEN}"}",
|
|
1769
|
-
)`;
|
|
1770
|
-
var snippetConversational = (model, accessToken, provider, providerModelId, opts) => {
|
|
1771
|
-
const streaming = opts?.streaming ?? true;
|
|
1772
|
-
const exampleMessages = getModelInputSnippet2(model);
|
|
1773
|
-
const messages = opts?.messages ?? exampleMessages;
|
|
1774
|
-
const messagesStr = stringifyMessages2(messages, { attributeKeyQuotes: true });
|
|
1775
|
-
const config = {
|
|
1776
|
-
...opts?.temperature ? { temperature: opts.temperature } : void 0,
|
|
1777
|
-
max_tokens: opts?.max_tokens ?? 500,
|
|
1778
|
-
...opts?.top_p ? { top_p: opts.top_p } : void 0
|
|
1779
|
-
};
|
|
1780
|
-
const configStr = stringifyGenerationConfig2(config, {
|
|
1781
|
-
indent: "\n ",
|
|
1782
|
-
attributeValueConnector: "="
|
|
1783
|
-
});
|
|
1784
|
-
if (streaming) {
|
|
1785
|
-
return [
|
|
1786
|
-
{
|
|
1787
|
-
client: "huggingface_hub",
|
|
1788
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
1789
|
-
|
|
1790
|
-
messages = ${messagesStr}
|
|
1791
|
-
|
|
1792
|
-
stream = client.chat.completions.create(
|
|
1793
|
-
model="${model.id}",
|
|
1794
|
-
messages=messages,
|
|
1795
|
-
${configStr}
|
|
1796
|
-
stream=True,
|
|
1797
|
-
)
|
|
1798
|
-
|
|
1799
|
-
for chunk in stream:
|
|
1800
|
-
print(chunk.choices[0].delta.content, end="")`
|
|
1801
|
-
},
|
|
1802
|
-
{
|
|
1803
|
-
client: "openai",
|
|
1804
|
-
content: `from openai import OpenAI
|
|
1805
|
-
|
|
1806
|
-
client = OpenAI(
|
|
1807
|
-
base_url="${openAIbaseUrl(provider)}",
|
|
1808
|
-
api_key="${accessToken || "{API_TOKEN}"}"
|
|
1809
|
-
)
|
|
1810
|
-
|
|
1811
|
-
messages = ${messagesStr}
|
|
1812
|
-
|
|
1813
|
-
stream = client.chat.completions.create(
|
|
1814
|
-
model="${providerModelId ?? model.id}",
|
|
1815
|
-
messages=messages,
|
|
1816
|
-
${configStr}
|
|
1817
|
-
stream=True
|
|
1818
|
-
)
|
|
1819
|
-
|
|
1820
|
-
for chunk in stream:
|
|
1821
|
-
print(chunk.choices[0].delta.content, end="")`
|
|
1822
|
-
}
|
|
1823
|
-
];
|
|
1824
|
-
} else {
|
|
1825
|
-
return [
|
|
1826
|
-
{
|
|
1827
|
-
client: "huggingface_hub",
|
|
1828
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
1829
|
-
|
|
1830
|
-
messages = ${messagesStr}
|
|
1831
|
-
|
|
1832
|
-
completion = client.chat.completions.create(
|
|
1833
|
-
model="${model.id}",
|
|
1834
|
-
messages=messages,
|
|
1835
|
-
${configStr}
|
|
1836
|
-
)
|
|
1837
|
-
|
|
1838
|
-
print(completion.choices[0].message)`
|
|
1839
|
-
},
|
|
1840
|
-
{
|
|
1841
|
-
client: "openai",
|
|
1842
|
-
content: `from openai import OpenAI
|
|
1843
|
-
|
|
1844
|
-
client = OpenAI(
|
|
1845
|
-
base_url="${openAIbaseUrl(provider)}",
|
|
1846
|
-
api_key="${accessToken || "{API_TOKEN}"}"
|
|
1847
|
-
)
|
|
1848
|
-
|
|
1849
|
-
messages = ${messagesStr}
|
|
1850
|
-
|
|
1851
|
-
completion = client.chat.completions.create(
|
|
1852
|
-
model="${providerModelId ?? model.id}",
|
|
1853
|
-
messages=messages,
|
|
1854
|
-
${configStr}
|
|
1855
|
-
)
|
|
1856
|
-
|
|
1857
|
-
print(completion.choices[0].message)`
|
|
1858
|
-
}
|
|
1859
|
-
];
|
|
1860
|
-
}
|
|
1861
|
-
};
|
|
1862
|
-
var snippetZeroShotClassification2 = (model) => {
|
|
1863
|
-
return [
|
|
1864
|
-
{
|
|
1865
|
-
client: "requests",
|
|
1866
|
-
content: `def query(payload):
|
|
1867
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
1868
|
-
return response.json()
|
|
1869
|
-
|
|
1870
|
-
output = query({
|
|
1871
|
-
"inputs": ${getModelInputSnippet2(model)},
|
|
1872
|
-
"parameters": {"candidate_labels": ["refund", "legal", "faq"]},
|
|
1873
|
-
})`
|
|
1874
|
-
}
|
|
1875
|
-
];
|
|
1876
|
-
};
|
|
1877
|
-
var snippetZeroShotImageClassification = (model) => {
|
|
1878
|
-
return [
|
|
1879
|
-
{
|
|
1880
|
-
client: "requests",
|
|
1881
|
-
content: `def query(data):
|
|
1882
|
-
with open(data["image_path"], "rb") as f:
|
|
1883
|
-
img = f.read()
|
|
1884
|
-
payload={
|
|
1885
|
-
"parameters": data["parameters"],
|
|
1886
|
-
"inputs": base64.b64encode(img).decode("utf-8")
|
|
1887
|
-
}
|
|
1888
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
1889
|
-
return response.json()
|
|
1890
|
-
|
|
1891
|
-
output = query({
|
|
1892
|
-
"image_path": ${getModelInputSnippet2(model)},
|
|
1893
|
-
"parameters": {"candidate_labels": ["cat", "dog", "llama"]},
|
|
1894
|
-
})`
|
|
1895
|
-
}
|
|
1896
|
-
];
|
|
1668
|
+
"zero-shot-image-classification": "zero_shot_image_classification"
|
|
1897
1669
|
};
|
|
1898
|
-
var
|
|
1899
|
-
|
|
1900
|
-
|
|
1901
|
-
|
|
1902
|
-
|
|
1903
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
1904
|
-
|
|
1905
|
-
result = client.${HFH_INFERENCE_CLIENT_METHODS[model.pipeline_tag]}(
|
|
1906
|
-
inputs=${getModelInputSnippet2(model)},
|
|
1907
|
-
model="${model.id}",
|
|
1908
|
-
)
|
|
1909
|
-
|
|
1910
|
-
print(result)
|
|
1911
|
-
`
|
|
1912
|
-
}
|
|
1913
|
-
] : [],
|
|
1914
|
-
{
|
|
1915
|
-
client: "requests",
|
|
1916
|
-
content: `def query(payload):
|
|
1917
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
1918
|
-
return response.json()
|
|
1919
|
-
|
|
1920
|
-
output = query({
|
|
1921
|
-
"inputs": ${getModelInputSnippet2(model)},
|
|
1922
|
-
})`
|
|
1923
|
-
}
|
|
1924
|
-
];
|
|
1925
|
-
};
|
|
1926
|
-
var snippetFile2 = (model) => {
|
|
1927
|
-
return [
|
|
1928
|
-
{
|
|
1929
|
-
client: "requests",
|
|
1930
|
-
content: `def query(filename):
|
|
1931
|
-
with open(filename, "rb") as f:
|
|
1932
|
-
data = f.read()
|
|
1933
|
-
response = requests.post(API_URL, headers=headers, data=data)
|
|
1934
|
-
return response.json()
|
|
1935
|
-
|
|
1936
|
-
output = query(${getModelInputSnippet2(model)})`
|
|
1937
|
-
}
|
|
1938
|
-
];
|
|
1939
|
-
};
|
|
1940
|
-
var snippetTextToImage = (model, accessToken, provider, providerModelId) => {
|
|
1941
|
-
return [
|
|
1942
|
-
{
|
|
1943
|
-
client: "huggingface_hub",
|
|
1944
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
1945
|
-
|
|
1946
|
-
# output is a PIL.Image object
|
|
1947
|
-
image = client.text_to_image(
|
|
1948
|
-
${getModelInputSnippet2(model)},
|
|
1949
|
-
model="${model.id}",
|
|
1950
|
-
)`
|
|
1951
|
-
},
|
|
1952
|
-
...provider === "fal-ai" ? [
|
|
1953
|
-
{
|
|
1954
|
-
client: "fal-client",
|
|
1955
|
-
content: `import fal_client
|
|
1956
|
-
|
|
1957
|
-
result = fal_client.subscribe(
|
|
1958
|
-
"${providerModelId ?? model.id}",
|
|
1959
|
-
arguments={
|
|
1960
|
-
"prompt": ${getModelInputSnippet2(model)},
|
|
1961
|
-
},
|
|
1962
|
-
)
|
|
1963
|
-
print(result)
|
|
1964
|
-
`
|
|
1965
|
-
}
|
|
1966
|
-
] : [],
|
|
1967
|
-
...provider === "hf-inference" ? [
|
|
1968
|
-
{
|
|
1969
|
-
client: "requests",
|
|
1970
|
-
content: `def query(payload):
|
|
1971
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
1972
|
-
return response.content
|
|
1973
|
-
|
|
1974
|
-
image_bytes = query({
|
|
1975
|
-
"inputs": ${getModelInputSnippet2(model)},
|
|
1976
|
-
})
|
|
1977
|
-
|
|
1978
|
-
# You can access the image with PIL.Image for example
|
|
1979
|
-
import io
|
|
1980
|
-
from PIL import Image
|
|
1981
|
-
image = Image.open(io.BytesIO(image_bytes))`
|
|
1982
|
-
}
|
|
1983
|
-
] : []
|
|
1984
|
-
];
|
|
1985
|
-
};
|
|
1986
|
-
var snippetTextToVideo = (model, accessToken, provider) => {
|
|
1987
|
-
return ["fal-ai", "replicate"].includes(provider) ? [
|
|
1988
|
-
{
|
|
1989
|
-
client: "huggingface_hub",
|
|
1990
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
1991
|
-
|
|
1992
|
-
video = client.text_to_video(
|
|
1993
|
-
${getModelInputSnippet2(model)},
|
|
1994
|
-
model="${model.id}",
|
|
1995
|
-
)`
|
|
1996
|
-
}
|
|
1997
|
-
] : [];
|
|
1998
|
-
};
|
|
1999
|
-
var snippetTabular = (model) => {
|
|
2000
|
-
return [
|
|
2001
|
-
{
|
|
2002
|
-
client: "requests",
|
|
2003
|
-
content: `def query(payload):
|
|
2004
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
2005
|
-
return response.content
|
|
2006
|
-
|
|
2007
|
-
response = query({
|
|
2008
|
-
"inputs": {"data": ${getModelInputSnippet2(model)}},
|
|
2009
|
-
})`
|
|
2010
|
-
}
|
|
2011
|
-
];
|
|
2012
|
-
};
|
|
2013
|
-
var snippetTextToAudio = (model) => {
|
|
2014
|
-
if (model.library_name === "transformers") {
|
|
2015
|
-
return [
|
|
2016
|
-
{
|
|
2017
|
-
client: "requests",
|
|
2018
|
-
content: `def query(payload):
|
|
2019
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
2020
|
-
return response.content
|
|
2021
|
-
|
|
2022
|
-
audio_bytes = query({
|
|
2023
|
-
"inputs": ${getModelInputSnippet2(model)},
|
|
2024
|
-
})
|
|
2025
|
-
# You can access the audio with IPython.display for example
|
|
2026
|
-
from IPython.display import Audio
|
|
2027
|
-
Audio(audio_bytes)`
|
|
2028
|
-
}
|
|
2029
|
-
];
|
|
2030
|
-
} else {
|
|
2031
|
-
return [
|
|
2032
|
-
{
|
|
2033
|
-
client: "requests",
|
|
2034
|
-
content: `def query(payload):
|
|
2035
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
2036
|
-
return response.json()
|
|
2037
|
-
|
|
2038
|
-
audio, sampling_rate = query({
|
|
2039
|
-
"inputs": ${getModelInputSnippet2(model)},
|
|
2040
|
-
})
|
|
2041
|
-
# You can access the audio with IPython.display for example
|
|
2042
|
-
from IPython.display import Audio
|
|
2043
|
-
Audio(audio, rate=sampling_rate)`
|
|
2044
|
-
}
|
|
2045
|
-
];
|
|
2046
|
-
}
|
|
2047
|
-
};
|
|
2048
|
-
var snippetAutomaticSpeechRecognition = (model, accessToken, provider) => {
|
|
2049
|
-
return [
|
|
2050
|
-
{
|
|
2051
|
-
client: "huggingface_hub",
|
|
2052
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
2053
|
-
output = client.automatic_speech_recognition(${getModelInputSnippet2(model)}, model="${model.id}")`
|
|
2054
|
-
},
|
|
2055
|
-
snippetFile2(model)[0]
|
|
2056
|
-
];
|
|
2057
|
-
};
|
|
2058
|
-
var snippetDocumentQuestionAnswering = (model, accessToken, provider) => {
|
|
2059
|
-
const inputsAsStr = getModelInputSnippet2(model);
|
|
2060
|
-
const inputsAsObj = JSON.parse(inputsAsStr);
|
|
2061
|
-
return [
|
|
2062
|
-
{
|
|
2063
|
-
client: "huggingface_hub",
|
|
2064
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
2065
|
-
output = client.document_question_answering(
|
|
2066
|
-
"${inputsAsObj.image}",
|
|
2067
|
-
question="${inputsAsObj.question}",
|
|
2068
|
-
model="${model.id}",
|
|
2069
|
-
)`
|
|
2070
|
-
},
|
|
2071
|
-
{
|
|
2072
|
-
client: "requests",
|
|
2073
|
-
content: `def query(payload):
|
|
2074
|
-
with open(payload["image"], "rb") as f:
|
|
2075
|
-
img = f.read()
|
|
2076
|
-
payload["image"] = base64.b64encode(img).decode("utf-8")
|
|
2077
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
2078
|
-
return response.json()
|
|
2079
|
-
|
|
2080
|
-
output = query({
|
|
2081
|
-
"inputs": ${inputsAsStr},
|
|
2082
|
-
})`
|
|
2083
|
-
}
|
|
2084
|
-
];
|
|
2085
|
-
};
|
|
2086
|
-
var snippetImageToImage = (model, accessToken, provider) => {
|
|
2087
|
-
const inputsAsStr = getModelInputSnippet2(model);
|
|
2088
|
-
const inputsAsObj = JSON.parse(inputsAsStr);
|
|
2089
|
-
return [
|
|
2090
|
-
{
|
|
2091
|
-
client: "huggingface_hub",
|
|
2092
|
-
content: `${snippetImportInferenceClient(accessToken, provider)}
|
|
2093
|
-
# output is a PIL.Image object
|
|
2094
|
-
image = client.image_to_image(
|
|
2095
|
-
"${inputsAsObj.image}",
|
|
2096
|
-
prompt="${inputsAsObj.prompt}",
|
|
2097
|
-
model="${model.id}",
|
|
2098
|
-
)`
|
|
2099
|
-
},
|
|
2100
|
-
{
|
|
2101
|
-
client: "requests",
|
|
2102
|
-
content: `def query(payload):
|
|
2103
|
-
with open(payload["inputs"], "rb") as f:
|
|
2104
|
-
img = f.read()
|
|
2105
|
-
payload["inputs"] = base64.b64encode(img).decode("utf-8")
|
|
2106
|
-
response = requests.post(API_URL, headers=headers, json=payload)
|
|
2107
|
-
return response.content
|
|
2108
|
-
|
|
2109
|
-
image_bytes = query({
|
|
2110
|
-
"inputs": "${inputsAsObj.image}",
|
|
2111
|
-
"parameters": {"prompt": "${inputsAsObj.prompt}"},
|
|
2112
|
-
})
|
|
2113
|
-
|
|
2114
|
-
# You can access the image with PIL.Image for example
|
|
2115
|
-
import io
|
|
2116
|
-
from PIL import Image
|
|
2117
|
-
image = Image.open(io.BytesIO(image_bytes))`
|
|
2118
|
-
}
|
|
2119
|
-
];
|
|
2120
|
-
};
|
|
2121
|
-
var pythonSnippets = {
|
|
2122
|
-
// Same order as in tasks/src/pipelines.ts
|
|
2123
|
-
"text-classification": snippetBasic2,
|
|
2124
|
-
"token-classification": snippetBasic2,
|
|
2125
|
-
"table-question-answering": snippetBasic2,
|
|
2126
|
-
"question-answering": snippetBasic2,
|
|
2127
|
-
"zero-shot-classification": snippetZeroShotClassification2,
|
|
2128
|
-
translation: snippetBasic2,
|
|
2129
|
-
summarization: snippetBasic2,
|
|
2130
|
-
"feature-extraction": snippetBasic2,
|
|
2131
|
-
"text-generation": snippetBasic2,
|
|
2132
|
-
"text2text-generation": snippetBasic2,
|
|
2133
|
-
"image-text-to-text": snippetConversational,
|
|
2134
|
-
"fill-mask": snippetBasic2,
|
|
2135
|
-
"sentence-similarity": snippetBasic2,
|
|
2136
|
-
"automatic-speech-recognition": snippetAutomaticSpeechRecognition,
|
|
2137
|
-
"text-to-image": snippetTextToImage,
|
|
2138
|
-
"text-to-video": snippetTextToVideo,
|
|
2139
|
-
"text-to-speech": snippetTextToAudio,
|
|
2140
|
-
"text-to-audio": snippetTextToAudio,
|
|
2141
|
-
"audio-to-audio": snippetFile2,
|
|
2142
|
-
"audio-classification": snippetFile2,
|
|
2143
|
-
"image-classification": snippetFile2,
|
|
2144
|
-
"tabular-regression": snippetTabular,
|
|
2145
|
-
"tabular-classification": snippetTabular,
|
|
2146
|
-
"object-detection": snippetFile2,
|
|
2147
|
-
"image-segmentation": snippetFile2,
|
|
2148
|
-
"document-question-answering": snippetDocumentQuestionAnswering,
|
|
2149
|
-
"image-to-text": snippetFile2,
|
|
2150
|
-
"image-to-image": snippetImageToImage,
|
|
2151
|
-
"zero-shot-image-classification": snippetZeroShotImageClassification
|
|
2152
|
-
};
|
|
2153
|
-
function getPythonInferenceSnippet(model, accessToken, provider, providerModelId, opts) {
|
|
2154
|
-
if (model.tags.includes("conversational")) {
|
|
2155
|
-
return snippetConversational(model, accessToken, provider, providerModelId, opts);
|
|
2156
|
-
} else {
|
|
2157
|
-
const snippets = model.pipeline_tag && model.pipeline_tag in pythonSnippets ? pythonSnippets[model.pipeline_tag]?.(model, accessToken, provider, providerModelId) ?? [] : [];
|
|
2158
|
-
return snippets.map((snippet) => {
|
|
2159
|
-
return {
|
|
2160
|
-
...snippet,
|
|
2161
|
-
content: addImportsToSnippet(snippet.content, model, accessToken)
|
|
2162
|
-
};
|
|
2163
|
-
});
|
|
2164
|
-
}
|
|
2165
|
-
}
|
|
2166
|
-
var addImportsToSnippet = (snippet, model, accessToken) => {
|
|
2167
|
-
if (snippet.includes("requests")) {
|
|
2168
|
-
snippet = `import requests
|
|
2169
|
-
|
|
2170
|
-
API_URL = "https://router.huggingface.co/hf-inference/models/${model.id}"
|
|
2171
|
-
headers = {"Authorization": ${accessToken ? `"Bearer ${accessToken}"` : `f"Bearer {API_TOKEN}"`}}
|
|
2172
|
-
|
|
2173
|
-
${snippet}`;
|
|
2174
|
-
}
|
|
2175
|
-
if (snippet.includes("base64")) {
|
|
2176
|
-
snippet = `import base64
|
|
2177
|
-
${snippet}`;
|
|
2178
|
-
}
|
|
2179
|
-
return snippet;
|
|
2180
|
-
};
|
|
2181
|
-
|
|
2182
|
-
// src/snippets/js.ts
|
|
2183
|
-
var js_exports = {};
|
|
2184
|
-
__export(js_exports, {
|
|
2185
|
-
getJsInferenceSnippet: () => getJsInferenceSnippet,
|
|
2186
|
-
jsSnippets: () => jsSnippets,
|
|
2187
|
-
snippetAutomaticSpeechRecognition: () => snippetAutomaticSpeechRecognition2,
|
|
2188
|
-
snippetBasic: () => snippetBasic3,
|
|
2189
|
-
snippetFile: () => snippetFile3,
|
|
2190
|
-
snippetTextGeneration: () => snippetTextGeneration2,
|
|
2191
|
-
snippetTextToAudio: () => snippetTextToAudio2,
|
|
2192
|
-
snippetTextToImage: () => snippetTextToImage2,
|
|
2193
|
-
snippetTextToVideo: () => snippetTextToVideo2,
|
|
2194
|
-
snippetZeroShotClassification: () => snippetZeroShotClassification3
|
|
2195
|
-
});
|
|
2196
|
-
import { openAIbaseUrl as openAIbaseUrl2 } from "@huggingface/tasks";
|
|
2197
|
-
import {
|
|
2198
|
-
getModelInputSnippet as getModelInputSnippet3,
|
|
2199
|
-
stringifyGenerationConfig as stringifyGenerationConfig3,
|
|
2200
|
-
stringifyMessages as stringifyMessages3
|
|
2201
|
-
} from "@huggingface/tasks";
|
|
2202
|
-
var HFJS_METHODS = {
|
|
2203
|
-
"text-classification": "textClassification",
|
|
2204
|
-
"token-classification": "tokenClassification",
|
|
2205
|
-
"table-question-answering": "tableQuestionAnswering",
|
|
1670
|
+
var HF_JS_METHODS = {
|
|
1671
|
+
"automatic-speech-recognition": "automaticSpeechRecognition",
|
|
1672
|
+
"feature-extraction": "featureExtraction",
|
|
1673
|
+
"fill-mask": "fillMask",
|
|
1674
|
+
"image-classification": "imageClassification",
|
|
2206
1675
|
"question-answering": "questionAnswering",
|
|
2207
|
-
|
|
1676
|
+
"sentence-similarity": "sentenceSimilarity",
|
|
2208
1677
|
summarization: "summarization",
|
|
2209
|
-
"
|
|
1678
|
+
"table-question-answering": "tableQuestionAnswering",
|
|
1679
|
+
"text-classification": "textClassification",
|
|
2210
1680
|
"text-generation": "textGeneration",
|
|
2211
1681
|
"text2text-generation": "textGeneration",
|
|
2212
|
-
"
|
|
2213
|
-
|
|
1682
|
+
"token-classification": "tokenClassification",
|
|
1683
|
+
translation: "translation"
|
|
2214
1684
|
};
|
|
2215
|
-
var
|
|
2216
|
-
return
|
|
2217
|
-
|
|
2218
|
-
|
|
2219
|
-
|
|
2220
|
-
|
|
2221
|
-
|
|
2222
|
-
const
|
|
2223
|
-
|
|
2224
|
-
|
|
2225
|
-
|
|
2226
|
-
|
|
2227
|
-
|
|
2228
|
-
|
|
2229
|
-
|
|
2230
|
-
|
|
2231
|
-
|
|
1685
|
+
var snippetGenerator = (templateName, inputPreparationFn) => {
|
|
1686
|
+
return (model, accessToken, provider, providerModelId, opts) => {
|
|
1687
|
+
if (model.pipeline_tag && ["text-generation", "image-text-to-text"].includes(model.pipeline_tag) && model.tags.includes("conversational")) {
|
|
1688
|
+
templateName = opts?.streaming ? "conversationalStream" : "conversational";
|
|
1689
|
+
inputPreparationFn = prepareConversationalInput;
|
|
1690
|
+
}
|
|
1691
|
+
const inputs = inputPreparationFn ? inputPreparationFn(model, opts) : { inputs: getModelInputSnippet(model) };
|
|
1692
|
+
const request2 = makeRequestOptionsFromResolvedModel(
|
|
1693
|
+
providerModelId ?? model.id,
|
|
1694
|
+
{ accessToken, provider, ...inputs },
|
|
1695
|
+
{ chatCompletion: templateName.includes("conversational"), task: model.pipeline_tag }
|
|
1696
|
+
);
|
|
1697
|
+
let providerInputs = inputs;
|
|
1698
|
+
const bodyAsObj = request2.info.body;
|
|
1699
|
+
if (typeof bodyAsObj === "string") {
|
|
1700
|
+
try {
|
|
1701
|
+
providerInputs = JSON.parse(bodyAsObj);
|
|
1702
|
+
} catch (e) {
|
|
1703
|
+
console.error("Failed to parse body as JSON", e);
|
|
2232
1704
|
}
|
|
2233
|
-
] : [],
|
|
2234
|
-
{
|
|
2235
|
-
client: "fetch",
|
|
2236
|
-
content: `async function query(data) {
|
|
2237
|
-
const response = await fetch(
|
|
2238
|
-
"https://router.huggingface.co/hf-inference/models/${model.id}",
|
|
2239
|
-
{
|
|
2240
|
-
headers: {
|
|
2241
|
-
Authorization: "Bearer ${accessToken || `{API_TOKEN}`}",
|
|
2242
|
-
"Content-Type": "application/json",
|
|
2243
|
-
},
|
|
2244
|
-
method: "POST",
|
|
2245
|
-
body: JSON.stringify(data),
|
|
2246
|
-
}
|
|
2247
|
-
);
|
|
2248
|
-
const result = await response.json();
|
|
2249
|
-
return result;
|
|
2250
|
-
}
|
|
2251
|
-
|
|
2252
|
-
query({"inputs": ${getModelInputSnippet3(model)}}).then((response) => {
|
|
2253
|
-
console.log(JSON.stringify(response));
|
|
2254
|
-
});`
|
|
2255
1705
|
}
|
|
2256
|
-
|
|
2257
|
-
|
|
2258
|
-
|
|
2259
|
-
|
|
2260
|
-
|
|
2261
|
-
|
|
2262
|
-
|
|
2263
|
-
|
|
2264
|
-
|
|
2265
|
-
|
|
2266
|
-
|
|
2267
|
-
|
|
1706
|
+
const params = {
|
|
1707
|
+
accessToken,
|
|
1708
|
+
authorizationHeader: request2.info.headers?.Authorization,
|
|
1709
|
+
baseUrl: removeSuffix(request2.url, "/chat/completions"),
|
|
1710
|
+
fullUrl: request2.url,
|
|
1711
|
+
inputs: {
|
|
1712
|
+
asObj: inputs,
|
|
1713
|
+
asCurlString: formatBody(inputs, "curl"),
|
|
1714
|
+
asJsonString: formatBody(inputs, "json"),
|
|
1715
|
+
asPythonString: formatBody(inputs, "python"),
|
|
1716
|
+
asTsString: formatBody(inputs, "ts")
|
|
1717
|
+
},
|
|
1718
|
+
providerInputs: {
|
|
1719
|
+
asObj: providerInputs,
|
|
1720
|
+
asCurlString: formatBody(providerInputs, "curl"),
|
|
1721
|
+
asJsonString: formatBody(providerInputs, "json"),
|
|
1722
|
+
asPythonString: formatBody(providerInputs, "python"),
|
|
1723
|
+
asTsString: formatBody(providerInputs, "ts")
|
|
1724
|
+
},
|
|
1725
|
+
model,
|
|
1726
|
+
provider,
|
|
1727
|
+
providerModelId: providerModelId ?? model.id
|
|
2268
1728
|
};
|
|
2269
|
-
|
|
2270
|
-
|
|
2271
|
-
|
|
2272
|
-
|
|
2273
|
-
if (streaming) {
|
|
2274
|
-
return [
|
|
2275
|
-
{
|
|
2276
|
-
client: "huggingface.js",
|
|
2277
|
-
content: `import { InferenceClient } from "@huggingface/inference";
|
|
2278
|
-
|
|
2279
|
-
const client = new InferenceClient("${accessToken || `{API_TOKEN}`}");
|
|
2280
|
-
|
|
2281
|
-
let out = "";
|
|
2282
|
-
|
|
2283
|
-
const stream = client.chatCompletionStream({
|
|
2284
|
-
model: "${model.id}",
|
|
2285
|
-
messages: ${messagesStr},
|
|
2286
|
-
provider: "${provider}",
|
|
2287
|
-
${configStr}
|
|
2288
|
-
});
|
|
2289
|
-
|
|
2290
|
-
for await (const chunk of stream) {
|
|
2291
|
-
if (chunk.choices && chunk.choices.length > 0) {
|
|
2292
|
-
const newContent = chunk.choices[0].delta.content;
|
|
2293
|
-
out += newContent;
|
|
2294
|
-
console.log(newContent);
|
|
2295
|
-
}
|
|
2296
|
-
}`
|
|
2297
|
-
},
|
|
2298
|
-
{
|
|
2299
|
-
client: "openai",
|
|
2300
|
-
content: `import { OpenAI } from "openai";
|
|
2301
|
-
|
|
2302
|
-
const client = new OpenAI({
|
|
2303
|
-
baseURL: "${openAIbaseUrl2(provider)}",
|
|
2304
|
-
apiKey: "${accessToken || `{API_TOKEN}`}"
|
|
2305
|
-
});
|
|
2306
|
-
|
|
2307
|
-
let out = "";
|
|
2308
|
-
|
|
2309
|
-
const stream = await client.chat.completions.create({
|
|
2310
|
-
model: "${providerModelId ?? model.id}",
|
|
2311
|
-
messages: ${messagesStr},
|
|
2312
|
-
${configStr}
|
|
2313
|
-
stream: true,
|
|
2314
|
-
});
|
|
2315
|
-
|
|
2316
|
-
for await (const chunk of stream) {
|
|
2317
|
-
if (chunk.choices && chunk.choices.length > 0) {
|
|
2318
|
-
const newContent = chunk.choices[0].delta.content;
|
|
2319
|
-
out += newContent;
|
|
2320
|
-
console.log(newContent);
|
|
2321
|
-
}
|
|
2322
|
-
}`
|
|
1729
|
+
return inferenceSnippetLanguages.map((language) => {
|
|
1730
|
+
return CLIENTS[language].map((client) => {
|
|
1731
|
+
if (!hasTemplate(language, client, templateName)) {
|
|
1732
|
+
return;
|
|
2323
1733
|
}
|
|
2324
|
-
|
|
2325
|
-
|
|
2326
|
-
|
|
2327
|
-
|
|
2328
|
-
|
|
2329
|
-
|
|
2330
|
-
|
|
2331
|
-
|
|
2332
|
-
|
|
2333
|
-
|
|
2334
|
-
|
|
2335
|
-
|
|
2336
|
-
|
|
2337
|
-
|
|
2338
|
-
|
|
2339
|
-
|
|
2340
|
-
|
|
2341
|
-
|
|
2342
|
-
|
|
2343
|
-
|
|
2344
|
-
client: "openai",
|
|
2345
|
-
content: `import { OpenAI } from "openai";
|
|
2346
|
-
|
|
2347
|
-
const client = new OpenAI({
|
|
2348
|
-
baseURL: "${openAIbaseUrl2(provider)}",
|
|
2349
|
-
apiKey: "${accessToken || `{API_TOKEN}`}"
|
|
2350
|
-
});
|
|
1734
|
+
const template = loadTemplate(language, client, templateName);
|
|
1735
|
+
if (client === "huggingface_hub" && templateName.includes("basic")) {
|
|
1736
|
+
if (!(model.pipeline_tag && model.pipeline_tag in HF_PYTHON_METHODS)) {
|
|
1737
|
+
return;
|
|
1738
|
+
}
|
|
1739
|
+
params["methodName"] = HF_PYTHON_METHODS[model.pipeline_tag];
|
|
1740
|
+
}
|
|
1741
|
+
if (client === "huggingface.js" && templateName.includes("basic")) {
|
|
1742
|
+
if (!(model.pipeline_tag && model.pipeline_tag in HF_JS_METHODS)) {
|
|
1743
|
+
return;
|
|
1744
|
+
}
|
|
1745
|
+
params["methodName"] = HF_JS_METHODS[model.pipeline_tag];
|
|
1746
|
+
}
|
|
1747
|
+
let snippet = template(params).trim();
|
|
1748
|
+
if (!snippet) {
|
|
1749
|
+
return;
|
|
1750
|
+
}
|
|
1751
|
+
if (client === "huggingface_hub") {
|
|
1752
|
+
const importSection = snippetImportPythonInferenceClient({ ...params });
|
|
1753
|
+
snippet = `${importSection}
|
|
2351
1754
|
|
|
2352
|
-
|
|
2353
|
-
|
|
2354
|
-
|
|
2355
|
-
|
|
2356
|
-
|
|
1755
|
+
${snippet}`;
|
|
1756
|
+
} else if (client === "requests") {
|
|
1757
|
+
const importSection = snippetImportRequests({
|
|
1758
|
+
...params,
|
|
1759
|
+
importBase64: snippet.includes("base64"),
|
|
1760
|
+
importJson: snippet.includes("json.")
|
|
1761
|
+
});
|
|
1762
|
+
snippet = `${importSection}
|
|
2357
1763
|
|
|
2358
|
-
|
|
2359
|
-
`
|
|
1764
|
+
${snippet}`;
|
|
2360
1765
|
}
|
|
2361
|
-
|
|
2362
|
-
|
|
2363
|
-
|
|
2364
|
-
|
|
2365
|
-
}
|
|
2366
|
-
};
|
|
2367
|
-
var snippetZeroShotClassification3 = (model, accessToken) => {
|
|
2368
|
-
return [
|
|
2369
|
-
{
|
|
2370
|
-
client: "fetch",
|
|
2371
|
-
content: `async function query(data) {
|
|
2372
|
-
const response = await fetch(
|
|
2373
|
-
"https://router.huggingface.co/hf-inference/models/${model.id}",
|
|
2374
|
-
{
|
|
2375
|
-
headers: {
|
|
2376
|
-
Authorization: "Bearer ${accessToken || `{API_TOKEN}`}",
|
|
2377
|
-
"Content-Type": "application/json",
|
|
2378
|
-
},
|
|
2379
|
-
method: "POST",
|
|
2380
|
-
body: JSON.stringify(data),
|
|
2381
|
-
}
|
|
2382
|
-
);
|
|
2383
|
-
const result = await response.json();
|
|
2384
|
-
return result;
|
|
2385
|
-
}
|
|
2386
|
-
|
|
2387
|
-
query({"inputs": ${getModelInputSnippet3(
|
|
2388
|
-
model
|
|
2389
|
-
)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}).then((response) => {
|
|
2390
|
-
console.log(JSON.stringify(response));
|
|
2391
|
-
});`
|
|
2392
|
-
}
|
|
2393
|
-
];
|
|
1766
|
+
return { language, client, content: snippet };
|
|
1767
|
+
}).filter((snippet) => snippet !== void 0);
|
|
1768
|
+
}).flat();
|
|
1769
|
+
};
|
|
2394
1770
|
};
|
|
2395
|
-
var
|
|
2396
|
-
return
|
|
2397
|
-
|
|
2398
|
-
|
|
2399
|
-
|
|
2400
|
-
|
|
2401
|
-
|
|
2402
|
-
|
|
2403
|
-
|
|
2404
|
-
|
|
2405
|
-
|
|
2406
|
-
|
|
2407
|
-
|
|
2408
|
-
}
|
|
2409
|
-
|
|
2410
|
-
|
|
2411
|
-
|
|
2412
|
-
|
|
2413
|
-
|
|
2414
|
-
|
|
2415
|
-
|
|
2416
|
-
|
|
2417
|
-
|
|
2418
|
-
|
|
2419
|
-
|
|
2420
|
-
|
|
2421
|
-
|
|
2422
|
-
|
|
2423
|
-
|
|
2424
|
-
|
|
2425
|
-
|
|
2426
|
-
|
|
2427
|
-
|
|
2428
|
-
|
|
1771
|
+
var prepareDocumentQuestionAnsweringInput = (model) => {
|
|
1772
|
+
return JSON.parse(getModelInputSnippet(model));
|
|
1773
|
+
};
|
|
1774
|
+
var prepareImageToImageInput = (model) => {
|
|
1775
|
+
const data = JSON.parse(getModelInputSnippet(model));
|
|
1776
|
+
return { inputs: data.image, parameters: { prompt: data.prompt } };
|
|
1777
|
+
};
|
|
1778
|
+
var prepareConversationalInput = (model, opts) => {
|
|
1779
|
+
return {
|
|
1780
|
+
messages: opts?.messages ?? getModelInputSnippet(model),
|
|
1781
|
+
...opts?.temperature ? { temperature: opts?.temperature } : void 0,
|
|
1782
|
+
max_tokens: opts?.max_tokens ?? 500,
|
|
1783
|
+
...opts?.top_p ? { top_p: opts?.top_p } : void 0
|
|
1784
|
+
};
|
|
1785
|
+
};
|
|
1786
|
+
var snippets = {
|
|
1787
|
+
"audio-classification": snippetGenerator("basicAudio"),
|
|
1788
|
+
"audio-to-audio": snippetGenerator("basicAudio"),
|
|
1789
|
+
"automatic-speech-recognition": snippetGenerator("basicAudio"),
|
|
1790
|
+
"document-question-answering": snippetGenerator("documentQuestionAnswering", prepareDocumentQuestionAnsweringInput),
|
|
1791
|
+
"feature-extraction": snippetGenerator("basic"),
|
|
1792
|
+
"fill-mask": snippetGenerator("basic"),
|
|
1793
|
+
"image-classification": snippetGenerator("basicImage"),
|
|
1794
|
+
"image-segmentation": snippetGenerator("basicImage"),
|
|
1795
|
+
"image-text-to-text": snippetGenerator("conversational"),
|
|
1796
|
+
"image-to-image": snippetGenerator("imageToImage", prepareImageToImageInput),
|
|
1797
|
+
"image-to-text": snippetGenerator("basicImage"),
|
|
1798
|
+
"object-detection": snippetGenerator("basicImage"),
|
|
1799
|
+
"question-answering": snippetGenerator("basic"),
|
|
1800
|
+
"sentence-similarity": snippetGenerator("basic"),
|
|
1801
|
+
summarization: snippetGenerator("basic"),
|
|
1802
|
+
"tabular-classification": snippetGenerator("tabular"),
|
|
1803
|
+
"tabular-regression": snippetGenerator("tabular"),
|
|
1804
|
+
"table-question-answering": snippetGenerator("basic"),
|
|
1805
|
+
"text-classification": snippetGenerator("basic"),
|
|
1806
|
+
"text-generation": snippetGenerator("basic"),
|
|
1807
|
+
"text-to-audio": snippetGenerator("textToAudio"),
|
|
1808
|
+
"text-to-image": snippetGenerator("textToImage"),
|
|
1809
|
+
"text-to-speech": snippetGenerator("textToAudio"),
|
|
1810
|
+
"text-to-video": snippetGenerator("textToVideo"),
|
|
1811
|
+
"text2text-generation": snippetGenerator("basic"),
|
|
1812
|
+
"token-classification": snippetGenerator("basic"),
|
|
1813
|
+
translation: snippetGenerator("basic"),
|
|
1814
|
+
"zero-shot-classification": snippetGenerator("zeroShotClassification"),
|
|
1815
|
+
"zero-shot-image-classification": snippetGenerator("zeroShotImageClassification")
|
|
1816
|
+
};
|
|
1817
|
+
function getInferenceSnippets(model, accessToken, provider, providerModelId, opts) {
|
|
1818
|
+
return model.pipeline_tag && model.pipeline_tag in snippets ? snippets[model.pipeline_tag]?.(model, accessToken, provider, providerModelId, opts) ?? [] : [];
|
|
2429
1819
|
}
|
|
2430
|
-
|
|
2431
|
-
|
|
2432
|
-
|
|
2433
|
-
|
|
2434
|
-
|
|
2435
|
-
|
|
2436
|
-
|
|
2437
|
-
|
|
2438
|
-
|
|
2439
|
-
|
|
2440
|
-
|
|
2441
|
-
|
|
2442
|
-
|
|
2443
|
-
|
|
2444
|
-
|
|
2445
|
-
|
|
2446
|
-
|
|
2447
|
-
provider: "${provider}",
|
|
2448
|
-
inputs: ${getModelInputSnippet3(model)},
|
|
2449
|
-
parameters: { num_inference_steps: 5 },
|
|
2450
|
-
});
|
|
2451
|
-
// Use the generated video (it's a Blob)
|
|
2452
|
-
`
|
|
2453
|
-
}
|
|
2454
|
-
] : [];
|
|
2455
|
-
};
|
|
2456
|
-
var snippetTextToAudio2 = (model, accessToken, provider) => {
|
|
2457
|
-
if (provider !== "hf-inference") {
|
|
2458
|
-
return [];
|
|
2459
|
-
}
|
|
2460
|
-
const commonSnippet = `async function query(data) {
|
|
2461
|
-
const response = await fetch(
|
|
2462
|
-
"https://router.huggingface.co/hf-inference/models/${model.id}",
|
|
2463
|
-
{
|
|
2464
|
-
headers: {
|
|
2465
|
-
Authorization: "Bearer ${accessToken || `{API_TOKEN}`}",
|
|
2466
|
-
"Content-Type": "application/json",
|
|
2467
|
-
},
|
|
2468
|
-
method: "POST",
|
|
2469
|
-
body: JSON.stringify(data),
|
|
2470
|
-
}
|
|
2471
|
-
);`;
|
|
2472
|
-
if (model.library_name === "transformers") {
|
|
2473
|
-
return [
|
|
2474
|
-
{
|
|
2475
|
-
client: "fetch",
|
|
2476
|
-
content: commonSnippet + `
|
|
2477
|
-
const result = await response.blob();
|
|
2478
|
-
return result;
|
|
2479
|
-
}
|
|
2480
|
-
query({"inputs": ${getModelInputSnippet3(model)}}).then((response) => {
|
|
2481
|
-
// Returns a byte object of the Audio wavform. Use it directly!
|
|
2482
|
-
});`
|
|
2483
|
-
}
|
|
2484
|
-
];
|
|
2485
|
-
} else {
|
|
2486
|
-
return [
|
|
2487
|
-
{
|
|
2488
|
-
client: "fetch",
|
|
2489
|
-
content: commonSnippet + `
|
|
2490
|
-
const result = await response.json();
|
|
2491
|
-
return result;
|
|
2492
|
-
}
|
|
2493
|
-
|
|
2494
|
-
query({"inputs": ${getModelInputSnippet3(model)}}).then((response) => {
|
|
2495
|
-
console.log(JSON.stringify(response));
|
|
2496
|
-
});`
|
|
2497
|
-
}
|
|
2498
|
-
];
|
|
1820
|
+
function formatBody(obj, format) {
|
|
1821
|
+
switch (format) {
|
|
1822
|
+
case "curl":
|
|
1823
|
+
return indentString(formatBody(obj, "json"));
|
|
1824
|
+
case "json":
|
|
1825
|
+
return JSON.stringify(obj, null, 4).split("\n").slice(1, -1).join("\n");
|
|
1826
|
+
case "python":
|
|
1827
|
+
return indentString(
|
|
1828
|
+
Object.entries(obj).map(([key, value]) => {
|
|
1829
|
+
const formattedValue = JSON.stringify(value, null, 4).replace(/"/g, '"');
|
|
1830
|
+
return `${key}=${formattedValue},`;
|
|
1831
|
+
}).join("\n")
|
|
1832
|
+
);
|
|
1833
|
+
case "ts":
|
|
1834
|
+
return formatTsObject(obj).split("\n").slice(1, -1).join("\n");
|
|
1835
|
+
default:
|
|
1836
|
+
throw new Error(`Unsupported format: ${format}`);
|
|
2499
1837
|
}
|
|
2500
|
-
};
|
|
2501
|
-
var snippetAutomaticSpeechRecognition2 = (model, accessToken, provider) => {
|
|
2502
|
-
return [
|
|
2503
|
-
{
|
|
2504
|
-
client: "huggingface.js",
|
|
2505
|
-
content: `import { InferenceClient } from "@huggingface/inference";
|
|
2506
|
-
|
|
2507
|
-
const client = new InferenceClient("${accessToken || `{API_TOKEN}`}");
|
|
2508
|
-
|
|
2509
|
-
const data = fs.readFileSync(${getModelInputSnippet3(model)});
|
|
2510
|
-
|
|
2511
|
-
const output = await client.automaticSpeechRecognition({
|
|
2512
|
-
data,
|
|
2513
|
-
model: "${model.id}",
|
|
2514
|
-
provider: "${provider}",
|
|
2515
|
-
});
|
|
2516
|
-
|
|
2517
|
-
console.log(output);
|
|
2518
|
-
`
|
|
2519
|
-
},
|
|
2520
|
-
...provider === "hf-inference" ? snippetFile3(model, accessToken, provider) : []
|
|
2521
|
-
];
|
|
2522
|
-
};
|
|
2523
|
-
var snippetFile3 = (model, accessToken, provider) => {
|
|
2524
|
-
if (provider !== "hf-inference") {
|
|
2525
|
-
return [];
|
|
2526
|
-
}
|
|
2527
|
-
return [
|
|
2528
|
-
{
|
|
2529
|
-
client: "fetch",
|
|
2530
|
-
content: `async function query(filename) {
|
|
2531
|
-
const data = fs.readFileSync(filename);
|
|
2532
|
-
const response = await fetch(
|
|
2533
|
-
"https://router.huggingface.co/hf-inference/models/${model.id}",
|
|
2534
|
-
{
|
|
2535
|
-
headers: {
|
|
2536
|
-
Authorization: "Bearer ${accessToken || `{API_TOKEN}`}",
|
|
2537
|
-
"Content-Type": "application/json",
|
|
2538
|
-
},
|
|
2539
|
-
method: "POST",
|
|
2540
|
-
body: data,
|
|
2541
|
-
}
|
|
2542
|
-
);
|
|
2543
|
-
const result = await response.json();
|
|
2544
|
-
return result;
|
|
2545
1838
|
}
|
|
2546
|
-
|
|
2547
|
-
|
|
2548
|
-
|
|
2549
|
-
|
|
2550
|
-
|
|
2551
|
-
|
|
2552
|
-
|
|
2553
|
-
|
|
2554
|
-
|
|
2555
|
-
|
|
2556
|
-
|
|
2557
|
-
|
|
2558
|
-
|
|
2559
|
-
|
|
2560
|
-
|
|
2561
|
-
|
|
2562
|
-
|
|
2563
|
-
|
|
2564
|
-
|
|
2565
|
-
"
|
|
2566
|
-
|
|
2567
|
-
|
|
2568
|
-
|
|
2569
|
-
|
|
2570
|
-
|
|
2571
|
-
"
|
|
2572
|
-
|
|
2573
|
-
|
|
2574
|
-
|
|
2575
|
-
"image-classification": snippetFile3,
|
|
2576
|
-
"image-to-text": snippetFile3,
|
|
2577
|
-
"object-detection": snippetFile3,
|
|
2578
|
-
"image-segmentation": snippetFile3
|
|
2579
|
-
};
|
|
2580
|
-
function getJsInferenceSnippet(model, accessToken, provider, providerModelId, opts) {
|
|
2581
|
-
return model.pipeline_tag && model.pipeline_tag in jsSnippets ? jsSnippets[model.pipeline_tag]?.(model, accessToken, provider, providerModelId, opts) ?? [] : [];
|
|
1839
|
+
function formatTsObject(obj, depth) {
|
|
1840
|
+
depth = depth ?? 0;
|
|
1841
|
+
if (typeof obj !== "object" || obj === null) {
|
|
1842
|
+
return JSON.stringify(obj);
|
|
1843
|
+
}
|
|
1844
|
+
if (Array.isArray(obj)) {
|
|
1845
|
+
const items = obj.map((item) => {
|
|
1846
|
+
const formatted = formatTsObject(item, depth + 1);
|
|
1847
|
+
return `${" ".repeat(4 * (depth + 1))}${formatted},`;
|
|
1848
|
+
}).join("\n");
|
|
1849
|
+
return `[
|
|
1850
|
+
${items}
|
|
1851
|
+
${" ".repeat(4 * depth)}]`;
|
|
1852
|
+
}
|
|
1853
|
+
const entries = Object.entries(obj);
|
|
1854
|
+
const lines = entries.map(([key, value]) => {
|
|
1855
|
+
const formattedValue = formatTsObject(value, depth + 1);
|
|
1856
|
+
const keyStr = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(key) ? key : `"${key}"`;
|
|
1857
|
+
return `${" ".repeat(4 * (depth + 1))}${keyStr}: ${formattedValue},`;
|
|
1858
|
+
}).join("\n");
|
|
1859
|
+
return `{
|
|
1860
|
+
${lines}
|
|
1861
|
+
${" ".repeat(4 * depth)}}`;
|
|
1862
|
+
}
|
|
1863
|
+
function indentString(str) {
|
|
1864
|
+
return str.split("\n").map((line) => " ".repeat(4) + line).join("\n");
|
|
1865
|
+
}
|
|
1866
|
+
function removeSuffix(str, suffix) {
|
|
1867
|
+
return str.endsWith(suffix) ? str.slice(0, -suffix.length) : str;
|
|
2582
1868
|
}
|
|
2583
1869
|
export {
|
|
2584
1870
|
HfInference,
|