@huggingface/tasks 0.13.17 → 0.15.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (195) hide show
  1. package/dist/commonjs/index.d.ts +1 -0
  2. package/dist/commonjs/index.d.ts.map +1 -1
  3. package/dist/commonjs/index.js +1 -0
  4. package/dist/commonjs/inference-providers.d.ts +10 -0
  5. package/dist/commonjs/inference-providers.d.ts.map +1 -0
  6. package/dist/commonjs/inference-providers.js +16 -0
  7. package/dist/commonjs/model-libraries-snippets.d.ts +2 -2
  8. package/dist/commonjs/model-libraries-snippets.d.ts.map +1 -1
  9. package/dist/commonjs/model-libraries-snippets.js +2 -2
  10. package/dist/commonjs/model-libraries.d.ts +2 -2
  11. package/dist/commonjs/snippets/curl.d.ts +8 -8
  12. package/dist/commonjs/snippets/curl.d.ts.map +1 -1
  13. package/dist/commonjs/snippets/curl.js +58 -30
  14. package/dist/commonjs/snippets/js.d.ts +11 -10
  15. package/dist/commonjs/snippets/js.d.ts.map +1 -1
  16. package/dist/commonjs/snippets/js.js +162 -53
  17. package/dist/commonjs/snippets/python.d.ts +12 -12
  18. package/dist/commonjs/snippets/python.d.ts.map +1 -1
  19. package/dist/commonjs/snippets/python.js +141 -71
  20. package/dist/commonjs/snippets/types.d.ts +1 -1
  21. package/dist/commonjs/snippets/types.d.ts.map +1 -1
  22. package/dist/commonjs/tasks/audio-classification/inference.d.ts +1 -1
  23. package/dist/commonjs/tasks/audio-classification/inference.d.ts.map +1 -1
  24. package/dist/commonjs/tasks/automatic-speech-recognition/inference.d.ts +1 -1
  25. package/dist/commonjs/tasks/automatic-speech-recognition/inference.d.ts.map +1 -1
  26. package/dist/commonjs/tasks/automatic-speech-recognition/inference.js +0 -5
  27. package/dist/commonjs/tasks/chat-completion/inference.d.ts.map +1 -1
  28. package/dist/commonjs/tasks/chat-completion/inference.js +0 -5
  29. package/dist/commonjs/tasks/depth-estimation/inference.d.ts.map +1 -1
  30. package/dist/commonjs/tasks/depth-estimation/inference.js +0 -5
  31. package/dist/commonjs/tasks/feature-extraction/inference.d.ts.map +1 -1
  32. package/dist/commonjs/tasks/feature-extraction/inference.js +0 -5
  33. package/dist/commonjs/tasks/image-classification/inference.d.ts +1 -1
  34. package/dist/commonjs/tasks/image-classification/inference.d.ts.map +1 -1
  35. package/dist/commonjs/tasks/image-segmentation/inference.d.ts +1 -1
  36. package/dist/commonjs/tasks/image-segmentation/inference.d.ts.map +1 -1
  37. package/dist/commonjs/tasks/image-to-image/inference.d.ts +3 -3
  38. package/dist/commonjs/tasks/image-to-image/inference.d.ts.map +1 -1
  39. package/dist/commonjs/tasks/image-to-image/inference.js +0 -5
  40. package/dist/commonjs/tasks/image-to-text/inference.d.ts +1 -1
  41. package/dist/commonjs/tasks/image-to-text/inference.d.ts.map +1 -1
  42. package/dist/commonjs/tasks/image-to-text/inference.js +0 -5
  43. package/dist/commonjs/tasks/index.d.ts +1 -0
  44. package/dist/commonjs/tasks/index.d.ts.map +1 -1
  45. package/dist/commonjs/tasks/object-detection/inference.d.ts +1 -1
  46. package/dist/commonjs/tasks/object-detection/inference.d.ts.map +1 -1
  47. package/dist/commonjs/tasks/sentence-similarity/inference.d.ts.map +1 -1
  48. package/dist/commonjs/tasks/sentence-similarity/inference.js +0 -5
  49. package/dist/commonjs/tasks/summarization/inference.d.ts.map +1 -1
  50. package/dist/commonjs/tasks/summarization/inference.js +0 -5
  51. package/dist/commonjs/tasks/text-generation/inference.d.ts.map +1 -1
  52. package/dist/commonjs/tasks/text-generation/inference.js +0 -5
  53. package/dist/commonjs/tasks/text-to-audio/inference.d.ts +14 -15
  54. package/dist/commonjs/tasks/text-to-audio/inference.d.ts.map +1 -1
  55. package/dist/commonjs/tasks/text-to-audio/inference.js +0 -5
  56. package/dist/commonjs/tasks/text-to-image/inference.d.ts +2 -2
  57. package/dist/commonjs/tasks/text-to-image/inference.d.ts.map +1 -1
  58. package/dist/commonjs/tasks/text-to-image/inference.js +0 -5
  59. package/dist/commonjs/tasks/text-to-speech/inference.d.ts +14 -17
  60. package/dist/commonjs/tasks/text-to-speech/inference.d.ts.map +1 -1
  61. package/dist/commonjs/tasks/text-to-speech/inference.js +0 -5
  62. package/dist/commonjs/tasks/text-to-video/inference.d.ts +58 -0
  63. package/dist/commonjs/tasks/text-to-video/inference.d.ts.map +1 -0
  64. package/dist/commonjs/tasks/text-to-video/inference.js +2 -0
  65. package/dist/commonjs/tasks/text2text-generation/inference.d.ts.map +1 -1
  66. package/dist/commonjs/tasks/text2text-generation/inference.js +0 -5
  67. package/dist/commonjs/tasks/translation/inference.d.ts.map +1 -1
  68. package/dist/commonjs/tasks/translation/inference.js +0 -5
  69. package/dist/commonjs/tasks/visual-question-answering/inference.d.ts +1 -1
  70. package/dist/commonjs/tasks/visual-question-answering/inference.d.ts.map +1 -1
  71. package/dist/commonjs/tasks/zero-shot-image-classification/inference.d.ts +1 -1
  72. package/dist/commonjs/tasks/zero-shot-image-classification/inference.d.ts.map +1 -1
  73. package/dist/commonjs/tasks/zero-shot-object-detection/inference.d.ts +1 -1
  74. package/dist/commonjs/tasks/zero-shot-object-detection/inference.d.ts.map +1 -1
  75. package/dist/esm/index.d.ts +1 -0
  76. package/dist/esm/index.d.ts.map +1 -1
  77. package/dist/esm/index.js +1 -0
  78. package/dist/esm/inference-providers.d.ts +10 -0
  79. package/dist/esm/inference-providers.d.ts.map +1 -0
  80. package/dist/esm/inference-providers.js +12 -0
  81. package/dist/esm/model-libraries-snippets.d.ts +2 -2
  82. package/dist/esm/model-libraries-snippets.d.ts.map +1 -1
  83. package/dist/esm/model-libraries-snippets.js +2 -2
  84. package/dist/esm/model-libraries.d.ts +2 -2
  85. package/dist/esm/snippets/curl.d.ts +8 -8
  86. package/dist/esm/snippets/curl.d.ts.map +1 -1
  87. package/dist/esm/snippets/curl.js +58 -29
  88. package/dist/esm/snippets/js.d.ts +11 -10
  89. package/dist/esm/snippets/js.d.ts.map +1 -1
  90. package/dist/esm/snippets/js.js +159 -50
  91. package/dist/esm/snippets/python.d.ts +12 -12
  92. package/dist/esm/snippets/python.d.ts.map +1 -1
  93. package/dist/esm/snippets/python.js +140 -69
  94. package/dist/esm/snippets/types.d.ts +1 -1
  95. package/dist/esm/snippets/types.d.ts.map +1 -1
  96. package/dist/esm/tasks/audio-classification/inference.d.ts +1 -1
  97. package/dist/esm/tasks/audio-classification/inference.d.ts.map +1 -1
  98. package/dist/esm/tasks/automatic-speech-recognition/inference.d.ts +1 -1
  99. package/dist/esm/tasks/automatic-speech-recognition/inference.d.ts.map +1 -1
  100. package/dist/esm/tasks/automatic-speech-recognition/inference.js +0 -5
  101. package/dist/esm/tasks/chat-completion/inference.d.ts.map +1 -1
  102. package/dist/esm/tasks/chat-completion/inference.js +0 -5
  103. package/dist/esm/tasks/depth-estimation/inference.d.ts.map +1 -1
  104. package/dist/esm/tasks/depth-estimation/inference.js +0 -5
  105. package/dist/esm/tasks/feature-extraction/inference.d.ts.map +1 -1
  106. package/dist/esm/tasks/feature-extraction/inference.js +0 -5
  107. package/dist/esm/tasks/image-classification/inference.d.ts +1 -1
  108. package/dist/esm/tasks/image-classification/inference.d.ts.map +1 -1
  109. package/dist/esm/tasks/image-segmentation/inference.d.ts +1 -1
  110. package/dist/esm/tasks/image-segmentation/inference.d.ts.map +1 -1
  111. package/dist/esm/tasks/image-to-image/inference.d.ts +3 -3
  112. package/dist/esm/tasks/image-to-image/inference.d.ts.map +1 -1
  113. package/dist/esm/tasks/image-to-image/inference.js +0 -5
  114. package/dist/esm/tasks/image-to-text/inference.d.ts +1 -1
  115. package/dist/esm/tasks/image-to-text/inference.d.ts.map +1 -1
  116. package/dist/esm/tasks/image-to-text/inference.js +0 -5
  117. package/dist/esm/tasks/index.d.ts +1 -0
  118. package/dist/esm/tasks/index.d.ts.map +1 -1
  119. package/dist/esm/tasks/object-detection/inference.d.ts +1 -1
  120. package/dist/esm/tasks/object-detection/inference.d.ts.map +1 -1
  121. package/dist/esm/tasks/sentence-similarity/inference.d.ts.map +1 -1
  122. package/dist/esm/tasks/sentence-similarity/inference.js +0 -5
  123. package/dist/esm/tasks/summarization/inference.d.ts.map +1 -1
  124. package/dist/esm/tasks/summarization/inference.js +0 -5
  125. package/dist/esm/tasks/text-generation/inference.d.ts.map +1 -1
  126. package/dist/esm/tasks/text-generation/inference.js +0 -5
  127. package/dist/esm/tasks/text-to-audio/inference.d.ts +14 -15
  128. package/dist/esm/tasks/text-to-audio/inference.d.ts.map +1 -1
  129. package/dist/esm/tasks/text-to-audio/inference.js +0 -5
  130. package/dist/esm/tasks/text-to-image/inference.d.ts +2 -2
  131. package/dist/esm/tasks/text-to-image/inference.d.ts.map +1 -1
  132. package/dist/esm/tasks/text-to-image/inference.js +0 -5
  133. package/dist/esm/tasks/text-to-speech/inference.d.ts +14 -17
  134. package/dist/esm/tasks/text-to-speech/inference.d.ts.map +1 -1
  135. package/dist/esm/tasks/text-to-speech/inference.js +0 -5
  136. package/dist/esm/tasks/text-to-video/inference.d.ts +58 -0
  137. package/dist/esm/tasks/text-to-video/inference.d.ts.map +1 -0
  138. package/dist/esm/tasks/text-to-video/inference.js +1 -0
  139. package/dist/esm/tasks/text2text-generation/inference.d.ts.map +1 -1
  140. package/dist/esm/tasks/text2text-generation/inference.js +0 -5
  141. package/dist/esm/tasks/translation/inference.d.ts.map +1 -1
  142. package/dist/esm/tasks/translation/inference.js +0 -5
  143. package/dist/esm/tasks/visual-question-answering/inference.d.ts +1 -1
  144. package/dist/esm/tasks/visual-question-answering/inference.d.ts.map +1 -1
  145. package/dist/esm/tasks/zero-shot-image-classification/inference.d.ts +1 -1
  146. package/dist/esm/tasks/zero-shot-image-classification/inference.d.ts.map +1 -1
  147. package/dist/esm/tasks/zero-shot-object-detection/inference.d.ts +1 -1
  148. package/dist/esm/tasks/zero-shot-object-detection/inference.d.ts.map +1 -1
  149. package/package.json +1 -1
  150. package/src/index.ts +2 -0
  151. package/src/inference-providers.ts +16 -0
  152. package/src/model-libraries-snippets.ts +3 -3
  153. package/src/snippets/curl.ts +72 -23
  154. package/src/snippets/js.ts +189 -56
  155. package/src/snippets/python.ts +154 -75
  156. package/src/snippets/types.ts +1 -1
  157. package/src/tasks/audio-classification/inference.ts +1 -1
  158. package/src/tasks/audio-classification/spec/input.json +2 -1
  159. package/src/tasks/automatic-speech-recognition/inference.ts +1 -7
  160. package/src/tasks/automatic-speech-recognition/spec/input.json +2 -1
  161. package/src/tasks/chat-completion/inference.ts +0 -33
  162. package/src/tasks/depth-estimation/inference.ts +3 -3
  163. package/src/tasks/document-question-answering/spec/input.json +2 -1
  164. package/src/tasks/feature-extraction/inference.ts +0 -3
  165. package/src/tasks/image-classification/inference.ts +1 -1
  166. package/src/tasks/image-classification/spec/input.json +2 -1
  167. package/src/tasks/image-segmentation/inference.ts +1 -1
  168. package/src/tasks/image-segmentation/spec/input.json +2 -1
  169. package/src/tasks/image-to-image/inference.ts +3 -7
  170. package/src/tasks/image-to-image/spec/input.json +4 -6
  171. package/src/tasks/image-to-text/inference.ts +1 -6
  172. package/src/tasks/image-to-text/spec/input.json +2 -1
  173. package/src/tasks/index.ts +1 -0
  174. package/src/tasks/object-detection/inference.ts +1 -1
  175. package/src/tasks/object-detection/spec/input.json +2 -1
  176. package/src/tasks/sentence-similarity/inference.ts +3 -4
  177. package/src/tasks/summarization/inference.ts +3 -5
  178. package/src/tasks/text-generation/inference.ts +0 -13
  179. package/src/tasks/text-to-audio/inference.ts +14 -20
  180. package/src/tasks/text-to-audio/spec/output.json +3 -2
  181. package/src/tasks/text-to-image/inference.ts +2 -6
  182. package/src/tasks/text-to-image/spec/input.json +2 -5
  183. package/src/tasks/text-to-speech/inference.ts +14 -22
  184. package/src/tasks/text-to-speech/spec/output.json +13 -2
  185. package/src/tasks/text-to-video/inference.ts +57 -0
  186. package/src/tasks/text-to-video/spec/input.json +49 -0
  187. package/src/tasks/text-to-video/spec/output.json +13 -0
  188. package/src/tasks/text2text-generation/inference.ts +3 -5
  189. package/src/tasks/translation/inference.ts +3 -5
  190. package/src/tasks/visual-question-answering/inference.ts +1 -1
  191. package/src/tasks/visual-question-answering/spec/input.json +4 -2
  192. package/src/tasks/zero-shot-image-classification/inference.ts +1 -1
  193. package/src/tasks/zero-shot-image-classification/spec/input.json +2 -1
  194. package/src/tasks/zero-shot-object-detection/inference.ts +1 -1
  195. package/src/tasks/zero-shot-object-detection/spec/input.json +2 -1
