@automatalabs/react-native-transformers 0.1.1 → 0.1.3

package/dist/src/transformers.js

@@ -0,0 +1,224 @@
+"use strict";
+const runtime_1 = require("./runtime");
+const TRANSFORMERS_MODULE_ID = '@automatalabs/react-native-transformers/internal-transformers-web';
+const TRANSFORMERS_REACT_NATIVE_PATCHED = Symbol.for('@automatalabs/react-native-transformers.patched-transformers');
+const TRANSFORMERS_REACT_NATIVE_WRAPPED = Symbol.for('@automatalabs/react-native-transformers.wrapped-from-pretrained');
+const ONNX_DTYPE_RETRY_ORDER = Object.freeze([
+    'fp32',
+    'q8',
+    'q4',
+    'q4f16',
+    'fp16',
+    'int8',
+    'uint8',
+    'bnb4',
+]);
+function isPlainObject(value) {
+    return typeof value === 'object' && value !== null && !Array.isArray(value);
+}
+function getErrorMessage(error) {
+    if (error instanceof Error) {
+        return error.stack ?? error.message;
+    }
+    if (typeof error === 'string') {
+        return error;
+    }
+    if (typeof error === 'number' ||
+        typeof error === 'boolean' ||
+        typeof error === 'bigint' ||
+        typeof error === 'symbol') {
+        return String(error);
+    }
+    if (error === null || error === undefined) {
+        return '';
+    }
+    if (isPlainObject(error)) {
+        try {
+            return JSON.stringify(error);
+        }
+        catch {
+            return '[object Object]';
+        }
+    }
+    return Object.prototype.toString.call(error);
+}
+function isLikelyMissingOnnxAssetError(error) {
+    const message = getErrorMessage(error).toLowerCase();
+    if (!message.includes('.onnx') && !message.includes('external data')) {
+        return false;
+    }
+    return (message.includes('404') ||
+        message.includes('not found') ||
+        message.includes('missing') ||
+        message.includes('could not locate file') ||
+        message.includes('local file missing'));
+}
+function getRetryDtypes(initialDtype) {
+    const triedDtypes = new Set();
+    if (typeof initialDtype === 'string') {
+        triedDtypes.add(initialDtype);
+    }
+    return ONNX_DTYPE_RETRY_ORDER.filter((dtype) => !triedDtypes.has(dtype));
+}
+function hasWrappedMarker(value) {
+    return (typeof value === 'function' &&
+        value[TRANSFORMERS_REACT_NATIVE_WRAPPED] === true);
+}
+function markWrapped(fn) {
+    Object.defineProperty(fn, TRANSFORMERS_REACT_NATIVE_WRAPPED, {
+        configurable: false,
+        enumerable: false,
+        writable: false,
+        value: true,
+    });
+    return fn;
+}
+function isFetchImplementation(value) {
+    return typeof value === 'function';
+}
+function getEnsureFetchOptions(options) {
+    if (isPlainObject(options) && isFetchImplementation(options.fetch)) {
+        return {
+            fetch: options.fetch,
+        };
+    }
+    return undefined;
+}
+function wrapPipeline(transformersModule, fn) {
+    if (hasWrappedMarker(fn)) {
+        return fn;
+    }
+    return markWrapped(function wrappedPipeline(task, model, options, ...rest) {
+        const fetchOptions = getEnsureFetchOptions(options);
+        if (fetchOptions) {
+            (0, runtime_1.ensureTransformersFetch)(transformersModule, fetchOptions);
+        }
+        else {
+            (0, runtime_1.ensureTransformersFetch)(transformersModule);
+        }
+        return fn.call(this, task, model, (0, runtime_1.normalizeTransformersOptions)(isPlainObject(options) ? options : {}), ...rest);
+    });
+}
+function wrapOptionsMethod(target, methodName, optionsIndex) {
+    if (!target || (typeof target !== 'function' && typeof target !== 'object')) {
+        return;
+    }
+    const patchableTarget = target;
+    const original = patchableTarget[methodName];
+    if (typeof original !== 'function' || hasWrappedMarker(original)) {
+        return;
+    }
+    const originalMethod = original;
+    patchableTarget[methodName] = markWrapped(function wrappedMethod(...args) {
+        const normalizedArgs = [...args];
+        while (normalizedArgs.length <= optionsIndex) {
+            normalizedArgs.push(undefined);
+        }
+        const rawOptions = normalizedArgs[optionsIndex];
+        normalizedArgs[optionsIndex] = (0, runtime_1.normalizeTransformersOptions)(isPlainObject(rawOptions) ? rawOptions : {});
+        return originalMethod.apply(this, normalizedArgs);
+    });
+}
+function wrapFromPretrained(target, transformersModule) {
+    if (typeof target !== 'function') {
+        return;
+    }
+    const patchableTarget = target;
+    const original = patchableTarget.from_pretrained;
+    if (typeof original !== 'function' || hasWrappedMarker(original)) {
+        return;
+    }
+    const originalFromPretrained = original;
+    patchableTarget.from_pretrained = markWrapped(async function wrappedFromPretrained(...args) {
+        const normalizedArgs = [...args];
+        while (normalizedArgs.length <= 1) {
+            normalizedArgs.push(undefined);
+        }
+        const rawOptions = isPlainObject(normalizedArgs[1]) ? normalizedArgs[1] : {};
+        const fetchOptions = getEnsureFetchOptions(rawOptions);
+        if (fetchOptions) {
+            (0, runtime_1.ensureTransformersFetch)(transformersModule, fetchOptions);
+        }
+        else {
+            (0, runtime_1.ensureTransformersFetch)(transformersModule);
+        }
+        const normalizedOptions = (0, runtime_1.normalizeTransformersOptions)(rawOptions);
+        normalizedArgs[1] = normalizedOptions;
+        try {
+            return await originalFromPretrained.apply(this, normalizedArgs);
+        }
+        catch (error) {
+            if (normalizedOptions.dtype != null || !isLikelyMissingOnnxAssetError(error)) {
+                throw error;
+            }
+            let lastError = error;
+            for (const dtype of getRetryDtypes(normalizedOptions.dtype)) {
+                try {
+                    normalizedArgs[1] = {
+                        ...normalizedOptions,
+                        dtype,
+                    };
+                    return await originalFromPretrained.apply(this, normalizedArgs);
+                }
+                catch (retryError) {
+                    if (!isLikelyMissingOnnxAssetError(retryError)) {
+                        throw retryError;
+                    }
+                    lastError = retryError;
+                }
+            }
+            throw lastError;
+        }
+    });
+}
+function wrapModelRegistry(modelRegistry) {
+    if (!modelRegistry || (typeof modelRegistry !== 'function' && typeof modelRegistry !== 'object')) {
+        return;
+    }
+    wrapOptionsMethod(modelRegistry, 'get_files', 1);
+    wrapOptionsMethod(modelRegistry, 'get_model_files', 1);
+    wrapOptionsMethod(modelRegistry, 'get_pipeline_files', 2);
+    wrapOptionsMethod(modelRegistry, 'is_cached', 1);
+    wrapOptionsMethod(modelRegistry, 'is_cached_files', 1);
+    wrapOptionsMethod(modelRegistry, 'is_pipeline_cached', 2);
+    wrapOptionsMethod(modelRegistry, 'is_pipeline_cached_files', 2);
+    wrapOptionsMethod(modelRegistry, 'clear_cache', 1);
+    wrapOptionsMethod(modelRegistry, 'clear_pipeline_cache', 2);
+}
+function patchTransformersReactNative(transformers, options = {}) {
+    if (typeof transformers !== 'object' || transformers === null) {
+        return transformers;
+    }
+    if (transformers[TRANSFORMERS_REACT_NATIVE_PATCHED]) {
+        return transformers;
+    }
+    (0, runtime_1.installTransformersReactNativeGlobals)(options.globals);
+    (0, runtime_1.configureTransformersEnvironment)(transformers, options.environment);
+    (0, runtime_1.ensureTransformersFetch)(transformers);
+    if (typeof transformers.pipeline === 'function') {
+        transformers.pipeline = wrapPipeline(transformers, transformers.pipeline);
+    }
+    for (const value of Object.values(transformers)) {
+        wrapFromPretrained(value, transformers);
+    }
+    wrapModelRegistry(transformers.ModelRegistry);
+    Object.assign(transformers, {
+        configureTransformersEnvironment: runtime_1.configureTransformersEnvironment,
+        ensureTransformersFetch: runtime_1.ensureTransformersFetch,
+        installTransformersReactNativeGlobals: runtime_1.installTransformersReactNativeGlobals,
+        normalizeTransformersOptions: runtime_1.normalizeTransformersOptions,
+        patchTransformersReactNative,
+        rawTransformers: transformers,
+    });
+    Object.defineProperty(transformers, TRANSFORMERS_REACT_NATIVE_PATCHED, {
+        configurable: false,
+        enumerable: false,
+        writable: false,
+        value: true,
+    });
+    return transformers;
+}
+(0, runtime_1.installTransformersReactNativeGlobals)();
+const transformers = patchTransformersReactNative(require(TRANSFORMERS_MODULE_ID));
+module.exports = transformers;
+//# sourceMappingURL=transformers.js.map

