@huggingface/inference 3.13.0 → 3.13.2

package/dist/index.cjs

@@ -235,6 +235,43 @@ var BaseTextGenerationTask = class extends TaskProviderHelper {
   }
 };
 
+// src/utils/base64FromBytes.ts
+function base64FromBytes(arr) {
+  if (globalThis.Buffer) {
+    return globalThis.Buffer.from(arr).toString("base64");
+  } else {
+    const bin = [];
+    arr.forEach((byte) => {
+      bin.push(String.fromCharCode(byte));
+    });
+    return globalThis.btoa(bin.join(""));
+  }
+}
+
+// src/utils/pick.ts
+function pick(o, props) {
+  return Object.assign(
+    {},
+    ...props.map((prop) => {
+      if (o[prop] !== void 0) {
+        return { [prop]: o[prop] };
+      }
+    })
+  );
+}
+
+// src/utils/typedInclude.ts
+function typedInclude(arr, v) {
+  return arr.includes(v);
+}
+
+// src/utils/omit.ts
+function omit(o, props) {
+  const propsArr = Array.isArray(props) ? props : [props];
+  const letsKeep = Object.keys(o).filter((prop) => !typedInclude(propsArr, prop));
+  return pick(o, letsKeep);
+}
+
 // src/providers/hf-inference.ts
 var EQUIVALENT_SENTENCE_TRANSFORMERS_TASKS = ["feature-extraction", "sentence-similarity"];
 var HFInferenceTask = class extends TaskProviderHelper {
@@ -342,6 +379,12 @@ var HFInferenceAutomaticSpeechRecognitionTask = class extends HFInferenceTask {
   async getResponse(response) {
     return response;
   }
+  async preparePayloadAsync(args) {
+    return "data" in args ? args : {
+      ...omit(args, "inputs"),
+      data: args.inputs
+    };
+  }
 };
 var HFInferenceAudioToAudioTask = class extends HFInferenceTask {
   async getResponse(response) {
@@ -410,6 +453,22 @@ var HFInferenceImageToTextTask = class extends HFInferenceTask {
   }
 };
 var HFInferenceImageToImageTask = class extends HFInferenceTask {
+  async preparePayloadAsync(args) {
+    if (!args.parameters) {
+      return {
+        ...args,
+        model: args.model,
+        data: args.inputs
+      };
+    } else {
+      return {
+        ...args,
+        inputs: base64FromBytes(
+          new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())
+        )
+      };
+    }
+  }
   async getResponse(response) {
     if (response instanceof Blob) {
       return response;
@@ -566,11 +625,6 @@ var HFInferenceTextToAudioTask = class extends HFInferenceTask {
   }
 };
 
-// src/utils/typedInclude.ts
-function typedInclude(arr, v) {
-  return arr.includes(v);
-}
-
 // src/lib/getInferenceProviderMapping.ts
 var inferenceProviderMappingCache = /* @__PURE__ */ new Map();
 async function fetchInferenceProviderMappingForModel(modelId, accessToken, options) {
@@ -656,25 +710,6 @@ function delay(ms) {
   });
 }
 
-// src/utils/pick.ts
-function pick(o, props) {
-  return Object.assign(
-    {},
-    ...props.map((prop) => {
-      if (o[prop] !== void 0) {
-        return { [prop]: o[prop] };
-      }
-    })
-  );
-}
-
-// src/utils/omit.ts
-function omit(o, props) {
-  const propsArr = Array.isArray(props) ? props : [props];
-  const letsKeep = Object.keys(o).filter((prop) => !typedInclude(propsArr, prop));
-  return pick(o, letsKeep);
-}
-
 // src/providers/black-forest-labs.ts
 var BLACK_FOREST_LABS_AI_API_BASE_URL = "https://api.us1.bfl.ai";
 var BlackForestLabsTextToImageTask = class extends TaskProviderHelper {
@@ -881,6 +916,27 @@ var FalAIAutomaticSpeechRecognitionTask = class extends FalAITask {
     }
     return { text: res.text };
   }
+  async preparePayloadAsync(args) {
+    const blob = "data" in args && args.data instanceof Blob ? args.data : "inputs" in args ? args.inputs : void 0;
+    const contentType = blob?.type;
+    if (!contentType) {
+      throw new Error(
+        `Unable to determine the input's content-type. Make sure your are passing a Blob when using provider fal-ai.`
+      );
+    }
+    if (!FAL_AI_SUPPORTED_BLOB_TYPES.includes(contentType)) {
+      throw new Error(
+        `Provider fal-ai does not support blob type ${contentType} - supported content types are: ${FAL_AI_SUPPORTED_BLOB_TYPES.join(
+          ", "
+        )}`
+      );
+    }
+    const base64audio = base64FromBytes(new Uint8Array(await blob.arrayBuffer()));
+    return {
+      ..."data" in args ? omit(args, "data") : omit(args, "inputs"),
+      audio_url: `data:${contentType};base64,${base64audio}`
+    };
+  }
 };
 var FalAITextToSpeechTask = class extends FalAITask {
   preparePayload(params) {
@@ -1467,10 +1523,8 @@ var PROVIDERS = {
   }
 };
 function getProviderHelper(provider, task) {
-  if (provider === "hf-inference") {
-    if (!task) {
-      return new HFInferenceTask();
-    }
+  if (provider === "hf-inference" && !task || provider === "auto") {
+    return new HFInferenceTask();
   }
   if (!task) {
     throw new Error("you need to provide a task name when using an external provider, e.g. 'text-to-image'");
@@ -1489,7 +1543,7 @@ function getProviderHelper(provider, task) {
 
 // package.json
 var name = "@huggingface/inference";
-var version = "3.13.0";
+var version = "3.13.2";
 
 // src/lib/makeRequestOptions.ts
 var tasks = null;
@@ -1890,24 +1944,11 @@ async function audioToAudio(args, options) {
   return providerHelper.getResponse(res);
 }
 
-// src/utils/base64FromBytes.ts
-function base64FromBytes(arr) {
-  if (globalThis.Buffer) {
-    return globalThis.Buffer.from(arr).toString("base64");
-  } else {
-    const bin = [];
-    arr.forEach((byte) => {
-      bin.push(String.fromCharCode(byte));
-    });
-    return globalThis.btoa(bin.join(""));
-  }
-}
-
 // src/tasks/audio/automaticSpeechRecognition.ts
 async function automaticSpeechRecognition(args, options) {
   const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
   const providerHelper = getProviderHelper(provider, "automatic-speech-recognition");
-  const payload = await buildPayload(args);
+  const payload = await providerHelper.preparePayloadAsync(args);
   const { data: res } = await innerRequest(payload, providerHelper, {
     ...options,
     task: "automatic-speech-recognition"
@@ -1918,31 +1959,6 @@ async function automaticSpeechRecognition(args, options) {
   }
   return providerHelper.getResponse(res);
 }
-async function buildPayload(args) {
-  if (args.provider === "fal-ai") {
-    const blob = "data" in args && args.data instanceof Blob ? args.data : "inputs" in args ? args.inputs : void 0;
-    const contentType = blob?.type;
-    if (!contentType) {
-      throw new Error(
-        `Unable to determine the input's content-type. Make sure your are passing a Blob when using provider fal-ai.`
-      );
-    }
-    if (!FAL_AI_SUPPORTED_BLOB_TYPES.includes(contentType)) {
-      throw new Error(
-        `Provider fal-ai does not support blob type ${contentType} - supported content types are: ${FAL_AI_SUPPORTED_BLOB_TYPES.join(
-          ", "
-        )}`
-      );
-    }
-    const base64audio = base64FromBytes(new Uint8Array(await blob.arrayBuffer()));
-    return {
-      ..."data" in args ? omit(args, "data") : omit(args, "inputs"),
-      audio_url: `data:${contentType};base64,${base64audio}`
-    };
-  } else {
-    return preparePayload(args);
-  }
-}
 
 // src/tasks/audio/textToSpeech.ts
 async function textToSpeech(args, options) {
@@ -1988,22 +2004,8 @@ async function imageSegmentation(args, options) {
 async function imageToImage(args, options) {
   const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
   const providerHelper = getProviderHelper(provider, "image-to-image");
-  let reqArgs;
-  if (!args.parameters) {
-    reqArgs = {
-      accessToken: args.accessToken,
-      model: args.model,
-      data: args.inputs
-    };
-  } else {
-    reqArgs = {
-      ...args,
-      inputs: base64FromBytes(
-        new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())
-      )
-    };
-  }
-  const { data: res } = await innerRequest(reqArgs, providerHelper, {
+  const payload = await providerHelper.preparePayloadAsync(args);
+  const { data: res } = await innerRequest(payload, providerHelper, {
     ...options,
     task: "image-to-image"
   });
@@ -2468,7 +2470,7 @@ const video = await client.textToVideo({
       "textToImage": '{% if provider == "fal-ai" %}\nimport fal_client\n\n{% if providerInputs.asObj.loras is defined and providerInputs.asObj.loras != none %}\nresult = fal_client.subscribe(\n    "{{ providerModelId }}",\n    arguments={\n        "prompt": {{ inputs.asObj.inputs }},\n        "loras":{{ providerInputs.asObj.loras | tojson }},\n    },\n)\n{% else %}\nresult = fal_client.subscribe(\n    "{{ providerModelId }}",\n    arguments={\n        "prompt": {{ inputs.asObj.inputs }},\n    },\n)\n{% endif %} \nprint(result)\n{% endif %} '
     },
     "huggingface_hub": {
-      "basic": 'result = client.{{ methodName }}(\n    inputs={{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n)',
+      "basic": 'result = client.{{ methodName }}(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n)',
       "basicAudio": 'output = client.{{ methodName }}({{ inputs.asObj.inputs }}, model="{{ model.id }}")',
       "basicImage": 'output = client.{{ methodName }}({{ inputs.asObj.inputs }}, model="{{ model.id }}")',
       "conversational": 'completion = client.chat.completions.create(\n    model="{{ model.id }}",\n{{ inputs.asPythonString }}\n)\n\nprint(completion.choices[0].message) ',
@@ -2476,6 +2478,8 @@ const video = await client.textToVideo({
       "documentQuestionAnswering": 'output = client.document_question_answering(\n    "{{ inputs.asObj.image }}",\n    question="{{ inputs.asObj.question }}",\n    model="{{ model.id }}",\n) ',
       "imageToImage": '# output is a PIL.Image object\nimage = client.image_to_image(\n    "{{ inputs.asObj.inputs }}",\n    prompt="{{ inputs.asObj.parameters.prompt }}",\n    model="{{ model.id }}",\n) ',
       "importInferenceClient": 'from huggingface_hub import InferenceClient\n\nclient = InferenceClient(\n    provider="{{ provider }}",\n    api_key="{{ accessToken }}",\n{% if billTo %}\n    bill_to="{{ billTo }}",\n{% endif %}\n)',
+      "questionAnswering": 'answer = client.question_answering(\n    question="{{ inputs.asObj.question }}",\n    context="{{ inputs.asObj.context }}",\n    model="{{ model.id }}",\n) ',
+      "tableQuestionAnswering": 'answer = client.question_answering(\n    query="{{ inputs.asObj.query }}",\n    table={{ inputs.asObj.table }},\n    model="{{ model.id }}",\n) ',
       "textToImage": '# output is a PIL.Image object\nimage = client.text_to_image(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n) ',
       "textToSpeech": '# audio is returned as bytes\naudio = client.text_to_speech(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n) \n',
       "textToVideo": 'video = client.text_to_video(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n) '
@@ -2617,7 +2621,7 @@ var snippetGenerator = (templateName, inputPreparationFn) => {
     }
     let providerHelper;
     try {
-      providerHelper = getProviderHelper(provider === "auto" ? "hf-inference" : provider, task);
+      providerHelper = getProviderHelper(provider, task);
     } catch (e) {
       console.error(`Failed to get provider helper for ${provider} (${task})`, e);
       return [];
@@ -2729,6 +2733,14 @@ var prepareConversationalInput = (model, opts) => {
     ...opts?.top_p ? { top_p: opts?.top_p } : void 0
   };
 };
+var prepareQuestionAnsweringInput = (model) => {
+  const data = JSON.parse((0, import_tasks.getModelInputSnippet)(model));
+  return { question: data.question, context: data.context };
+};
+var prepareTableQuestionAnsweringInput = (model) => {
+  const data = JSON.parse((0, import_tasks.getModelInputSnippet)(model));
+  return { query: data.query, table: JSON.stringify(data.table) };
+};
 var snippets = {
   "audio-classification": snippetGenerator("basicAudio"),
   "audio-to-audio": snippetGenerator("basicAudio"),
@@ -2742,12 +2754,12 @@ var snippets = {
   "image-to-image": snippetGenerator("imageToImage", prepareImageToImageInput),
   "image-to-text": snippetGenerator("basicImage"),
   "object-detection": snippetGenerator("basicImage"),
-  "question-answering": snippetGenerator("basic"),
+  "question-answering": snippetGenerator("questionAnswering", prepareQuestionAnsweringInput),
   "sentence-similarity": snippetGenerator("basic"),
   summarization: snippetGenerator("basic"),
   "tabular-classification": snippetGenerator("tabular"),
   "tabular-regression": snippetGenerator("tabular"),
-  "table-question-answering": snippetGenerator("basic"),
+  "table-question-answering": snippetGenerator("tableQuestionAnswering", prepareTableQuestionAnsweringInput),
   "text-classification": snippetGenerator("basic"),
   "text-generation": snippetGenerator("basic"),
   "text-to-audio": snippetGenerator("textToAudio"),

package/dist/index.js

@@ -177,6 +177,43 @@ var BaseTextGenerationTask = class extends TaskProviderHelper {
   }
 };
 
+// src/utils/base64FromBytes.ts
+function base64FromBytes(arr) {
+  if (globalThis.Buffer) {
+    return globalThis.Buffer.from(arr).toString("base64");
+  } else {
+    const bin = [];
+    arr.forEach((byte) => {
+      bin.push(String.fromCharCode(byte));
+    });
+    return globalThis.btoa(bin.join(""));
+  }
+}
+
+// src/utils/pick.ts
+function pick(o, props) {
+  return Object.assign(
+    {},
+    ...props.map((prop) => {
+      if (o[prop] !== void 0) {
+        return { [prop]: o[prop] };
+      }
+    })
+  );
+}
+
+// src/utils/typedInclude.ts
+function typedInclude(arr, v) {
+  return arr.includes(v);
+}
+
+// src/utils/omit.ts
+function omit(o, props) {
+  const propsArr = Array.isArray(props) ? props : [props];
+  const letsKeep = Object.keys(o).filter((prop) => !typedInclude(propsArr, prop));
+  return pick(o, letsKeep);
+}
+
 // src/providers/hf-inference.ts
 var EQUIVALENT_SENTENCE_TRANSFORMERS_TASKS = ["feature-extraction", "sentence-similarity"];
 var HFInferenceTask = class extends TaskProviderHelper {
@@ -284,6 +321,12 @@ var HFInferenceAutomaticSpeechRecognitionTask = class extends HFInferenceTask {
   async getResponse(response) {
     return response;
   }
+  async preparePayloadAsync(args) {
+    return "data" in args ? args : {
+      ...omit(args, "inputs"),
+      data: args.inputs
+    };
+  }
 };
 var HFInferenceAudioToAudioTask = class extends HFInferenceTask {
   async getResponse(response) {
@@ -352,6 +395,22 @@ var HFInferenceImageToTextTask = class extends HFInferenceTask {
   }
 };
 var HFInferenceImageToImageTask = class extends HFInferenceTask {
+  async preparePayloadAsync(args) {
+    if (!args.parameters) {
+      return {
+        ...args,
+        model: args.model,
+        data: args.inputs
+      };
+    } else {
+      return {
+        ...args,
+        inputs: base64FromBytes(
+          new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())
+        )
+      };
+    }
+  }
   async getResponse(response) {
     if (response instanceof Blob) {
       return response;
@@ -508,11 +567,6 @@ var HFInferenceTextToAudioTask = class extends HFInferenceTask {
   }
 };
 
-// src/utils/typedInclude.ts
-function typedInclude(arr, v) {
-  return arr.includes(v);
-}
-
 // src/lib/getInferenceProviderMapping.ts
 var inferenceProviderMappingCache = /* @__PURE__ */ new Map();
 async function fetchInferenceProviderMappingForModel(modelId, accessToken, options) {
@@ -598,25 +652,6 @@ function delay(ms) {
   });
 }
 
-// src/utils/pick.ts
-function pick(o, props) {
-  return Object.assign(
-    {},
-    ...props.map((prop) => {
-      if (o[prop] !== void 0) {
-        return { [prop]: o[prop] };
-      }
-    })
-  );
-}
-
-// src/utils/omit.ts
-function omit(o, props) {
-  const propsArr = Array.isArray(props) ? props : [props];
-  const letsKeep = Object.keys(o).filter((prop) => !typedInclude(propsArr, prop));
-  return pick(o, letsKeep);
-}
-
 // src/providers/black-forest-labs.ts
 var BLACK_FOREST_LABS_AI_API_BASE_URL = "https://api.us1.bfl.ai";
 var BlackForestLabsTextToImageTask = class extends TaskProviderHelper {
@@ -823,6 +858,27 @@ var FalAIAutomaticSpeechRecognitionTask = class extends FalAITask {
     }
     return { text: res.text };
   }
+  async preparePayloadAsync(args) {
+    const blob = "data" in args && args.data instanceof Blob ? args.data : "inputs" in args ? args.inputs : void 0;
+    const contentType = blob?.type;
+    if (!contentType) {
+      throw new Error(
+        `Unable to determine the input's content-type. Make sure your are passing a Blob when using provider fal-ai.`
+      );
+    }
+    if (!FAL_AI_SUPPORTED_BLOB_TYPES.includes(contentType)) {
+      throw new Error(
+        `Provider fal-ai does not support blob type ${contentType} - supported content types are: ${FAL_AI_SUPPORTED_BLOB_TYPES.join(
+          ", "
+        )}`
+      );
+    }
+    const base64audio = base64FromBytes(new Uint8Array(await blob.arrayBuffer()));
+    return {
+      ..."data" in args ? omit(args, "data") : omit(args, "inputs"),
+      audio_url: `data:${contentType};base64,${base64audio}`
+    };
+  }
 };
 var FalAITextToSpeechTask = class extends FalAITask {
   preparePayload(params) {
@@ -1409,10 +1465,8 @@ var PROVIDERS = {
   }
 };
 function getProviderHelper(provider, task) {
-  if (provider === "hf-inference") {
-    if (!task) {
-      return new HFInferenceTask();
-    }
+  if (provider === "hf-inference" && !task || provider === "auto") {
+    return new HFInferenceTask();
   }
   if (!task) {
     throw new Error("you need to provide a task name when using an external provider, e.g. 'text-to-image'");
@@ -1431,7 +1485,7 @@ function getProviderHelper(provider, task) {
 
 // package.json
 var name = "@huggingface/inference";
-var version = "3.13.0";
+var version = "3.13.2";
 
 // src/lib/makeRequestOptions.ts
 var tasks = null;
@@ -1832,24 +1886,11 @@ async function audioToAudio(args, options) {
   return providerHelper.getResponse(res);
 }
 
-// src/utils/base64FromBytes.ts
-function base64FromBytes(arr) {
-  if (globalThis.Buffer) {
-    return globalThis.Buffer.from(arr).toString("base64");
-  } else {
-    const bin = [];
-    arr.forEach((byte) => {
-      bin.push(String.fromCharCode(byte));
-    });
-    return globalThis.btoa(bin.join(""));
-  }
-}
-
 // src/tasks/audio/automaticSpeechRecognition.ts
 async function automaticSpeechRecognition(args, options) {
   const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
   const providerHelper = getProviderHelper(provider, "automatic-speech-recognition");
-  const payload = await buildPayload(args);
+  const payload = await providerHelper.preparePayloadAsync(args);
   const { data: res } = await innerRequest(payload, providerHelper, {
     ...options,
     task: "automatic-speech-recognition"
@@ -1860,31 +1901,6 @@ async function automaticSpeechRecognition(args, options) {
   }
   return providerHelper.getResponse(res);
 }
-async function buildPayload(args) {
-  if (args.provider === "fal-ai") {
-    const blob = "data" in args && args.data instanceof Blob ? args.data : "inputs" in args ? args.inputs : void 0;
-    const contentType = blob?.type;
-    if (!contentType) {
-      throw new Error(
-        `Unable to determine the input's content-type. Make sure your are passing a Blob when using provider fal-ai.`
-      );
-    }
-    if (!FAL_AI_SUPPORTED_BLOB_TYPES.includes(contentType)) {
-      throw new Error(
-        `Provider fal-ai does not support blob type ${contentType} - supported content types are: ${FAL_AI_SUPPORTED_BLOB_TYPES.join(
-          ", "
-        )}`
-      );
-    }
-    const base64audio = base64FromBytes(new Uint8Array(await blob.arrayBuffer()));
-    return {
-      ..."data" in args ? omit(args, "data") : omit(args, "inputs"),
-      audio_url: `data:${contentType};base64,${base64audio}`
-    };
-  } else {
-    return preparePayload(args);
-  }
-}
 
 // src/tasks/audio/textToSpeech.ts
 async function textToSpeech(args, options) {
@@ -1930,22 +1946,8 @@ async function imageSegmentation(args, options) {
 async function imageToImage(args, options) {
   const provider = await resolveProvider(args.provider, args.model, args.endpointUrl);
   const providerHelper = getProviderHelper(provider, "image-to-image");
-  let reqArgs;
-  if (!args.parameters) {
-    reqArgs = {
-      accessToken: args.accessToken,
-      model: args.model,
-      data: args.inputs
-    };
-  } else {
-    reqArgs = {
-      ...args,
-      inputs: base64FromBytes(
-        new Uint8Array(args.inputs instanceof ArrayBuffer ? args.inputs : await args.inputs.arrayBuffer())
-      )
-    };
-  }
-  const { data: res } = await innerRequest(reqArgs, providerHelper, {
+  const payload = await providerHelper.preparePayloadAsync(args);
+  const { data: res } = await innerRequest(payload, providerHelper, {
     ...options,
     task: "image-to-image"
   });
@@ -2413,7 +2415,7 @@ const video = await client.textToVideo({
       "textToImage": '{% if provider == "fal-ai" %}\nimport fal_client\n\n{% if providerInputs.asObj.loras is defined and providerInputs.asObj.loras != none %}\nresult = fal_client.subscribe(\n    "{{ providerModelId }}",\n    arguments={\n        "prompt": {{ inputs.asObj.inputs }},\n        "loras":{{ providerInputs.asObj.loras | tojson }},\n    },\n)\n{% else %}\nresult = fal_client.subscribe(\n    "{{ providerModelId }}",\n    arguments={\n        "prompt": {{ inputs.asObj.inputs }},\n    },\n)\n{% endif %} \nprint(result)\n{% endif %} '
     },
     "huggingface_hub": {
-      "basic": 'result = client.{{ methodName }}(\n    inputs={{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n)',
+      "basic": 'result = client.{{ methodName }}(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n)',
       "basicAudio": 'output = client.{{ methodName }}({{ inputs.asObj.inputs }}, model="{{ model.id }}")',
       "basicImage": 'output = client.{{ methodName }}({{ inputs.asObj.inputs }}, model="{{ model.id }}")',
       "conversational": 'completion = client.chat.completions.create(\n    model="{{ model.id }}",\n{{ inputs.asPythonString }}\n)\n\nprint(completion.choices[0].message) ',
@@ -2421,6 +2423,8 @@ const video = await client.textToVideo({
       "documentQuestionAnswering": 'output = client.document_question_answering(\n    "{{ inputs.asObj.image }}",\n    question="{{ inputs.asObj.question }}",\n    model="{{ model.id }}",\n) ',
       "imageToImage": '# output is a PIL.Image object\nimage = client.image_to_image(\n    "{{ inputs.asObj.inputs }}",\n    prompt="{{ inputs.asObj.parameters.prompt }}",\n    model="{{ model.id }}",\n) ',
       "importInferenceClient": 'from huggingface_hub import InferenceClient\n\nclient = InferenceClient(\n    provider="{{ provider }}",\n    api_key="{{ accessToken }}",\n{% if billTo %}\n    bill_to="{{ billTo }}",\n{% endif %}\n)',
+      "questionAnswering": 'answer = client.question_answering(\n    question="{{ inputs.asObj.question }}",\n    context="{{ inputs.asObj.context }}",\n    model="{{ model.id }}",\n) ',
+      "tableQuestionAnswering": 'answer = client.question_answering(\n    query="{{ inputs.asObj.query }}",\n    table={{ inputs.asObj.table }},\n    model="{{ model.id }}",\n) ',
       "textToImage": '# output is a PIL.Image object\nimage = client.text_to_image(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n) ',
       "textToSpeech": '# audio is returned as bytes\naudio = client.text_to_speech(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n) \n',
       "textToVideo": 'video = client.text_to_video(\n    {{ inputs.asObj.inputs }},\n    model="{{ model.id }}",\n) '
@@ -2562,7 +2566,7 @@ var snippetGenerator = (templateName, inputPreparationFn) => {
     }
     let providerHelper;
     try {
-      providerHelper = getProviderHelper(provider === "auto" ? "hf-inference" : provider, task);
+      providerHelper = getProviderHelper(provider, task);
     } catch (e) {
       console.error(`Failed to get provider helper for ${provider} (${task})`, e);
       return [];
@@ -2674,6 +2678,14 @@ var prepareConversationalInput = (model, opts) => {
     ...opts?.top_p ? { top_p: opts?.top_p } : void 0
   };
 };
+var prepareQuestionAnsweringInput = (model) => {
+  const data = JSON.parse(getModelInputSnippet(model));
+  return { question: data.question, context: data.context };
+};
+var prepareTableQuestionAnsweringInput = (model) => {
+  const data = JSON.parse(getModelInputSnippet(model));
+  return { query: data.query, table: JSON.stringify(data.table) };
+};
 var snippets = {
   "audio-classification": snippetGenerator("basicAudio"),
   "audio-to-audio": snippetGenerator("basicAudio"),
@@ -2687,12 +2699,12 @@ var snippets = {
   "image-to-image": snippetGenerator("imageToImage", prepareImageToImageInput),
   "image-to-text": snippetGenerator("basicImage"),
   "object-detection": snippetGenerator("basicImage"),
-  "question-answering": snippetGenerator("basic"),
+  "question-answering": snippetGenerator("questionAnswering", prepareQuestionAnsweringInput),
   "sentence-similarity": snippetGenerator("basic"),
   summarization: snippetGenerator("basic"),
   "tabular-classification": snippetGenerator("tabular"),
   "tabular-regression": snippetGenerator("tabular"),
-  "table-question-answering": snippetGenerator("basic"),
+  "table-question-answering": snippetGenerator("tableQuestionAnswering", prepareTableQuestionAnsweringInput),
   "text-classification": snippetGenerator("basic"),
   "text-generation": snippetGenerator("basic"),
   "text-to-audio": snippetGenerator("textToAudio"),

package/dist/src/lib/getDefaultTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"getDefaultTask.d.ts","sourceRoot":"","sources":["../../../src/lib/getDefaultTask.ts"],"names":[],"mappings":"AAYA,MAAM,WAAW,kBAAkB;IAClC,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;CACrB;AAED;;;;;GAKG;AACH,wBAAsB,cAAc,CACnC,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,GAAG,SAAS,EAC/B,OAAO,CAAC,EAAE,kBAAkB,GAC1B,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAkCxB"}
+{"version":3,"file":"getDefaultTask.d.ts","sourceRoot":"","sources":["../../../src/lib/getDefaultTask.ts"],"names":[],"mappings":"AAYA,MAAM,WAAW,kBAAkB;IAClC,KAAK,CAAC,EAAE,OAAO,KAAK,CAAC;CACrB;AAED;;;;;GAKG;AACH,wBAAsB,cAAc,CACnC,KAAK,EAAE,MAAM,EACb,WAAW,EAAE,MAAM,GAAG,SAAS,EAC/B,OAAO,CAAC,EAAE,kBAAkB,GAC1B,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAmCxB"}