@@ -19,4 +19,5 @@ export { LOCAL_APPS } from "./local-apps.js";
19
19
  export type { LocalApp, LocalAppKey, LocalAppSnippet } from "./local-apps.js";
20
20
  export { DATASET_LIBRARIES_UI_ELEMENTS } from "./dataset-libraries.js";
21
21
  export type { DatasetLibraryUiElement, DatasetLibraryKey } from "./dataset-libraries.js";
22
+ export * from "./inference-providers.js";
22
23
  //# sourceMappingURL=index.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACvF,cAAc,kBAAkB,CAAC;AACjC,OAAO,EACN,aAAa,EACb,cAAc,EACd,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,YAAY,EACjB,KAAK,QAAQ,EACb,UAAU,EACV,eAAe,EACf,aAAa,EACb,kBAAkB,GAClB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACN,8BAA8B,EAC9B,sBAAsB,EACtB,2BAA2B,GAC3B,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC9E,YAAY,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACzF,YAAY,EACX,aAAa,EACb,sBAAsB,EACtB,gCAAgC,EAChC,8BAA8B,EAC9B,kCAAkC,EAClC,uBAAuB,EACvB,sBAAsB,EACtB,oCAAoC,EACpC,gCAAgC,EAChC,2BAA2B,EAC3B,gCAAgC,EAChC,8BAA8B,EAC9B,sBAAsB,EACtB,8BAA8B,EAC9B,mBAAmB,EACnB,sBAAsB,EACtB,yBAAyB,EACzB,8BAA8B,EAC9B,uBAAuB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAEhE,OAAO,KAAK,QAAQ,MAAM,qBAAqB,CAAC;AAChD,cAAc,WAAW,CAAC;AAE1B,OAAO,EAAE,QAAQ,EAAE,CAAC;AACpB,YAAY,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE5D,OAAO,EAAE,IAAI,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAC7D,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAE9E,OAAO,EAAE,6BAA6B,EAAE,MAAM,wBAAwB,CAAC;AACvE,YAAY,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC"}
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,YAAY,EAAE,QAAQ,EAAE,QAAQ,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACvF,cAAc,kBAAkB,CAAC;AACjC,OAAO,EACN,aAAa,EACb,cAAc,EACd,KAAK,UAAU,EACf,KAAK,YAAY,EACjB,KAAK,YAAY,EACjB,KAAK,QAAQ,EACb,UAAU,EACV,eAAe,EACf,aAAa,EACb,kBAAkB,GAClB,MAAM,gBAAgB,CAAC;AACxB,OAAO,EACN,8BAA8B,EAC9B,sBAAsB,EACtB,2BAA2B,GAC3B,MAAM,sBAAsB,CAAC;AAC9B,YAAY,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAC9E,YAAY,EAAE,SAAS,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnE,YAAY,EAAE,UAAU,EAAE,gBAAgB,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AACzF,YAAY,EACX,aAAa,EACb,sBAAsB,EACtB,gCAAgC,EAChC,8BAA8B,EAC9B,kCAAkC,EAClC,uBAAuB,EACvB,sBAAsB,EACtB,oCAAoC,EACpC,gCAAgC,EAChC,2BAA2B,EAC3B,gCAAgC,EAChC,8BAA8B,EAC9B,sBAAsB,EACtB,8BAA8B,EAC9B,mBAAmB,EACnB,sBAAsB,EACtB,yBAAyB,EACzB,8BAA8B,EAC9B,uBAAuB,GACvB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,yBAAyB,EAAE,MAAM,qBAAqB,CAAC;AAEhE,OAAO,KAAK,QAAQ,MAAM,qBAAqB,CAAC;AAChD,cAAc,WAAW,CAAC;AAE1B,OAAO,EAAE,QAAQ,EAAE,CAAC;AACpB,YAAY,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAE5D,OAAO,EAAE,IAAI,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAC7D,YAAY,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,YAAY,EAAE,QAAQ,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAE9E,OAAO,EAAE,6BAA6B,EAAE,MAAM,wBAAwB,CAAC;AACvE,YAAY,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAEzF,cAAc,0BAA0B,CAAC"}
@@ -55,3 +55,4 @@ var local_apps_js_1 = require("./local-apps.js");
55
55
  Object.defineProperty(exports, "LOCAL_APPS", { enumerable: true, get: function () { return local_apps_js_1.LOCAL_APPS; } });