package/dist/src/transformers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"transformers.js","sourceRoot":"","sources":["../../src/transformers.ts"],"names":[],"mappings":";AAAA,uCAKmB;AAWnB,MAAM,sBAAsB,GAAG,mEAAmE,CAAC;AACnG,MAAM,iCAAiC,GAAG,MAAM,CAAC,GAAG,CAClD,8DAA8D,CAC/D,CAAC;AACF,MAAM,iCAAiC,GAAG,MAAM,CAAC,GAAG,CAClD,iEAAiE,CAClE,CAAC;AACF,MAAM,sBAAsB,GAAG,MAAM,CAAC,MAAM,CAAC;IAC3C,MAAM;IACN,IAAI;IACJ,IAAI;IACJ,OAAO;IACP,MAAM;IACN,MAAM;IACN,OAAO;IACP,MAAM;CACE,CAAC,CAAC;AAqBZ,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,eAAe,CAAC,KAAc;IACrC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC;IACtC,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IACE,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,SAAS;QAC1B,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAO,KAAK,KAAK,QAAQ,EACzB,CAAC;QACD,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IACvB,CAAC;IAED,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;QAC1C,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;QACzB,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,iBAAiB,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,6BAA6B,CAAC,KAAc;IACnD,MAAM,OAAO,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IAErD,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;QACrE,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,CACL,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC;QACvB,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC7B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC3B,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACzC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,CACvC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,YAAqB;IAC3C,MAAM,WAAW,GAAG,IAAI,GAAG,EAAU,CAAC;IAEtC,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;QACrC,WAAW,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,sBAAsB,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC;AAC3E,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAc;IACtC,OAAO,CACL,OAAO,KAAK,KAAK,UAAU;QAC1B,KAAiD,CAAC,iCAAiC,CAAC,KAAK,IAAI,CAC/F,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAqB,EAAK;IAC5C,MAAM,CAAC,cAAc,CAAC,EAAE,EAAE,iCAAiC,EAAE;QAC3D,YAAY,EAAE,KAAK;QACnB,UAAU,EAAE,KAAK;QACjB,QAAQ,EAAE,KAAK;QACf,KAAK,EAAE,IAAI;KACZ,CAAC,CAAC;IAEH,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,OAAO,OAAO,KAAK,KAAK,UAAU,CAAC;AACrC,CAAC;AAED,SAAS,qBAAqB,CAAC,OAAgB;IAC7C,IAAI,aAAa,CAAC,OAAO,CAAC,IAAI,qBAAqB,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACnE,OAAO;YACL,KAAK,EAAE,OAAO,CAAC,KAAK;SACrB,CAAC;IACJ,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,YAAY,CACnB,kBAA+C,EAC/C,EAAwB;IAExB,IAAI,gBAAgB,CAAC,EAAE,CAAC,EAAE,CAAC;QACzB,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,OAAO,WAAW,CAAC,SAAS,eAAe,CAEzC,IAAY,EACZ,KAAc,EACd,OAA6B,EAC7B,GAAG,IAAe;QAElB,MAAM,YAAY,GAAG,qBAAqB,CAAC,OAAO,CAAC,CAAC;QAEpD,IAAI,YAAY,EAAE,CAAC;YACjB,IAAA,iCAAuB,EAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC;QAC5D,CAAC;aAAM,CAAC;YACN,IAAA,iCAAuB,EAAC,kBAAkB,CAAC,CAAC;QAC9C,CAAC;QAED,OAAO,EAAE,CAAC,IAAI,CACZ,IAAI,EACJ,IAAI,EACJ,KAAK,EACL,IAAA,sCAA4B,EAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,EACnE,GAAG,IAAI,CACR,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB,CAAC,MAA8C,EAAE,UAAkB,EAAE,YAAoB;IACjH,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO,MAAM,KAAK,UAAU,IAAI,OAAO,MAAM,KAAK,QAAQ,CAAC,EAAE,CAAC;QAC5E,OAAO;IACT,CAAC;IAED,MAAM,eAAe,GAAG,MAAyB,CAAC;IAClD,MAAM,QAAQ,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC;IAC7C,IAAI,OAAO,QAAQ,KAAK,UAAU,IAAI,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjE,OAAO;IACT,CAAC;IAED,MAAM,cAAc,GAAG,QAA2B,CAAC;IACnD,eAAe,CAAC,UAAU,CAAC,GAAG,WAAW,CAAC,SAAS,aAAa,CAE9D,GAAG,IAAe;QAElB,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QACjC,OAAO,cAAc,CAAC,MAAM,IAAI,YAAY,EAAE,CAAC;YAC7C,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,UAAU,GAAG,cAAc,CAAC,YAAY,CAAC,CAAC;QAChD,cAAc,CAAC,YAAY,CAAC,GAAG,IAAA,sCAA4B,EACzD,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAC5C,CAAC;QAEF,OAAO,cAAc,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,kBAAkB,CACzB,MAAe,EACf,kBAA+C;IAE/C,IAAI,OAAO,MAAM,KAAK,UAAU,EAAE,CAAC;QACjC,OAAO;IACT,CAAC;IAED,MAAM,eAAe,GAAG,MAAyB,CAAC;IAClD,MAAM,QAAQ,GAAG,eAAe,CAAC,eAAe,CAAC;IACjD,IAAI,OAAO,QAAQ,KAAK,UAAU,IAAI,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC;QACjE,OAAO;IACT,CAAC;IAED,MAAM,sBAAsB,GAAG,QAAQ,CAAC;IACxC,eAAe,CAAC,eAAe,GAAG,WAAW,CAAC,KAAK,UAAU,qBAAqB,CAEhF,GAAG,IAAe;QAElB,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;QACjC,OAAO,cAAc,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YAClC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC;QAED,MAAM,UAAU,GAAG,aAAa,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7E,MAAM,YAAY,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;QAEvD,IAAI,YAAY,EAAE,CAAC;YACjB,IAAA,iCAAuB,EAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC;QAC5D,CAAC;aAAM,CAAC;YACN,IAAA,iCAAuB,EAAC,kBAAkB,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,iBAAiB,GAAG,IAAA,sCAA4B,EAAC,UAAU,CAAC,CAAC;QACnE,cAAc,CAAC,CAAC,CAAC,GAAG,iBAAiB,CAAC;QAEtC,IAAI,CAAC;YACH,OAAO,MAAM,sBAAsB,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QAClE,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACxB,IAAI,iBAAiB,CAAC,KAAK,IAAI,IAAI,IAAI,CAAC,6BAA6B,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC7E,MAAM,KAAK,CAAC;YACd,CAAC;YAED,IAAI,SAAS,GAAG,KAAK,CAAC;YAEtB,KAAK,MAAM,KAAK,IAAI,cAAc,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5D,IAAI,CAAC;oBACH,cAAc,CAAC,CAAC,CAAC,GAAG;wBAClB,GAAG,iBAAiB;wBACpB,KAAK;qBACN,CAAC;oBAEF,OAAO,MAAM,sBAAsB,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;gBAClE,CAAC;gBAAC,OAAO,UAAmB,EAAE,CAAC;oBAC7B,IAAI,CAAC,6BAA6B,CAAC,UAAU,CAAC,EAAE,CAAC;wBAC/C,MAAM,UAAU,CAAC;oBACnB,CAAC;oBAED,SAAS,GAAG,UAAU,CAAC;gBACzB,CAAC;YACH,CAAC;YAED,MAAM,SAAS,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,iBAAiB,CAAC,aAAsB;IAC/C,IAAI,CAAC,aAAa,IAAI,CAAC,OAAO,aAAa,KAAK,UAAU,IAAI,OAAO,aAAa,KAAK,QAAQ,CAAC,EAAE,CAAC;QACjG,OAAO;IACT,CAAC;IAED,iBAAiB,CAAC,aAAgC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;IACpE,iBAAiB,CAAC,aAAgC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC;IAC1E,iBAAiB,CAAC,aAAgC,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;IAC7E,iBAAiB,CAAC,aAAgC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC;IACpE,iBAAiB,CAAC,aAAgC,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC;IAC1E,iBAAiB,CAAC,aAAgC,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC;IAC7E,iBAAiB,CAAC,aAAgC,EAAE,0BAA0B,EAAE,CAAC,CAAC,CAAC;IACnF,iBAAiB,CAAC,aAAgC,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC;IACtE,iBAAiB,CAAC,aAAgC,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC;AACjF,CAAC;AAED,SAAS,4BAA4B,CACnC,YAAyC,EACzC,UAA+C,EAAE;IAEjD,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;QAC9D,OAAO,YAAyC,CAAC;IACnD,CAAC;IAED,IAAI,YAAY,CAAC,iCAAiC,CAAC,EAAE,CAAC;QACpD,OAAO,YAAyC,CAAC;IACnD,CAAC;IAED,IAAA,+CAAqC,EAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACvD,IAAA,0CAAgC,EAAC,YAAY,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IACpE,IAAA,iCAAuB,EAAC,YAAY,CAAC,CAAC;IAEtC,IAAI,OAAO,YAAY,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;QAChD,YAAY,CAAC,QAAQ,GAAG,YAAY,CAAC,YAAY,EAAE,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC5E,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,CAAC;QAChD,kBAAkB,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;IAC1C,CAAC;IAED,iBAAiB,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC;IAE9C,MAAM,CAAC,MAAM,CAAC,YAAY,EAAE;QAC1B,gCAAgC,EAAhC,0CAAgC;QAChC,uBAAuB,EAAvB,iCAAuB;QACvB,qCAAqC,EAArC,+CAAqC;QACrC,4BAA4B,EAA5B,sCAA4B;QAC5B,4BAA4B;QAC5B,eAAe,EAAE,YAAY;KAC9B,CAAC,CAAC;IAEH,MAAM,CAAC,cAAc,CAAC,YAAY,EAAE,iCAAiC,EAAE;QACrE,YAAY,EAAE,KAAK;QACnB,UAAU,EAAE,KAAK;QACjB,QAAQ,EAAE,KAAK;QACf,KAAK,EAAE,IAAI;KACZ,CAAC,CAAC;IAEH,OAAO,YAAyC,CAAC;AACnD,CAAC;AAED,IAAA,+CAAqC,GAAE,CAAC;AAExC,MAAM,YAAY,GAAG,4BAA4B,CAC/C,OAAO,CAAC,sBAAsB,CAAgC,CAC/D,CAAC;AAEF,iBAAS,YAAY,CAAC"}

