@huggingface/inference 1.6.1 → 1.6.2

package/dist/index.js

@@ -23,7 +23,7 @@ __export(src_exports, {
 });
 module.exports = __toCommonJS(src_exports);
 
-// src/utils.ts
+// src/utils/to-array.ts
 function toArray(obj) {
   if (Array.isArray(obj)) {
     return obj;

package/dist/index.mjs

@@ -1,4 +1,4 @@
-// src/utils.ts
+// src/utils/to-array.ts
 function toArray(obj) {
   if (Array.isArray(obj)) {
     return obj;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@huggingface/inference",
-  "version": "1.6.1",
+  "version": "1.6.2",
   "license": "MIT",
   "author": "Tim Mikeladze <tim.mikeladze@gmail.com>",
   "description": "Typescript wrapper for the Hugging Face Inference API",

package/src/HfInference.ts

@@ -1,4 +1,4 @@
-import { toArray } from "./utils";
+import { toArray } from "./utils/to-array";
 
 export interface Options {
 	/**
@@ -526,7 +526,7 @@ export class HfInference {
 	 * This task is well known to summarize longer text into shorter text. Be careful, some models have a maximum length of input. That means that the summary cannot handle full books for instance. Be careful when choosing your model.
 	 */
 	public async summarization(args: SummarizationArgs, options?: Options): Promise<SummarizationReturn> {
-		return (await this.request(args, options))?.[0];
+		return (await this.request<SummarizationReturn[]>(args, options))?.[0];
 	}
 
 	/**
@@ -550,14 +550,14 @@ export class HfInference {
 	 * Usually used for sentiment-analysis this will output the likelihood of classes of an input. Recommended model: distilbert-base-uncased-finetuned-sst-2-english
 	 */
 	public async textClassification(args: TextClassificationArgs, options?: Options): Promise<TextClassificationReturn> {
-		return (await this.request(args, options))?.[0];
+		return (await this.request<TextClassificationReturn[]>(args, options))?.[0];
 	}
 
 	/**
 	 * Use to continue text from a prompt. This is a very generic task. Recommended model: gpt2 (it’s a simple model, but fun to play with).
 	 */
 	public async textGeneration(args: TextGenerationArgs, options?: Options): Promise<TextGenerationReturn> {
-		return (await this.request(args, options))?.[0];
+		return (await this.request<TextGenerationReturn[]>(args, options))?.[0];
 	}
 
 	/**
@@ -574,7 +574,7 @@ export class HfInference {
 	 * This task is well known to translate text from one language to another. Recommended model: Helsinki-NLP/opus-mt-ru-en.
 	 */
 	public async translation(args: TranslationArgs, options?: Options): Promise<TranslationReturn> {
-		return (await this.request(args, options))?.[0];
+		return (await this.request<TranslationReturn[]>(args, options))?.[0];
 	}
 
 	/**

package/src/utils/env-predicates.ts

@@ -0,0 +1,7 @@
+const isBrowser = typeof window !== "undefined" && typeof window.document !== "undefined";
+
+const isWebWorker =
+	typeof self === "object" && self.constructor && self.constructor.name === "DedicatedWorkerGlobalScope";
+
+export const isFrontend = isBrowser || isWebWorker;
+export const isBackend = !isBrowser && !isWebWorker;