56
56
  var dataset_libraries_js_1 = require("./dataset-libraries.js");
57
57
  Object.defineProperty(exports, "DATASET_LIBRARIES_UI_ELEMENTS", { enumerable: true, get: function () { return dataset_libraries_js_1.DATASET_LIBRARIES_UI_ELEMENTS; } });
58
+ __exportStar(require("./inference-providers.js"), exports);
@@ -0,0 +1,10 @@
1
+ export declare const INFERENCE_PROVIDERS: readonly ["hf-inference", "fal-ai", "replicate", "sambanova", "together"];
2
+ export type InferenceProvider = (typeof INFERENCE_PROVIDERS)[number];
3
+ export declare const HF_HUB_INFERENCE_PROXY_TEMPLATE = "https://huggingface.co/api/inference-proxy/{{PROVIDER}}";
4
+ /**
5
+ * URL to set as baseUrl in the OpenAI SDK.
6
+ *
7
+ * TODO(Expose this from HfInference in the future?)
8
+ */
9
+ export declare function openAIbaseUrl(provider: InferenceProvider): string;
10
+ //# sourceMappingURL=inference-providers.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"inference-providers.d.ts","sourceRoot":"","sources":["../../src/inference-providers.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,mBAAmB,2EAA4E,CAAC;AAE7G,MAAM,MAAM,iBAAiB,GAAG,CAAC,OAAO,mBAAmB,CAAC,CAAC,MAAM,CAAC,CAAC;AAErE,eAAO,MAAM,+BAA+B,4DAA4D,CAAC;AAEzG;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,QAAQ,EAAE,iBAAiB,GAAG,MAAM,CAIjE"}
@@ -0,0 +1,16 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.HF_HUB_INFERENCE_PROXY_TEMPLATE = exports.INFERENCE_PROVIDERS = void 0;
4
+ exports.openAIbaseUrl = openAIbaseUrl;
5
+ exports.INFERENCE_PROVIDERS = ["hf-inference", "fal-ai", "replicate", "sambanova", "together"];
6
+ exports.HF_HUB_INFERENCE_PROXY_TEMPLATE = `https://huggingface.co/api/inference-proxy/{{PROVIDER}}`;
7
+ /**
8
+ * URL to set as baseUrl in the OpenAI SDK.
9
+ *
10
+ * TODO(Expose this from HfInference in the future?)
11
+ */
12
+ function openAIbaseUrl(provider) {
13
+ return provider === "hf-inference"
14
+ ? "https://api-inference.huggingface.co/v1/"
15
+ : exports.HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider);
16
+ }
@@ -5,10 +5,10 @@ export declare const asteroid: (model: ModelData) => string[];
5
5
  export declare const audioseal: (model: ModelData) => string[];