package/dist/src/types.d.ts

@@ -0,0 +1,96 @@
+import type { InferenceSession } from 'onnxruntime-react-native';
+export type FetchImplementation = typeof fetch;
+export type RequestLike = string | URL | Request;
+export type ExecutionProviderLike = InferenceSession.ExecutionProviderConfig;
+export type SessionOptions = InferenceSession.SessionOptions;
+export type RunOptions = InferenceSession.RunOptions;
+export type ValueMetadata = InferenceSession.ValueMetadata;
+export type FeedsType = InferenceSession.FeedsType;
+export type FetchesType = InferenceSession.FetchesType;
+export type InferenceReturnType = InferenceSession.ReturnType;
+export type ReactNativeExecutionProviderName = 'cpu' | 'coreml' | 'xnnpack' | 'nnapi' | 'qnn';
+export interface CacheProgressInfo {
+    progress: number;
+    loaded: number;
+    total: number;
+}
+export type CacheProgressCallback = (info: CacheProgressInfo) => void;
+export interface CacheLike {
+    match(request: RequestLike): Promise<Response | undefined>;
+    put(request: RequestLike, response: Response, progressCallback?: CacheProgressCallback): Promise<void>;
+    delete(request: RequestLike): Promise<boolean>;
+}
+export interface TransformersEnvironmentLike {
+    useFS: boolean;
+    useFSCache: boolean;
+    useBrowserCache: boolean;
+    useWasmCache: boolean;
+    allowLocalModels?: boolean;
+    allowRemoteModels?: boolean;
+    localModelPath?: string;
+    customCache?: CacheLike | null;
+    useCustomCache?: boolean;
+    fetch?: FetchImplementation;
+    [key: string]: unknown;
+}
+export interface TransformersJsConfig {
+    device?: unknown;
+    device_config?: Record<string, Record<string, unknown> | undefined>;
+    [key: string]: unknown;
+}
+export interface TransformersConfig {
+    'transformers.js_config'?: TransformersJsConfig;
+    [key: string]: unknown;
+}
+export interface TransformersOptions {
+    config?: TransformersConfig;
+    device?: unknown;
+    dtype?: string | Record<string, string>;
+    session_options?: SessionOptions;
+    [key: string]: unknown;
+}
+export interface InstallGlobalsOptions {
+    installNavigatorGpuShim?: boolean;
+}
+export interface InstallGlobalsResult {
+    installedSelfAlias: boolean;
+    installedWindowAlias: boolean;
+    installedNavigatorGpuShim: boolean;
+}
+export interface ConfigureTransformersEnvironmentOptions {
+    allowLocalModels?: boolean | undefined;
+    allowRemoteModels?: boolean | undefined;
+    localModelPath?: string | undefined;
+    customCache?: CacheLike | null | undefined;
+    enableCustomCache?: boolean | undefined;
+    fetch?: FetchImplementation | undefined;
+}
+export interface EnsureTransformersFetchOptions {
+    fetch?: FetchImplementation | undefined;
+}
+export interface GetDefaultExecutionProvidersOptions {
+    platform?: string | undefined;
+    availableNames?: Iterable<string> | undefined;
+}
+export type RewriteExecutionProviderOptions = GetDefaultExecutionProvidersOptions;
+export interface TransformersReactNativeMetadata {
+    preferredExecutionProviders?: ExecutionProviderLike[];
+    originalDevice?: unknown;
+}
+export interface PipelineFunctionLike {
+    (task: string, model?: string, options?: TransformersOptions, ...rest: unknown[]): unknown;
+}
+export interface ModelRegistryLike {
+    [key: string]: unknown;
+}
+export interface TransformersModuleLike {
+    env?: TransformersEnvironmentLike;
+    pipeline?: PipelineFunctionLike;
+    ModelRegistry?: ModelRegistryLike;
+    [key: string]: unknown;
+}
+export interface PatchTransformersReactNativeOptions {
+    globals?: InstallGlobalsOptions | undefined;
+    environment?: ConfigureTransformersEnvironmentOptions | undefined;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAEjE,MAAM,MAAM,mBAAmB,GAAG,OAAO,KAAK,CAAC;AAC/C,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC;AACjD,MAAM,MAAM,qBAAqB,GAAG,gBAAgB,CAAC,uBAAuB,CAAC;AAC7E,MAAM,MAAM,cAAc,GAAG,gBAAgB,CAAC,cAAc,CAAC;AAC7D,MAAM,MAAM,UAAU,GAAG,gBAAgB,CAAC,UAAU,CAAC;AACrD,MAAM,MAAM,aAAa,GAAG,gBAAgB,CAAC,aAAa,CAAC;AAC3D,MAAM,MAAM,SAAS,GAAG,gBAAgB,CAAC,SAAS,CAAC;AACnD,MAAM,MAAM,WAAW,GAAG,gBAAgB,CAAC,WAAW,CAAC;AACvD,MAAM,MAAM,mBAAmB,GAAG,gBAAgB,CAAC,UAAU,CAAC;AAE9D,MAAM,MAAM,gCAAgC,GAAG,KAAK,GAAG,QAAQ,GAAG,SAAS,GAAG,OAAO,GAAG,KAAK,CAAC;AAE9F,MAAM,WAAW,iBAAiB;IAChC,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,MAAM,qBAAqB,GAAG,CAAC,IAAI,EAAE,iBAAiB,KAAK,IAAI,CAAC;AAEtE,MAAM,WAAW,SAAS;IACxB,KAAK,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,GAAG,SAAS,CAAC,CAAC;IAC3D,GAAG,CACD,OAAO,EAAE,WAAW,EACpB,QAAQ,EAAE,QAAQ,EAClB,gBAAgB,CAAC,EAAE,qBAAqB,GACvC,OAAO,CAAC,IAAI,CAAC,CAAC;IACjB,MAAM,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC;CAChD;AAED,MAAM,WAAW,2BAA2B;IAC1C,KAAK,EAAE,OAAO,CAAC;IACf,UAAU,EAAE,OAAO,CAAC;IACpB,eAAe,EAAE,OAAO,CAAC;IACzB,YAAY,EAAE,OAAO,CAAC;IACtB,gBAAgB,CAAC,EAAE,OAAO,CAAC;IAC3B,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,SAAS,GAAG,IAAI,CAAC;IAC/B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,KAAK,CAAC,EAAE,mBAAmB,CAAC;IAC5B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,oBAAoB;IACnC,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,SAAS,CAAC,CAAC;IACpE,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,kBAAkB;IACjC,wBAAwB,CAAC,EAAE,oBAAoB,CAAC;IAChD,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,mBAAmB;IAClC,MAAM,CAAC,EAAE,kBAAkB,CAAC;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACxC,eAAe,CAAC,EAAE,cAAc,CAAC;IACjC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,qBAAqB;IACpC,uBAAuB,CAAC,EAAE,OAAO,CAAC;CACnC;AAED,MAAM,WAAW,oBAAoB;IACnC,kBAAkB,EAAE,OAAO,CAAC;IAC5B,oBAAoB,EAAE,OAAO,CAAC;IAC9B,yBAAyB,EAAE,OAAO,CAAC;CACpC;AAED,MAAM,WAAW,uCAAuC;IACtD,gBAAgB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACvC,iBAAiB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACxC,cAAc,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACpC,WAAW,CAAC,EAAE,SAAS,GAAG,IAAI,GAAG,SAAS,CAAC;IAC3C,iBAAiB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACxC,KAAK,CAAC,EAAE,mBAAmB,GAAG,SAAS,CAAC;CACzC;AAED,MAAM,WAAW,8BAA8B;IAC7C,KAAK,CAAC,EAAE,mBAAmB,GAAG,SAAS,CAAC;CACzC;AAED,MAAM,WAAW,mCAAmC;IAClD,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,cAAc,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,GAAG,SAAS,CAAC;CAC/C;AAED,MAAM,MAAM,+BAA+B,GAAG,mCAAmC,CAAC;AAElF,MAAM,WAAW,+BAA+B;IAC9C,2BAA2B,CAAC,EAAE,qBAAqB,EAAE,CAAC;IACtD,cAAc,CAAC,EAAE,OAAO,CAAC;CAC1B;AAED,MAAM,WAAW,oBAAoB;IACnC,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,mBAAmB,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;CAC5F;AAED,MAAM,WAAW,iBAAiB;IAChC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,sBAAsB;IACrC,GAAG,CAAC,EAAE,2BAA2B,CAAC;IAClC,QAAQ,CAAC,EAAE,oBAAoB,CAAC;IAChC,aAAa,CAAC,EAAE,iBAAiB,CAAC;IAClC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,mCAAmC;IAClD,OAAO,CAAC,EAAE,qBAAqB,GAAG,SAAS,CAAC;IAC5C,WAAW,CAAC,EAAE,uCAAuC,GAAG,SAAS,CAAC;CACnE"}

package/dist/src/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/src/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":""}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@automatalabs/react-native-transformers",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Use @huggingface/transformers in Expo and React Native apps through onnxruntime-react-native.",
   "license": "MIT",
   "author": "Automata Labs",
@@ -29,27 +29,58 @@
     "on-device-ai"
   ],
   "scripts": {
-    "check": "node --check app.plugin.js plugin/src/index.js src/index.js src/runtime.js src/expoFileSystemCache.js src/metro.js src/transformers.js src/adapter/onnxruntime-web-webgpu.js scripts/sync-transformers-peer.mjs",
+    "clean": "rm -rf dist tsconfig.tsbuildinfo",
+    "build": "npm run clean && tsc -p tsconfig.json",
+    "typecheck": "tsc -p tsconfig.json --noEmit",
+    "lint": "eslint app.plugin.js app.plugin.ts eslint.config.mjs plugin src scripts types",
+    "check": "npm run check:transformers-peer && npm run lint && npm run typecheck && npm run build",
     "check:transformers-peer": "node ./scripts/sync-transformers-peer.mjs",
     "sync:transformers-peer": "node ./scripts/sync-transformers-peer.mjs --write",
+    "prepare": "npm run build",
+    "prepack": "npm run build",
     "prepublishOnly": "npm run check",
     "example": "npm --prefix example run start",
     "example:ios": "npm --prefix example run ios"
   },
