@huggingface/inference 4.13.9 → 4.13.11
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/commonjs/lib/getProviderHelper.d.ts.map +1 -1
- package/dist/commonjs/lib/getProviderHelper.js +8 -4
- package/dist/commonjs/lib/makeRequestOptions.d.ts +2 -1
- package/dist/commonjs/lib/makeRequestOptions.d.ts.map +1 -1
- package/dist/commonjs/lib/makeRequestOptions.js +4 -2
- package/dist/commonjs/package.d.ts +1 -1
- package/dist/commonjs/package.d.ts.map +1 -1
- package/dist/commonjs/package.js +1 -1
- package/dist/commonjs/providers/fal-ai.d.ts +13 -3
- package/dist/commonjs/providers/fal-ai.d.ts.map +1 -1
- package/dist/commonjs/providers/fal-ai.js +49 -2
- package/dist/commonjs/providers/hf-inference.d.ts +3 -2
- package/dist/commonjs/providers/hf-inference.d.ts.map +1 -1
- package/dist/commonjs/providers/hf-inference.js +14 -8
- package/dist/commonjs/providers/hyperbolic.d.ts +2 -2
- package/dist/commonjs/providers/hyperbolic.d.ts.map +1 -1
- package/dist/commonjs/providers/hyperbolic.js +4 -1
- package/dist/commonjs/providers/nebius.d.ts +5 -4
- package/dist/commonjs/providers/nebius.d.ts.map +1 -1
- package/dist/commonjs/providers/nebius.js +11 -8
- package/dist/commonjs/providers/nscale.d.ts +2 -2
- package/dist/commonjs/providers/nscale.d.ts.map +1 -1
- package/dist/commonjs/providers/nscale.js +4 -1
- package/dist/commonjs/providers/providerHelper.d.ts +3 -3
- package/dist/commonjs/providers/providerHelper.d.ts.map +1 -1
- package/dist/commonjs/providers/providerHelper.js +3 -0
- package/dist/commonjs/providers/replicate.d.ts +2 -2
- package/dist/commonjs/providers/replicate.d.ts.map +1 -1
- package/dist/commonjs/providers/replicate.js +11 -3
- package/dist/commonjs/providers/together.d.ts +5 -4
- package/dist/commonjs/providers/together.d.ts.map +1 -1
- package/dist/commonjs/providers/together.js +11 -8
- package/dist/commonjs/providers/wavespeed.d.ts +14 -4
- package/dist/commonjs/providers/wavespeed.d.ts.map +1 -1
- package/dist/commonjs/providers/wavespeed.js +30 -2
- package/dist/commonjs/providers/zai-org.d.ts +2 -2
- package/dist/commonjs/providers/zai-org.d.ts.map +1 -1
- package/dist/commonjs/providers/zai-org.js +3 -1
- package/dist/commonjs/snippets/getInferenceSnippets.js +11 -20
- package/dist/commonjs/tasks/cv/textToImage.d.ts +5 -2
- package/dist/commonjs/tasks/cv/textToImage.d.ts.map +1 -1
- package/dist/commonjs/types.d.ts +4 -0
- package/dist/commonjs/types.d.ts.map +1 -1
- package/dist/commonjs/utils/dataUrlFromBlob.d.ts +2 -0
- package/dist/commonjs/utils/dataUrlFromBlob.d.ts.map +1 -0
- package/dist/commonjs/utils/dataUrlFromBlob.js +9 -0
- package/dist/esm/lib/getProviderHelper.d.ts.map +1 -1
- package/dist/esm/lib/getProviderHelper.js +8 -4
- package/dist/esm/lib/makeRequestOptions.d.ts +2 -1
- package/dist/esm/lib/makeRequestOptions.d.ts.map +1 -1
- package/dist/esm/lib/makeRequestOptions.js +4 -2
- package/dist/esm/package.d.ts +1 -1
- package/dist/esm/package.d.ts.map +1 -1
- package/dist/esm/package.js +1 -1
- package/dist/esm/providers/fal-ai.d.ts +13 -3
- package/dist/esm/providers/fal-ai.d.ts.map +1 -1
- package/dist/esm/providers/fal-ai.js +46 -1
- package/dist/esm/providers/hf-inference.d.ts +3 -2
- package/dist/esm/providers/hf-inference.d.ts.map +1 -1
- package/dist/esm/providers/hf-inference.js +15 -9
- package/dist/esm/providers/hyperbolic.d.ts +2 -2
- package/dist/esm/providers/hyperbolic.d.ts.map +1 -1
- package/dist/esm/providers/hyperbolic.js +5 -2
- package/dist/esm/providers/nebius.d.ts +5 -4
- package/dist/esm/providers/nebius.d.ts.map +1 -1
- package/dist/esm/providers/nebius.js +11 -8
- package/dist/esm/providers/nscale.d.ts +2 -2
- package/dist/esm/providers/nscale.d.ts.map +1 -1
- package/dist/esm/providers/nscale.js +5 -2
- package/dist/esm/providers/providerHelper.d.ts +3 -3
- package/dist/esm/providers/providerHelper.d.ts.map +1 -1
- package/dist/esm/providers/providerHelper.js +3 -0
- package/dist/esm/providers/replicate.d.ts +2 -2
- package/dist/esm/providers/replicate.d.ts.map +1 -1
- package/dist/esm/providers/replicate.js +11 -3
- package/dist/esm/providers/together.d.ts +5 -4
- package/dist/esm/providers/together.d.ts.map +1 -1
- package/dist/esm/providers/together.js +11 -8
- package/dist/esm/providers/wavespeed.d.ts +14 -4
- package/dist/esm/providers/wavespeed.d.ts.map +1 -1
- package/dist/esm/providers/wavespeed.js +27 -1
- package/dist/esm/providers/zai-org.d.ts +2 -2
- package/dist/esm/providers/zai-org.d.ts.map +1 -1
- package/dist/esm/providers/zai-org.js +3 -1
- package/dist/esm/snippets/getInferenceSnippets.js +11 -20
- package/dist/esm/tasks/cv/textToImage.d.ts +5 -2
- package/dist/esm/tasks/cv/textToImage.d.ts.map +1 -1
- package/dist/esm/types.d.ts +4 -0
- package/dist/esm/types.d.ts.map +1 -1
- package/dist/esm/utils/dataUrlFromBlob.d.ts +2 -0
- package/dist/esm/utils/dataUrlFromBlob.d.ts.map +1 -0
- package/dist/esm/utils/dataUrlFromBlob.js +6 -0
- package/package.json +34 -34
- package/src/InferenceClient.ts +2 -2
- package/src/errors.ts +1 -1
- package/src/lib/getDefaultTask.ts +1 -1
- package/src/lib/getInferenceProviderMapping.ts +11 -11
- package/src/lib/getProviderHelper.ts +45 -41
- package/src/lib/makeRequestOptions.ts +17 -14
- package/src/package.ts +1 -1
- package/src/providers/black-forest-labs.ts +3 -3
- package/src/providers/fal-ai.ts +92 -36
- package/src/providers/featherless-ai.ts +1 -1
- package/src/providers/hf-inference.ts +66 -57
- package/src/providers/hyperbolic.ts +10 -5
- package/src/providers/nebius.ts +19 -13
- package/src/providers/novita.ts +7 -7
- package/src/providers/nscale.ts +9 -4
- package/src/providers/ovhcloud.ts +1 -1
- package/src/providers/providerHelper.ts +19 -8
- package/src/providers/replicate.ts +14 -6
- package/src/providers/sambanova.ts +1 -1
- package/src/providers/together.ts +19 -13
- package/src/providers/wavespeed.ts +56 -12
- package/src/providers/zai-org.ts +11 -9
- package/src/snippets/getInferenceSnippets.ts +26 -26
- package/src/tasks/audio/audioClassification.ts +1 -1
- package/src/tasks/audio/automaticSpeechRecognition.ts +1 -1
- package/src/tasks/audio/utils.ts +1 -1
- package/src/tasks/custom/request.ts +2 -2
- package/src/tasks/custom/streamingRequest.ts +2 -2
- package/src/tasks/cv/imageClassification.ts +1 -1
- package/src/tasks/cv/imageSegmentation.ts +1 -1
- package/src/tasks/cv/textToImage.ts +10 -6
- package/src/tasks/cv/textToVideo.ts +1 -1
- package/src/tasks/cv/zeroShotImageClassification.ts +3 -3
- package/src/tasks/multimodal/documentQuestionAnswering.ts +2 -2
- package/src/tasks/multimodal/visualQuestionAnswering.ts +1 -1
- package/src/tasks/nlp/chatCompletion.ts +1 -1
- package/src/tasks/nlp/chatCompletionStream.ts +1 -1
- package/src/tasks/nlp/featureExtraction.ts +1 -1
- package/src/tasks/nlp/questionAnswering.ts +2 -2
- package/src/tasks/nlp/sentenceSimilarity.ts +1 -1
- package/src/tasks/nlp/tableQuestionAnswering.ts +2 -2
- package/src/tasks/nlp/textClassification.ts +1 -1
- package/src/tasks/nlp/textGeneration.ts +1 -1
- package/src/tasks/nlp/textGenerationStream.ts +1 -1
- package/src/tasks/nlp/tokenClassification.ts +2 -2
- package/src/tasks/nlp/zeroShotClassification.ts +2 -2
- package/src/tasks/tabular/tabularClassification.ts +1 -1
- package/src/tasks/tabular/tabularRegression.ts +1 -1
- package/src/types.ts +5 -0
- package/src/utils/dataUrlFromBlob.ts +7 -0
- package/src/utils/pick.ts +1 -1
- package/src/utils/request.ts +20 -20
- package/src/utils/typedEntries.ts +1 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"getProviderHelper.d.ts","sourceRoot":"","sources":["../../../src/lib/getProviderHelper.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EACX,6BAA6B,EAC7B,sBAAsB,EACtB,oCAAoC,EACpC,wBAAwB,EACxB,mCAAmC,EACnC,2BAA2B,EAC3B,kBAAkB,EAClB,6BAA6B,EAC7B,2BAA2B,EAC3B,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,0BAA0B,EAC1B,0BAA0B,EAC1B,yBAAyB,EACzB,2BAA2B,EAC3B,4BAA4B,EAC5B,uBAAuB,EACvB,gCAAgC,EAChC,+BAA+B,EAC/B,2BAA2B,EAC3B,kBAAkB,EAClB,4BAA4B,EAC5B,wBAAwB,EACxB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,6BAA6B,EAC7B,qBAAqB,EACrB,iCAAiC,EACjC,gCAAgC,EAChC,qCAAqC,EACrC,MAAM,gCAAgC,CAAC;AAOxC,OAAO,KAAK,EAAE,iBAAiB,EAAE,yBAAyB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG/F,eAAO,MAAM,SAAS,EAAE,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,MAAM,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC,
|
|
1
|
+
{"version":3,"file":"getProviderHelper.d.ts","sourceRoot":"","sources":["../../../src/lib/getProviderHelper.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EACX,6BAA6B,EAC7B,sBAAsB,EACtB,oCAAoC,EACpC,wBAAwB,EACxB,mCAAmC,EACnC,2BAA2B,EAC3B,kBAAkB,EAClB,6BAA6B,EAC7B,2BAA2B,EAC3B,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,0BAA0B,EAC1B,0BAA0B,EAC1B,yBAAyB,EACzB,2BAA2B,EAC3B,4BAA4B,EAC5B,uBAAuB,EACvB,gCAAgC,EAChC,+BAA+B,EAC/B,2BAA2B,EAC3B,kBAAkB,EAClB,4BAA4B,EAC5B,wBAAwB,EACxB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,6BAA6B,EAC7B,qBAAqB,EACrB,iCAAiC,EACjC,gCAAgC,EAChC,qCAAqC,EACrC,MAAM,gCAAgC,CAAC;AAOxC,OAAO,KAAK,EAAE,iBAAiB,EAAE,yBAAyB,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAG/F,eAAO,MAAM,SAAS,EAAE,MAAM,CAAC,iBAAiB,EAAE,OAAO,CAAC,MAAM,CAAC,aAAa,EAAE,kBAAkB,CAAC,CAAC,CAmInG,CAAC;AAEF;;GAEG;AACH,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,eAAe,GACnB,qBAAqB,GAAG,kBAAkB,CAAC;AAC9C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,gBAAgB,GACpB,wBAAwB,GAAG,kBAAkB,CAAC;AACjD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,iBAAiB,GACrB,wBAAwB,GAAG,kBAAkB,CAAC;AACjD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,gBAAgB,GACpB,sBAAsB,GAAG,kBAAkB,CAAC;AAC/C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,eAAe,GACnB,qBAAqB,GAAG,kBAAkB,CAAC;AAC9C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,8BAA8B,GAClC,oCAAoC,GAAG,kBAAkB,CAAC;AAC7D,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,eAAe,GACnB,qBAAqB,GAAG,kBAAkB,CAAC;AAC9C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,qBAAqB,GACzB,4BAA4B,GAAG,kBAAkB,CAAC;AACrD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,oBAAoB,GACxB,2BAA2B,GAAG,kBAAkB,CAAC;AACpD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,sBAAsB,GAC1B,6BAA6B,GAAG,kBAAkB,CAAC;AACtD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,gBAAgB,GACpB,sBAAsB,GAAG,kBAAkB,CAAC;AAC/C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,WAAW,GACf,kBAAkB,GAAG,kBAAkB,CAAC;AAC3C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,oBAAoB,GACxB,2BAA2B,GAAG,kBAAkB,CAAC;AACpD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,sBAAsB,GAC1B,6BAA6B,GAAG,kBAAkB,CAAC;AACtD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,oBAAoB,GACxB,2BAA2B,GAAG,kBAAkB,CAAC;AACpD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,6BAA6B,GACjC,mCAAmC,GAAG,kBAAkB,CAAC;AAC5D,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,eAAe,GACnB,qBAAqB,GAAG,kBAAkB,CAAC;AAC9C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,kBAAkB,GACtB,yBAAyB,GAAG,kBAAkB,CAAC;AAClD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,gCAAgC,GACpC,qCAAqC,GAAG,kBAAkB,CAAC;AAC9D,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,0BAA0B,GAC9B,gCAAgC,GAAG,kBAAkB,CAAC;AACzD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,gBAAgB,GACpB,sBAAsB,GAAG,kBAAkB,CAAC;AAC/C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,gBAAgB,GACpB,sBAAsB,GAAG,kBAAkB,CAAC;AAC/C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,qBAAqB,GACzB,0BAA0B,GAAG,kBAAkB,CAAC;AACnD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,qBAAqB,GACzB,0BAA0B,GAAG,kBAAkB,CAAC;AACnD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,qBAAqB,GACzB,4BAA4B,GAAG,kBAAkB,CAAC;AACrD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,0BAA0B,GAC9B,gCAAgC,GAAG,kBAAkB,CAAC;AACzD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,wBAAwB,GAC5B,+BAA+B,GAAG,kBAAkB,CAAC;AACxD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,oBAAoB,GACxB,2BAA2B,GAAG,kBAAkB,CAAC;AACpD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,sBAAsB,GAC1B,6BAA6B,GAAG,kBAAkB,CAAC;AACtD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,aAAa,GACjB,qBAAqB,GAAG,kBAAkB,CAAC;AAC9C,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,eAAe,GACnB,uBAAuB,GAAG,kBAAkB,CAAC;AAChD,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,2BAA2B,GAC/B,iCAAiC,GAAG,kBAAkB,CAAC;AAC1D,wBAAgB,iBAAiB,CAChC,QAAQ,EAAE,yBAAyB,EACnC,IAAI,EAAE,aAAa,GAAG,SAAS,GAC7B,kBAAkB,CAAC"}
|
|
@@ -76,13 +76,15 @@ exports.PROVIDERS = {
|
|
|
76
76
|
conversational: new Cohere.CohereConversationalTask(),
|
|
77
77
|
},
|
|
78
78
|
"fal-ai": {
|
|
79
|
-
"text-to-image": new FalAI.FalAITextToImageTask(),
|
|
80
|
-
"text-to-speech": new FalAI.FalAITextToSpeechTask(),
|
|
81
|
-
"text-to-video": new FalAI.FalAITextToVideoTask(),
|
|
82
|
-
"image-to-image": new FalAI.FalAIImageToImageTask(),
|
|
83
79
|
"automatic-speech-recognition": new FalAI.FalAIAutomaticSpeechRecognitionTask(),
|
|
80
|
+
"image-text-to-image": new FalAI.FalAIImageTextToImageTask(),
|
|
81
|
+
"image-text-to-video": new FalAI.FalAIImageTextToVideoTask(),
|
|
82
|
+
"image-to-image": new FalAI.FalAIImageToImageTask(),
|
|
84
83
|
"image-segmentation": new FalAI.FalAIImageSegmentationTask(),
|
|
85
84
|
"image-to-video": new FalAI.FalAIImageToVideoTask(),
|
|
85
|
+
"text-to-image": new FalAI.FalAITextToImageTask(),
|
|
86
|
+
"text-to-speech": new FalAI.FalAITextToSpeechTask(),
|
|
87
|
+
"text-to-video": new FalAI.FalAITextToVideoTask(),
|
|
86
88
|
},
|
|
87
89
|
"featherless-ai": {
|
|
88
90
|
conversational: new FeatherlessAI.FeatherlessAIConversationalTask(),
|
|
@@ -181,6 +183,8 @@ exports.PROVIDERS = {
|
|
|
181
183
|
"text-to-video": new Wavespeed.WavespeedAITextToVideoTask(),
|
|
182
184
|
"image-to-image": new Wavespeed.WavespeedAIImageToImageTask(),
|
|
183
185
|
"image-to-video": new Wavespeed.WavespeedAIImageToVideoTask(),
|
|
186
|
+
"image-text-to-image": new Wavespeed.WavespeedAIImageTextToImageTask(),
|
|
187
|
+
"image-text-to-video": new Wavespeed.WavespeedAIImageTextToVideoTask(),
|
|
184
188
|
},
|
|
185
189
|
"zai-org": {
|
|
186
190
|
conversational: new Zai.ZaiConversationalTask(),
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { InferenceTask, InferenceProviderMappingEntry, Options, RequestArgs } from "../types.js";
|
|
1
|
+
import type { InferenceTask, InferenceProviderMappingEntry, Options, OutputType, RequestArgs } from "../types.js";
|
|
2
2
|
import type { getProviderHelper } from "./getProviderHelper.js";
|
|
3
3
|
/**
|
|
4
4
|
* Helper that prepares request arguments.
|
|
@@ -23,6 +23,7 @@ export declare function makeRequestOptionsFromResolvedModel(resolvedModel: strin
|
|
|
23
23
|
stream?: boolean;
|
|
24
24
|
}, mapping: InferenceProviderMappingEntry | undefined, options?: Options & {
|
|
25
25
|
task?: InferenceTask;
|
|
26
|
+
outputType?: OutputType;
|
|
26
27
|
}): {
|
|
27
28
|
url: string;
|
|
28
29
|
info: RequestInit;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"makeRequestOptions.d.ts","sourceRoot":"","sources":["../../../src/lib/makeRequestOptions.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,6BAA6B,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"makeRequestOptions.d.ts","sourceRoot":"","sources":["../../../src/lib/makeRequestOptions.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,aAAa,EAAE,6BAA6B,EAAE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAElH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAUhE;;;GAGG;AACH,wBAAsB,kBAAkB,CACvC,IAAI,EAAE,WAAW,GAAG;IACnB,IAAI,CAAC,EAAE,IAAI,GAAG,WAAW,CAAC;IAC1B,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB,EACD,cAAc,EAAE,UAAU,CAAC,OAAO,iBAAiB,CAAC,EACpD,OAAO,CAAC,EAAE,OAAO,GAAG;IACnB,oEAAoE;IACpE,IAAI,CAAC,EAAE,aAAa,CAAC;CACrB,GACC,OAAO,CAAC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,WAAW,CAAA;CAAE,CAAC,CAsE7C;AAED;;;GAGG;AACH,wBAAgB,mCAAmC,CAClD,aAAa,EAAE,MAAM,EACrB,cAAc,EAAE,UAAU,CAAC,OAAO,iBAAiB,CAAC,EACpD,IAAI,EAAE,WAAW,GAAG;IACnB,IAAI,CAAC,EAAE,IAAI,GAAG,WAAW,CAAC;IAC1B,MAAM,CAAC,EAAE,OAAO,CAAC;CACjB,EACD,OAAO,EAAE,6BAA6B,GAAG,SAAS,EAClD,OAAO,CAAC,EAAE,OAAO,GAAG;IACnB,IAAI,CAAC,EAAE,aAAa,CAAC;IACrB,UAAU,CAAC,EAAE,UAAU,CAAC;CACxB,GACC;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,WAAW,CAAA;CAAE,CAgFpC"}
|
|
@@ -68,11 +68,11 @@ async function makeRequestOptions(args, providerHelper, options) {
|
|
|
68
68
|
* This sync version skips the model ID resolution step
|
|
69
69
|
*/
|
|
70
70
|
function makeRequestOptionsFromResolvedModel(resolvedModel, providerHelper, args, mapping, options) {
|
|
71
|
-
const { accessToken, endpointUrl, provider: maybeProvider, model, ...remainingArgs } = args;
|
|
71
|
+
const { accessToken, endpointUrl, provider: maybeProvider, model, urlTransform, ...remainingArgs } = args;
|
|
72
72
|
void model;
|
|
73
73
|
void maybeProvider;
|
|
74
74
|
const provider = providerHelper.provider;
|
|
75
|
-
const { includeCredentials, task, signal, billTo } = options ?? {};
|
|
75
|
+
const { includeCredentials, task, signal, billTo, outputType } = options ?? {};
|
|
76
76
|
const authMethod = (() => {
|
|
77
77
|
if (providerHelper.clientSideRoutingOnly) {
|
|
78
78
|
// Closed-source providers require an accessToken (cannot be routed).
|
|
@@ -95,6 +95,7 @@ function makeRequestOptionsFromResolvedModel(resolvedModel, providerHelper, args
|
|
|
95
95
|
authMethod,
|
|
96
96
|
model: modelId,
|
|
97
97
|
task,
|
|
98
|
+
urlTransform,
|
|
98
99
|
});
|
|
99
100
|
// Make headers
|
|
100
101
|
const headers = providerHelper.prepareHeaders({
|
|
@@ -117,6 +118,7 @@ function makeRequestOptionsFromResolvedModel(resolvedModel, providerHelper, args
|
|
|
117
118
|
model: resolvedModel,
|
|
118
119
|
task,
|
|
119
120
|
mapping,
|
|
121
|
+
outputType,
|
|
120
122
|
});
|
|
121
123
|
/**
|
|
122
124
|
* For edge runtimes, leave 'credentials' undefined, otherwise cloudflare workers will error
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"package.d.ts","sourceRoot":"","sources":["../../src/package.ts"],"names":[],"mappings":"AACA,eAAO,MAAM,eAAe,
|
|
1
|
+
{"version":3,"file":"package.d.ts","sourceRoot":"","sources":["../../src/package.ts"],"names":[],"mappings":"AACA,eAAO,MAAM,eAAe,YAAY,CAAC;AACzC,eAAO,MAAM,YAAY,2BAA2B,CAAC"}
|
package/dist/commonjs/package.js
CHANGED
|
@@ -2,5 +2,5 @@
|
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
3
|
exports.PACKAGE_NAME = exports.PACKAGE_VERSION = void 0;
|
|
4
4
|
// Generated file from package.json. Issues importing JSON directly when publishing on commonjs/ESM - see https://github.com/microsoft/TypeScript/issues/51783
|
|
5
|
-
exports.PACKAGE_VERSION = "4.13.
|
|
5
|
+
exports.PACKAGE_VERSION = "4.13.11";
|
|
6
6
|
exports.PACKAGE_NAME = "@huggingface/inference";
|
|
@@ -1,9 +1,11 @@
|
|
|
1
1
|
import type { AutomaticSpeechRecognitionOutput, ImageSegmentationOutput } from "@huggingface/tasks";
|
|
2
|
-
import type { BodyParams, HeaderParams, InferenceTask, RequestArgs, UrlParams } from "../types.js";
|
|
3
|
-
import type { ImageSegmentationTaskHelper, ImageToImageTaskHelper } from "./providerHelper.js";
|
|
2
|
+
import type { BodyParams, HeaderParams, InferenceTask, OutputType, RequestArgs, UrlParams } from "../types.js";
|
|
3
|
+
import type { ImageSegmentationTaskHelper, ImageToImageTaskHelper, ImageTextToImageTaskHelper, ImageTextToVideoTaskHelper } from "./providerHelper.js";
|
|
4
4
|
import { type AutomaticSpeechRecognitionTaskHelper, TaskProviderHelper, type TextToImageTaskHelper, type TextToVideoTaskHelper, type ImageToVideoTaskHelper } from "./providerHelper.js";
|
|
5
5
|
import type { AutomaticSpeechRecognitionArgs } from "../tasks/audio/automaticSpeechRecognition.js";
|
|
6
6
|
import type { ImageToImageArgs, ImageToVideoArgs } from "../tasks/index.js";
|
|
7
|
+
import type { ImageTextToImageArgs } from "../tasks/cv/imageTextToImage.js";
|
|
8
|
+
import type { ImageTextToVideoArgs } from "../tasks/cv/imageTextToVideo.js";
|
|
7
9
|
import type { ImageSegmentationArgs } from "../tasks/cv/imageSegmentation.js";
|
|
8
10
|
export interface FalAiQueueOutput {
|
|
9
11
|
request_id: string;
|
|
@@ -26,7 +28,7 @@ export declare class FalAITextToImageTask extends FalAiQueueTask implements Text
|
|
|
26
28
|
task: InferenceTask;
|
|
27
29
|
constructor();
|
|
28
30
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
29
|
-
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>, outputType?:
|
|
31
|
+
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
30
32
|
}
|
|
31
33
|
export declare class FalAIImageToImageTask extends FalAiQueueTask implements ImageToImageTaskHelper {
|
|
32
34
|
task: InferenceTask;
|
|
@@ -35,6 +37,10 @@ export declare class FalAIImageToImageTask extends FalAiQueueTask implements Ima
|
|
|
35
37
|
preparePayloadAsync(args: ImageToImageArgs): Promise<RequestArgs>;
|
|
36
38
|
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<Blob>;
|
|
37
39
|
}
|
|
40
|
+
export declare class FalAIImageTextToImageTask extends FalAIImageToImageTask implements ImageTextToImageTaskHelper {
|
|
41
|
+
constructor();
|
|
42
|
+
preparePayloadAsync(args: ImageTextToImageArgs): Promise<RequestArgs>;
|
|
43
|
+
}
|
|
38
44
|
export declare class FalAITextToVideoTask extends FalAiQueueTask implements TextToVideoTaskHelper {
|
|
39
45
|
task: InferenceTask;
|
|
40
46
|
constructor();
|
|
@@ -51,6 +57,10 @@ export declare class FalAIImageToVideoTask extends FalAiQueueTask implements Ima
|
|
|
51
57
|
/** Queue polling + final download – mirrors Text‑to‑Video */
|
|
52
58
|
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<Blob>;
|
|
53
59
|
}
|
|
60
|
+
export declare class FalAIImageTextToVideoTask extends FalAIImageToVideoTask implements ImageTextToVideoTaskHelper {
|
|
61
|
+
constructor();
|
|
62
|
+
preparePayloadAsync(args: ImageTextToVideoArgs): Promise<RequestArgs>;
|
|
63
|
+
}
|
|
54
64
|
export declare class FalAIAutomaticSpeechRecognitionTask extends FalAITask implements AutomaticSpeechRecognitionTaskHelper {
|
|
55
65
|
prepareHeaders(params: HeaderParams, binary: boolean): Record<string, string>;
|
|
56
66
|
getResponse(response: unknown): Promise<AutomaticSpeechRecognitionOutput>;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"fal-ai.d.ts","sourceRoot":"","sources":["../../../src/providers/fal-ai.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"fal-ai.d.ts","sourceRoot":"","sources":["../../../src/providers/fal-ai.ts"],"names":[],"mappings":"AAmBA,OAAO,KAAK,EAAE,gCAAgC,EAAE,uBAAuB,EAAE,MAAM,oBAAoB,CAAC;AAEpG,OAAO,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,aAAa,EAAW,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxH,OAAO,KAAK,EACX,2BAA2B,EAC3B,sBAAsB,EACtB,0BAA0B,EAC1B,0BAA0B,EAC1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACN,KAAK,oCAAoC,EACzC,kBAAkB,EAClB,KAAK,qBAAqB,EAC1B,KAAK,qBAAqB,EAC1B,KAAK,sBAAsB,EAC3B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,8CAA8C,CAAC;AAMnG,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAC5E,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AAC5E,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,iCAAiC,CAAC;AAC5E,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AAE9E,MAAM,WAAW,gBAAgB;IAChC,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;CACrB;AAkBD,eAAO,MAAM,2BAA2B,UAA0D,CAAC;AAEnG,uBAAe,SAAU,SAAQ,kBAAkB;gBACtC,GAAG,CAAC,EAAE,MAAM;IAIxB,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAG3D,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAG3B,cAAc,CAAC,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;CAUtF;AAED,uBAAe,cAAe,SAAQ,SAAS;IAC9C,QAAQ,CAAC,IAAI,EAAE,aAAa,CAAC;IAEpB,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAMvC,uBAAuB,CAC5B,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,OAAO,CAAC;CA4DnB;AAMD,qBAAa,oBAAqB,SAAQ,cAAe,YAAW,qBAAqB;IACxF,IAAI,EAAE,aAAa,CAAC;;IAOX,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAsBrD,WAAW,CACzB,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAChC,UAAU,CAAC,EAAE,UAAU,GACrB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CA4BnD;AAED,qBAAa,qBAAsB,SAAQ,cAAe,YAAW,sBAAsB;IAC1F,IAAI,EAAE,aAAa,CAAC;;IAMX,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAa9D,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IAexD,WAAW,CACzB,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAyBhB;AAED,qBAAa,yBAA0B,SAAQ,qBAAsB,YAAW,0BAA0B;;IAM1F,mBAAmB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,WAAW,CAAC;CAgBpF;AAED,qBAAa,oBAAqB,SAAQ,cAAe,YAAW,qBAAqB;IACxF,IAAI,EAAE,aAAa,CAAC;;IAMX,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAQrD,WAAW,CACzB,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAuBhB;AAED,qBAAa,qBAAsB,SAAQ,cAAe,YAAW,sBAAsB;IAC1F,IAAI,EAAE,aAAa,CAAC;;IAOpB,gEAAgE;IACvD,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IASpE,kDAAkD;IAC5C,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IAYvE,6DAA6D;IAC9C,WAAW,CACzB,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,IAAI,CAAC;CAwBhB;AAED,qBAAa,yBAA0B,SAAQ,qBAAsB,YAAW,0BAA0B;;IAM1F,mBAAmB,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,WAAW,CAAC;CAepF;AAED,qBAAa,mCAAoC,SAAQ,SAAU,YAAW,oCAAoC;IACxG,cAAc,CAAC,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAKvE,WAAW,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC,gCAAgC,CAAC;IAYlF,mBAAmB,CAAC,IAAI,EAAE,8BAA8B,GAAG,OAAO,CAAC,WAAW,CAAC;CAqBrF;AAED,qBAAa,qBAAsB,SAAQ,SAAS;IAC1C,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAQrD,WAAW,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;CAmC5D;AACD,qBAAa,0BAA2B,SAAQ,cAAe,YAAW,2BAA2B;IACpG,IAAI,EAAE,aAAa,CAAC;;IAMX,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAQ9D,mBAAmB,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,WAAW,CAAC;IAe7D,WAAW,CACzB,QAAQ,EAAE,gBAAgB,EAC1B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAC9B,OAAO,CAAC,uBAAuB,CAAC;CA0CnC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.FalAIImageSegmentationTask = exports.FalAITextToSpeechTask = exports.FalAIAutomaticSpeechRecognitionTask = exports.FalAIImageToVideoTask = exports.FalAITextToVideoTask = exports.FalAIImageToImageTask = exports.FalAITextToImageTask = exports.FAL_AI_SUPPORTED_BLOB_TYPES = void 0;
|
|
3
|
+
exports.FalAIImageSegmentationTask = exports.FalAITextToSpeechTask = exports.FalAIAutomaticSpeechRecognitionTask = exports.FalAIImageTextToVideoTask = exports.FalAIImageToVideoTask = exports.FalAITextToVideoTask = exports.FalAIImageTextToImageTask = exports.FalAIImageToImageTask = exports.FalAITextToImageTask = exports.FAL_AI_SUPPORTED_BLOB_TYPES = void 0;
|
|
4
4
|
/**
|
|
5
5
|
* See the registered mapping of HF model ID => Fal model ID here:
|
|
6
6
|
*
|
|
@@ -18,6 +18,7 @@ exports.FalAIImageSegmentationTask = exports.FalAITextToSpeechTask = exports.Fal
|
|
|
18
18
|
* Thanks!
|
|
19
19
|
*/
|
|
20
20
|
const base64FromBytes_js_1 = require("../utils/base64FromBytes.js");
|
|
21
|
+
const dataUrlFromBlob_js_1 = require("../utils/dataUrlFromBlob.js");
|
|
21
22
|
const isUrl_js_1 = require("../lib/isUrl.js");
|
|
22
23
|
const delay_js_1 = require("../utils/delay.js");
|
|
23
24
|
const omit_js_1 = require("../utils/omit.js");
|
|
@@ -141,7 +142,8 @@ class FalAITextToImageTask extends FalAiQueueTask {
|
|
|
141
142
|
return result.images[0].url;
|
|
142
143
|
}
|
|
143
144
|
const urlResponse = await fetch(result.images[0].url);
|
|
144
|
-
|
|
145
|
+
const blob = await urlResponse.blob();
|
|
146
|
+
return outputType === "dataUrl" ? (0, dataUrlFromBlob_js_1.dataUrlFromBlob)(blob) : blob;
|
|
145
147
|
}
|
|
146
148
|
throw new errors_js_1.InferenceClientProviderOutputError(`Received malformed response from Fal.ai text-to-image API: expected { images: Array<{ url: string }> } result format, got instead: ${JSON.stringify(result)}`);
|
|
147
149
|
}
|
|
@@ -198,6 +200,29 @@ class FalAIImageToImageTask extends FalAiQueueTask {
|
|
|
198
200
|
}
|
|
199
201
|
}
|
|
200
202
|
exports.FalAIImageToImageTask = FalAIImageToImageTask;
|
|
203
|
+
class FalAIImageTextToImageTask extends FalAIImageToImageTask {
|
|
204
|
+
constructor() {
|
|
205
|
+
super();
|
|
206
|
+
this.task = "image-text-to-image";
|
|
207
|
+
}
|
|
208
|
+
async preparePayloadAsync(args) {
|
|
209
|
+
if (args.inputs) {
|
|
210
|
+
return super.preparePayloadAsync(args);
|
|
211
|
+
}
|
|
212
|
+
return {
|
|
213
|
+
...(0, omit_js_1.omit)(args, ["inputs", "parameters"]),
|
|
214
|
+
...args.parameters,
|
|
215
|
+
prompt: args.parameters?.prompt,
|
|
216
|
+
urlTransform: (url) => {
|
|
217
|
+
const urlObj = new URL(url);
|
|
218
|
+
// Strip last path segment: fal-ai/flux-2/edit => fal-ai/flux-2
|
|
219
|
+
urlObj.pathname = urlObj.pathname.split("/").slice(0, -1).join("/");
|
|
220
|
+
return urlObj.toString();
|
|
221
|
+
},
|
|
222
|
+
};
|
|
223
|
+
}
|
|
224
|
+
}
|
|
225
|
+
exports.FalAIImageTextToImageTask = FalAIImageTextToImageTask;
|
|
201
226
|
class FalAITextToVideoTask extends FalAiQueueTask {
|
|
202
227
|
task;
|
|
203
228
|
constructor() {
|
|
@@ -274,6 +299,28 @@ class FalAIImageToVideoTask extends FalAiQueueTask {
|
|
|
274
299
|
}
|
|
275
300
|
}
|
|
276
301
|
exports.FalAIImageToVideoTask = FalAIImageToVideoTask;
|
|
302
|
+
class FalAIImageTextToVideoTask extends FalAIImageToVideoTask {
|
|
303
|
+
constructor() {
|
|
304
|
+
super();
|
|
305
|
+
this.task = "image-text-to-video";
|
|
306
|
+
}
|
|
307
|
+
async preparePayloadAsync(args) {
|
|
308
|
+
if (args.inputs) {
|
|
309
|
+
return super.preparePayloadAsync(args);
|
|
310
|
+
}
|
|
311
|
+
return {
|
|
312
|
+
...(0, omit_js_1.omit)(args, ["inputs", "parameters"]),
|
|
313
|
+
...args.parameters,
|
|
314
|
+
prompt: args.parameters?.prompt,
|
|
315
|
+
urlTransform: (url) => {
|
|
316
|
+
const urlObj = new URL(url);
|
|
317
|
+
urlObj.pathname = urlObj.pathname.split("/").slice(0, -1).join("/");
|
|
318
|
+
return urlObj.toString();
|
|
319
|
+
},
|
|
320
|
+
};
|
|
321
|
+
}
|
|
322
|
+
}
|
|
323
|
+
exports.FalAIImageTextToVideoTask = FalAIImageTextToVideoTask;
|
|
277
324
|
class FalAIAutomaticSpeechRecognitionTask extends FalAITask {
|
|
278
325
|
prepareHeaders(params, binary) {
|
|
279
326
|
const headers = super.prepareHeaders(params, binary);
|
|
@@ -12,7 +12,7 @@
|
|
|
12
12
|
*/
|
|
13
13
|
import type { AudioClassificationOutput, AutomaticSpeechRecognitionOutput, ChatCompletionOutput, DocumentQuestionAnsweringOutput, FeatureExtractionOutput, FillMaskOutput, ImageClassificationOutput, ImageSegmentationOutput, ImageToTextOutput, ObjectDetectionOutput, QuestionAnsweringOutput, QuestionAnsweringOutputElement, SentenceSimilarityOutput, SummarizationOutput, TableQuestionAnsweringOutput, TextClassificationOutput, TextGenerationOutput, TokenClassificationOutput, TranslationOutput, VisualQuestionAnsweringOutput, ZeroShotClassificationOutput, ZeroShotImageClassificationOutput } from "@huggingface/tasks";
|
|
14
14
|
import type { TabularClassificationOutput } from "../tasks/tabular/tabularClassification.js";
|
|
15
|
-
import type { BodyParams, RequestArgs, UrlParams } from "../types.js";
|
|
15
|
+
import type { BodyParams, OutputType, RequestArgs, UrlParams } from "../types.js";
|
|
16
16
|
import type { AudioClassificationTaskHelper, AudioToAudioTaskHelper, AutomaticSpeechRecognitionTaskHelper, ConversationalTaskHelper, DocumentQuestionAnsweringTaskHelper, FeatureExtractionTaskHelper, FillMaskTaskHelper, ImageClassificationTaskHelper, ImageSegmentationTaskHelper, ImageToImageTaskHelper, ImageToTextTaskHelper, ObjectDetectionTaskHelper, QuestionAnsweringTaskHelper, SentenceSimilarityTaskHelper, SummarizationTaskHelper, TableQuestionAnsweringTaskHelper, TabularClassificationTaskHelper, TabularRegressionTaskHelper, TextClassificationTaskHelper, TextGenerationTaskHelper, TextToAudioTaskHelper, TextToImageTaskHelper, TextToSpeechTaskHelper, TokenClassificationTaskHelper, TranslationTaskHelper, VisualQuestionAnsweringTaskHelper, ZeroShotClassificationTaskHelper, ZeroShotImageClassificationTaskHelper } from "./providerHelper.js";
|
|
17
17
|
import { TaskProviderHelper } from "./providerHelper.js";
|
|
18
18
|
import type { ImageToImageArgs } from "../tasks/cv/imageToImage.js";
|
|
@@ -40,7 +40,8 @@ export declare class HFInferenceTask extends TaskProviderHelper {
|
|
|
40
40
|
getResponse(response: unknown): Promise<unknown>;
|
|
41
41
|
}
|
|
42
42
|
export declare class HFInferenceTextToImageTask extends HFInferenceTask implements TextToImageTaskHelper {
|
|
43
|
-
|
|
43
|
+
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
44
|
+
getResponse(response: Base64ImageGeneration | OutputUrlImageGeneration, url?: string, headers?: HeadersInit, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
44
45
|
}
|
|
45
46
|
export declare class HFInferenceConversationalTask extends HFInferenceTask implements ConversationalTaskHelper {
|
|
46
47
|
makeUrl(params: UrlParams): string;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hf-inference.d.ts","sourceRoot":"","sources":["../../../src/providers/hf-inference.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,KAAK,EACX,yBAAyB,EACzB,gCAAgC,EAChC,oBAAoB,EACpB,+BAA+B,EAC/B,uBAAuB,EACvB,cAAc,EACd,yBAAyB,EACzB,uBAAuB,EACvB,iBAAiB,EACjB,qBAAqB,EACrB,uBAAuB,EACvB,8BAA8B,EAC9B,wBAAwB,EACxB,mBAAmB,EACnB,4BAA4B,EAC5B,wBAAwB,EACxB,oBAAoB,EACpB,yBAAyB,EACzB,iBAAiB,EACjB,6BAA6B,EAC7B,4BAA4B,EAE5B,iCAAiC,EACjC,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,2CAA2C,CAAC;AAC7F,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"hf-inference.d.ts","sourceRoot":"","sources":["../../../src/providers/hf-inference.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AACH,OAAO,KAAK,EACX,yBAAyB,EACzB,gCAAgC,EAChC,oBAAoB,EACpB,+BAA+B,EAC/B,uBAAuB,EACvB,cAAc,EACd,yBAAyB,EACzB,uBAAuB,EACvB,iBAAiB,EACjB,qBAAqB,EACrB,uBAAuB,EACvB,8BAA8B,EAC9B,wBAAwB,EACxB,mBAAmB,EACnB,4BAA4B,EAC5B,wBAAwB,EACxB,oBAAoB,EACpB,yBAAyB,EACzB,iBAAiB,EACjB,6BAA6B,EAC7B,4BAA4B,EAE5B,iCAAiC,EACjC,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,2BAA2B,EAAE,MAAM,2CAA2C,CAAC;AAC7F,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAElF,OAAO,KAAK,EACX,6BAA6B,EAC7B,sBAAsB,EACtB,oCAAoC,EACpC,wBAAwB,EACxB,mCAAmC,EACnC,2BAA2B,EAC3B,kBAAkB,EAClB,6BAA6B,EAC7B,2BAA2B,EAC3B,sBAAsB,EACtB,qBAAqB,EACrB,yBAAyB,EACzB,2BAA2B,EAC3B,4BAA4B,EAC5B,uBAAuB,EACvB,gCAAgC,EAChC,+BAA+B,EAC/B,2BAA2B,EAC3B,4BAA4B,EAC5B,wBAAwB,EACxB,qBAAqB,EACrB,qBAAqB,EACrB,sBAAsB,EACtB,6BAA6B,EAC7B,qBAAqB,EACrB,iCAAiC,EACjC,gCAAgC,EAChC,qCAAqC,EACrC,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAGzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,8BAA8B,EAAE,MAAM,8CAA8C,CAAC;AAEnG,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AAC9E,UAAU,qBAAqB;IAC9B,IAAI,EAAE,KAAK,CAAC;QACX,QAAQ,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;CACH;AAED,UAAU,wBAAwB;IACjC,MAAM,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,UAAU,kBAAkB;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,cAAc,EAAE,MAAM,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;CACd;AAED,eAAO,MAAM,sCAAsC,wDAAyD,CAAC;AAE7G,qBAAa,eAAgB,SAAQ,kBAAkB;;IAItD,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAGlD,OAAO,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAO3C,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAQrB,WAAW,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;CAG/D;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;IACtF,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IASrD,WAAW,CACzB,QAAQ,EAAE,qBAAqB,GAAG,wBAAwB,EAC1D,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,WAAW,EACrB,UAAU,CAAC,EAAE,UAAU,GACrB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAqCnD;AAED,qBAAa,6BAA8B,SAAQ,eAAgB,YAAW,wBAAwB;IAC5F,OAAO,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAkBlC,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAOrD,WAAW,CAAC,QAAQ,EAAE,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,CAAC;CAGzF;AAED,qBAAa,6BAA8B,SAAQ,eAAgB,YAAW,wBAAwB;IACtF,WAAW,CAAC,QAAQ,EAAE,oBAAoB,GAAG,oBAAoB,EAAE,GAAG,OAAO,CAAC,oBAAoB,CAAC;CASlH;AAED,qBAAa,kCAAmC,SAAQ,eAAgB,YAAW,6BAA6B;IAChG,WAAW,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAcjF;AAED,qBAAa,yCACZ,SAAQ,eACR,YAAW,oCAAoC;IAEhC,WAAW,CAAC,QAAQ,EAAE,gCAAgC,GAAG,OAAO,CAAC,gCAAgC,CAAC;IAI3G,mBAAmB,CAAC,IAAI,EAAE,8BAA8B,GAAG,OAAO,CAAC,WAAW,CAAC;CAQrF;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;IAClF,WAAW,CAAC,QAAQ,EAAE,kBAAkB,EAAE,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;CA0BzF;AAED,qBAAa,wCACZ,SAAQ,eACR,YAAW,mCAAmC;IAE/B,WAAW,CACzB,QAAQ,EAAE,+BAA+B,GACvC,OAAO,CAAC,+BAA+B,CAAC,MAAM,CAAC,CAAC;CAmBnD;AAED,qBAAa,gCAAiC,SAAQ,eAAgB,YAAW,2BAA2B;IAC5F,WAAW,CAAC,QAAQ,EAAE,uBAAuB,GAAG,OAAO,CAAC,uBAAuB,CAAC;CAgB/F;AAED,qBAAa,kCAAmC,SAAQ,eAAgB,YAAW,6BAA6B;IAChG,WAAW,CAAC,QAAQ,EAAE,yBAAyB,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAQnG;AAED,qBAAa,gCAAiC,SAAQ,eAAgB,YAAW,2BAA2B;IAC5F,WAAW,CAAC,QAAQ,EAAE,uBAAuB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAiBzF,mBAAmB,CAAC,IAAI,EAAE,qBAAqB,GAAG,OAAO,CAAC,WAAW,CAAC;CAQ5E;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;IAChF,WAAW,CAAC,QAAQ,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;CAQnF;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;IAC3F,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,WAAW,CAAC;IAiBxD,WAAW,CAAC,QAAQ,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;CAQzD;AAED,qBAAa,8BAA+B,SAAQ,eAAgB,YAAW,yBAAyB;IACxF,WAAW,CAAC,QAAQ,EAAE,qBAAqB,GAAG,OAAO,CAAC,qBAAqB,CAAC;CAmB3F;AAED,qBAAa,0CACZ,SAAQ,eACR,YAAW,qCAAqC;IAEjC,WAAW,CAAC,QAAQ,EAAE,iCAAiC,GAAG,OAAO,CAAC,iCAAiC,CAAC;CAQnH;AAED,qBAAa,iCAAkC,SAAQ,eAAgB,YAAW,4BAA4B;IAC9F,WAAW,CAAC,QAAQ,EAAE,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;CASjG;AAED,qBAAa,gCAAiC,SAAQ,eAAgB,YAAW,2BAA2B;IAC5F,WAAW,CACzB,QAAQ,EAAE,uBAAuB,GAAG,uBAAuB,CAAC,MAAM,CAAC,GACjE,OAAO,CAAC,8BAA8B,CAAC;CAyB1C;AAED,qBAAa,uBAAwB,SAAQ,eAAgB,YAAW,kBAAkB;IAC1E,WAAW,CAAC,QAAQ,EAAE,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;CAiB7E;AAED,qBAAa,qCAAsC,SAAQ,eAAgB,YAAW,gCAAgC;IACtG,WAAW,CAAC,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAC,4BAA4B,CAAC;IA4BpF,OAAO,CAAC,MAAM,CAAC,qBAAqB;CAUpC;AAED,qBAAa,iCAAkC,SAAQ,eAAgB,YAAW,4BAA4B;IAC9F,WAAW,CAAC,QAAQ,EAAE,wBAAwB,GAAG,OAAO,CAAC,wBAAwB,CAAC;CAQjG;AAED,qBAAa,qCAAsC,SAAQ,eAAgB,YAAW,gCAAgC;IACrH,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,GAAG,IAAI,IAAI,4BAA4B,CAAC,MAAM,CAAC;IAkB7D,WAAW,CAAC,QAAQ,EAAE,4BAA4B,GAAG,OAAO,CAAC,4BAA4B,CAAC,MAAM,CAAC,CAAC;CAYjH;AAED,qBAAa,kCAAmC,SAAQ,eAAgB,YAAW,6BAA6B;IAChG,WAAW,CAAC,QAAQ,EAAE,yBAAyB,GAAG,OAAO,CAAC,yBAAyB,CAAC;CAkBnG;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;IAChF,WAAW,CAAC,QAAQ,EAAE,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,CAAC;CAQnF;AAED,qBAAa,4BAA6B,SAAQ,eAAgB,YAAW,uBAAuB;IACpF,WAAW,CAAC,QAAQ,EAAE,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,CAAC;CAQvF;AAED,qBAAa,2BAA4B,SAAQ,eAAgB,YAAW,sBAAsB;IAClF,WAAW,CAAC,QAAQ,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;CAGzD;AAED,qBAAa,oCAAqC,SAAQ,eAAgB,YAAW,+BAA+B;IACpG,WAAW,CAAC,QAAQ,EAAE,2BAA2B,GAAG,OAAO,CAAC,2BAA2B,CAAC;CAQvG;AAED,qBAAa,sCACZ,SAAQ,eACR,YAAW,iCAAiC;IAE7B,WAAW,CAAC,QAAQ,EAAE,6BAA6B,GAAG,OAAO,CAAC,6BAA6B,CAAC,MAAM,CAAC,CAAC;CAcnH;AAED,qBAAa,gCAAiC,SAAQ,eAAgB,YAAW,2BAA2B;IAC5F,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;CAQjE;AAED,qBAAa,0BAA2B,SAAQ,eAAgB,YAAW,qBAAqB;IAChF,WAAW,CAAC,QAAQ,EAAE,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;CAGzD"}
|
|
@@ -6,6 +6,7 @@ const errors_js_1 = require("../errors.js");
|
|
|
6
6
|
const toArray_js_1 = require("../utils/toArray.js");
|
|
7
7
|
const providerHelper_js_1 = require("./providerHelper.js");
|
|
8
8
|
const base64FromBytes_js_1 = require("../utils/base64FromBytes.js");
|
|
9
|
+
const dataUrlFromBlob_js_1 = require("../utils/dataUrlFromBlob.js");
|
|
9
10
|
const omit_js_1 = require("../utils/omit.js");
|
|
10
11
|
exports.EQUIVALENT_SENTENCE_TRANSFORMERS_TASKS = ["feature-extraction", "sentence-similarity"];
|
|
11
12
|
class HFInferenceTask extends providerHelper_js_1.TaskProviderHelper {
|
|
@@ -34,6 +35,12 @@ class HFInferenceTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
34
35
|
}
|
|
35
36
|
exports.HFInferenceTask = HFInferenceTask;
|
|
36
37
|
class HFInferenceTextToImageTask extends HFInferenceTask {
|
|
38
|
+
preparePayload(params) {
|
|
39
|
+
if (params.outputType === "url") {
|
|
40
|
+
throw new errors_js_1.InferenceClientInputError("hf-inference provider does not support URL output. Use outputType 'blob', 'dataUrl' or 'json' instead.");
|
|
41
|
+
}
|
|
42
|
+
return params.args;
|
|
43
|
+
}
|
|
37
44
|
async getResponse(response, url, headers, outputType) {
|
|
38
45
|
if (!response) {
|
|
39
46
|
throw new errors_js_1.InferenceClientProviderOutputError("Received malformed response from HF-Inference text-to-image API: response is undefined");
|
|
@@ -44,25 +51,24 @@ class HFInferenceTextToImageTask extends HFInferenceTask {
|
|
|
44
51
|
}
|
|
45
52
|
if ("data" in response && Array.isArray(response.data) && response.data[0].b64_json) {
|
|
46
53
|
const base64Data = response.data[0].b64_json;
|
|
47
|
-
if (outputType === "
|
|
54
|
+
if (outputType === "dataUrl") {
|
|
48
55
|
return `data:image/jpeg;base64,${base64Data}`;
|
|
49
56
|
}
|
|
50
57
|
const base64Response = await fetch(`data:image/jpeg;base64,${base64Data}`);
|
|
51
58
|
return await base64Response.blob();
|
|
52
59
|
}
|
|
53
60
|
if ("output" in response && Array.isArray(response.output)) {
|
|
54
|
-
if (outputType === "url") {
|
|
55
|
-
return response.output[0];
|
|
56
|
-
}
|
|
57
61
|
const urlResponse = await fetch(response.output[0]);
|
|
58
62
|
const blob = await urlResponse.blob();
|
|
59
|
-
return blob;
|
|
63
|
+
return outputType === "dataUrl" ? (0, dataUrlFromBlob_js_1.dataUrlFromBlob)(blob) : blob;
|
|
60
64
|
}
|
|
61
65
|
}
|
|
62
66
|
if (response instanceof Blob) {
|
|
63
|
-
if (outputType === "
|
|
64
|
-
|
|
65
|
-
|
|
67
|
+
if (outputType === "dataUrl") {
|
|
68
|
+
return (0, dataUrlFromBlob_js_1.dataUrlFromBlob)(response);
|
|
69
|
+
}
|
|
70
|
+
if (outputType === "json") {
|
|
71
|
+
return { output: await (0, dataUrlFromBlob_js_1.dataUrlFromBlob)(response) };
|
|
66
72
|
}
|
|
67
73
|
return response;
|
|
68
74
|
}
|
|
@@ -15,7 +15,7 @@
|
|
|
15
15
|
* Thanks!
|
|
16
16
|
*/
|
|
17
17
|
import type { ChatCompletionOutput, TextGenerationOutput } from "@huggingface/tasks";
|
|
18
|
-
import type { BodyParams, UrlParams } from "../types.js";
|
|
18
|
+
import type { BodyParams, OutputType, UrlParams } from "../types.js";
|
|
19
19
|
import { BaseConversationalTask, BaseTextGenerationTask, TaskProviderHelper, type TextToImageTaskHelper } from "./providerHelper.js";
|
|
20
20
|
export interface HyperbolicTextCompletionOutput extends Omit<ChatCompletionOutput, "choices"> {
|
|
21
21
|
choices: Array<{
|
|
@@ -42,7 +42,7 @@ export declare class HyperbolicTextToImageTask extends TaskProviderHelper implem
|
|
|
42
42
|
constructor();
|
|
43
43
|
makeRoute(params: UrlParams): string;
|
|
44
44
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
45
|
-
getResponse(response: HyperbolicTextToImageOutput, url?: string, headers?: HeadersInit, outputType?:
|
|
45
|
+
getResponse(response: HyperbolicTextToImageOutput, url?: string, headers?: HeadersInit, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
46
46
|
}
|
|
47
47
|
export {};
|
|
48
48
|
//# sourceMappingURL=hyperbolic.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"hyperbolic.d.ts","sourceRoot":"","sources":["../../../src/providers/hyperbolic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,KAAK,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACrF,OAAO,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"hyperbolic.d.ts","sourceRoot":"","sources":["../../../src/providers/hyperbolic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,KAAK,EAAE,oBAAoB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACrF,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAErE,OAAO,EACN,sBAAsB,EACtB,sBAAsB,EACtB,kBAAkB,EAClB,KAAK,qBAAqB,EAC1B,MAAM,qBAAqB,CAAC;AAI7B,MAAM,WAAW,8BAA+B,SAAQ,IAAI,CAAC,oBAAoB,EAAE,SAAS,CAAC;IAC5F,OAAO,EAAE,KAAK,CAAC;QACd,OAAO,EAAE;YAAE,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC;KAC7B,CAAC,CAAC;CACH;AAED,UAAU,2BAA2B;IACpC,MAAM,EAAE,KAAK,CAAC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;CACjC;AAED,qBAAa,4BAA6B,SAAQ,sBAAsB;;CAIvE;AAED,qBAAa,4BAA6B,SAAQ,sBAAsB;;IAK9D,SAAS,IAAI,MAAM;IAInB,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAcrD,WAAW,CAAC,QAAQ,EAAE,8BAA8B,GAAG,OAAO,CAAC,oBAAoB,CAAC;CAenG;AAED,qBAAa,yBAA0B,SAAQ,kBAAmB,YAAW,qBAAqB;;IAKjG,SAAS,CAAC,MAAM,EAAE,SAAS,GAAG,MAAM;IAKpC,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAcrD,WAAW,CAChB,QAAQ,EAAE,2BAA2B,EACrC,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,WAAW,EACrB,UAAU,CAAC,EAAE,UAAU,GACrB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAmBnD"}
|
|
@@ -54,6 +54,9 @@ class HyperbolicTextToImageTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
54
54
|
return `/v1/images/generations`;
|
|
55
55
|
}
|
|
56
56
|
preparePayload(params) {
|
|
57
|
+
if (params.outputType === "url") {
|
|
58
|
+
throw new errors_js_1.InferenceClientInputError("hyperbolic provider does not support URL output. Use outputType 'blob', 'dataUrl' or 'json' instead.");
|
|
59
|
+
}
|
|
57
60
|
return {
|
|
58
61
|
...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
|
|
59
62
|
...params.args.parameters,
|
|
@@ -70,7 +73,7 @@ class HyperbolicTextToImageTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
70
73
|
if (outputType === "json") {
|
|
71
74
|
return { ...response };
|
|
72
75
|
}
|
|
73
|
-
if (outputType === "
|
|
76
|
+
if (outputType === "dataUrl") {
|
|
74
77
|
return `data:image/jpeg;base64,${response.images[0].image}`;
|
|
75
78
|
}
|
|
76
79
|
return fetch(`data:image/jpeg;base64,${response.images[0].image}`).then((res) => res.blob());
|
|
@@ -15,12 +15,13 @@
|
|
|
15
15
|
* Thanks!
|
|
16
16
|
*/
|
|
17
17
|
import type { FeatureExtractionOutput, TextGenerationOutput } from "@huggingface/tasks";
|
|
18
|
-
import type { BodyParams } from "../types.js";
|
|
18
|
+
import type { BodyParams, OutputType } from "../types.js";
|
|
19
19
|
import { BaseConversationalTask, BaseTextGenerationTask, TaskProviderHelper, type FeatureExtractionTaskHelper, type TextToImageTaskHelper } from "./providerHelper.js";
|
|
20
20
|
import type { ChatCompletionInput } from "../../../tasks/dist/commonjs/index.js";
|
|
21
|
-
interface
|
|
21
|
+
interface NebiusImageGeneration {
|
|
22
22
|
data: Array<{
|
|
23
|
-
b64_json
|
|
23
|
+
b64_json?: string;
|
|
24
|
+
url?: string;
|
|
24
25
|
}>;
|
|
25
26
|
}
|
|
26
27
|
interface NebiusEmbeddingsResponse {
|
|
@@ -46,7 +47,7 @@ export declare class NebiusTextToImageTask extends TaskProviderHelper implements
|
|
|
46
47
|
constructor();
|
|
47
48
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
48
49
|
makeRoute(): string;
|
|
49
|
-
getResponse(response:
|
|
50
|
+
getResponse(response: NebiusImageGeneration, url?: string, headers?: HeadersInit, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
50
51
|
}
|
|
51
52
|
export declare class NebiusFeatureExtractionTask extends TaskProviderHelper implements FeatureExtractionTaskHelper {
|
|
52
53
|
constructor();
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"nebius.d.ts","sourceRoot":"","sources":["../../../src/providers/nebius.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"nebius.d.ts","sourceRoot":"","sources":["../../../src/providers/nebius.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,KAAK,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACxF,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAE1D,OAAO,EACN,sBAAsB,EACtB,sBAAsB,EACtB,kBAAkB,EAClB,KAAK,2BAA2B,EAChC,KAAK,qBAAqB,EAC1B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AAIjF,UAAU,qBAAqB;IAC9B,IAAI,EAAE,KAAK,CAAC;QACX,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,GAAG,CAAC,EAAE,MAAM,CAAC;KACb,CAAC,CAAC;CACH;AAED,UAAU,wBAAwB;IACjC,IAAI,EAAE,KAAK,CAAC;QACX,SAAS,EAAE,MAAM,EAAE,CAAC;KACpB,CAAC,CAAC;CACH;AAED,UAAU,0BAA2B,SAAQ,IAAI,CAAC,oBAAoB,EAAE,SAAS,CAAC;IACjF,OAAO,EAAE,KAAK,CAAC;QACd,IAAI,EAAE,MAAM,CAAC;KACb,CAAC,CAAC;CACH;AAED,qBAAa,wBAAyB,SAAQ,sBAAsB;;IAK1D,cAAc,CAAC,MAAM,EAAE,UAAU,CAAC,mBAAmB,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;CAUzF;AAED,qBAAa,wBAAyB,SAAQ,sBAAsB;;IAK1D,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAQrD,WAAW,CAAC,QAAQ,EAAE,0BAA0B,GAAG,OAAO,CAAC,oBAAoB,CAAC;CAc/F;AAED,qBAAa,qBAAsB,SAAQ,kBAAmB,YAAW,qBAAqB;;IAK7F,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAU3D,SAAS,IAAI,MAAM;IAIb,WAAW,CAChB,QAAQ,EAAE,qBAAqB,EAC/B,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,WAAW,EACrB,UAAU,CAAC,EAAE,UAAU,GACrB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CA0BnD;AAED,qBAAa,2BAA4B,SAAQ,kBAAmB,YAAW,2BAA2B;;IAKzG,cAAc,CAAC,MAAM,EAAE,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAO3D,SAAS,IAAI,MAAM;IAIb,WAAW,CAAC,QAAQ,EAAE,wBAAwB,GAAG,OAAO,CAAC,uBAAuB,CAAC;CAGvF"}
|
|
@@ -52,7 +52,7 @@ class NebiusTextToImageTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
52
52
|
return {
|
|
53
53
|
...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
|
|
54
54
|
...params.args.parameters,
|
|
55
|
-
response_format: "b64_json",
|
|
55
|
+
response_format: params.outputType === "url" ? "url" : "b64_json",
|
|
56
56
|
prompt: params.args.inputs,
|
|
57
57
|
model: params.model,
|
|
58
58
|
};
|
|
@@ -64,17 +64,20 @@ class NebiusTextToImageTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
64
64
|
if (typeof response === "object" &&
|
|
65
65
|
"data" in response &&
|
|
66
66
|
Array.isArray(response.data) &&
|
|
67
|
-
response.data.length > 0
|
|
68
|
-
"b64_json" in response.data[0] &&
|
|
69
|
-
typeof response.data[0].b64_json === "string") {
|
|
67
|
+
response.data.length > 0) {
|
|
70
68
|
if (outputType === "json") {
|
|
71
69
|
return { ...response };
|
|
72
70
|
}
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
71
|
+
if ("url" in response.data[0] && typeof response.data[0].url === "string") {
|
|
72
|
+
return response.data[0].url;
|
|
73
|
+
}
|
|
74
|
+
if ("b64_json" in response.data[0] && typeof response.data[0].b64_json === "string") {
|
|
75
|
+
const base64Data = response.data[0].b64_json;
|
|
76
|
+
if (outputType === "dataUrl") {
|
|
77
|
+
return `data:image/jpeg;base64,${base64Data}`;
|
|
78
|
+
}
|
|
79
|
+
return fetch(`data:image/jpeg;base64,${base64Data}`).then((res) => res.blob());
|
|
76
80
|
}
|
|
77
|
-
return fetch(`data:image/jpeg;base64,${base64Data}`).then((res) => res.blob());
|
|
78
81
|
}
|
|
79
82
|
throw new errors_js_1.InferenceClientProviderOutputError("Received malformed response from Nebius text-to-image API");
|
|
80
83
|
}
|
|
@@ -15,7 +15,7 @@
|
|
|
15
15
|
* Thanks!
|
|
16
16
|
*/
|
|
17
17
|
import type { TextToImageInput } from "@huggingface/tasks";
|
|
18
|
-
import type { BodyParams } from "../types.js";
|
|
18
|
+
import type { BodyParams, OutputType } from "../types.js";
|
|
19
19
|
import { BaseConversationalTask, TaskProviderHelper, type TextToImageTaskHelper } from "./providerHelper.js";
|
|
20
20
|
interface NscaleCloudBase64ImageGeneration {
|
|
21
21
|
data: Array<{
|
|
@@ -29,7 +29,7 @@ export declare class NscaleTextToImageTask extends TaskProviderHelper implements
|
|
|
29
29
|
constructor();
|
|
30
30
|
preparePayload(params: BodyParams<TextToImageInput>): Record<string, unknown>;
|
|
31
31
|
makeRoute(): string;
|
|
32
|
-
getResponse(response: NscaleCloudBase64ImageGeneration, url?: string, headers?: HeadersInit, outputType?:
|
|
32
|
+
getResponse(response: NscaleCloudBase64ImageGeneration, url?: string, headers?: HeadersInit, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
33
33
|
}
|
|
34
34
|
export {};
|
|
35
35
|
//# sourceMappingURL=nscale.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"nscale.d.ts","sourceRoot":"","sources":["../../../src/providers/nscale.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;
|
|
1
|
+
{"version":3,"file":"nscale.d.ts","sourceRoot":"","sources":["../../../src/providers/nscale.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AACH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAE1D,OAAO,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,KAAK,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAK7G,UAAU,gCAAgC;IACzC,IAAI,EAAE,KAAK,CAAC;QACX,QAAQ,EAAE,MAAM,CAAC;KACjB,CAAC,CAAC;CACH;AAED,qBAAa,wBAAyB,SAAQ,sBAAsB;;CAInE;AAED,qBAAa,qBAAsB,SAAQ,kBAAmB,YAAW,qBAAqB;;IAK7F,cAAc,CAAC,MAAM,EAAE,UAAU,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAe7E,SAAS,IAAI,MAAM;IAIb,WAAW,CAChB,QAAQ,EAAE,gCAAgC,EAC1C,GAAG,CAAC,EAAE,MAAM,EACZ,OAAO,CAAC,EAAE,WAAW,EACrB,UAAU,CAAC,EAAE,UAAU,GACrB,OAAO,CAAC,MAAM,GAAG,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAqBnD"}
|
|
@@ -16,6 +16,9 @@ class NscaleTextToImageTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
16
16
|
super("nscale", NSCALE_API_BASE_URL);
|
|
17
17
|
}
|
|
18
18
|
preparePayload(params) {
|
|
19
|
+
if (params.outputType === "url") {
|
|
20
|
+
throw new errors_js_1.InferenceClientInputError("nscale provider does not support URL output. Use outputType 'blob', 'dataUrl' or 'json' instead.");
|
|
21
|
+
}
|
|
19
22
|
return {
|
|
20
23
|
...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
|
|
21
24
|
...params.args.parameters,
|
|
@@ -38,7 +41,7 @@ class NscaleTextToImageTask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
38
41
|
return { ...response };
|
|
39
42
|
}
|
|
40
43
|
const base64Data = response.data[0].b64_json;
|
|
41
|
-
if (outputType === "
|
|
44
|
+
if (outputType === "dataUrl") {
|
|
42
45
|
return `data:image/jpeg;base64,${base64Data}`;
|
|
43
46
|
}
|
|
44
47
|
return fetch(`data:image/jpeg;base64,${base64Data}`).then((res) => res.blob());
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import type { AudioClassificationInput, AudioClassificationOutput, AutomaticSpeechRecognitionInput, AutomaticSpeechRecognitionOutput, ChatCompletionInput, ChatCompletionOutput, DocumentQuestionAnsweringInput, DocumentQuestionAnsweringOutput, FeatureExtractionInput, FeatureExtractionOutput, FillMaskInput, FillMaskOutput, ImageClassificationInput, ImageClassificationOutput, ImageSegmentationInput, ImageSegmentationOutput, ImageToImageInput, ImageToTextInput, ImageToTextOutput, ImageToVideoInput, ImageTextToImageInput, ImageTextToVideoInput, ObjectDetectionInput, ObjectDetectionOutput, QuestionAnsweringInput, QuestionAnsweringOutput, SentenceSimilarityInput, SentenceSimilarityOutput, SummarizationInput, SummarizationOutput, TableQuestionAnsweringInput, TableQuestionAnsweringOutput, TextClassificationOutput, TextGenerationInput, TextGenerationOutput, TextToImageInput, TextToSpeechInput, TextToVideoInput, TokenClassificationInput, TokenClassificationOutput, TranslationInput, TranslationOutput, VisualQuestionAnsweringInput, VisualQuestionAnsweringOutput, ZeroShotClassificationInput, ZeroShotClassificationOutput, ZeroShotImageClassificationInput, ZeroShotImageClassificationOutput } from "@huggingface/tasks";
|
|
2
2
|
import type { AudioToAudioOutput } from "../tasks/audio/audioToAudio.js";
|
|
3
|
-
import type { BaseArgs, BodyParams, HeaderParams, InferenceProvider, RequestArgs, UrlParams } from "../types.js";
|
|
3
|
+
import type { BaseArgs, BodyParams, HeaderParams, InferenceProvider, OutputType, RequestArgs, UrlParams } from "../types.js";
|
|
4
4
|
import type { ImageToImageArgs } from "../tasks/cv/imageToImage.js";
|
|
5
5
|
import type { AutomaticSpeechRecognitionArgs } from "../tasks/audio/automaticSpeechRecognition.js";
|
|
6
6
|
import type { ImageToVideoArgs } from "../tasks/cv/imageToVideo.js";
|
|
@@ -19,7 +19,7 @@ export declare abstract class TaskProviderHelper {
|
|
|
19
19
|
* Return the response in the expected format.
|
|
20
20
|
* Needs to be implemented in the subclasses.
|
|
21
21
|
*/
|
|
22
|
-
abstract getResponse(response: unknown, url?: string, headers?: HeadersInit, outputType?:
|
|
22
|
+
abstract getResponse(response: unknown, url?: string, headers?: HeadersInit, outputType?: OutputType): Promise<unknown>;
|
|
23
23
|
/**
|
|
24
24
|
* Prepare the route for the request
|
|
25
25
|
* Needs to be implemented in the subclasses.
|
|
@@ -48,7 +48,7 @@ export declare abstract class TaskProviderHelper {
|
|
|
48
48
|
prepareHeaders(params: HeaderParams, isBinary: boolean): Record<string, string>;
|
|
49
49
|
}
|
|
50
50
|
export interface TextToImageTaskHelper {
|
|
51
|
-
getResponse(response: unknown, url?: string, headers?: HeadersInit, outputType?:
|
|
51
|
+
getResponse(response: unknown, url?: string, headers?: HeadersInit, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
52
52
|
preparePayload(params: BodyParams<TextToImageInput & BaseArgs>): Record<string, unknown>;
|
|
53
53
|
}
|
|
54
54
|
export interface TextToVideoTaskHelper {
|