6
6
  export declare const bertopic: (model: ModelData) => string[];
7
7
  export declare const bm25s: (model: ModelData) => string[];
8
- export declare const cxr_foundation: (model: ModelData) => string[];
8
+ export declare const cxr_foundation: () => string[];
9
9
  export declare const depth_anything_v2: (model: ModelData) => string[];
10
10
  export declare const depth_pro: (model: ModelData) => string[];
11
- export declare const derm_foundation: (model: ModelData) => string[];
11
+ export declare const derm_foundation: () => string[];
12
12
  export declare const diffusers: (model: ModelData) => string[];
13
13
  export declare const diffusionkit: (model: ModelData) => string[];
14
14
  export declare const cartesia_pytorch: (model: ModelData) => string[];
@@ -1 +1 @@
1
- {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAaF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAqBvD,CAAC;AAEF,eAAO,MAAM,iBAAiB,UAAW,SAAS,KAAG,MAAM,EA6C1D,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAEF,eAAO,MAAM,eAAe,UAAW,SAAS,KAAG,MAAM,EAoBxD,CAAA;AAuCD,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAwCrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAmBrD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AA4EF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EA0BzD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EA2B7C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AASF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAoB7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAiB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAiBpD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAYjD,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC"}
1
+ {"version":3,"file":"model-libraries-snippets.d.ts","sourceRoot":"","sources":["../../src/model-libraries-snippets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAkBjD,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAkBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAkBlD,CAAC;AAaF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,cAAc,QAAO,MAAM,EAqBvC,CAAC;AAEF,eAAO,MAAM,iBAAiB,UAAW,SAAS,KAAG,MAAM,EA6C1D,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAwBlD,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,MAAM,EAoBxC,CAAC;AAuCF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAUlD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAwCrD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAgBzD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EAmBrD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAgB/C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAMlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EASlD,CAAC;AAIF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAMhD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAS9C,CAAC;AA4EF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EA0BzD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAOjD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,MAAM,EAQtC,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAKlD,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAsBlD,CAAC;AAEF,eAAO,MAAM,uBAAuB,UAAW,SAAS,KAAG,MAAM,EAehE,CAAC;AAiBF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAyBF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAOtD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,OAAO,QAA6B,MAAM,EAQtD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAanC,CAAC;AAsCF,eAAO,MAAM,OAAO,UAAW,SAAS,KAAG,MAAM,EAehD,CAAC;AAEF,eAAO,MAAM,kBAAkB,UAAW,SAAS,KAAG,MAAM,EAmC3D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EA2B7C,CAAC;AAEF,eAAO,MAAM,aAAa,UAAW,SAAS,KAAG,MAAM,EAEtD,CAAC;AASF,eAAO,MAAM,oBAAoB,UAAW,SAAS,KAAG,MAAM,EAoB7D,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAU9C,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAIpD,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAK/C,CAAC;AAkBF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAkBpD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,YAAY,UAAW,SAAS,KAAG,MAAM,EA4CrD,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAcvD,CAAC;AAiBF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAiB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAKjD,CAAC;AAEF,eAAO,MAAM,gBAAgB,UAAW,SAAS,KAAG,MAAM,EAMzD,CAAC;AAgBF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAEjD,CAAC;AAEF,eAAO,MAAM,MAAM,QAA6B,MAAM,EAMrD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAkB7C,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAIjD,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAInD,CAAC;AAEF,eAAO,MAAM,OAAO,QAAO,MAAM,EAYhC,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,SAAS,KAAG,MAAM,EAiBpD,CAAC;AAEF,eAAO,MAAM,QAAQ,UAAW,SAAS,KAAG,MAAM,EAYjD,CAAC;AAEF,eAAO,MAAM,GAAG,UAAW,SAAS,KAAG,MAAM,EAK5C,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC;AAEF,eAAO,MAAM,SAAS,UAAW,SAAS,KAAG,MAAM,EAIlD,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAQ7C,CAAC;AAEF,eAAO,MAAM,IAAI,UAAW,SAAS,KAAG,MAAM,EAI7C,CAAC;AAEF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAI/C,CAAC;AA4BF,eAAO,MAAM,MAAM,UAAW,SAAS,KAAG,MAAM,EAO/C,CAAC;AAEF,eAAO,MAAM,UAAU,UAAW,SAAS,KAAG,MAAM,EAUnD,CAAC;AAEF,eAAO,MAAM,UAAU,QAAO,MAAM,EAYnC,CAAC;AAEF,eAAO,MAAM,cAAc,UAAW,SAAS,KAAG,MAAM,EAKvD,CAAC;AAEF,eAAO,MAAM,KAAK,UAAW,SAAS,KAAG,MAAM,EAI9C,CAAC"}
@@ -86,7 +86,7 @@ const bm25s = (model) => [
86
86
  retriever = BM25HF.load_from_hub("${model.id}")`,
87
87
  ];
88
88
  exports.bm25s = bm25s;
89
- const cxr_foundation = (model) => [
89
+ const cxr_foundation = () => [
90
90
  `!git clone https://github.com/Google-Health/cxr-foundation.git
91
91
  import tensorflow as tf, sys, requests
92
92
  sys.path.append('cxr-foundation/python/')
@@ -179,7 +179,7 @@ focallength_px = prediction["focallength_px"]`;
179
179
  return [installSnippet, inferenceSnippet];
180
180
  };
181
181
  exports.depth_pro = depth_pro;
182
- const derm_foundation = (model) => [
182
+ const derm_foundation = () => [
183
183
  `from huggingface_hub import from_pretrained_keras
184
184
  import tensorflow as tf, requests
185
185
 
@@ -172,7 +172,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
172
172
  prettyLabel: string;
173
173
  repoName: string;
174
174
  repoUrl: string;
175
- snippets: (model: ModelData) => string[];
175
+ snippets: () => string[];
176
176
  filter: false;
177
177
  countDownloads: string;
178
178
  };
@@ -203,7 +203,7 @@ export declare const MODEL_LIBRARIES_UI_ELEMENTS: {
203
203
  prettyLabel: string;
204
204
  repoName: string;
205
205
  repoUrl: string;
206
- snippets: (model: ModelData) => string[];
206
+ snippets: () => string[];
207
207
  filter: false;
208
208
  countDownloads: string;
209
209
  };
@@ -1,17 +1,17 @@
1
+ import { type InferenceProvider } from "../inference-providers.js";
1
2
  import type { PipelineType } from "../pipelines.js";
2
3
  import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
3
4
  import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
4
- export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
5
- export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToken: string, opts?: {
5
+ export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
6
+ export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider, opts?: {
6
7
  streaming?: boolean;
7
8
  messages?: ChatCompletionInputMessage[];
8
9
  temperature?: GenerationParameters["temperature"];
9
10
  max_tokens?: GenerationParameters["max_tokens"];
10
11
  top_p?: GenerationParameters["top_p"];
11
- }) => InferenceSnippet;
12
- export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
13
- export declare const snippetFile: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
14
- export declare const curlSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, opts?: Record<string, unknown>) => InferenceSnippet>>;
15
- export declare function getCurlInferenceSnippet(model: ModelDataMinimal, accessToken: string, opts?: Record<string, unknown>): InferenceSnippet;
16
- export declare function hasCurlInferenceSnippet(model: Pick<ModelDataMinimal, "pipeline_tag">): boolean;
12
+ }) => InferenceSnippet[];
13
+ export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
14
+ export declare const snippetFile: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
15
+ export declare const curlSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider, opts?: Record<string, unknown>) => InferenceSnippet[]>>;
16
+ export declare function getCurlInferenceSnippet(model: ModelDataMinimal, accessToken: string, provider: InferenceProvider, opts?: Record<string, unknown>): InferenceSnippet[];
17
17
  //# sourceMappingURL=curl.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAErE,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAM1E,CAAC;AAEH,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;IACN,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACxC,WAAW,CAAC,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAClD,UAAU,CAAC,EAAE,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;CACtC,KACC,gBAkCF,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAM3F,CAAC;AAEH,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAKzE,CAAC;AAEH,eAAO,MAAM,YAAY,EAAE,OAAO,CACjC,MAAM,CACL,YAAY,EACZ,CAAC,KAAK,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,gBAAgB,CAClG,CA0BD,CAAC;AAEF,wBAAgB,uBAAuB,CACtC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,CAIlB;AAED,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,cAAc,CAAC,GAAG,OAAO,CAE9F"}