-  "main": "./src/index.js",
-  "react-native": "./src/index.js",
+  "main": "./dist/src/index.js",
+  "types": "./dist/src/index.d.ts",
+  "react-native": "./dist/src/index.js",
   "exports": {
-    ".": "./src/index.js",
-    "./metro": "./src/metro.js",
-    "./plugin": "./plugin/src/index.js",
-    "./transformers": "./src/transformers.js",
-    "./adapter/onnxruntime-web-webgpu": "./src/adapter/onnxruntime-web-webgpu.js",
-    "./app.plugin": "./app.plugin.js",
+    ".": {
+      "types": "./dist/src/index.d.ts",
+      "require": "./dist/src/index.js",
+      "default": "./dist/src/index.js"
+    },
+    "./metro": {
+      "types": "./dist/src/metro.d.ts",
+      "require": "./dist/src/metro.js",
+      "default": "./dist/src/metro.js"
+    },
+    "./plugin": {
+      "types": "./dist/plugin/src/index.d.ts",
+      "require": "./dist/plugin/src/index.js",
+      "default": "./dist/plugin/src/index.js"
+    },
+    "./transformers": {
+      "types": "./dist/src/transformers.d.ts",
+      "require": "./dist/src/transformers.js",
+      "default": "./dist/src/transformers.js"
+    },
+    "./adapter/onnxruntime-web-webgpu": {
+      "types": "./dist/src/adapter/onnxruntime-web-webgpu.d.ts",
+      "require": "./dist/src/adapter/onnxruntime-web-webgpu.js",
+      "default": "./dist/src/adapter/onnxruntime-web-webgpu.js"
+    },
+    "./app.plugin": {
+      "types": "./dist/app.plugin.d.ts",
+      "require": "./dist/app.plugin.js",
+      "default": "./dist/app.plugin.js"
+    },
+    "./app.plugin.js": "./app.plugin.js",
     "./package.json": "./package.json"
   },
   "files": [
-    "plugin",
-    "src",
+    "dist",
     "app.plugin.js",
     "README.md",
     "LICENSE"
@@ -70,5 +101,19 @@
       "optional": true
     }
   },
