@huggingface/inference 4.13.8 → 4.13.10
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 +17 -14
- package/dist/commonjs/providers/fal-ai.d.ts.map +1 -1
- package/dist/commonjs/providers/fal-ai.js +73 -36
- 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 +5 -2
- 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 +17 -4
- package/dist/commonjs/providers/wavespeed.d.ts.map +1 -1
- package/dist/commonjs/providers/wavespeed.js +50 -5
- 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/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 +17 -14
- package/dist/esm/providers/fal-ai.d.ts.map +1 -1
- package/dist/esm/providers/fal-ai.js +70 -35
- 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 +5 -2
- 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 +17 -4
- package/dist/esm/providers/wavespeed.d.ts.map +1 -1
- package/dist/esm/providers/wavespeed.js +47 -4
- 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/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 +2 -2
- package/src/lib/getProviderHelper.ts +8 -4
- package/src/lib/makeRequestOptions.ts +6 -3
- package/src/package.ts +1 -1
- package/src/providers/fal-ai.ts +93 -43
- package/src/providers/hf-inference.ts +20 -11
- package/src/providers/hyperbolic.ts +9 -4
- package/src/providers/nebius.ts +19 -13
- package/src/providers/nscale.ts +9 -4
- package/src/providers/providerHelper.ts +14 -3
- package/src/providers/replicate.ts +7 -4
- package/src/providers/together.ts +19 -13
- package/src/providers/wavespeed.ts +86 -7
- package/src/providers/zai-org.ts +5 -3
- package/src/tasks/cv/textToImage.ts +6 -2
- package/src/types.ts +5 -0
- package/src/utils/dataUrlFromBlob.ts +7 -0
|
@@ -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.10";
|
|
6
6
|
exports.PACKAGE_NAME = "@huggingface/inference";
|
|
@@ -1,20 +1,17 @@
|
|
|
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;
|
|
10
12
|
status: string;
|
|
11
13
|
response_url: string;
|
|
12
14
|
}
|
|
13
|
-
interface FalAITextToImageOutput {
|
|
14
|
-
images: Array<{
|
|
15
|
-
url: string;
|
|
16
|
-
}>;
|
|
17
|
-
}
|
|
18
15
|
export declare const FAL_AI_SUPPORTED_BLOB_TYPES: string[];
|
|
19
16
|
declare abstract class FalAITask extends TaskProviderHelper {
|
|
20
17
|
constructor(url?: string);
|
|
@@ -24,32 +21,35 @@ declare abstract class FalAITask extends TaskProviderHelper {
|
|
|
24
21
|
}
|
|
25
22
|
declare abstract class FalAiQueueTask extends FalAITask {
|
|
26
23
|
abstract task: InferenceTask;
|
|
24
|
+
makeRoute(params: UrlParams): string;
|
|
27
25
|
getResponseFromQueueApi(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<unknown>;
|
|
28
26
|
}
|
|
29
|
-
export declare class FalAITextToImageTask extends
|
|
27
|
+
export declare class FalAITextToImageTask extends FalAiQueueTask implements TextToImageTaskHelper {
|
|
28
|
+
task: InferenceTask;
|
|
29
|
+
constructor();
|
|
30
30
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
31
|
-
getResponse(response:
|
|
31
|
+
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>, outputType?: OutputType): Promise<string | Blob | Record<string, unknown>>;
|
|
32
32
|
}
|
|
33
33
|
export declare class FalAIImageToImageTask extends FalAiQueueTask implements ImageToImageTaskHelper {
|
|
34
34
|
task: InferenceTask;
|
|
35
35
|
constructor();
|
|
36
|
-
makeRoute(params: UrlParams): string;
|
|
37
36
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
38
37
|
preparePayloadAsync(args: ImageToImageArgs): Promise<RequestArgs>;
|
|
39
38
|
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<Blob>;
|
|
40
39
|
}
|
|
40
|
+
export declare class FalAIImageTextToImageTask extends FalAIImageToImageTask implements ImageTextToImageTaskHelper {
|
|
41
|
+
constructor();
|
|
42
|
+
preparePayloadAsync(args: ImageTextToImageArgs): Promise<RequestArgs>;
|
|
43
|
+
}
|
|
41
44
|
export declare class FalAITextToVideoTask extends FalAiQueueTask implements TextToVideoTaskHelper {
|
|
42
45
|
task: InferenceTask;
|
|
43
46
|
constructor();
|
|
44
|
-
makeRoute(params: UrlParams): string;
|
|
45
47
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
46
48
|
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<Blob>;
|
|
47
49
|
}
|
|
48
50
|
export declare class FalAIImageToVideoTask extends FalAiQueueTask implements ImageToVideoTaskHelper {
|
|
49
51
|
task: InferenceTask;
|
|
50
52
|
constructor();
|
|
51
|
-
/** Same queue routing rule as the other Fal queue tasks */
|
|
52
|
-
makeRoute(params: UrlParams): string;
|
|
53
53
|
/** Synchronous case – caller already gave us base64 or a URL */
|
|
54
54
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
55
55
|
/** Asynchronous helper – caller gave us a Blob */
|
|
@@ -57,6 +57,10 @@ export declare class FalAIImageToVideoTask extends FalAiQueueTask implements Ima
|
|
|
57
57
|
/** Queue polling + final download – mirrors Text‑to‑Video */
|
|
58
58
|
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<Blob>;
|
|
59
59
|
}
|
|
60
|
+
export declare class FalAIImageTextToVideoTask extends FalAIImageToVideoTask implements ImageTextToVideoTaskHelper {
|
|
61
|
+
constructor();
|
|
62
|
+
preparePayloadAsync(args: ImageTextToVideoArgs): Promise<RequestArgs>;
|
|
63
|
+
}
|
|
60
64
|
export declare class FalAIAutomaticSpeechRecognitionTask extends FalAITask implements AutomaticSpeechRecognitionTaskHelper {
|
|
61
65
|
prepareHeaders(params: HeaderParams, binary: boolean): Record<string, string>;
|
|
62
66
|
getResponse(response: unknown): Promise<AutomaticSpeechRecognitionOutput>;
|
|
@@ -69,7 +73,6 @@ export declare class FalAITextToSpeechTask extends FalAITask {
|
|
|
69
73
|
export declare class FalAIImageSegmentationTask extends FalAiQueueTask implements ImageSegmentationTaskHelper {
|
|
70
74
|
task: InferenceTask;
|
|
71
75
|
constructor();
|
|
72
|
-
makeRoute(params: UrlParams): string;
|
|
73
76
|
preparePayload(params: BodyParams): Record<string, unknown>;
|
|
74
77
|
preparePayloadAsync(args: ImageSegmentationArgs): Promise<RequestArgs>;
|
|
75
78
|
getResponse(response: FalAiQueueOutput, url?: string, headers?: Record<string, string>): Promise<ImageSegmentationOutput>;
|
|
@@ -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");
|
|
@@ -46,6 +47,12 @@ class FalAITask extends providerHelper_js_1.TaskProviderHelper {
|
|
|
46
47
|
}
|
|
47
48
|
}
|
|
48
49
|
class FalAiQueueTask extends FalAITask {
|
|
50
|
+
makeRoute(params) {
|
|
51
|
+
if (params.authMethod !== "provider-key") {
|
|
52
|
+
return `/${params.model}?_subdomain=queue`;
|
|
53
|
+
}
|
|
54
|
+
return `/${params.model}`;
|
|
55
|
+
}
|
|
49
56
|
async getResponseFromQueueApi(response, url, headers) {
|
|
50
57
|
if (!url || !headers) {
|
|
51
58
|
throw new errors_js_1.InferenceClientInputError(`URL and headers are required for ${this.task} task`);
|
|
@@ -94,12 +101,16 @@ class FalAiQueueTask extends FalAITask {
|
|
|
94
101
|
function buildLoraPath(modelId, adapterWeightsPath) {
|
|
95
102
|
return `${config_js_1.HF_HUB_URL}/${modelId}/resolve/main/${adapterWeightsPath}`;
|
|
96
103
|
}
|
|
97
|
-
class FalAITextToImageTask extends
|
|
104
|
+
class FalAITextToImageTask extends FalAiQueueTask {
|
|
105
|
+
task;
|
|
106
|
+
constructor() {
|
|
107
|
+
super("https://queue.fal.run");
|
|
108
|
+
this.task = "text-to-image";
|
|
109
|
+
}
|
|
98
110
|
preparePayload(params) {
|
|
99
111
|
const payload = {
|
|
100
112
|
...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
|
|
101
113
|
...params.args.parameters,
|
|
102
|
-
sync_mode: true,
|
|
103
114
|
prompt: params.args.inputs,
|
|
104
115
|
};
|
|
105
116
|
if (params.mapping?.adapter === "lora" && params.mapping.adapterWeightsPath) {
|
|
@@ -116,22 +127,25 @@ class FalAITextToImageTask extends FalAITask {
|
|
|
116
127
|
return payload;
|
|
117
128
|
}
|
|
118
129
|
async getResponse(response, url, headers, outputType) {
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
130
|
+
const result = (await this.getResponseFromQueueApi(response, url, headers));
|
|
131
|
+
if (typeof result === "object" &&
|
|
132
|
+
"images" in result &&
|
|
133
|
+
Array.isArray(result.images) &&
|
|
134
|
+
result.images.length > 0 &&
|
|
135
|
+
"url" in result.images[0] &&
|
|
136
|
+
typeof result.images[0].url === "string" &&
|
|
137
|
+
(0, isUrl_js_1.isUrl)(result.images[0].url)) {
|
|
125
138
|
if (outputType === "json") {
|
|
126
|
-
return { ...
|
|
139
|
+
return { ...result };
|
|
127
140
|
}
|
|
128
141
|
if (outputType === "url") {
|
|
129
|
-
return
|
|
142
|
+
return result.images[0].url;
|
|
130
143
|
}
|
|
131
|
-
const urlResponse = await fetch(
|
|
132
|
-
|
|
144
|
+
const urlResponse = await fetch(result.images[0].url);
|
|
145
|
+
const blob = await urlResponse.blob();
|
|
146
|
+
return outputType === "dataUrl" ? (0, dataUrlFromBlob_js_1.dataUrlFromBlob)(blob) : blob;
|
|
133
147
|
}
|
|
134
|
-
throw new errors_js_1.InferenceClientProviderOutputError(
|
|
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)}`);
|
|
135
149
|
}
|
|
136
150
|
}
|
|
137
151
|
exports.FalAITextToImageTask = FalAITextToImageTask;
|
|
@@ -141,12 +155,6 @@ class FalAIImageToImageTask extends FalAiQueueTask {
|
|
|
141
155
|
super("https://queue.fal.run");
|
|
142
156
|
this.task = "image-to-image";
|
|
143
157
|
}
|
|
144
|
-
makeRoute(params) {
|
|
145
|
-
if (params.authMethod !== "provider-key") {
|
|
146
|
-
return `/${params.model}?_subdomain=queue`;
|
|
147
|
-
}
|
|
148
|
-
return `/${params.model}`;
|
|
149
|
-
}
|
|
150
158
|
preparePayload(params) {
|
|
151
159
|
const payload = params.args;
|
|
152
160
|
if (params.mapping?.adapter === "lora" && params.mapping.adapterWeightsPath) {
|
|
@@ -192,18 +200,35 @@ class FalAIImageToImageTask extends FalAiQueueTask {
|
|
|
192
200
|
}
|
|
193
201
|
}
|
|
194
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;
|
|
195
226
|
class FalAITextToVideoTask extends FalAiQueueTask {
|
|
196
227
|
task;
|
|
197
228
|
constructor() {
|
|
198
229
|
super("https://queue.fal.run");
|
|
199
230
|
this.task = "text-to-video";
|
|
200
231
|
}
|
|
201
|
-
makeRoute(params) {
|
|
202
|
-
if (params.authMethod !== "provider-key") {
|
|
203
|
-
return `/${params.model}?_subdomain=queue`;
|
|
204
|
-
}
|
|
205
|
-
return `/${params.model}`;
|
|
206
|
-
}
|
|
207
232
|
preparePayload(params) {
|
|
208
233
|
return {
|
|
209
234
|
...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
|
|
@@ -236,10 +261,6 @@ class FalAIImageToVideoTask extends FalAiQueueTask {
|
|
|
236
261
|
super("https://queue.fal.run");
|
|
237
262
|
this.task = "image-to-video";
|
|
238
263
|
}
|
|
239
|
-
/** Same queue routing rule as the other Fal queue tasks */
|
|
240
|
-
makeRoute(params) {
|
|
241
|
-
return params.authMethod !== "provider-key" ? `/${params.model}?_subdomain=queue` : `/${params.model}`;
|
|
242
|
-
}
|
|
243
264
|
/** Synchronous case – caller already gave us base64 or a URL */
|
|
244
265
|
preparePayload(params) {
|
|
245
266
|
return {
|
|
@@ -278,6 +299,28 @@ class FalAIImageToVideoTask extends FalAiQueueTask {
|
|
|
278
299
|
}
|
|
279
300
|
}
|
|
280
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;
|
|
281
324
|
class FalAIAutomaticSpeechRecognitionTask extends FalAITask {
|
|
282
325
|
prepareHeaders(params, binary) {
|
|
283
326
|
const headers = super.prepareHeaders(params, binary);
|
|
@@ -348,12 +391,6 @@ class FalAIImageSegmentationTask extends FalAiQueueTask {
|
|
|
348
391
|
super("https://queue.fal.run");
|
|
349
392
|
this.task = "image-segmentation";
|
|
350
393
|
}
|
|
351
|
-
makeRoute(params) {
|
|
352
|
-
if (params.authMethod !== "provider-key") {
|
|
353
|
-
return `/${params.model}?_subdomain=queue`;
|
|
354
|
-
}
|
|
355
|
-
return `/${params.model}`;
|
|
356
|
-
}
|
|
357
394
|
preparePayload(params) {
|
|
358
395
|
return {
|
|
359
396
|
...(0, omit_js_1.omit)(params.args, ["inputs", "parameters"]),
|
|
@@ -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"}
|