1
+ {"version":3,"file":"curl.d.ts","sourceRoot":"","sources":["../../../src/snippets/curl.ts"],"names":[],"mappings":"AAAA,OAAO,EAAmC,KAAK,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AACpG,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAErE,eAAO,MAAM,YAAY,UACjB,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EAelB,CAAC;AAEF,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,YACT,iBAAiB,SACpB;IACN,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACxC,WAAW,CAAC,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAClD,UAAU,CAAC,EAAE,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;CACtC,KACC,gBAAgB,EA0ClB,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAClC,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EAclB,CAAC;AAEF,eAAO,MAAM,WAAW,UAChB,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EAalB,CAAC;AAEF,eAAO,MAAM,YAAY,EAAE,OAAO,CACjC,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,iBAAiB,EAC3B,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,EAAE,CACvB,CA0BD,CAAC;AAEF,wBAAgB,uBAAuB,CACtC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,iBAAiB,EAC3B,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,EAAE,CAIpB"}
@@ -2,19 +2,31 @@
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.curlSnippets = exports.snippetFile = exports.snippetZeroShotClassification = exports.snippetTextGeneration = exports.snippetBasic = void 0;
4
4
  exports.getCurlInferenceSnippet = getCurlInferenceSnippet;
5
- exports.hasCurlInferenceSnippet = hasCurlInferenceSnippet;
5
+ const inference_providers_js_1 = require("../inference-providers.js");
6
6
  const common_js_1 = require("./common.js");
