react-native-executorch 0.4.8 → 0.5.0

package/lib/module/index.js

@@ -1,6 +1,16 @@
 "use strict";
 
-import { SpeechToTextLanguage } from './types/stt';
+import { ETInstallerNativeModule } from './native/RnExecutorchModules';
+
+// eslint-disable no-var
+
+// eslint-disable no-var
+if (global.loadStyleTransfer == null || global.loadImageSegmentation == null || global.loadExecutorchModule == null || global.loadClassification == null || global.loadObjectDetection == null || global.loadTokenizerModule == null || global.loadTextEmbeddings == null || global.loadImageEmbeddings == null || global.loadSpeechToText == null || global.loadOCR == null || global.loadVerticalOCR == null || global.loadImageEmbeddings == null) {
+  if (!ETInstallerNativeModule) {
+    throw new Error(`Failed to install react-native-executorch: The native module could not be found.`);
+  }
+  ETInstallerNativeModule.install();
+}
 
 // hooks
 export * from './hooks/computer_vision/useClassification';
@@ -9,6 +19,7 @@ export * from './hooks/computer_vision/useStyleTransfer';
 export * from './hooks/computer_vision/useImageSegmentation';
 export * from './hooks/computer_vision/useOCR';
 export * from './hooks/computer_vision/useVerticalOCR';
+export * from './hooks/computer_vision/useImageEmbeddings';
 export * from './hooks/natural_language_processing/useLLM';
 export * from './hooks/natural_language_processing/useSpeechToText';
 export * from './hooks/natural_language_processing/useTextEmbeddings';
@@ -22,25 +33,25 @@ export * from './modules/computer_vision/StyleTransferModule';
 export * from './modules/computer_vision/ImageSegmentationModule';
 export * from './modules/computer_vision/OCRModule';
 export * from './modules/computer_vision/VerticalOCRModule';
+export * from './modules/general/ExecutorchModule';
+export * from './modules/computer_vision/ImageEmbeddingsModule';
 export * from './modules/natural_language_processing/LLMModule';
 export * from './modules/natural_language_processing/SpeechToTextModule';
 export * from './modules/natural_language_processing/TextEmbeddingsModule';
 export * from './modules/natural_language_processing/TokenizerModule';
-export * from './modules/general/ExecutorchModule';
 
 // utils
 export * from './utils/ResourceFetcher';
+export * from './utils/llm';
 
 // types
 export * from './types/objectDetection';
 export * from './types/ocr';
 export * from './types/imageSegmentation';
 export * from './types/llm';
-export { SpeechToTextLanguage };
 
 // constants
 export * from './constants/modelUrls';
 export * from './constants/ocr/models';
 export * from './constants/llmDefaults';
-export { STREAMING_ACTION, MODES } from './constants/sttDefaults';
 //# sourceMappingURL=index.js.map

package/lib/module/index.js.map

@@ -1 +1 @@
-{"version":3,"names":["SpeechToTextLanguage","STREAMING_ACTION","MODES"],"sourceRoot":"../../src","sources":["index.tsx"],"mappings":";;AAAA,SAASA,oBAAoB,QAAQ,aAAa;;AAElD;AACA,cAAc,2CAA2C;AACzD,cAAc,4CAA4C;AAC1D,cAAc,0CAA0C;AACxD,cAAc,8CAA8C;AAC5D,cAAc,gCAAgC;AAC9C,cAAc,wCAAwC;AAEtD,cAAc,4CAA4C;AAC1D,cAAc,qDAAqD;AACnE,cAAc,uDAAuD;AACrE,cAAc,kDAAkD;AAEhE,cAAc,qCAAqC;;AAEnD;AACA,cAAc,gDAAgD;AAC9D,cAAc,iDAAiD;AAC/D,cAAc,+CAA+C;AAC7D,cAAc,mDAAmD;AACjE,cAAc,qCAAqC;AACnD,cAAc,6CAA6C;AAE3D,cAAc,iDAAiD;AAC/D,cAAc,0DAA0D;AACxE,cAAc,4DAA4D;AAC1E,cAAc,uDAAuD;AAErE,cAAc,oCAAoC;;AAElD;AACA,cAAc,yBAAyB;;AAEvC;AACA,cAAc,yBAAyB;AACvC,cAAc,aAAa;AAC3B,cAAc,2BAA2B;AACzC,cAAc,aAAa;AAC3B,SAASA,oBAAoB;;AAE7B;AACA,cAAc,uBAAuB;AACrC,cAAc,wBAAwB;AACtC,cAAc,yBAAyB;AACvC,SAASC,gBAAgB,EAAEC,KAAK,QAAQ,yBAAyB","ignoreList":[]}
+{"version":3,"names":["ETInstallerNativeModule","global","loadStyleTransfer","loadImageSegmentation","loadExecutorchModule","loadClassification","loadObjectDetection","loadTokenizerModule","loadTextEmbeddings","loadImageEmbeddings","loadSpeechToText","loadOCR","loadVerticalOCR","Error","install"],"sourceRoot":"../../src","sources":["index.ts"],"mappings":";;AAAA,SAASA,uBAAuB,QAAQ,8BAA8B;;AAEtE;;AA8BA;AACA,IACEC,MAAM,CAACC,iBAAiB,IAAI,IAAI,IAChCD,MAAM,CAACE,qBAAqB,IAAI,IAAI,IACpCF,MAAM,CAACG,oBAAoB,IAAI,IAAI,IACnCH,MAAM,CAACI,kBAAkB,IAAI,IAAI,IACjCJ,MAAM,CAACK,mBAAmB,IAAI,IAAI,IAClCL,MAAM,CAACM,mBAAmB,IAAI,IAAI,IAClCN,MAAM,CAACO,kBAAkB,IAAI,IAAI,IACjCP,MAAM,CAACQ,mBAAmB,IAAI,IAAI,IAClCR,MAAM,CAACS,gBAAgB,IAAI,IAAI,IAC/BT,MAAM,CAACU,OAAO,IAAI,IAAI,IACtBV,MAAM,CAACW,eAAe,IAAI,IAAI,IAC9BX,MAAM,CAACQ,mBAAmB,IAAI,IAAI,EAClC;EACA,IAAI,CAACT,uBAAuB,EAAE;IAC5B,MAAM,IAAIa,KAAK,CACb,kFACF,CAAC;EACH;EACAb,uBAAuB,CAACc,OAAO,CAAC,CAAC;AACnC;;AAEA;AACA,cAAc,2CAA2C;AACzD,cAAc,4CAA4C;AAC1D,cAAc,0CAA0C;AACxD,cAAc,8CAA8C;AAC5D,cAAc,gCAAgC;AAC9C,cAAc,wCAAwC;AACtD,cAAc,4CAA4C;AAE1D,cAAc,4CAA4C;AAC1D,cAAc,qDAAqD;AACnE,cAAc,uDAAuD;AACrE,cAAc,kDAAkD;AAEhE,cAAc,qCAAqC;;AAEnD;AACA,cAAc,gDAAgD;AAC9D,cAAc,iDAAiD;AAC/D,cAAc,+CAA+C;AAC7D,cAAc,mDAAmD;AACjE,cAAc,qCAAqC;AACnD,cAAc,6CAA6C;AAC3D,cAAc,oCAAoC;AAClD,cAAc,iDAAiD;AAE/D,cAAc,iDAAiD;AAC/D,cAAc,0DAA0D;AACxE,cAAc,4DAA4D;AAC1E,cAAc,uDAAuD;;AAErE;AACA,cAAc,yBAAyB;AACvC,cAAc,aAAa;;AAE3B;AACA,cAAc,yBAAyB;AACvC,cAAc,aAAa;AAC3B,cAAc,2BAA2B;AACzC,cAAc,aAAa;;AAE3B;AACA,cAAc,uBAAuB;AACrC,cAAc,wBAAwB;AACtC,cAAc,yBAAyB","ignoreList":[]}