-  "dependencies": {}
+  "dependencies": {},
+  "devDependencies": {
+    "@eslint/js": "^10.0.1",
+    "@huggingface/transformers": "^4.0.1",
+    "@types/node": "^25.5.2",
+    "@types/react": "^19.2.14",
+    "@typescript-eslint/eslint-plugin": "^8.58.1",
+    "@typescript-eslint/parser": "^8.58.1",
+    "eslint": "^10.2.0",
+    "globals": "^17.4.0",
+    "onnxruntime-react-native": "^1.24.3",
+    "react": "^19.2.5",
+    "react-native": "^0.85.0",
+    "typescript": "^6.0.2"
+  }
 }

package/plugin/src/index.js

@@ -1,13 +0,0 @@
-function requireFromProject(moduleId) {
-  return require(require.resolve(moduleId, { paths: [process.cwd(), __dirname] }));
-}
-
-function withReactNativeTransformers(config) {
-  const onnxruntimeReactNativePlugin = requireFromProject('onnxruntime-react-native/app.plugin');
-  const withOnnxruntimeReactNative =
-    onnxruntimeReactNativePlugin.default ?? onnxruntimeReactNativePlugin;
-
-  return withOnnxruntimeReactNative(config);
-}
-
-module.exports = withReactNativeTransformers;