7
7
  const inputs_js_1 = require("./inputs.js");
8
- const snippetBasic = (model, accessToken) => ({
9
- content: `curl https://api-inference.huggingface.co/models/${model.id} \\
8
+ const snippetBasic = (model, accessToken, provider) => {
9
+ if (provider !== "hf-inference") {
10
+ return [];
11
+ }
12
+ return [
13
+ {
14
+ client: "curl",
15
+ content: `\
16
+ curl https://api-inference.huggingface.co/models/${model.id} \\
10
17
  -X POST \\
11
18
  -d '{"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model, true)}}' \\
12
19
  -H 'Content-Type: application/json' \\
13
20
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`,
14
- });
21
+ },
22
+ ];
23
+ };
15
24
  exports.snippetBasic = snippetBasic;
16
- const snippetTextGeneration = (model, accessToken, opts) => {
25
+ const snippetTextGeneration = (model, accessToken, provider, opts) => {
17
26
  if (model.tags.includes("conversational")) {
27
+ const baseUrl = provider === "hf-inference"
28
+ ? `https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions`
29
+ : inference_providers_js_1.HF_HUB_INFERENCE_PROXY_TEMPLATE.replace("{{PROVIDER}}", provider) + "/v1/chat/completions";
18
30
  // Conversational model detected, so we display a code snippet that features the Messages API
19
31
  const streaming = opts?.streaming ?? true;
20
32
  const exampleMessages = (0, inputs_js_1.getModelInputSnippet)(model);
@@ -24,48 +36,67 @@ const snippetTextGeneration = (model, accessToken, opts) => {
24
36
  max_tokens: opts?.max_tokens ?? 500,
25
37
  ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
26
38
  };
27
- return {
28
- content: `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\
39
+ return [
40
+ {
41
+ client: "curl",
42
+ content: `curl '${baseUrl}' \\
29
43
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}' \\
30
44
  -H 'Content-Type: application/json' \\
31
45
  --data '{
32
46
  "model": "${model.id}",
33
47
  "messages": ${(0, common_js_1.stringifyMessages)(messages, {
34
- indent: "\t",
35
- attributeKeyQuotes: true,
36
- customContentEscaper: (str) => str.replace(/'/g, "'\\''"),
37
- })},
48
+ indent: "\t",
49
+ attributeKeyQuotes: true,
50
+ customContentEscaper: (str) => str.replace(/'/g, "'\\''"),
51
+ })},
38
52
  ${(0, common_js_1.stringifyGenerationConfig)(config, {
39
- indent: "\n ",
40
- attributeKeyQuotes: true,
41
- attributeValueConnector: ": ",
42
- })},
53
+ indent: "\n ",
54
+ attributeKeyQuotes: true,
55
+ attributeValueConnector: ": ",
56
+ })},
43
57
  "stream": ${!!streaming}
44
58
  }'`,
45
- };
59
+ },
60
+ ];
46
61
  }