package/lib/module/modules/BaseModule.js

@@ -4,10 +4,13 @@ import { ResourceFetcher } from '../utils/ResourceFetcher';
 import { getError } from '../Error';
 export class BaseModule {
   static onDownloadProgressCallback = () => {};
-  static async load(...sources) {
+  static async load(sources, ...loadArgs) {
     try {
-      const paths = await ResourceFetcher.fetchMultipleResources(this.onDownloadProgressCallback, ...sources);
-      await this.nativeModule.loadModule(...paths);
+      const paths = await ResourceFetcher.fetch(this.onDownloadProgressCallback, ...sources);
+      if (paths === null || paths.length < sources.length) {
+        throw new Error('Download interrupted.');
+      }
+      await this.nativeModule.loadModule(...paths, ...loadArgs);
     } catch (error) {
       throw new Error(getError(error));
     }

package/lib/module/modules/BaseModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["ResourceFetcher","getError","BaseModule","onDownloadProgressCallback","load","sources","paths","fetchMultipleResources","nativeModule","loadModule","error","Error","forward","_args","onDownloadProgress","callback"],"sourceRoot":"../../../src","sources":["modules/BaseModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,0BAA0B;AAC1D,SAASC,QAAQ,QAAQ,UAAU;AAGnC,OAAO,MAAMC,UAAU,CAAC;EAEtB,OAAOC,0BAA0B,GAC/BA,CAAA,KAAM,CAAC,CAAC;EAEV,aAAaC,IAAIA,CAAC,GAAGC,OAAyB,EAAiB;IAC7D,IAAI;MACF,MAAMC,KAAK,GAAG,MAAMN,eAAe,CAACO,sBAAsB,CACxD,IAAI,CAACJ,0BAA0B,EAC/B,GAAGE,OACL,CAAC;MACD,MAAM,IAAI,CAACG,YAAY,CAACC,UAAU,CAAC,GAAGH,KAAK,CAAC;IAC9C,CAAC,CAAC,OAAOI,KAAK,EAAE;MACd,MAAM,IAAIC,KAAK,CAACV,QAAQ,CAACS,KAAK,CAAC,CAAC;IAClC;EACF;EAEA,aAAuBE,OAAOA,CAAC,GAAGC,KAAY,EAAgB;IAC5D,MAAM,IAAIF,KAAK,CACb,sGACF,CAAC;EACH;EAEA,OAAOG,kBAAkBA,CAACC,QAA4C,EAAE;IACtE,IAAI,CAACZ,0BAA0B,GAAGY,QAAQ;EAC5C;AACF","ignoreList":[]}
+{"version":3,"names":["ResourceFetcher","getError","BaseModule","onDownloadProgressCallback","load","sources","loadArgs","paths","fetch","length","Error","nativeModule","loadModule","error","forward","_args","onDownloadProgress","callback"],"sourceRoot":"../../../src","sources":["modules/BaseModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,0BAA0B;AAC1D,SAASC,QAAQ,QAAQ,UAAU;AAGnC,OAAO,MAAMC,UAAU,CAAC;EAEtB,OAAOC,0BAA0B,GAC/BA,CAAA,KAAM,CAAC,CAAC;EAEV,aAAaC,IAAIA,CACfC,OAAyB,EACzB,GAAGC,QAAe,EACH;IACf,IAAI;MACF,MAAMC,KAAK,GAAG,MAAMP,eAAe,CAACQ,KAAK,CACvC,IAAI,CAACL,0BAA0B,EAC/B,GAAGE,OACL,CAAC;MACD,IAAIE,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACE,MAAM,GAAGJ,OAAO,CAACI,MAAM,EAAE;QACnD,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;MAC1C;MACA,MAAM,IAAI,CAACC,YAAY,CAACC,UAAU,CAAC,GAAGL,KAAK,EAAE,GAAGD,QAAQ,CAAC;IAC3D,CAAC,CAAC,OAAOO,KAAK,EAAE;MACd,MAAM,IAAIH,KAAK,CAACT,QAAQ,CAACY,KAAK,CAAC,CAAC;IAClC;EACF;EAEA,aAAuBC,OAAOA,CAAC,GAAGC,KAAY,EAAgB;IAC5D,MAAM,IAAIL,KAAK,CACb,sGACF,CAAC;EACH;EAEA,OAAOM,kBAAkBA,CAACC,QAA4C,EAAE;IACtE,IAAI,CAACd,0BAA0B,GAAGc,QAAQ;EAC5C;AACF","ignoreList":[]}

package/lib/module/modules/BaseNonStaticModule.js

@@ -0,0 +1,17 @@
+"use strict";
+
+export class BaseNonStaticModule {
+  nativeModule = null;
+  async forwardET(inputTensor) {
+    return await this.nativeModule.forward(inputTensor);
+  }
+  async getInputShape(methodName, index) {
+    return this.nativeModule.getInputShape(methodName, index);
+  }
+  delete() {
+    if (this.nativeModule !== null) {
+      this.nativeModule.unload();
+    }
+  }
+}
+//# sourceMappingURL=BaseNonStaticModule.js.map

package/lib/module/modules/BaseNonStaticModule.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["BaseNonStaticModule","nativeModule","forwardET","inputTensor","forward","getInputShape","methodName","index","delete","unload"],"sourceRoot":"../../../src","sources":["modules/BaseNonStaticModule.ts"],"mappings":";;AAGA,OAAO,MAAeA,mBAAmB,CAAC;EACxCC,YAAY,GAAQ,IAAI;EAQxB,MAAgBC,SAASA,CAACC,WAAwB,EAAwB;IACxE,OAAO,MAAM,IAAI,CAACF,YAAY,CAACG,OAAO,CAACD,WAAW,CAAC;EACrD;EAEA,MAAME,aAAaA,CAACC,UAAkB,EAAEC,KAAa,EAAqB;IACxE,OAAO,IAAI,CAACN,YAAY,CAACI,aAAa,CAACC,UAAU,EAAEC,KAAK,CAAC;EAC3D;EAEAC,MAAMA,CAAA,EAAG;IACP,IAAI,IAAI,CAACP,YAAY,KAAK,IAAI,EAAE;MAC9B,IAAI,CAACA,YAAY,CAACQ,MAAM,CAAC,CAAC;IAC5B;EACF;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/ClassificationModule.js

@@ -1,14 +1,19 @@
 "use strict";
 
-import { ClassificationNativeModule } from '../../native/RnExecutorchModules';
-import { BaseModule } from '../BaseModule';
-export class ClassificationModule extends BaseModule {
-  static nativeModule = ClassificationNativeModule;
-  static async load(modelSource) {
-    await super.load(modelSource);
+import { ResourceFetcher } from '../../utils/ResourceFetcher';
+import { ETError, getError } from '../../Error';
+import { BaseNonStaticModule } from '../BaseNonStaticModule';
+export class ClassificationModule extends BaseNonStaticModule {
+  async load(model, onDownloadProgressCallback = () => {}) {
+    const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, model.modelSource);
+    if (paths === null || paths.length < 1) {
+      throw new Error('Download interrupted.');
+    }
+    this.nativeModule = global.loadClassification(paths[0] || '');
   }
-  static async forward(input) {
-    return await this.nativeModule.forward(input);
+  async forward(imageSource) {
+    if (this.nativeModule == null) throw new Error(getError(ETError.ModuleNotLoaded));
+    return await this.nativeModule.generate(imageSource);
   }
 }
 //# sourceMappingURL=ClassificationModule.js.map

package/lib/module/modules/computer_vision/ClassificationModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["ClassificationNativeModule","BaseModule","ClassificationModule","nativeModule","load","modelSource","forward","input"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ClassificationModule.ts"],"mappings":";;AAAA,SAASA,0BAA0B,QAAQ,kCAAkC;AAE7E,SAASC,UAAU,QAAQ,eAAe;AAE1C,OAAO,MAAMC,oBAAoB,SAASD,UAAU,CAAC;EACnD,OAA0BE,YAAY,GAAGH,0BAA0B;EAEnE,aAAsBI,IAAIA,CAACC,WAA2B,EAAE;IACtD,MAAM,KAAK,CAACD,IAAI,CAACC,WAAW,CAAC;EAC/B;EAEA,aAAsBC,OAAOA,CAC3BC,KAAa,EAC0C;IACvD,OAAO,MAAM,IAAI,CAACJ,YAAY,CAACG,OAAO,CAACC,KAAK,CAAC;EAC/C;AACF","ignoreList":[]}
+{"version":3,"names":["ResourceFetcher","ETError","getError","BaseNonStaticModule","ClassificationModule","load","model","onDownloadProgressCallback","paths","fetch","modelSource","length","Error","nativeModule","global","loadClassification","forward","imageSource","ModuleNotLoaded","generate"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ClassificationModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,6BAA6B;AAE7D,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,mBAAmB,QAAQ,wBAAwB;AAE5D,OAAO,MAAMC,oBAAoB,SAASD,mBAAmB,CAAC;EAC5D,MAAME,IAAIA,CACRC,KAAsC,EACtCC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EAClD;IACf,MAAMC,KAAK,GAAG,MAAMR,eAAe,CAACS,KAAK,CACvCF,0BAA0B,EAC1BD,KAAK,CAACI,WACR,CAAC;IACD,IAAIF,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACG,MAAM,GAAG,CAAC,EAAE;MACtC,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;IAC1C;IACA,IAAI,CAACC,YAAY,GAAGC,MAAM,CAACC,kBAAkB,CAACP,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;EAC/D;EAEA,MAAMQ,OAAOA,CAACC,WAAmB,EAAE;IACjC,IAAI,IAAI,CAACJ,YAAY,IAAI,IAAI,EAC3B,MAAM,IAAID,KAAK,CAACV,QAAQ,CAACD,OAAO,CAACiB,eAAe,CAAC,CAAC;IACpD,OAAO,MAAM,IAAI,CAACL,YAAY,CAACM,QAAQ,CAACF,WAAW,CAAC;EACtD;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/ImageEmbeddingsModule.js

@@ -0,0 +1,19 @@
+"use strict";
+
+import { ResourceFetcher } from '../../utils/ResourceFetcher';
+import { ETError, getError } from '../../Error';
+import { BaseNonStaticModule } from '../BaseNonStaticModule';
+export class ImageEmbeddingsModule extends BaseNonStaticModule {
+  async load(model, onDownloadProgressCallback = () => {}) {
+    const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, model.modelSource);
+    if (paths === null || paths.length < 1) {
+      throw new Error('Download interrupted.');
+    }
+    this.nativeModule = global.loadImageEmbeddings(paths[0] || '');
+  }
+  async forward(imageSource) {
+    if (this.nativeModule == null) throw new Error(getError(ETError.ModuleNotLoaded));
+    return new Float32Array(await this.nativeModule.generate(imageSource));
+  }
+}
+//# sourceMappingURL=ImageEmbeddingsModule.js.map

package/lib/module/modules/computer_vision/ImageEmbeddingsModule.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["ResourceFetcher","ETError","getError","BaseNonStaticModule","ImageEmbeddingsModule","load","model","onDownloadProgressCallback","paths","fetch","modelSource","length","Error","nativeModule","global","loadImageEmbeddings","forward","imageSource","ModuleNotLoaded","Float32Array","generate"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ImageEmbeddingsModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,6BAA6B;AAE7D,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,mBAAmB,QAAQ,wBAAwB;AAE5D,OAAO,MAAMC,qBAAqB,SAASD,mBAAmB,CAAC;EAC7D,MAAME,IAAIA,CACRC,KAAsC,EACtCC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EAClD;IACf,MAAMC,KAAK,GAAG,MAAMR,eAAe,CAACS,KAAK,CACvCF,0BAA0B,EAC1BD,KAAK,CAACI,WACR,CAAC;IACD,IAAIF,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACG,MAAM,GAAG,CAAC,EAAE;MACtC,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;IAC1C;IACA,IAAI,CAACC,YAAY,GAAGC,MAAM,CAACC,mBAAmB,CAACP,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;EAChE;EAEA,MAAMQ,OAAOA,CAACC,WAAmB,EAAyB;IACxD,IAAI,IAAI,CAACJ,YAAY,IAAI,IAAI,EAC3B,MAAM,IAAID,KAAK,CAACV,QAAQ,CAACD,OAAO,CAACiB,eAAe,CAAC,CAAC;IACpD,OAAO,IAAIC,YAAY,CAAC,MAAM,IAAI,CAACN,YAAY,CAACO,QAAQ,CAACH,WAAW,CAAC,CAAC;EACxE;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/ImageSegmentationModule.js

@@ -1,28 +1,30 @@
 "use strict";
 
-import { BaseModule } from '../BaseModule';
-import { getError } from '../../Error';
+import { ResourceFetcher } from '../../utils/ResourceFetcher';
 import { DeeplabLabel } from '../../types/imageSegmentation';
-import { ImageSegmentationNativeModule } from '../../native/RnExecutorchModules';
-export class ImageSegmentationModule extends BaseModule {
-  static nativeModule = ImageSegmentationNativeModule;
-  static async load(modelSource) {
-    return await super.load(modelSource);
+import { ETError, getError } from '../../Error';
+import { BaseNonStaticModule } from '../BaseNonStaticModule';
+export class ImageSegmentationModule extends BaseNonStaticModule {
+  async load(model, onDownloadProgressCallback = () => {}) {
+    const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, model.modelSource);
+    if (paths === null || paths.length < 1) {
+      throw new Error('Download interrupted.');
+    }
+    this.nativeModule = global.loadImageSegmentation(paths[0] || '');
   }
-  static async forward(input, classesOfInterest, resize) {
-    try {
-      const stringDict = await this.nativeModule.forward(input, (classesOfInterest || []).map(label => DeeplabLabel[label]), resize || false);
-      let enumDict = {};
-      for (const key in stringDict) {
-        if (key in DeeplabLabel) {
-          const enumKey = DeeplabLabel[key];
-          enumDict[enumKey] = stringDict[key];
-        }
+  async forward(imageSource, classesOfInterest, resize) {
+    if (this.nativeModule == null) {
+      throw new Error(getError(ETError.ModuleNotLoaded));
+    }
+    const stringDict = await this.nativeModule.generate(imageSource, (classesOfInterest || []).map(label => DeeplabLabel[label]), resize || false);
+    let enumDict = {};
+    for (const key in stringDict) {
+      if (key in DeeplabLabel) {
+        const enumKey = DeeplabLabel[key];
+        enumDict[enumKey] = stringDict[key];
       }
-      return enumDict;
-    } catch (e) {
-      throw new Error(getError(e));
     }
+    return enumDict;
   }
 }
 //# sourceMappingURL=ImageSegmentationModule.js.map

package/lib/module/modules/computer_vision/ImageSegmentationModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["BaseModule","getError","DeeplabLabel","ImageSegmentationNativeModule","ImageSegmentationModule","nativeModule","load","modelSource","forward","input","classesOfInterest","resize","stringDict","map","label","enumDict","key","enumKey","e","Error"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ImageSegmentationModule.ts"],"mappings":";;AAAA,SAASA,UAAU,QAAQ,eAAe;AAC1C,SAASC,QAAQ,QAAQ,aAAa;AACtC,SAASC,YAAY,QAAQ,+BAA+B;AAE5D,SAASC,6BAA6B,QAAQ,kCAAkC;AAEhF,OAAO,MAAMC,uBAAuB,SAASJ,UAAU,CAAC;EACtD,OAA0BK,YAAY,GAAGF,6BAA6B;EAEtE,aAAsBG,IAAIA,CAACC,WAA2B,EAAE;IACtD,OAAO,MAAM,KAAK,CAACD,IAAI,CAACC,WAAW,CAAC;EACtC;EAEA,aAAsBC,OAAOA,CAC3BC,KAAa,EACbC,iBAAkC,EAClCC,MAAgB,EAChB;IACA,IAAI;MACF,MAAMC,UAAU,GAAG,MAAO,IAAI,CAACP,YAAY,CAACG,OAAO,CACjDC,KAAK,EACL,CAACC,iBAAiB,IAAI,EAAE,EAAEG,GAAG,CAAEC,KAAK,IAAKZ,YAAY,CAACY,KAAK,CAAC,CAAC,EAC7DH,MAAM,IAAI,KACZ,CAAuD;MAEvD,IAAII,QAA8C,GAAG,CAAC,CAAC;MAEvD,KAAK,MAAMC,GAAG,IAAIJ,UAAU,EAAE;QAC5B,IAAII,GAAG,IAAId,YAAY,EAAE;UACvB,MAAMe,OAAO,GAAGf,YAAY,CAACc,GAAG,CAA8B;UAC9DD,QAAQ,CAACE,OAAO,CAAC,GAAGL,UAAU,CAACI,GAAG,CAAC;QACrC;MACF;MACA,OAAOD,QAAQ;IACjB,CAAC,CAAC,OAAOG,CAAC,EAAE;MACV,MAAM,IAAIC,KAAK,CAAClB,QAAQ,CAACiB,CAAC,CAAC,CAAC;IAC9B;EACF;AACF","ignoreList":[]}
+{"version":3,"names":["ResourceFetcher","DeeplabLabel","ETError","getError","BaseNonStaticModule","ImageSegmentationModule","load","model","onDownloadProgressCallback","paths","fetch","modelSource","length","Error","nativeModule","global","loadImageSegmentation","forward","imageSource","classesOfInterest","resize","ModuleNotLoaded","stringDict","generate","map","label","enumDict","key","enumKey"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ImageSegmentationModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,6BAA6B;AAE7D,SAASC,YAAY,QAAQ,+BAA+B;AAC5D,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,mBAAmB,QAAQ,wBAAwB;AAE5D,OAAO,MAAMC,uBAAuB,SAASD,mBAAmB,CAAC;EAC/D,MAAME,IAAIA,CACRC,KAAsC,EACtCC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EAClD;IACf,MAAMC,KAAK,GAAG,MAAMT,eAAe,CAACU,KAAK,CACvCF,0BAA0B,EAC1BD,KAAK,CAACI,WACR,CAAC;IACD,IAAIF,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACG,MAAM,GAAG,CAAC,EAAE;MACtC,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;IAC1C;IACA,IAAI,CAACC,YAAY,GAAGC,MAAM,CAACC,qBAAqB,CAACP,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;EAClE;EAEA,MAAMQ,OAAOA,CACXC,WAAmB,EACnBC,iBAAkC,EAClCC,MAAgB,EAC+B;IAC/C,IAAI,IAAI,CAACN,YAAY,IAAI,IAAI,EAAE;MAC7B,MAAM,IAAID,KAAK,CAACV,QAAQ,CAACD,OAAO,CAACmB,eAAe,CAAC,CAAC;IACpD;IAEA,MAAMC,UAAU,GAAG,MAAM,IAAI,CAACR,YAAY,CAACS,QAAQ,CACjDL,WAAW,EACX,CAACC,iBAAiB,IAAI,EAAE,EAAEK,GAAG,CAAEC,KAAK,IAAKxB,YAAY,CAACwB,KAAK,CAAC,CAAC,EAC7DL,MAAM,IAAI,KACZ,CAAC;IAED,IAAIM,QAA8C,GAAG,CAAC,CAAC;IAEvD,KAAK,MAAMC,GAAG,IAAIL,UAAU,EAAE;MAC5B,IAAIK,GAAG,IAAI1B,YAAY,EAAE;QACvB,MAAM2B,OAAO,GAAG3B,YAAY,CAAC0B,GAAG,CAA8B;QAC9DD,QAAQ,CAACE,OAAO,CAAC,GAAGN,UAAU,CAACK,GAAG,CAAC;MACrC;IACF;IACA,OAAOD,QAAQ;EACjB;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/OCRModule.js

@@ -2,18 +2,21 @@
 
 import { OCRController } from '../../controllers/OCRController';
 export class OCRModule {
-  static onDownloadProgressCallback = _downloadProgress => {};
-  static async load(detectorSource, recognizerSources, language = 'en') {
-    this.module = new OCRController({
-      modelDownloadProgressCallback: this.onDownloadProgressCallback
-    });
-    await this.module.loadModel(detectorSource, recognizerSources, language);
+  constructor() {
+    this.controller = new OCRController();
   }
-  static async forward(input) {
-    return await this.module.forward(input);
+  async load(model, onDownloadProgressCallback = () => {}) {
+    await this.controller.load(model.detectorSource, {
+      recognizerLarge: model.recognizerLarge,
+      recognizerMedium: model.recognizerMedium,
+      recognizerSmall: model.recognizerSmall
+    }, model.language, onDownloadProgressCallback);
   }
-  static onDownloadProgress(callback) {
-    this.onDownloadProgressCallback = callback;
+  async forward(input) {
+    return await this.controller.forward(input);
+  }
+  delete() {
+    this.controller.delete();
   }
 }
 //# sourceMappingURL=OCRModule.js.map

package/lib/module/modules/computer_vision/OCRModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["OCRController","OCRModule","onDownloadProgressCallback","_downloadProgress","load","detectorSource","recognizerSources","language","module","modelDownloadProgressCallback","loadModel","forward","input","onDownloadProgress","callback"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/OCRModule.ts"],"mappings":";;AAAA,SAASA,aAAa,QAAQ,iCAAiC;AAI/D,OAAO,MAAMC,SAAS,CAAC;EAGrB,OAAOC,0BAA0B,GAAIC,iBAAyB,IAAK,CAAC,CAAC;EAErE,aAAaC,IAAIA,CACfC,cAA8B,EAC9BC,iBAIC,EACDC,QAAqB,GAAG,IAAI,EAC5B;IACA,IAAI,CAACC,MAAM,GAAG,IAAIR,aAAa,CAAC;MAC9BS,6BAA6B,EAAE,IAAI,CAACP;IACtC,CAAC,CAAC;IAEF,MAAM,IAAI,CAACM,MAAM,CAACE,SAAS,CAACL,cAAc,EAAEC,iBAAiB,EAAEC,QAAQ,CAAC;EAC1E;EAEA,aAAaI,OAAOA,CAACC,KAAa,EAAE;IAClC,OAAO,MAAM,IAAI,CAACJ,MAAM,CAACG,OAAO,CAACC,KAAK,CAAC;EACzC;EAEA,OAAOC,kBAAkBA,CAACC,QAA4C,EAAE;IACtE,IAAI,CAACZ,0BAA0B,GAAGY,QAAQ;EAC5C;AACF","ignoreList":[]}
+{"version":3,"names":["OCRController","OCRModule","constructor","controller","load","model","onDownloadProgressCallback","detectorSource","recognizerLarge","recognizerMedium","recognizerSmall","language","forward","input","delete"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/OCRModule.ts"],"mappings":";;AAAA,SAASA,aAAa,QAAQ,iCAAiC;AAI/D,OAAO,MAAMC,SAAS,CAAC;EAGrBC,WAAWA,CAAA,EAAG;IACZ,IAAI,CAACC,UAAU,GAAG,IAAIH,aAAa,CAAC,CAAC;EACvC;EAEA,MAAMI,IAAIA,CACRC,KAMC,EACDC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EACjE;IACA,MAAM,IAAI,CAACH,UAAU,CAACC,IAAI,CACxBC,KAAK,CAACE,cAAc,EACpB;MACEC,eAAe,EAAEH,KAAK,CAACG,eAAe;MACtCC,gBAAgB,EAAEJ,KAAK,CAACI,gBAAgB;MACxCC,eAAe,EAAEL,KAAK,CAACK;IACzB,CAAC,EACDL,KAAK,CAACM,QAAQ,EACdL,0BACF,CAAC;EACH;EAEA,MAAMM,OAAOA,CAACC,KAAa,EAAE;IAC3B,OAAO,MAAM,IAAI,CAACV,UAAU,CAACS,OAAO,CAACC,KAAK,CAAC;EAC7C;EAEAC,MAAMA,CAAA,EAAG;IACP,IAAI,CAACX,UAAU,CAACW,MAAM,CAAC,CAAC;EAC1B;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/ObjectDetectionModule.js

@@ -1,14 +1,19 @@
 "use strict";
 
-import { ObjectDetectionNativeModule } from '../../native/RnExecutorchModules';
-import { BaseModule } from '../BaseModule';
-export class ObjectDetectionModule extends BaseModule {
-  static nativeModule = ObjectDetectionNativeModule;
-  static async load(modelSource) {
-    return await super.load(modelSource);
+import { ResourceFetcher } from '../../utils/ResourceFetcher';
+import { ETError, getError } from '../../Error';
+import { BaseNonStaticModule } from '../BaseNonStaticModule';
+export class ObjectDetectionModule extends BaseNonStaticModule {
+  async load(model, onDownloadProgressCallback = () => {}) {
+    const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, model.modelSource);
+    if (paths === null || paths.length < 1) {
+      throw new Error('Download interrupted.');
+    }
+    this.nativeModule = global.loadObjectDetection(paths[0] || '');
   }
-  static async forward(input) {
-    return await this.nativeModule.forward(input);
+  async forward(imageSource, detectionThreshold = 0.7) {
+    if (this.nativeModule == null) throw new Error(getError(ETError.ModuleNotLoaded));
+    return await this.nativeModule.generate(imageSource, detectionThreshold);
   }
 }
 //# sourceMappingURL=ObjectDetectionModule.js.map

package/lib/module/modules/computer_vision/ObjectDetectionModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["ObjectDetectionNativeModule","BaseModule","ObjectDetectionModule","nativeModule","load","modelSource","forward","input"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ObjectDetectionModule.ts"],"mappings":";;AAAA,SAASA,2BAA2B,QAAQ,kCAAkC;AAE9E,SAASC,UAAU,QAAQ,eAAe;AAE1C,OAAO,MAAMC,qBAAqB,SAASD,UAAU,CAAC;EACpD,OAA0BE,YAAY,GAAGH,2BAA2B;EAEpE,aAAsBI,IAAIA,CAACC,WAA2B,EAAE;IACtD,OAAO,MAAM,KAAK,CAACD,IAAI,CAACC,WAAW,CAAC;EACtC;EAEA,aAAsBC,OAAOA,CAC3BC,KAAa,EACiC;IAC9C,OAAO,MAAM,IAAI,CAACJ,YAAY,CAACG,OAAO,CAACC,KAAK,CAAC;EAC/C;AACF","ignoreList":[]}
+{"version":3,"names":["ResourceFetcher","ETError","getError","BaseNonStaticModule","ObjectDetectionModule","load","model","onDownloadProgressCallback","paths","fetch","modelSource","length","Error","nativeModule","global","loadObjectDetection","forward","imageSource","detectionThreshold","ModuleNotLoaded","generate"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/ObjectDetectionModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,6BAA6B;AAG7D,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,mBAAmB,QAAQ,wBAAwB;AAE5D,OAAO,MAAMC,qBAAqB,SAASD,mBAAmB,CAAC;EAC7D,MAAME,IAAIA,CACRC,KAAsC,EACtCC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EAClD;IACf,MAAMC,KAAK,GAAG,MAAMR,eAAe,CAACS,KAAK,CACvCF,0BAA0B,EAC1BD,KAAK,CAACI,WACR,CAAC;IACD,IAAIF,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACG,MAAM,GAAG,CAAC,EAAE;MACtC,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;IAC1C;IACA,IAAI,CAACC,YAAY,GAAGC,MAAM,CAACC,mBAAmB,CAACP,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;EAChE;EAEA,MAAMQ,OAAOA,CACXC,WAAmB,EACnBC,kBAA0B,GAAG,GAAG,EACV;IACtB,IAAI,IAAI,CAACL,YAAY,IAAI,IAAI,EAC3B,MAAM,IAAID,KAAK,CAACV,QAAQ,CAACD,OAAO,CAACkB,eAAe,CAAC,CAAC;IACpD,OAAO,MAAM,IAAI,CAACN,YAAY,CAACO,QAAQ,CAACH,WAAW,EAAEC,kBAAkB,CAAC;EAC1E;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/StyleTransferModule.js

@@ -1,14 +1,19 @@
 "use strict";
 
-import { StyleTransferNativeModule } from '../../native/RnExecutorchModules';
-import { BaseModule } from '../BaseModule';
-export class StyleTransferModule extends BaseModule {
-  static nativeModule = StyleTransferNativeModule;
-  static async load(modelSource) {
-    return await super.load(modelSource);
+import { ResourceFetcher } from '../../utils/ResourceFetcher';
+import { ETError, getError } from '../../Error';
+import { BaseNonStaticModule } from '../BaseNonStaticModule';
+export class StyleTransferModule extends BaseNonStaticModule {
+  async load(model, onDownloadProgressCallback = () => {}) {
+    const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, model.modelSource);
+    if (paths === null || paths.length < 1) {
+      throw new Error('Download interrupted.');
+    }
+    this.nativeModule = global.loadStyleTransfer(paths[0] || '');
   }
-  static async forward(input) {
-    return await this.nativeModule.forward(input);
+  async forward(imageSource) {
+    if (this.nativeModule == null) throw new Error(getError(ETError.ModuleNotLoaded));
+    return await this.nativeModule.generate(imageSource);
   }
 }
 //# sourceMappingURL=StyleTransferModule.js.map

package/lib/module/modules/computer_vision/StyleTransferModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["StyleTransferNativeModule","BaseModule","StyleTransferModule","nativeModule","load","modelSource","forward","input"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/StyleTransferModule.ts"],"mappings":";;AAAA,SAASA,yBAAyB,QAAQ,kCAAkC;AAE5E,SAASC,UAAU,QAAQ,eAAe;AAE1C,OAAO,MAAMC,mBAAmB,SAASD,UAAU,CAAC;EAClD,OAA0BE,YAAY,GAAGH,yBAAyB;EAElE,aAAsBI,IAAIA,CAACC,WAA2B,EAAE;IACtD,OAAO,MAAM,KAAK,CAACD,IAAI,CAACC,WAAW,CAAC;EACtC;EAEA,aAAsBC,OAAOA,CAC3BC,KAAa,EACiC;IAC9C,OAAO,MAAM,IAAI,CAACJ,YAAY,CAACG,OAAO,CAACC,KAAK,CAAC;EAC/C;AACF","ignoreList":[]}
+{"version":3,"names":["ResourceFetcher","ETError","getError","BaseNonStaticModule","StyleTransferModule","load","model","onDownloadProgressCallback","paths","fetch","modelSource","length","Error","nativeModule","global","loadStyleTransfer","forward","imageSource","ModuleNotLoaded","generate"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/StyleTransferModule.ts"],"mappings":";;AAAA,SAASA,eAAe,QAAQ,6BAA6B;AAE7D,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,mBAAmB,QAAQ,wBAAwB;AAE5D,OAAO,MAAMC,mBAAmB,SAASD,mBAAmB,CAAC;EAC3D,MAAME,IAAIA,CACRC,KAAsC,EACtCC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EAClD;IACf,MAAMC,KAAK,GAAG,MAAMR,eAAe,CAACS,KAAK,CACvCF,0BAA0B,EAC1BD,KAAK,CAACI,WACR,CAAC;IACD,IAAIF,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACG,MAAM,GAAG,CAAC,EAAE;MACtC,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;IAC1C;IACA,IAAI,CAACC,YAAY,GAAGC,MAAM,CAACC,iBAAiB,CAACP,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;EAC9D;EAEA,MAAMQ,OAAOA,CAACC,WAAmB,EAAmB;IAClD,IAAI,IAAI,CAACJ,YAAY,IAAI,IAAI,EAC3B,MAAM,IAAID,KAAK,CAACV,QAAQ,CAACD,OAAO,CAACiB,eAAe,CAAC,CAAC;IACpD,OAAO,MAAM,IAAI,CAACL,YAAY,CAACM,QAAQ,CAACF,WAAW,CAAC;EACtD;AACF","ignoreList":[]}

package/lib/module/modules/computer_vision/VerticalOCRModule.js

@@ -2,18 +2,23 @@
 
 import { VerticalOCRController } from '../../controllers/VerticalOCRController';
 export class VerticalOCRModule {
-  static onDownloadProgressCallback = _downloadProgress => {};
-  static async load(detectorSources, recognizerSources, language = 'en', independentCharacters = false) {
-    this.module = new VerticalOCRController({
-      modelDownloadProgressCallback: this.onDownloadProgressCallback
-    });
-    await this.module.loadModel(detectorSources, recognizerSources, language, independentCharacters);
+  constructor() {
+    this.controller = new VerticalOCRController();
   }
-  static async forward(input) {
-    return await this.module.forward(input);
+  async load(model, independentCharacters, onDownloadProgressCallback = () => {}) {
+    await this.controller.load({
+      detectorLarge: model.detectorLarge,
+      detectorNarrow: model.detectorNarrow
+    }, {
+      recognizerLarge: model.recognizerLarge,
+      recognizerSmall: model.recognizerSmall
+    }, model.language, independentCharacters, onDownloadProgressCallback);
   }
-  static onDownloadProgress(callback) {
-    this.onDownloadProgressCallback = callback;
+  async forward(input) {
+    return await this.controller.forward(input);
+  }
+  delete() {
+    this.controller.delete();
   }
 }
 //# sourceMappingURL=VerticalOCRModule.js.map

package/lib/module/modules/computer_vision/VerticalOCRModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["VerticalOCRController","VerticalOCRModule","onDownloadProgressCallback","_downloadProgress","load","detectorSources","recognizerSources","language","independentCharacters","module","modelDownloadProgressCallback","loadModel","forward","input","onDownloadProgress","callback"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/VerticalOCRModule.ts"],"mappings":";;AAAA,SAASA,qBAAqB,QAAQ,yCAAyC;AAI/E,OAAO,MAAMC,iBAAiB,CAAC;EAG7B,OAAOC,0BAA0B,GAAIC,iBAAyB,IAAK,CAAC,CAAC;EAErE,aAAaC,IAAIA,CACfC,eAGC,EACDC,iBAGC,EACDC,QAAqB,GAAG,IAAI,EAC5BC,qBAA8B,GAAG,KAAK,EACtC;IACA,IAAI,CAACC,MAAM,GAAG,IAAIT,qBAAqB,CAAC;MACtCU,6BAA6B,EAAE,IAAI,CAACR;IACtC,CAAC,CAAC;IAEF,MAAM,IAAI,CAACO,MAAM,CAACE,SAAS,CACzBN,eAAe,EACfC,iBAAiB,EACjBC,QAAQ,EACRC,qBACF,CAAC;EACH;EAEA,aAAaI,OAAOA,CAACC,KAAa,EAAE;IAClC,OAAO,MAAM,IAAI,CAACJ,MAAM,CAACG,OAAO,CAACC,KAAK,CAAC;EACzC;EAEA,OAAOC,kBAAkBA,CAACC,QAA4C,EAAE;IACtE,IAAI,CAACb,0BAA0B,GAAGa,QAAQ;EAC5C;AACF","ignoreList":[]}
+{"version":3,"names":["VerticalOCRController","VerticalOCRModule","constructor","controller","load","model","independentCharacters","onDownloadProgressCallback","detectorLarge","detectorNarrow","recognizerLarge","recognizerSmall","language","forward","input","delete"],"sourceRoot":"../../../../src","sources":["modules/computer_vision/VerticalOCRModule.ts"],"mappings":";;AAAA,SAASA,qBAAqB,QAAQ,yCAAyC;AAI/E,OAAO,MAAMC,iBAAiB,CAAC;EAG7BC,WAAWA,CAAA,EAAG;IACZ,IAAI,CAACC,UAAU,GAAG,IAAIH,qBAAqB,CAAC,CAAC;EAC/C;EAEA,MAAMI,IAAIA,CACRC,KAMC,EACDC,qBAA8B,EAC9BC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EACjE;IACA,MAAM,IAAI,CAACJ,UAAU,CAACC,IAAI,CACxB;MACEI,aAAa,EAAEH,KAAK,CAACG,aAAa;MAClCC,cAAc,EAAEJ,KAAK,CAACI;IACxB,CAAC,EACD;MACEC,eAAe,EAAEL,KAAK,CAACK,eAAe;MACtCC,eAAe,EAAEN,KAAK,CAACM;IACzB,CAAC,EACDN,KAAK,CAACO,QAAQ,EACdN,qBAAqB,EACrBC,0BACF,CAAC;EACH;EAEA,MAAMM,OAAOA,CAACC,KAAa,EAAE;IAC3B,OAAO,MAAM,IAAI,CAACX,UAAU,CAACU,OAAO,CAACC,KAAK,CAAC;EAC7C;EAEAC,MAAMA,CAAA,EAAG;IACP,IAAI,CAACZ,UAAU,CAACY,MAAM,CAAC,CAAC;EAC1B;AACF","ignoreList":[]}

package/lib/module/modules/general/ExecutorchModule.js

@@ -1,43 +1,17 @@
 "use strict";
 
-import { ETError, getError } from '../../Error';
-import { ETModuleNativeModule } from '../../native/RnExecutorchModules';
-import { getTypeIdentifier } from '../../types/common';
-import { BaseModule } from '../BaseModule';
-export class ExecutorchModule extends BaseModule {
-  static nativeModule = ETModuleNativeModule;
-  static async load(modelSource) {
-    return await super.load(modelSource);
-  }
-  static async forward(input, shape) {
-    if (!Array.isArray(input)) {
-      input = [input];
-    }
-    const inputTypeIdentifiers = [];
-    const modelInputs = [];
-    for (const subInput of input) {
-      const typeIdentifier = getTypeIdentifier(subInput);
-      if (typeIdentifier === -1) {
-        throw new Error(getError(ETError.InvalidArgument));
-      }
-      inputTypeIdentifiers.push(typeIdentifier);
-      modelInputs.push(Array.from(subInput, x => Number(x)));
-    }
-    try {
-      return await this.nativeModule.forward(modelInputs, shape, inputTypeIdentifiers);
-    } catch (e) {
-      throw new Error(getError(e));
-    }
-  }
-  static async loadMethod(methodName) {
-    try {
-      await this.nativeModule.loadMethod(methodName);
-    } catch (e) {
-      throw new Error(getError(e));
+import { BaseNonStaticModule } from '../BaseNonStaticModule';
+import { ResourceFetcher } from '../../utils/ResourceFetcher';
+export class ExecutorchModule extends BaseNonStaticModule {
+  async load(modelSource, onDownloadProgressCallback = () => {}) {
+    const paths = await ResourceFetcher.fetch(onDownloadProgressCallback, modelSource);
+    if (paths === null || paths.length < 1) {
+      throw new Error('Download interrupted.');
     }
+    this.nativeModule = global.loadExecutorchModule(paths[0] || '');
   }
-  static async loadForward() {
-    await this.loadMethod('forward');
+  async forward(inputTensor) {
+    return await this.forwardET(inputTensor);
   }
 }
 //# sourceMappingURL=ExecutorchModule.js.map

package/lib/module/modules/general/ExecutorchModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["ETError","getError","ETModuleNativeModule","getTypeIdentifier","BaseModule","ExecutorchModule","nativeModule","load","modelSource","forward","input","shape","Array","isArray","inputTypeIdentifiers","modelInputs","subInput","typeIdentifier","Error","InvalidArgument","push","from","x","Number","e","loadMethod","methodName","loadForward"],"sourceRoot":"../../../../src","sources":["modules/general/ExecutorchModule.ts"],"mappings":";;AAAA,SAASA,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,oBAAoB,QAAQ,kCAAkC;AAGvE,SAASC,iBAAiB,QAAQ,oBAAoB;AACtD,SAASC,UAAU,QAAQ,eAAe;AAE1C,OAAO,MAAMC,gBAAgB,SAASD,UAAU,CAAC;EAC/C,OAA0BE,YAAY,GAAGJ,oBAAoB;EAE7D,aAAsBK,IAAIA,CAACC,WAA2B,EAAE;IACtD,OAAO,MAAM,KAAK,CAACD,IAAI,CAACC,WAAW,CAAC;EACtC;EAEA,aAAsBC,OAAOA,CAACC,KAA0B,EAAEC,KAAiB,EAAE;IAC3E,IAAI,CAACC,KAAK,CAACC,OAAO,CAACH,KAAK,CAAC,EAAE;MACzBA,KAAK,GAAG,CAACA,KAAK,CAAC;IACjB;IAEA,MAAMI,oBAA8B,GAAG,EAAE;IACzC,MAAMC,WAAuB,GAAG,EAAE;IAElC,KAAK,MAAMC,QAAQ,IAAIN,KAAK,EAAE;MAC5B,MAAMO,cAAc,GAAGd,iBAAiB,CAACa,QAAQ,CAAC;MAClD,IAAIC,cAAc,KAAK,CAAC,CAAC,EAAE;QACzB,MAAM,IAAIC,KAAK,CAACjB,QAAQ,CAACD,OAAO,CAACmB,eAAe,CAAC,CAAC;MACpD;MACAL,oBAAoB,CAACM,IAAI,CAACH,cAAc,CAAC;MACzCF,WAAW,CAACK,IAAI,CAACR,KAAK,CAACS,IAAI,CAACL,QAAQ,EAAGM,CAAkB,IAAKC,MAAM,CAACD,CAAC,CAAC,CAAC,CAAC;IAC3E;IAEA,IAAI;MACF,OAAO,MAAM,IAAI,CAAChB,YAAY,CAACG,OAAO,CACpCM,WAAW,EACXJ,KAAK,EACLG,oBACF,CAAC;IACH,CAAC,CAAC,OAAOU,CAAC,EAAE;MACV,MAAM,IAAIN,KAAK,CAACjB,QAAQ,CAACuB,CAAC,CAAC,CAAC;IAC9B;EACF;EAEA,aAAaC,UAAUA,CAACC,UAAkB,EAAE;IAC1C,IAAI;MACF,MAAM,IAAI,CAACpB,YAAY,CAACmB,UAAU,CAACC,UAAU,CAAC;IAChD,CAAC,CAAC,OAAOF,CAAC,EAAE;MACV,MAAM,IAAIN,KAAK,CAACjB,QAAQ,CAACuB,CAAC,CAAC,CAAC;IAC9B;EACF;EAEA,aAAaG,WAAWA,CAAA,EAAG;IACzB,MAAM,IAAI,CAACF,UAAU,CAAC,SAAS,CAAC;EAClC;AACF","ignoreList":[]}
+{"version":3,"names":["BaseNonStaticModule","ResourceFetcher","ExecutorchModule","load","modelSource","onDownloadProgressCallback","paths","fetch","length","Error","nativeModule","global","loadExecutorchModule","forward","inputTensor","forwardET"],"sourceRoot":"../../../../src","sources":["modules/general/ExecutorchModule.ts"],"mappings":";;AACA,SAASA,mBAAmB,QAAQ,wBAAwB;AAE5D,SAASC,eAAe,QAAQ,6BAA6B;AAE7D,OAAO,MAAMC,gBAAgB,SAASF,mBAAmB,CAAC;EACxD,MAAMG,IAAIA,CACRC,WAA2B,EAC3BC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EAClD;IACf,MAAMC,KAAK,GAAG,MAAML,eAAe,CAACM,KAAK,CACvCF,0BAA0B,EAC1BD,WACF,CAAC;IACD,IAAIE,KAAK,KAAK,IAAI,IAAIA,KAAK,CAACE,MAAM,GAAG,CAAC,EAAE;MACtC,MAAM,IAAIC,KAAK,CAAC,uBAAuB,CAAC;IAC1C;IACA,IAAI,CAACC,YAAY,GAAGC,MAAM,CAACC,oBAAoB,CAACN,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;EACjE;EAEA,MAAMO,OAAOA,CAACC,WAAwB,EAAwB;IAC5D,OAAO,MAAM,IAAI,CAACC,SAAS,CAACD,WAAW,CAAC;EAC1C;AACF","ignoreList":[]}

package/lib/module/modules/natural_language_processing/LLMModule.js

@@ -2,33 +2,29 @@
 
 import { LLMController } from '../../controllers/LLMController';
 export class LLMModule {
-  static async load({
-    modelSource,
-    tokenizerSource,
-    tokenizerConfigSource,
-    onDownloadProgressCallback,
+  constructor({
     tokenCallback,
     responseCallback,
     messageHistoryCallback
-  }) {
+  } = {}) {
     this.controller = new LLMController({
-      tokenCallback: tokenCallback,
-      responseCallback: responseCallback,
-      messageHistoryCallback: messageHistoryCallback,
-      onDownloadProgressCallback: onDownloadProgressCallback
+      tokenCallback,
+      responseCallback,
+      messageHistoryCallback
     });
+  }
+  async load(model, onDownloadProgressCallback = () => {}) {
     await this.controller.load({
-      modelSource,
-      tokenizerSource,
-      tokenizerConfigSource
+      ...model,
+      onDownloadProgressCallback
     });
   }
-  static setTokenCallback({
+  setTokenCallback({
     tokenCallback
   }) {
     this.controller.setTokenCallback(tokenCallback);
   }
-  static configure({
+  configure({
     chatConfig,
     toolsConfig
   }) {
@@ -37,26 +33,26 @@ export class LLMModule {
       toolsConfig
     });
   }
-  static async forward(input) {
+  async forward(input) {
     await this.controller.forward(input);
     return this.controller.response;
   }
-  static async generate(messages, tools) {
+  async generate(messages, tools) {
     await this.controller.generate(messages, tools);
     return this.controller.response;
   }
-  static async sendMessage(message) {
+  async sendMessage(message) {
     await this.controller.sendMessage(message);
     return this.controller.messageHistory;
   }
-  static async deleteMessage(index) {
-    await this.controller.deleteMessage(index);
+  deleteMessage(index) {
+    this.controller.deleteMessage(index);
     return this.controller.messageHistory;
   }
-  static interrupt() {
+  interrupt() {
     this.controller.interrupt();
   }
-  static delete() {
+  delete() {
     this.controller.delete();
   }
 }

package/lib/module/modules/natural_language_processing/LLMModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["LLMController","LLMModule","load","modelSource","tokenizerSource","tokenizerConfigSource","onDownloadProgressCallback","tokenCallback","responseCallback","messageHistoryCallback","controller","setTokenCallback","configure","chatConfig","toolsConfig","forward","input","response","generate","messages","tools","sendMessage","message","messageHistory","deleteMessage","index","interrupt","delete"],"sourceRoot":"../../../../src","sources":["modules/natural_language_processing/LLMModule.ts"],"mappings":";;AAAA,SAASA,aAAa,QAAQ,iCAAiC;AAI/D,OAAO,MAAMC,SAAS,CAAC;EAGrB,aAAaC,IAAIA,CAAC;IAChBC,WAAW;IACXC,eAAe;IACfC,qBAAqB;IACrBC,0BAA0B;IAC1BC,aAAa;IACbC,gBAAgB;IAChBC;EASF,CAAC,EAAE;IACD,IAAI,CAACC,UAAU,GAAG,IAAIV,aAAa,CAAC;MAClCO,aAAa,EAAEA,aAAa;MAC5BC,gBAAgB,EAAEA,gBAAgB;MAClCC,sBAAsB,EAAEA,sBAAsB;MAC9CH,0BAA0B,EAAEA;IAC9B,CAAC,CAAC;IACF,MAAM,IAAI,CAACI,UAAU,CAACR,IAAI,CAAC;MACzBC,WAAW;MACXC,eAAe;MACfC;IACF,CAAC,CAAC;EACJ;EAEA,OAAOM,gBAAgBA,CAAC;IACtBJ;EAGF,CAAC,EAAE;IACD,IAAI,CAACG,UAAU,CAACC,gBAAgB,CAACJ,aAAa,CAAC;EACjD;EAEA,OAAOK,SAASA,CAAC;IACfC,UAAU;IACVC;EAIF,CAAC,EAAE;IACD,IAAI,CAACJ,UAAU,CAACE,SAAS,CAAC;MAAEC,UAAU;MAAEC;IAAY,CAAC,CAAC;EACxD;EAEA,aAAaC,OAAOA,CAACC,KAAa,EAAmB;IACnD,MAAM,IAAI,CAACN,UAAU,CAACK,OAAO,CAACC,KAAK,CAAC;IACpC,OAAO,IAAI,CAACN,UAAU,CAACO,QAAQ;EACjC;EAEA,aAAaC,QAAQA,CACnBC,QAAmB,EACnBC,KAAiB,EACA;IACjB,MAAM,IAAI,CAACV,UAAU,CAACQ,QAAQ,CAACC,QAAQ,EAAEC,KAAK,CAAC;IAC/C,OAAO,IAAI,CAACV,UAAU,CAACO,QAAQ;EACjC;EAEA,aAAaI,WAAWA,CAACC,OAAe,EAAsB;IAC5D,MAAM,IAAI,CAACZ,UAAU,CAACW,WAAW,CAACC,OAAO,CAAC;IAC1C,OAAO,IAAI,CAACZ,UAAU,CAACa,cAAc;EACvC;EAEA,aAAaC,aAAaA,CAACC,KAAa,EAAsB;IAC5D,MAAM,IAAI,CAACf,UAAU,CAACc,aAAa,CAACC,KAAK,CAAC;IAC1C,OAAO,IAAI,CAACf,UAAU,CAACa,cAAc;EACvC;EAEA,OAAOG,SAASA,CAAA,EAAG;IACjB,IAAI,CAAChB,UAAU,CAACgB,SAAS,CAAC,CAAC;EAC7B;EAEA,OAAOC,MAAMA,CAAA,EAAG;IACd,IAAI,CAACjB,UAAU,CAACiB,MAAM,CAAC,CAAC;EAC1B;AACF","ignoreList":[]}
+{"version":3,"names":["LLMController","LLMModule","constructor","tokenCallback","responseCallback","messageHistoryCallback","controller","load","model","onDownloadProgressCallback","setTokenCallback","configure","chatConfig","toolsConfig","forward","input","response","generate","messages","tools","sendMessage","message","messageHistory","deleteMessage","index","interrupt","delete"],"sourceRoot":"../../../../src","sources":["modules/natural_language_processing/LLMModule.ts"],"mappings":";;AAAA,SAASA,aAAa,QAAQ,iCAAiC;AAI/D,OAAO,MAAMC,SAAS,CAAC;EAGrBC,WAAWA,CAAC;IACVC,aAAa;IACbC,gBAAgB;IAChBC;EAKF,CAAC,GAAG,CAAC,CAAC,EAAE;IACN,IAAI,CAACC,UAAU,GAAG,IAAIN,aAAa,CAAC;MAClCG,aAAa;MACbC,gBAAgB;MAChBC;IACF,CAAC,CAAC;EACJ;EAEA,MAAME,IAAIA,CACRC,KAIC,EACDC,0BAAsD,GAAGA,CAAA,KAAM,CAAC,CAAC,EACjE;IACA,MAAM,IAAI,CAACH,UAAU,CAACC,IAAI,CAAC;MACzB,GAAGC,KAAK;MACRC;IACF,CAAC,CAAC;EACJ;EAEAC,gBAAgBA,CAAC;IACfP;EAGF,CAAC,EAAE;IACD,IAAI,CAACG,UAAU,CAACI,gBAAgB,CAACP,aAAa,CAAC;EACjD;EAEAQ,SAASA,CAAC;IACRC,UAAU;IACVC;EAIF,CAAC,EAAE;IACD,IAAI,CAACP,UAAU,CAACK,SAAS,CAAC;MAAEC,UAAU;MAAEC;IAAY,CAAC,CAAC;EACxD;EAEA,MAAMC,OAAOA,CAACC,KAAa,EAAmB;IAC5C,MAAM,IAAI,CAACT,UAAU,CAACQ,OAAO,CAACC,KAAK,CAAC;IACpC,OAAO,IAAI,CAACT,UAAU,CAACU,QAAQ;EACjC;EAEA,MAAMC,QAAQA,CAACC,QAAmB,EAAEC,KAAiB,EAAmB;IACtE,MAAM,IAAI,CAACb,UAAU,CAACW,QAAQ,CAACC,QAAQ,EAAEC,KAAK,CAAC;IAC/C,OAAO,IAAI,CAACb,UAAU,CAACU,QAAQ;EACjC;EAEA,MAAMI,WAAWA,CAACC,OAAe,EAAsB;IACrD,MAAM,IAAI,CAACf,UAAU,CAACc,WAAW,CAACC,OAAO,CAAC;IAC1C,OAAO,IAAI,CAACf,UAAU,CAACgB,cAAc;EACvC;EAEAC,aAAaA,CAACC,KAAa,EAAa;IACtC,IAAI,CAAClB,UAAU,CAACiB,aAAa,CAACC,KAAK,CAAC;IACpC,OAAO,IAAI,CAAClB,UAAU,CAACgB,cAAc;EACvC;EAEAG,SAASA,CAAA,EAAG;IACV,IAAI,CAACnB,UAAU,CAACmB,SAAS,CAAC,CAAC;EAC7B;EAEAC,MAAMA,CAAA,EAAG;IACP,IAAI,CAACpB,UAAU,CAACoB,MAAM,CAAC,CAAC;EAC1B;AACF","ignoreList":[]}