react-native-executorch 0.3.2 → 0.4.1

package/lib/module/hooks/computer_vision/useOCR.js

@@ -1,29 +1,34 @@
 "use strict";
 
-import { useEffect, useState } from 'react';
+import { useEffect, useMemo, useState } from 'react';
 import { OCRController } from '../../controllers/OCRController';
 export const useOCR = ({
   detectorSource,
   recognizerSources,
-  language = 'en'
+  language = 'en',
+  preventLoad = false
 }) => {
   const [error, setError] = useState(null);
   const [isReady, setIsReady] = useState(false);
   const [isGenerating, setIsGenerating] = useState(false);
   const [downloadProgress, setDownloadProgress] = useState(0);
-  const [model, _] = useState(() => new OCRController({
+  const model = useMemo(() => new OCRController({
     modelDownloadProgressCallback: setDownloadProgress,
     isReadyCallback: setIsReady,
     isGeneratingCallback: setIsGenerating,
     errorCallback: setError
-  }));
+  }), []);
   useEffect(() => {
     const loadModel = async () => {
       await model.loadModel(detectorSource, recognizerSources, language);
     };
-    loadModel();
+    if (!preventLoad) {
+      loadModel();
+    }
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [detectorSource, language, JSON.stringify(recognizerSources)]);
+  }, [model, detectorSource, language,
+  // eslint-disable-next-line react-hooks/exhaustive-deps
+  JSON.stringify(recognizerSources), preventLoad]);
   return {
     error,
     isReady,

package/lib/module/hooks/computer_vision/useOCR.js.map

@@ -1 +1 @@
-{"version":3,"names":["useEffect","useState","OCRController","useOCR","detectorSource","recognizerSources","language","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","model","_","modelDownloadProgressCallback","isReadyCallback","isGeneratingCallback","errorCallback","loadModel","JSON","stringify","forward"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useOCR.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,QAAQ,QAAQ,OAAO;AAG3C,SAASC,aAAa,QAAQ,iCAAiC;AAU/D,OAAO,MAAMC,MAAM,GAAGA,CAAC;EACrBC,cAAc;EACdC,iBAAiB;EACjBC,QAAQ,GAAG;AASb,CAAC,KAAgB;EACf,MAAM,CAACC,KAAK,EAAEC,QAAQ,CAAC,GAAGP,QAAQ,CAAgB,IAAI,CAAC;EACvD,MAAM,CAACQ,OAAO,EAAEC,UAAU,CAAC,GAAGT,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACU,YAAY,EAAEC,eAAe,CAAC,GAAGX,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACY,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGb,QAAQ,CAAC,CAAC,CAAC;EAE3D,MAAM,CAACc,KAAK,EAAEC,CAAC,CAAC,GAAGf,QAAQ,CACzB,MACE,IAAIC,aAAa,CAAC;IAChBe,6BAA6B,EAAEH,mBAAmB;IAClDI,eAAe,EAAER,UAAU;IAC3BS,oBAAoB,EAAEP,eAAe;IACrCQ,aAAa,EAAEZ;EACjB,CAAC,CACL,CAAC;EAEDR,SAAS,CAAC,MAAM;IACd,MAAMqB,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,MAAMN,KAAK,CAACM,SAAS,CAACjB,cAAc,EAAEC,iBAAiB,EAAEC,QAAQ,CAAC;IACpE,CAAC;IAEDe,SAAS,CAAC,CAAC;IACX;EACF,CAAC,EAAE,CAACjB,cAAc,EAAEE,QAAQ,EAAEgB,IAAI,CAACC,SAAS,CAAClB,iBAAiB,CAAC,CAAC,CAAC;EAEjE,OAAO;IACLE,KAAK;IACLE,OAAO;IACPE,YAAY;IACZa,OAAO,EAAET,KAAK,CAACS,OAAO;IACtBX;EACF,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["useEffect","useMemo","useState","OCRController","useOCR","detectorSource","recognizerSources","language","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","model","modelDownloadProgressCallback","isReadyCallback","isGeneratingCallback","errorCallback","loadModel","JSON","stringify","forward"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useOCR.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AAGpD,SAASC,aAAa,QAAQ,iCAAiC;AAU/D,OAAO,MAAMC,MAAM,GAAGA,CAAC;EACrBC,cAAc;EACdC,iBAAiB;EACjBC,QAAQ,GAAG,IAAI;EACfC,WAAW,GAAG;AAUhB,CAAC,KAAgB;EACf,MAAM,CAACC,KAAK,EAAEC,QAAQ,CAAC,GAAGR,QAAQ,CAAgB,IAAI,CAAC;EACvD,MAAM,CAACS,OAAO,EAAEC,UAAU,CAAC,GAAGV,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACW,YAAY,EAAEC,eAAe,CAAC,GAAGZ,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACa,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGd,QAAQ,CAAC,CAAC,CAAC;EAE3D,MAAMe,KAAK,GAAGhB,OAAO,CACnB,MACE,IAAIE,aAAa,CAAC;IAChBe,6BAA6B,EAAEF,mBAAmB;IAClDG,eAAe,EAAEP,UAAU;IAC3BQ,oBAAoB,EAAEN,eAAe;IACrCO,aAAa,EAAEX;EACjB,CAAC,CAAC,EACJ,EACF,CAAC;EAEDV,SAAS,CAAC,MAAM;IACd,MAAMsB,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,MAAML,KAAK,CAACK,SAAS,CAACjB,cAAc,EAAEC,iBAAiB,EAAEC,QAAQ,CAAC;IACpE,CAAC;IAED,IAAI,CAACC,WAAW,EAAE;MAChBc,SAAS,CAAC,CAAC;IACb;IACA;EACF,CAAC,EAAE,CACDL,KAAK,EACLZ,cAAc,EACdE,QAAQ;EACR;EACAgB,IAAI,CAACC,SAAS,CAAClB,iBAAiB,CAAC,EACjCE,WAAW,CACZ,CAAC;EAEF,OAAO;IACLC,KAAK;IACLE,OAAO;IACPE,YAAY;IACZY,OAAO,EAAER,KAAK,CAACQ,OAAO;IACtBV;EACF,CAAC;AACH,CAAC","ignoreList":[]}

package/lib/module/hooks/computer_vision/useObjectDetection.js

@@ -1,28 +1,13 @@
 "use strict";
 
-import { useState } from 'react';
-import { _ObjectDetectionModule } from '../../native/RnExecutorchModules';
 import { useModule } from '../useModule';
+import { ObjectDetectionModule } from '../../modules/computer_vision/ObjectDetectionModule';
 export const useObjectDetection = ({
-  modelSource
-}) => {
-  const [module, _] = useState(() => new _ObjectDetectionModule());
-  const {
-    error,
-    isReady,
-    isGenerating,
-    downloadProgress,
-    forwardImage: forward
-  } = useModule({
-    modelSource,
-    module
-  });
-  return {
-    error,
-    isReady,
-    isGenerating,
-    downloadProgress,
-    forward
-  };
-};
+  modelSource,
+  preventLoad = false
+}) => useModule({
+  module: ObjectDetectionModule,
+  loadArgs: [modelSource],
+  preventLoad
+});
 //# sourceMappingURL=useObjectDetection.js.map

package/lib/module/hooks/computer_vision/useObjectDetection.js.map

@@ -1 +1 @@
-{"version":3,"names":["useState","_ObjectDetectionModule","useModule","useObjectDetection","modelSource","module","_","error","isReady","isGenerating","downloadProgress","forwardImage","forward"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useObjectDetection.ts"],"mappings":";;AAAA,SAASA,QAAQ,QAAQ,OAAO;AAChC,SAASC,sBAAsB,QAAQ,kCAAkC;AACzE,SAASC,SAAS,QAAQ,cAAc;AAOxC,OAAO,MAAMC,kBAAkB,GAAGA,CAAC;EACjCC;AACK,CAAC,KAMH;EACH,MAAM,CAACC,MAAM,EAAEC,CAAC,CAAC,GAAGN,QAAQ,CAAC,MAAM,IAAIC,sBAAsB,CAAC,CAAC,CAAC;EAChE,MAAM;IACJM,KAAK;IACLC,OAAO;IACPC,YAAY;IACZC,gBAAgB;IAChBC,YAAY,EAAEC;EAChB,CAAC,GAAGV,SAAS,CAAC;IACZE,WAAW;IACXC;EACF,CAAC,CAAC;EAEF,OAAO;IAAEE,KAAK;IAAEC,OAAO;IAAEC,YAAY;IAAEC,gBAAgB;IAAEE;EAAQ,CAAC;AACpE,CAAC","ignoreList":[]}
+{"version":3,"names":["useModule","ObjectDetectionModule","useObjectDetection","modelSource","preventLoad","module","loadArgs"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useObjectDetection.ts"],"mappings":";;AACA,SAASA,SAAS,QAAQ,cAAc;AACxC,SAASC,qBAAqB,QAAQ,qDAAqD;AAO3F,OAAO,MAAMC,kBAAkB,GAAGA,CAAC;EACjCC,WAAW;EACXC,WAAW,GAAG;AACT,CAAC,KACNJ,SAAS,CAAC;EACRK,MAAM,EAAEJ,qBAAqB;EAC7BK,QAAQ,EAAE,CAACH,WAAW,CAAC;EACvBC;AACF,CAAC,CAAC","ignoreList":[]}

package/lib/module/hooks/computer_vision/useStyleTransfer.js

@@ -1,28 +1,13 @@
 "use strict";
 
-import { useState } from 'react';
-import { _StyleTransferModule } from '../../native/RnExecutorchModules';
 import { useModule } from '../useModule';
+import { StyleTransferModule } from '../../modules/computer_vision/StyleTransferModule';
 export const useStyleTransfer = ({
-  modelSource
-}) => {
-  const [module, _] = useState(() => new _StyleTransferModule());
-  const {
-    error,
-    isReady,
-    isGenerating,
-    downloadProgress,
-    forwardImage: forward
-  } = useModule({
-    modelSource,
-    module
-  });
-  return {
-    error,
-    isReady,
-    isGenerating,
-    downloadProgress,
-    forward
-  };
-};
+  modelSource,
+  preventLoad = false
+}) => useModule({
+  module: StyleTransferModule,
+  loadArgs: [modelSource],
+  preventLoad
+});
 //# sourceMappingURL=useStyleTransfer.js.map

package/lib/module/hooks/computer_vision/useStyleTransfer.js.map

@@ -1 +1 @@
-{"version":3,"names":["useState","_StyleTransferModule","useModule","useStyleTransfer","modelSource","module","_","error","isReady","isGenerating","downloadProgress","forwardImage","forward"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useStyleTransfer.ts"],"mappings":";;AAAA,SAASA,QAAQ,QAAQ,OAAO;AAChC,SAASC,oBAAoB,QAAQ,kCAAkC;AACvE,SAASC,SAAS,QAAQ,cAAc;AAMxC,OAAO,MAAMC,gBAAgB,GAAGA,CAAC;EAC/BC;AACK,CAAC,KAMH;EACH,MAAM,CAACC,MAAM,EAAEC,CAAC,CAAC,GAAGN,QAAQ,CAAC,MAAM,IAAIC,oBAAoB,CAAC,CAAC,CAAC;EAC9D,MAAM;IACJM,KAAK;IACLC,OAAO;IACPC,YAAY;IACZC,gBAAgB;IAChBC,YAAY,EAAEC;EAChB,CAAC,GAAGV,SAAS,CAAC;IACZE,WAAW;IACXC;EACF,CAAC,CAAC;EAEF,OAAO;IAAEE,KAAK;IAAEC,OAAO;IAAEC,YAAY;IAAEC,gBAAgB;IAAEE;EAAQ,CAAC;AACpE,CAAC","ignoreList":[]}
+{"version":3,"names":["useModule","StyleTransferModule","useStyleTransfer","modelSource","preventLoad","module","loadArgs"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useStyleTransfer.ts"],"mappings":";;AACA,SAASA,SAAS,QAAQ,cAAc;AACxC,SAASC,mBAAmB,QAAQ,mDAAmD;AAOvF,OAAO,MAAMC,gBAAgB,GAAGA,CAAC;EAAEC,WAAW;EAAEC,WAAW,GAAG;AAAa,CAAC,KAC1EJ,SAAS,CAAC;EACRK,MAAM,EAAEJ,mBAAmB;EAC3BK,QAAQ,EAAE,CAACH,WAAW,CAAC;EACvBC;AACF,CAAC,CAAC","ignoreList":[]}

package/lib/module/hooks/computer_vision/useVerticalOCR.js

@@ -1,34 +1,37 @@
 "use strict";
 
-import { useEffect, useState } from 'react';
+import { useEffect, useMemo, useState } from 'react';
 import { VerticalOCRController } from '../../controllers/VerticalOCRController';
 export const useVerticalOCR = ({
   detectorSources,
   recognizerSources,
   language = 'en',
-  independentCharacters = false
+  independentCharacters = false,
+  preventLoad = false
 }) => {
   const [error, setError] = useState(null);
   const [isReady, setIsReady] = useState(false);
   const [isGenerating, setIsGenerating] = useState(false);
   const [downloadProgress, setDownloadProgress] = useState(0);
-  const [model, _] = useState(() => new VerticalOCRController({
+  const model = useMemo(() => new VerticalOCRController({
     modelDownloadProgressCallback: setDownloadProgress,
     isReadyCallback: setIsReady,
     isGeneratingCallback: setIsGenerating,
     errorCallback: setError
-  }));
+  }), []);
   useEffect(() => {
     const loadModel = async () => {
       await model.loadModel(detectorSources, recognizerSources, language, independentCharacters);
     };
-    loadModel();
+    if (!preventLoad) {
+      loadModel();
+    }
     // eslint-disable-next-line react-hooks/exhaustive-deps
-  }, [
+  }, [model,
   // eslint-disable-next-line react-hooks/exhaustive-deps
   JSON.stringify(detectorSources), language, independentCharacters,
   // eslint-disable-next-line react-hooks/exhaustive-deps
-  JSON.stringify(recognizerSources)]);
+  JSON.stringify(recognizerSources), preventLoad]);
   return {
     error,
     isReady,

package/lib/module/hooks/computer_vision/useVerticalOCR.js.map

@@ -1 +1 @@
-{"version":3,"names":["useEffect","useState","VerticalOCRController","useVerticalOCR","detectorSources","recognizerSources","language","independentCharacters","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","model","_","modelDownloadProgressCallback","isReadyCallback","isGeneratingCallback","errorCallback","loadModel","JSON","stringify","forward"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useVerticalOCR.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,QAAQ,QAAQ,OAAO;AAG3C,SAASC,qBAAqB,QAAQ,yCAAyC;AAU/E,OAAO,MAAMC,cAAc,GAAGA,CAAC;EAC7BC,eAAe;EACfC,iBAAiB;EACjBC,QAAQ,GAAG,IAAI;EACfC,qBAAqB,GAAG;AAY1B,CAAC,KAAgB;EACf,MAAM,CAACC,KAAK,EAAEC,QAAQ,CAAC,GAAGR,QAAQ,CAAgB,IAAI,CAAC;EACvD,MAAM,CAACS,OAAO,EAAEC,UAAU,CAAC,GAAGV,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACW,YAAY,EAAEC,eAAe,CAAC,GAAGZ,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACa,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGd,QAAQ,CAAC,CAAC,CAAC;EAE3D,MAAM,CAACe,KAAK,EAAEC,CAAC,CAAC,GAAGhB,QAAQ,CACzB,MACE,IAAIC,qBAAqB,CAAC;IACxBgB,6BAA6B,EAAEH,mBAAmB;IAClDI,eAAe,EAAER,UAAU;IAC3BS,oBAAoB,EAAEP,eAAe;IACrCQ,aAAa,EAAEZ;EACjB,CAAC,CACL,CAAC;EAEDT,SAAS,CAAC,MAAM;IACd,MAAMsB,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,MAAMN,KAAK,CAACM,SAAS,CACnBlB,eAAe,EACfC,iBAAiB,EACjBC,QAAQ,EACRC,qBACF,CAAC;IACH,CAAC;IAEDe,SAAS,CAAC,CAAC;IACX;EACF,CAAC,EAAE;EACD;EACAC,IAAI,CAACC,SAAS,CAACpB,eAAe,CAAC,EAC/BE,QAAQ,EACRC,qBAAqB;EACrB;EACAgB,IAAI,CAACC,SAAS,CAACnB,iBAAiB,CAAC,CAClC,CAAC;EAEF,OAAO;IACLG,KAAK;IACLE,OAAO;IACPE,YAAY;IACZa,OAAO,EAAET,KAAK,CAACS,OAAO;IACtBX;EACF,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["useEffect","useMemo","useState","VerticalOCRController","useVerticalOCR","detectorSources","recognizerSources","language","independentCharacters","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","model","modelDownloadProgressCallback","isReadyCallback","isGeneratingCallback","errorCallback","loadModel","JSON","stringify","forward"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useVerticalOCR.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AAGpD,SAASC,qBAAqB,QAAQ,yCAAyC;AAU/E,OAAO,MAAMC,cAAc,GAAGA,CAAC;EAC7BC,eAAe;EACfC,iBAAiB;EACjBC,QAAQ,GAAG,IAAI;EACfC,qBAAqB,GAAG,KAAK;EAC7BC,WAAW,GAAG;AAahB,CAAC,KAAgB;EACf,MAAM,CAACC,KAAK,EAAEC,QAAQ,CAAC,GAAGT,QAAQ,CAAgB,IAAI,CAAC;EACvD,MAAM,CAACU,OAAO,EAAEC,UAAU,CAAC,GAAGX,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACY,YAAY,EAAEC,eAAe,CAAC,GAAGb,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACc,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGf,QAAQ,CAAC,CAAC,CAAC;EAE3D,MAAMgB,KAAK,GAAGjB,OAAO,CACnB,MACE,IAAIE,qBAAqB,CAAC;IACxBgB,6BAA6B,EAAEF,mBAAmB;IAClDG,eAAe,EAAEP,UAAU;IAC3BQ,oBAAoB,EAAEN,eAAe;IACrCO,aAAa,EAAEX;EACjB,CAAC,CAAC,EACJ,EACF,CAAC;EAEDX,SAAS,CAAC,MAAM;IACd,MAAMuB,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,MAAML,KAAK,CAACK,SAAS,CACnBlB,eAAe,EACfC,iBAAiB,EACjBC,QAAQ,EACRC,qBACF,CAAC;IACH,CAAC;IAED,IAAI,CAACC,WAAW,EAAE;MAChBc,SAAS,CAAC,CAAC;IACb;IACA;EACF,CAAC,EAAE,CACDL,KAAK;EACL;EACAM,IAAI,CAACC,SAAS,CAACpB,eAAe,CAAC,EAC/BE,QAAQ,EACRC,qBAAqB;EACrB;EACAgB,IAAI,CAACC,SAAS,CAACnB,iBAAiB,CAAC,EACjCG,WAAW,CACZ,CAAC;EAEF,OAAO;IACLC,KAAK;IACLE,OAAO;IACPE,YAAY;IACZY,OAAO,EAAER,KAAK,CAACQ,OAAO;IACtBV;EACF,CAAC;AACH,CAAC","ignoreList":[]}

package/lib/module/hooks/general/useExecutorchModule.js

@@ -1,41 +1,13 @@
 "use strict";
 
-import { useState } from 'react';
-import { _ETModule } from '../../native/RnExecutorchModules';
+import { ExecutorchModule } from '../../modules/general/ExecutorchModule';
 import { useModule } from '../useModule';
-import { getError } from '../../Error';
 export const useExecutorchModule = ({
-  modelSource
-}) => {
-  const [module] = useState(() => new _ETModule());
-  const {
-    error,
-    isReady,
-    isGenerating,
-    downloadProgress,
-    forwardETInput: forward
-  } = useModule({
-    modelSource,
-    module
-  });
-  const loadMethod = async methodName => {
-    try {
-      await module.loadMethod(methodName);
-    } catch (e) {
-      throw new Error(getError(e));
-    }
-  };
-  const loadForward = async () => {
-    await loadMethod('forward');
-  };
-  return {
-    error,
-    isReady,
-    isGenerating,
-    downloadProgress,
-    forward,
-    loadMethod,
-    loadForward
-  };
-};
+  modelSource,
+  preventLoad = false
+}) => useModule({
+  module: ExecutorchModule,
+  loadArgs: [modelSource],
+  preventLoad
+});
 //# sourceMappingURL=useExecutorchModule.js.map

package/lib/module/hooks/general/useExecutorchModule.js.map

@@ -1 +1 @@
-{"version":3,"names":["useState","_ETModule","useModule","getError","useExecutorchModule","modelSource","module","error","isReady","isGenerating","downloadProgress","forwardETInput","forward","loadMethod","methodName","e","Error","loadForward"],"sourceRoot":"../../../../src","sources":["hooks/general/useExecutorchModule.ts"],"mappings":";;AAAA,SAASA,QAAQ,QAAQ,OAAO;AAChC,SAASC,SAAS,QAAQ,kCAAkC;AAC5D,SAASC,SAAS,QAAQ,cAAc;AAExC,SAASC,QAAQ,QAAQ,aAAa;AAMtC,OAAO,MAAMC,mBAAmB,GAAGA,CAAC;EAClCC;AACK,CAAC,KAWH;EACH,MAAM,CAACC,MAAM,CAAC,GAAGN,QAAQ,CAAC,MAAM,IAAIC,SAAS,CAAC,CAAC,CAAC;EAChD,MAAM;IACJM,KAAK;IACLC,OAAO;IACPC,YAAY;IACZC,gBAAgB;IAChBC,cAAc,EAAEC;EAClB,CAAC,GAAGV,SAAS,CAAC;IACZG,WAAW;IACXC;EACF,CAAC,CAAC;EAEF,MAAMO,UAAU,GAAG,MAAOC,UAAkB,IAAK;IAC/C,IAAI;MACF,MAAMR,MAAM,CAACO,UAAU,CAACC,UAAU,CAAC;IACrC,CAAC,CAAC,OAAOC,CAAC,EAAE;MACV,MAAM,IAAIC,KAAK,CAACb,QAAQ,CAACY,CAAC,CAAC,CAAC;IAC9B;EACF,CAAC;EAED,MAAME,WAAW,GAAG,MAAAA,CAAA,KAAY;IAC9B,MAAMJ,UAAU,CAAC,SAAS,CAAC;EAC7B,CAAC;EAED,OAAO;IACLN,KAAK;IACLC,OAAO;IACPC,YAAY;IACZC,gBAAgB;IAChBE,OAAO;IACPC,UAAU;IACVI;EACF,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["ExecutorchModule","useModule","useExecutorchModule","modelSource","preventLoad","module","loadArgs"],"sourceRoot":"../../../../src","sources":["hooks/general/useExecutorchModule.ts"],"mappings":";;AAAA,SAASA,gBAAgB,QAAQ,wCAAwC;AAEzE,SAASC,SAAS,QAAQ,cAAc;AAOxC,OAAO,MAAMC,mBAAmB,GAAGA,CAAC;EAClCC,WAAW;EACXC,WAAW,GAAG;AACT,CAAC,KACNH,SAAS,CAAC;EACRI,MAAM,EAAEL,gBAAgB;EACxBM,QAAQ,EAAE,CAACH,WAAW,CAAC;EACvBC;AACF,CAAC,CAAC","ignoreList":[]}

package/lib/module/hooks/natural_language_processing/useLLM.js

@@ -1,84 +1,75 @@
 "use strict";
 
-import { useCallback, useEffect, useRef, useState } from 'react';
-import { LLM } from '../../native/RnExecutorchModules';
-import { fetchResource } from '../../utils/fetchResource';
-import { DEFAULT_CONTEXT_WINDOW_LENGTH, DEFAULT_MESSAGE_HISTORY, DEFAULT_SYSTEM_PROMPT, EOT_TOKEN } from '../../constants/llamaDefaults';
-const interrupt = () => {
-  LLM.interrupt();
-};
+import { useCallback, useEffect, useMemo, useState } from 'react';
+import { LLMController } from '../../controllers/LLMController';
+
+/*
+Hook version of LLMController
+*/
 export const useLLM = ({
   modelSource,
   tokenizerSource,
-  systemPrompt = DEFAULT_SYSTEM_PROMPT,
-  messageHistory = DEFAULT_MESSAGE_HISTORY,
-  contextWindowLength = DEFAULT_CONTEXT_WINDOW_LENGTH
+  tokenizerConfigSource,
+  preventLoad = false
 }) => {
-  const [error, setError] = useState(null);
+  const [response, setResponse] = useState('');
+  const [messageHistory, setMessageHistory] = useState([]);
   const [isReady, setIsReady] = useState(false);
   const [isGenerating, setIsGenerating] = useState(false);
-  const [response, setResponse] = useState('');
   const [downloadProgress, setDownloadProgress] = useState(0);
-  const tokenGeneratedListener = useRef(null);
+  const [error, setError] = useState(null);
+  const model = useMemo(() => new LLMController({
+    responseCallback: setResponse,
+    messageHistoryCallback: setMessageHistory,
+    isReadyCallback: setIsReady,
+    isGeneratingCallback: setIsGenerating,
+    onDownloadProgressCallback: setDownloadProgress
+  }), []);
   useEffect(() => {
-    const loadModel = async () => {
+    setDownloadProgress(0);
+    setError(null);
+    if (!preventLoad) {
       try {
-        setIsReady(false);
-        const tokenizerFileUri = await fetchResource(tokenizerSource);
-        const modelFileUri = await fetchResource(modelSource, setDownloadProgress);
-        await LLM.loadLLM(modelFileUri, tokenizerFileUri, systemPrompt, messageHistory, contextWindowLength);
-        setIsReady(true);
-        tokenGeneratedListener.current = LLM.onToken(data => {
-          if (!data) {
-            return;
-          }
-          if (data !== EOT_TOKEN) {
-            setResponse(prevResponse => prevResponse + data);
-          } else {
-            setIsGenerating(false);
-          }
-        });
-      } catch (err) {
-        const message = err.message;
-        setIsReady(false);
-        setError(message);
-      } finally {
-        setDownloadProgress(0);
+        (async () => {
+          await model.load({
+            modelSource,
+            tokenizerSource,
+            tokenizerConfigSource
+          });
+        })();
+      } catch (e) {
+        setError(e);
       }
-    };
-    loadModel();
+    }
     return () => {
-      tokenGeneratedListener.current?.remove();
-      tokenGeneratedListener.current = null;
-      LLM.deleteModule();
+      model.delete();
     };
-  }, [modelSource, tokenizerSource, systemPrompt, messageHistory, contextWindowLength]);
-  const generate = useCallback(async input => {
-    if (!isReady) {
-      throw new Error('Model is still loading');
-    }
-    if (error) {
-      throw new Error(error);
-    }
-    try {
-      setResponse('');
-      setIsGenerating(true);
-      await LLM.runInference(input);
-    } catch (err) {
-      setIsGenerating(false);
-      throw new Error(err.message);
-    }
-  }, [isReady, error]);
+  }, [modelSource, tokenizerSource, tokenizerConfigSource, preventLoad, model]);
+
+  // memoization of returned functions
+  const configure = useCallback(({
+    chatConfig,
+    toolsConfig
+  }) => model.configure({
+    chatConfig,
+    toolsConfig
+  }), [model]);
+  const generate = useCallback((messages, tools) => model.generate(messages, tools), [model]);
+  const sendMessage = useCallback(message => model.sendMessage(message), [model]);
+  const deleteMessage = useCallback(index => model.deleteMessage(index), [model]);
+  const interrupt = useCallback(() => model.interrupt(), [model]);
   return {
-    generate,
-    error,
+    messageHistory,
+    response,
     isReady,
     isGenerating,
-    isModelReady: isReady,
-    isModelGenerating: isGenerating,
-    response,
     downloadProgress,
-    interrupt
+    error,
+    configure: configure,
+    generate: generate,
+    sendMessage: sendMessage,
+    deleteMessage: deleteMessage,
+    interrupt: interrupt
   };
 };
 //# sourceMappingURL=useLLM.js.map

package/lib/module/hooks/natural_language_processing/useLLM.js.map

@@ -1 +1 @@
-{"version":3,"names":["useCallback","useEffect","useRef","useState","LLM","fetchResource","DEFAULT_CONTEXT_WINDOW_LENGTH","DEFAULT_MESSAGE_HISTORY","DEFAULT_SYSTEM_PROMPT","EOT_TOKEN","interrupt","useLLM","modelSource","tokenizerSource","systemPrompt","messageHistory","contextWindowLength","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","response","setResponse","downloadProgress","setDownloadProgress","tokenGeneratedListener","loadModel","tokenizerFileUri","modelFileUri","loadLLM","current","onToken","data","prevResponse","err","message","remove","deleteModule","generate","input","Error","runInference","isModelReady","isModelGenerating"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useLLM.ts"],"mappings":";;AAAA,SAASA,WAAW,EAAEC,SAAS,EAAEC,MAAM,EAAEC,QAAQ,QAAQ,OAAO;AAEhE,SAASC,GAAG,QAAQ,kCAAkC;AACtD,SAASC,aAAa,QAAQ,2BAA2B;AAEzD,SACEC,6BAA6B,EAC7BC,uBAAuB,EACvBC,qBAAqB,EACrBC,SAAS,QACJ,+BAA+B;AAEtC,MAAMC,SAAS,GAAGA,CAAA,KAAM;EACtBN,GAAG,CAACM,SAAS,CAAC,CAAC;AACjB,CAAC;AAED,OAAO,MAAMC,MAAM,GAAGA,CAAC;EACrBC,WAAW;EACXC,eAAe;EACfC,YAAY,GAAGN,qBAAqB;EACpCO,cAAc,GAAGR,uBAAuB;EACxCS,mBAAmB,GAAGV;AAOxB,CAAC,KAAY;EACX,MAAM,CAACW,KAAK,EAAEC,QAAQ,CAAC,GAAGf,QAAQ,CAAgB,IAAI,CAAC;EACvD,MAAM,CAACgB,OAAO,EAAEC,UAAU,CAAC,GAAGjB,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACkB,YAAY,EAAEC,eAAe,CAAC,GAAGnB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACoB,QAAQ,EAAEC,WAAW,CAAC,GAAGrB,QAAQ,CAAC,EAAE,CAAC;EAC5C,MAAM,CAACsB,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGvB,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAMwB,sBAAsB,GAAGzB,MAAM,CAA2B,IAAI,CAAC;EAErED,SAAS,CAAC,MAAM;IACd,MAAM2B,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,IAAI;QACFR,UAAU,CAAC,KAAK,CAAC;QAEjB,MAAMS,gBAAgB,GAAG,MAAMxB,aAAa,CAACQ,eAAe,CAAC;QAC7D,MAAMiB,YAAY,GAAG,MAAMzB,aAAa,CACtCO,WAAW,EACXc,mBACF,CAAC;QAED,MAAMtB,GAAG,CAAC2B,OAAO,CACfD,YAAY,EACZD,gBAAgB,EAChBf,YAAY,EACZC,cAAc,EACdC,mBACF,CAAC;QAEDI,UAAU,CAAC,IAAI,CAAC;QAEhBO,sBAAsB,CAACK,OAAO,GAAG5B,GAAG,CAAC6B,OAAO,CACzCC,IAAwB,IAAK;UAC5B,IAAI,CAACA,IAAI,EAAE;YACT;UACF;UACA,IAAIA,IAAI,KAAKzB,SAAS,EAAE;YACtBe,WAAW,CAAEW,YAAY,IAAKA,YAAY,GAAGD,IAAI,CAAC;UACpD,CAAC,MAAM;YACLZ,eAAe,CAAC,KAAK,CAAC;UACxB;QACF,CACF,CAAC;MACH,CAAC,CAAC,OAAOc,GAAG,EAAE;QACZ,MAAMC,OAAO,GAAID,GAAG,CAAWC,OAAO;QACtCjB,UAAU,CAAC,KAAK,CAAC;QACjBF,QAAQ,CAACmB,OAAO,CAAC;MACnB,CAAC,SAAS;QACRX,mBAAmB,CAAC,CAAC,CAAC;MACxB;IACF,CAAC;IAEDE,SAAS,CAAC,CAAC;IAEX,OAAO,MAAM;MACXD,sBAAsB,CAACK,OAAO,EAAEM,MAAM,CAAC,CAAC;MACxCX,sBAAsB,CAACK,OAAO,GAAG,IAAI;MACrC5B,GAAG,CAACmC,YAAY,CAAC,CAAC;IACpB,CAAC;EACH,CAAC,EAAE,CACD3B,WAAW,EACXC,eAAe,EACfC,YAAY,EACZC,cAAc,EACdC,mBAAmB,CACpB,CAAC;EAEF,MAAMwB,QAAQ,GAAGxC,WAAW,CAC1B,MAAOyC,KAAa,IAAoB;IACtC,IAAI,CAACtB,OAAO,EAAE;MACZ,MAAM,IAAIuB,KAAK,CAAC,wBAAwB,CAAC;IAC3C;IACA,IAAIzB,KAAK,EAAE;MACT,MAAM,IAAIyB,KAAK,CAACzB,KAAK,CAAC;IACxB;IAEA,IAAI;MACFO,WAAW,CAAC,EAAE,CAAC;MACfF,eAAe,CAAC,IAAI,CAAC;MACrB,MAAMlB,GAAG,CAACuC,YAAY,CAACF,KAAK,CAAC;IAC/B,CAAC,CAAC,OAAOL,GAAG,EAAE;MACZd,eAAe,CAAC,KAAK,CAAC;MACtB,MAAM,IAAIoB,KAAK,CAAEN,GAAG,CAAWC,OAAO,CAAC;IACzC;EACF,CAAC,EACD,CAAClB,OAAO,EAAEF,KAAK,CACjB,CAAC;EAED,OAAO;IACLuB,QAAQ;IACRvB,KAAK;IACLE,OAAO;IACPE,YAAY;IACZuB,YAAY,EAAEzB,OAAO;IACrB0B,iBAAiB,EAAExB,YAAY;IAC/BE,QAAQ;IACRE,gBAAgB;IAChBf;EACF,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["useCallback","useEffect","useMemo","useState","LLMController","useLLM","modelSource","tokenizerSource","tokenizerConfigSource","preventLoad","response","setResponse","messageHistory","setMessageHistory","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","error","setError","model","responseCallback","messageHistoryCallback","isReadyCallback","isGeneratingCallback","onDownloadProgressCallback","load","e","delete","configure","chatConfig","toolsConfig","generate","messages","tools","sendMessage","message","deleteMessage","index","interrupt"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useLLM.ts"],"mappings":";;AAAA,SAASA,WAAW,EAAEC,SAAS,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AASjE,SAASC,aAAa,QAAQ,iCAAiC;;AAE/D;AACA;AACA;AACA,OAAO,MAAMC,MAAM,GAAGA,CAAC;EACrBC,WAAW;EACXC,eAAe;EACfC,qBAAqB;EACrBC,WAAW,GAAG;AAMhB,CAAC,KAAc;EACb,MAAM,CAACC,QAAQ,EAAEC,WAAW,CAAC,GAAGR,QAAQ,CAAC,EAAE,CAAC;EAC5C,MAAM,CAACS,cAAc,EAAEC,iBAAiB,CAAC,GAAGV,QAAQ,CAAY,EAAE,CAAC;EACnE,MAAM,CAACW,OAAO,EAAEC,UAAU,CAAC,GAAGZ,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACa,YAAY,EAAEC,eAAe,CAAC,GAAGd,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACe,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGhB,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAM,CAACiB,KAAK,EAAEC,QAAQ,CAAC,GAAGlB,QAAQ,CAAM,IAAI,CAAC;EAE7C,MAAMmB,KAAK,GAAGpB,OAAO,CACnB,MACE,IAAIE,aAAa,CAAC;IAChBmB,gBAAgB,EAAEZ,WAAW;IAC7Ba,sBAAsB,EAAEX,iBAAiB;IACzCY,eAAe,EAAEV,UAAU;IAC3BW,oBAAoB,EAAET,eAAe;IACrCU,0BAA0B,EAAER;EAC9B,CAAC,CAAC,EACJ,EACF,CAAC;EAEDlB,SAAS,CAAC,MAAM;IACdkB,mBAAmB,CAAC,CAAC,CAAC;IACtBE,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAI,CAACZ,WAAW,EAAE;MAChB,IAAI;QACF,CAAC,YAAY;UACX,MAAMa,KAAK,CAACM,IAAI,CAAC;YACftB,WAAW;YACXC,eAAe;YACfC;UACF,CAAC,CAAC;QACJ,CAAC,EAAE,CAAC;MACN,CAAC,CAAC,OAAOqB,CAAC,EAAE;QACVR,QAAQ,CAACQ,CAAC,CAAC;MACb;IACF;IAEA,OAAO,MAAM;MACXP,KAAK,CAACQ,MAAM,CAAC,CAAC;IAChB,CAAC;EACH,CAAC,EAAE,CAACxB,WAAW,EAAEC,eAAe,EAAEC,qBAAqB,EAAEC,WAAW,EAAEa,KAAK,CAAC,CAAC;;EAE7E;EACA,MAAMS,SAAS,GAAG/B,WAAW,CAC3B,CAAC;IACCgC,UAAU;IACVC;EAIF,CAAC,KAAKX,KAAK,CAACS,SAAS,CAAC;IAAEC,UAAU;IAAEC;EAAY,CAAC,CAAC,EAClD,CAACX,KAAK,CACR,CAAC;EAED,MAAMY,QAAQ,GAAGlC,WAAW,CAC1B,CAACmC,QAAmB,EAAEC,KAAiB,KAAKd,KAAK,CAACY,QAAQ,CAACC,QAAQ,EAAEC,KAAK,CAAC,EAC3E,CAACd,KAAK,CACR,CAAC;EAED,MAAMe,WAAW,GAAGrC,WAAW,CAC5BsC,OAAe,IAAKhB,KAAK,CAACe,WAAW,CAACC,OAAO,CAAC,EAC/C,CAAChB,KAAK,CACR,CAAC;EAED,MAAMiB,aAAa,GAAGvC,WAAW,CAC9BwC,KAAa,IAAKlB,KAAK,CAACiB,aAAa,CAACC,KAAK,CAAC,EAC7C,CAAClB,KAAK,CACR,CAAC;EACD,MAAMmB,SAAS,GAAGzC,WAAW,CAAC,MAAMsB,KAAK,CAACmB,SAAS,CAAC,CAAC,EAAE,CAACnB,KAAK,CAAC,CAAC;EAE/D,OAAO;IACLV,cAAc;IACdF,QAAQ;IACRI,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBE,KAAK;IACLW,SAAS,EAAEA,SAAS;IACpBG,QAAQ,EAAEA,QAAQ;IAClBG,WAAW,EAAEA,WAAW;IACxBE,aAAa,EAAEA,aAAa;IAC5BE,SAAS,EAAEA;EACb,CAAC;AACH,CAAC","ignoreList":[]}

package/lib/module/hooks/natural_language_processing/useSpeechToText.js

@@ -1,6 +1,6 @@
 "use strict";
 
-import { useEffect, useState } from 'react';
+import { useEffect, useMemo, useState } from 'react';
 import { SpeechToTextController } from '../../controllers/SpeechToTextController';
 export const useSpeechToText = ({
   modelName,
@@ -9,29 +9,32 @@ export const useSpeechToText = ({
   tokenizerSource,
   overlapSeconds,
   windowSize,
-  streamingConfig
+  streamingConfig,
+  preventLoad = false
 }) => {
   const [sequence, setSequence] = useState('');
   const [isReady, setIsReady] = useState(false);
   const [downloadProgress, setDownloadProgress] = useState(0);
   const [isGenerating, setIsGenerating] = useState(false);
   const [error, setError] = useState();
-  const [model, _] = useState(() => new SpeechToTextController({
+  const model = useMemo(() => new SpeechToTextController({
     transcribeCallback: setSequence,
     isReadyCallback: setIsReady,
     isGeneratingCallback: setIsGenerating,
     onErrorCallback: setError,
-    modelDownloadProgessCallback: setDownloadProgress,
-    overlapSeconds: overlapSeconds,
-    windowSize: windowSize,
-    streamingConfig: streamingConfig
-  }));
+    modelDownloadProgressCallback: setDownloadProgress
+  }), []);
+  useEffect(() => {
+    model.configureStreaming(overlapSeconds, windowSize, streamingConfig);
+  }, [model, overlapSeconds, windowSize, streamingConfig]);
   useEffect(() => {
     const loadModel = async () => {
       await model.loadModel(modelName, encoderSource, decoderSource, tokenizerSource);
     };
-    loadModel();
-  }, [model, modelName, encoderSource, decoderSource, tokenizerSource]);
+    if (!preventLoad) {
+      loadModel();
+    }
+  }, [model, modelName, encoderSource, decoderSource, tokenizerSource, preventLoad]);
   return {
     isReady,
     isGenerating,
@@ -39,7 +42,8 @@ export const useSpeechToText = ({
     configureStreaming: model.configureStreaming,
     sequence,
     error,
-    transcribe: waveform => model.transcribe(waveform)
+    transcribe: (waveform, audioLanguage) => model.transcribe(waveform, audioLanguage),
+    streamingTranscribe: (streamAction, waveform, audioLanguage) => model.streamingTranscribe(streamAction, waveform, audioLanguage)
   };
 };
 //# sourceMappingURL=useSpeechToText.js.map

package/lib/module/hooks/natural_language_processing/useSpeechToText.js.map

@@ -1 +1 @@
-{"version":3,"names":["useEffect","useState","SpeechToTextController","useSpeechToText","modelName","encoderSource","decoderSource","tokenizerSource","overlapSeconds","windowSize","streamingConfig","sequence","setSequence","isReady","setIsReady","downloadProgress","setDownloadProgress","isGenerating","setIsGenerating","error","setError","model","_","transcribeCallback","isReadyCallback","isGeneratingCallback","onErrorCallback","modelDownloadProgessCallback","loadModel","configureStreaming","transcribe","waveform"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useSpeechToText.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,QAAQ,QAAQ,OAAO;AAC3C,SAASC,sBAAsB,QAAQ,0CAA0C;AAejF,OAAO,MAAMC,eAAe,GAAGA,CAAC;EAC9BC,SAAS;EACTC,aAAa;EACbC,aAAa;EACbC,eAAe;EACfC,cAAc;EACdC,UAAU;EACVC;AAeF,CAAC,KAAyB;EACxB,MAAM,CAACC,QAAQ,EAAEC,WAAW,CAAC,GAAGX,QAAQ,CAAS,EAAE,CAAC;EACpD,MAAM,CAACY,OAAO,EAAEC,UAAU,CAAC,GAAGb,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACc,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGf,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAM,CAACgB,YAAY,EAAEC,eAAe,CAAC,GAAGjB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACkB,KAAK,EAAEC,QAAQ,CAAC,GAAGnB,QAAQ,CAAoB,CAAC;EAEvD,MAAM,CAACoB,KAAK,EAAEC,CAAC,CAAC,GAAGrB,QAAQ,CACzB,MACE,IAAIC,sBAAsB,CAAC;IACzBqB,kBAAkB,EAAEX,WAAW;IAC/BY,eAAe,EAAEV,UAAU;IAC3BW,oBAAoB,EAAEP,eAAe;IACrCQ,eAAe,EAAEN,QAAQ;IACzBO,4BAA4B,EAAEX,mBAAmB;IACjDR,cAAc,EAAEA,cAAc;IAC9BC,UAAU,EAAEA,UAAU;IACtBC,eAAe,EAAEA;EACnB,CAAC,CACL,CAAC;EAEDV,SAAS,CAAC,MAAM;IACd,MAAM4B,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,MAAMP,KAAK,CAACO,SAAS,CACnBxB,SAAS,EACTC,aAAa,EACbC,aAAa,EACbC,eACF,CAAC;IACH,CAAC;IACDqB,SAAS,CAAC,CAAC;EACb,CAAC,EAAE,CAACP,KAAK,EAAEjB,SAAS,EAAEC,aAAa,EAAEC,aAAa,EAAEC,eAAe,CAAC,CAAC;EAErE,OAAO;IACLM,OAAO;IACPI,YAAY;IACZF,gBAAgB;IAChBc,kBAAkB,EAAER,KAAK,CAACQ,kBAAkB;IAC5ClB,QAAQ;IACRQ,KAAK;IACLW,UAAU,EAAGC,QAAkB,IAAKV,KAAK,CAACS,UAAU,CAACC,QAAQ;EAC/D,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["useEffect","useMemo","useState","SpeechToTextController","useSpeechToText","modelName","encoderSource","decoderSource","tokenizerSource","overlapSeconds","windowSize","streamingConfig","preventLoad","sequence","setSequence","isReady","setIsReady","downloadProgress","setDownloadProgress","isGenerating","setIsGenerating","error","setError","model","transcribeCallback","isReadyCallback","isGeneratingCallback","onErrorCallback","modelDownloadProgressCallback","configureStreaming","loadModel","transcribe","waveform","audioLanguage","streamingTranscribe","streamAction"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useSpeechToText.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AACpD,SAASC,sBAAsB,QAAQ,0CAA0C;AAuBjF,OAAO,MAAMC,eAAe,GAAGA,CAAC;EAC9BC,SAAS;EACTC,aAAa;EACbC,aAAa;EACbC,eAAe;EACfC,cAAc;EACdC,UAAU;EACVC,eAAe;EACfC,WAAW,GAAG;AAgBhB,CAAC,KAAyB;EACxB,MAAM,CAACC,QAAQ,EAAEC,WAAW,CAAC,GAAGZ,QAAQ,CAAS,EAAE,CAAC;EACpD,MAAM,CAACa,OAAO,EAAEC,UAAU,CAAC,GAAGd,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACe,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGhB,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAM,CAACiB,YAAY,EAAEC,eAAe,CAAC,GAAGlB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACmB,KAAK,EAAEC,QAAQ,CAAC,GAAGpB,QAAQ,CAAoB,CAAC;EAEvD,MAAMqB,KAAK,GAAGtB,OAAO,CACnB,MACE,IAAIE,sBAAsB,CAAC;IACzBqB,kBAAkB,EAAEV,WAAW;IAC/BW,eAAe,EAAET,UAAU;IAC3BU,oBAAoB,EAAEN,eAAe;IACrCO,eAAe,EAAEL,QAAQ;IACzBM,6BAA6B,EAAEV;EACjC,CAAC,CAAC,EACJ,EACF,CAAC;EAEDlB,SAAS,CAAC,MAAM;IACduB,KAAK,CAACM,kBAAkB,CAACpB,cAAc,EAAEC,UAAU,EAAEC,eAAe,CAAC;EACvE,CAAC,EAAE,CAACY,KAAK,EAAEd,cAAc,EAAEC,UAAU,EAAEC,eAAe,CAAC,CAAC;EAExDX,SAAS,CAAC,MAAM;IACd,MAAM8B,SAAS,GAAG,MAAAA,CAAA,KAAY;MAC5B,MAAMP,KAAK,CAACO,SAAS,CACnBzB,SAAS,EACTC,aAAa,EACbC,aAAa,EACbC,eACF,CAAC;IACH,CAAC;IACD,IAAI,CAACI,WAAW,EAAE;MAChBkB,SAAS,CAAC,CAAC;IACb;EACF,CAAC,EAAE,CACDP,KAAK,EACLlB,SAAS,EACTC,aAAa,EACbC,aAAa,EACbC,eAAe,EACfI,WAAW,CACZ,CAAC;EAEF,OAAO;IACLG,OAAO;IACPI,YAAY;IACZF,gBAAgB;IAChBY,kBAAkB,EAAEN,KAAK,CAACM,kBAAkB;IAC5ChB,QAAQ;IACRQ,KAAK;IACLU,UAAU,EAAEA,CAACC,QAAkB,EAAEC,aAAoC,KACnEV,KAAK,CAACQ,UAAU,CAACC,QAAQ,EAAEC,aAAa,CAAC;IAC3CC,mBAAmB,EAAEA,CACnBC,YAA8B,EAC9BH,QAAmB,EACnBC,aAAoC,KACjCV,KAAK,CAACW,mBAAmB,CAACC,YAAY,EAAEH,QAAQ,EAAEC,aAAa;EACtE,CAAC;AACH,CAAC","ignoreList":[]}

package/lib/module/hooks/natural_language_processing/useTextEmbeddings.js

@@ -0,0 +1,14 @@
+"use strict";
+
+import { TextEmbeddingsModule } from '../../modules/natural_language_processing/TextEmbeddingsModule';
+import { useModule } from '../useModule';
+export const useTextEmbeddings = ({
+  modelSource,
+  tokenizerSource,
+  preventLoad = false
+}) => useModule({
+  module: TextEmbeddingsModule,
+  loadArgs: [modelSource, tokenizerSource],
+  preventLoad
+});
+//# sourceMappingURL=useTextEmbeddings.js.map

package/lib/module/hooks/natural_language_processing/useTextEmbeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["TextEmbeddingsModule","useModule","useTextEmbeddings","modelSource","tokenizerSource","preventLoad","module","loadArgs"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useTextEmbeddings.ts"],"mappings":";;AAAA,SAASA,oBAAoB,QAAQ,gEAAgE;AAErG,SAASC,SAAS,QAAQ,cAAc;AAExC,OAAO,MAAMC,iBAAiB,GAAGA,CAAC;EAChCC,WAAW;EACXC,eAAe;EACfC,WAAW,GAAG;AAKhB,CAAC,KACCJ,SAAS,CAAC;EACRK,MAAM,EAAEN,oBAAoB;EAC5BO,QAAQ,EAAE,CAACJ,WAAW,EAAEC,eAAe,CAAC;EACxCC;AACF,CAAC,CAAC","ignoreList":[]}

package/lib/module/hooks/natural_language_processing/useTokenizer.js

@@ -0,0 +1,54 @@
+"use strict";
+
+import { useEffect, useState } from 'react';
+import { TokenizerModule } from '../../modules/natural_language_processing/TokenizerModule';
+import { ETError, getError } from '../../Error';
+export const useTokenizer = ({
+  tokenizerSource,
+  preventLoad = false
+}) => {
+  const [error, setError] = useState(null);
+  const [isReady, setIsReady] = useState(false);
+  const [isGenerating, setIsGenerating] = useState(false);
+  const [downloadProgress, setDownloadProgress] = useState(0);
+  useEffect(() => {
+    const loadModule = async () => {
+      try {
+        setIsReady(false);
+        TokenizerModule.onDownloadProgress(setDownloadProgress);
+        await TokenizerModule.load(tokenizerSource);
+        setIsReady(true);
+      } catch (err) {
+        setError(err.message);
+      }
+    };
+    if (!preventLoad) {
+      loadModule();
+    }
+  }, [tokenizerSource, preventLoad]);
+  const stateWrapper = fn => {
+    const boundFn = fn.bind(TokenizerModule);
+    return async (...args) => {
+      if (!isReady) throw new Error(getError(ETError.ModuleNotLoaded));
+      if (isGenerating) throw new Error(getError(ETError.ModelGenerating));
+      setIsGenerating(true);
+      try {
+        return await boundFn(...args);
+      } finally {
+        setIsGenerating(false);
+      }
+    };
+  };
+  return {
+    error,
+    isReady,
+    isGenerating,
+    downloadProgress,
+    decode: stateWrapper(TokenizerModule.decode),
+    encode: stateWrapper(TokenizerModule.encode),
+    getVocabSize: stateWrapper(TokenizerModule.getVocabSize),
+    idToToken: stateWrapper(TokenizerModule.idToToken),
+    tokenToId: stateWrapper(TokenizerModule.tokenToId)
+  };
+};
+//# sourceMappingURL=useTokenizer.js.map

package/lib/module/hooks/natural_language_processing/useTokenizer.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["useEffect","useState","TokenizerModule","ETError","getError","useTokenizer","tokenizerSource","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","loadModule","onDownloadProgress","load","err","message","stateWrapper","fn","boundFn","bind","args","Error","ModuleNotLoaded","ModelGenerating","decode","encode","getVocabSize","idToToken","tokenToId"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useTokenizer.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,QAAQ,QAAQ,OAAO;AAC3C,SAASC,eAAe,QAAQ,2DAA2D;AAE3F,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAE/C,OAAO,MAAMC,YAAY,GAAGA,CAAC;EAC3BC,eAAe;EACfC,WAAW,GAAG;AAIhB,CAAC,KAAK;EACJ,MAAM,CAACC,KAAK,EAAEC,QAAQ,CAAC,GAAGR,QAAQ,CAAgB,IAAI,CAAC;EACvD,MAAM,CAACS,OAAO,EAAEC,UAAU,CAAC,GAAGV,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACW,YAAY,EAAEC,eAAe,CAAC,GAAGZ,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACa,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGd,QAAQ,CAAC,CAAC,CAAC;EAE3DD,SAAS,CAAC,MAAM;IACd,MAAMgB,UAAU,GAAG,MAAAA,CAAA,KAAY;MAC7B,IAAI;QACFL,UAAU,CAAC,KAAK,CAAC;QACjBT,eAAe,CAACe,kBAAkB,CAACF,mBAAmB,CAAC;QACvD,MAAMb,eAAe,CAACgB,IAAI,CAACZ,eAAe,CAAC;QAC3CK,UAAU,CAAC,IAAI,CAAC;MAClB,CAAC,CAAC,OAAOQ,GAAG,EAAE;QACZV,QAAQ,CAAEU,GAAG,CAAWC,OAAO,CAAC;MAClC;IACF,CAAC;IACD,IAAI,CAACb,WAAW,EAAE;MAChBS,UAAU,CAAC,CAAC;IACd;EACF,CAAC,EAAE,CAACV,eAAe,EAAEC,WAAW,CAAC,CAAC;EAElC,MAAMc,YAAY,GAAgDC,EAAK,IAAK;IAC1E,MAAMC,OAAO,GAAGD,EAAE,CAACE,IAAI,CAACtB,eAAe,CAAC;IAExC,OAAO,OAAO,GAAGuB,IAAmB,KAA6B;MAC/D,IAAI,CAACf,OAAO,EAAE,MAAM,IAAIgB,KAAK,CAACtB,QAAQ,CAACD,OAAO,CAACwB,eAAe,CAAC,CAAC;MAChE,IAAIf,YAAY,EAAE,MAAM,IAAIc,KAAK,CAACtB,QAAQ,CAACD,OAAO,CAACyB,eAAe,CAAC,CAAC;MAEpEf,eAAe,CAAC,IAAI,CAAC;MACrB,IAAI;QACF,OAAO,MAAMU,OAAO,CAAC,GAAGE,IAAI,CAAC;MAC/B,CAAC,SAAS;QACRZ,eAAe,CAAC,KAAK,CAAC;MACxB;IACF,CAAC;EACH,CAAC;EAED,OAAO;IACLL,KAAK;IACLE,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBe,MAAM,EAAER,YAAY,CAACnB,eAAe,CAAC2B,MAAM,CAAC;IAC5CC,MAAM,EAAET,YAAY,CAACnB,eAAe,CAAC4B,MAAM,CAAC;IAC5CC,YAAY,EAAEV,YAAY,CAACnB,eAAe,CAAC6B,YAAY,CAAC;IACxDC,SAAS,EAAEX,YAAY,CAACnB,eAAe,CAAC8B,SAAS,CAAC;IAClDC,SAAS,EAAEZ,YAAY,CAACnB,eAAe,CAAC+B,SAAS;EACnD,CAAC;AACH,CAAC","ignoreList":[]}