47
62
  else {
48
- return (0, exports.snippetBasic)(model, accessToken);
63
+ return (0, exports.snippetBasic)(model, accessToken, provider);
49
64
  }
50
65
  };
51
66
  exports.snippetTextGeneration = snippetTextGeneration;
52
- const snippetZeroShotClassification = (model, accessToken) => ({
53
- content: `curl https://api-inference.huggingface.co/models/${model.id} \\
67
+ const snippetZeroShotClassification = (model, accessToken, provider) => {
68
+ if (provider !== "hf-inference") {
69
+ return [];
70
+ }
71
+ return [
72
+ {
73
+ client: "curl",
74
+ content: `curl https://api-inference.huggingface.co/models/${model.id} \\
54
75
  -X POST \\
55
76
  -d '{"inputs": ${(0, inputs_js_1.getModelInputSnippet)(model, true)}, "parameters": {"candidate_labels": ["refund", "legal", "faq"]}}' \\
56
77
  -H 'Content-Type: application/json' \\
57
78
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`,
58
- });
79
+ },
80
+ ];
81
+ };
59
82
  exports.snippetZeroShotClassification = snippetZeroShotClassification;
60
- const snippetFile = (model, accessToken) => ({
61
- content: `curl https://api-inference.huggingface.co/models/${model.id} \\
83
+ const snippetFile = (model, accessToken, provider) => {
84
+ if (provider !== "hf-inference") {
85
+ return [];
86
+ }
87
+ return [
88
+ {
89
+ client: "curl",
90
+ content: `curl https://api-inference.huggingface.co/models/${model.id} \\
62
91
  -X POST \\
63
92
  --data-binary '@${(0, inputs_js_1.getModelInputSnippet)(model, true, true)}' \\
64
93
  -H 'Authorization: Bearer ${accessToken || `{API_TOKEN}`}'`,
65
- });
94
+ },
95
+ ];
96
+ };
66
97
  exports.snippetFile = snippetFile;
67
98
  exports.curlSnippets = {
68
- // Same order as in js/src/lib/interfaces/Types.ts
99
+ // Same order as in tasks/src/pipelines.ts
69
100
  "text-classification": exports.snippetBasic,
70
101
  "token-classification": exports.snippetBasic,
71
102
  "table-question-answering": exports.snippetBasic,
@@ -90,11 +121,8 @@ exports.curlSnippets = {
90
121
  "object-detection": exports.snippetFile,
91
122
  "image-segmentation": exports.snippetFile,
92
123
  };
93
- function getCurlInferenceSnippet(model, accessToken, opts) {
124
+ function getCurlInferenceSnippet(model, accessToken, provider, opts) {
94
125
  return model.pipeline_tag && model.pipeline_tag in exports.curlSnippets
95
- ? exports.curlSnippets[model.pipeline_tag]?.(model, accessToken, opts) ?? { content: "" }
96
- : { content: "" };
97
- }
98
- function hasCurlInferenceSnippet(model) {
99
- return !!model.pipeline_tag && model.pipeline_tag in exports.curlSnippets;
126
+ ? exports.curlSnippets[model.pipeline_tag]?.(model, accessToken, provider, opts) ?? []
127
+ : [];
100
128
  }
@@ -1,19 +1,20 @@
1
+ import { type InferenceProvider } from "../inference-providers.js";
1
2
  import type { PipelineType } from "../pipelines.js";
2
3
  import type { ChatCompletionInputMessage, GenerationParameters } from "../tasks/index.js";
3
4
  import type { InferenceSnippet, ModelDataMinimal } from "./types.js";
4
- export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
5
- export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToken: string, opts?: {
5
+ export declare const snippetBasic: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
6
+ export declare const snippetTextGeneration: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider, opts?: {
6
7
  streaming?: boolean;
7
8
  messages?: ChatCompletionInputMessage[];
8
9
  temperature?: GenerationParameters["temperature"];
9
10
  max_tokens?: GenerationParameters["max_tokens"];
10
11
  top_p?: GenerationParameters["top_p"];
11
- }) => InferenceSnippet | InferenceSnippet[];
12
- export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
13
- export declare const snippetTextToImage: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
14
- export declare const snippetTextToAudio: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
15
- export declare const snippetFile: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet;
16
- export declare const jsSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, opts?: Record<string, unknown>) => InferenceSnippet | InferenceSnippet[]>>;
17
- export declare function getJsInferenceSnippet(model: ModelDataMinimal, accessToken: string, opts?: Record<string, unknown>): InferenceSnippet | InferenceSnippet[];
18
- export declare function hasJsInferenceSnippet(model: ModelDataMinimal): boolean;
12
+ }) => InferenceSnippet[];
13
+ export declare const snippetZeroShotClassification: (model: ModelDataMinimal, accessToken: string) => InferenceSnippet[];
14
+ export declare const snippetTextToImage: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
15
+ export declare const snippetTextToAudio: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
16
+ export declare const snippetAutomaticSpeechRecognition: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
17
+ export declare const snippetFile: (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider) => InferenceSnippet[];
18
+ export declare const jsSnippets: Partial<Record<PipelineType, (model: ModelDataMinimal, accessToken: string, provider: InferenceProvider, opts?: Record<string, unknown>) => InferenceSnippet[]>>;
19
+ export declare function getJsInferenceSnippet(model: ModelDataMinimal, accessToken: string, provider: InferenceProvider, opts?: Record<string, unknown>): InferenceSnippet[];
19
20
  //# sourceMappingURL=js.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAErE,eAAO,MAAM,YAAY,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAoB1E,CAAC;AAEH,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,SACZ;IACN,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACxC,WAAW,CAAC,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAClD,UAAU,CAAC,EAAE,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;CACtC,KACC,gBAAgB,GAAG,gBAAgB,EA2GrC,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAsB3F,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAmBhF,CAAC;AAEH,eAAO,MAAM,kBAAkB,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAuCjF,CAAC;AAEF,eAAO,MAAM,WAAW,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAqBzE,CAAC;AAEH,eAAO,MAAM,UAAU,EAAE,OAAO,CAC/B,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,GAAG,gBAAgB,EAAE,CAC1C,CA0BD,CAAC;AAEF,wBAAgB,qBAAqB,CACpC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,GAAG,gBAAgB,EAAE,CAIvC;AAED,wBAAgB,qBAAqB,CAAC,KAAK,EAAE,gBAAgB,GAAG,OAAO,CAEtE"}
1
+ {"version":3,"file":"js.d.ts","sourceRoot":"","sources":["../../../src/snippets/js.ts"],"names":[],"mappings":"AAAA,OAAO,EAAiB,KAAK,iBAAiB,EAAE,MAAM,2BAA2B,CAAC;AAClF,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,KAAK,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAG1F,OAAO,KAAK,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAgBrE,eAAO,MAAM,YAAY,UACjB,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EA8ClB,CAAC;AAEF,eAAO,MAAM,qBAAqB,UAC1B,gBAAgB,eACV,MAAM,YACT,iBAAiB,SACpB;IACN,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB,QAAQ,CAAC,EAAE,0BAA0B,EAAE,CAAC;IACxC,WAAW,CAAC,EAAE,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAClD,UAAU,CAAC,EAAE,oBAAoB,CAAC,YAAY,CAAC,CAAC;IAChD,KAAK,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,CAAC;CACtC,KACC,gBAAgB,EA6GlB,CAAC;AAEF,eAAO,MAAM,6BAA6B,UAAW,gBAAgB,eAAe,MAAM,KAAG,gBAAgB,EA2B5G,CAAC;AAEF,eAAO,MAAM,kBAAkB,UACvB,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EA4ClB,CAAC;AAEF,eAAO,MAAM,kBAAkB,UACvB,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EAgDlB,CAAC;AAEF,eAAO,MAAM,iCAAiC,UACtC,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EAsBlB,CAAC;AAEF,eAAO,MAAM,WAAW,UAChB,gBAAgB,eACV,MAAM,YACT,iBAAiB,KACzB,gBAAgB,EA6BlB,CAAC;AAEF,eAAO,MAAM,UAAU,EAAE,OAAO,CAC/B,MAAM,CACL,YAAY,EACZ,CACC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,iBAAiB,EAC3B,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAC1B,gBAAgB,EAAE,CACvB,CA0BD,CAAC;AAEF,wBAAgB,qBAAqB,CACpC,KAAK,EAAE,gBAAgB,EACvB,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,iBAAiB,EAC3B,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC5B,gBAAgB,EAAE,CAIpB"}