react-native-executorch 0.4.8 → 0.5.0

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

@@ -1,13 +1,13 @@
 "use strict";
 
-import { useModule } from '../useModule';
+import { useNonStaticModule } from '../useNonStaticModule';
 import { ObjectDetectionModule } from '../../modules/computer_vision/ObjectDetectionModule';
 export const useObjectDetection = ({
-  modelSource,
+  model,
   preventLoad = false
-}) => useModule({
+}) => useNonStaticModule({
   module: ObjectDetectionModule,
-  loadArgs: [modelSource],
-  preventLoad
+  model,
+  preventLoad: preventLoad
 });
 //# sourceMappingURL=useObjectDetection.js.map

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

@@ -1 +1 @@
-{"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":[]}
+{"version":3,"names":["useNonStaticModule","ObjectDetectionModule","useObjectDetection","model","preventLoad","module"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useObjectDetection.ts"],"mappings":";;AACA,SAASA,kBAAkB,QAAQ,uBAAuB;AAC1D,SAASC,qBAAqB,QAAQ,qDAAqD;AAO3F,OAAO,MAAMC,kBAAkB,GAAGA,CAAC;EAAEC,KAAK;EAAEC,WAAW,GAAG;AAAa,CAAC,KACtEJ,kBAAkB,CAAC;EACjBK,MAAM,EAAEJ,qBAAqB;EAC7BE,KAAK;EACLC,WAAW,EAAEA;AACf,CAAC,CAAC","ignoreList":[]}

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

@@ -1,13 +1,13 @@
 "use strict";
 
-import { useModule } from '../useModule';
+import { useNonStaticModule } from '../useNonStaticModule';
 import { StyleTransferModule } from '../../modules/computer_vision/StyleTransferModule';
 export const useStyleTransfer = ({
-  modelSource,
+  model,
   preventLoad = false
-}) => useModule({
+}) => useNonStaticModule({
   module: StyleTransferModule,
-  loadArgs: [modelSource],
-  preventLoad
+  model,
+  preventLoad: preventLoad
 });
 //# sourceMappingURL=useStyleTransfer.js.map

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

@@ -1 +1 @@
-{"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":[]}
+{"version":3,"names":["useNonStaticModule","StyleTransferModule","useStyleTransfer","model","preventLoad","module"],"sourceRoot":"../../../../src","sources":["hooks/computer_vision/useStyleTransfer.ts"],"mappings":";;AACA,SAASA,kBAAkB,QAAQ,uBAAuB;AAC1D,SAASC,mBAAmB,QAAQ,mDAAmD;AAOvF,OAAO,MAAMC,gBAAgB,GAAGA,CAAC;EAAEC,KAAK;EAAEC,WAAW,GAAG;AAAa,CAAC,KACpEJ,kBAAkB,CAAC;EACjBK,MAAM,EAAEJ,mBAAmB;EAC3BE,KAAK;EACLC,WAAW,EAAEA;AACf,CAAC,CAAC","ignoreList":[]}

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

@@ -3,9 +3,7 @@
 import { useEffect, useMemo, useState } from 'react';
 import { VerticalOCRController } from '../../controllers/VerticalOCRController';
 export const useVerticalOCR = ({
-  detectorSources,
-  recognizerSources,
-  language = 'en',
+  model,
   independentCharacters = false,
   preventLoad = false
 }) => {
@@ -13,30 +11,31 @@ export const useVerticalOCR = ({
   const [isReady, setIsReady] = useState(false);
   const [isGenerating, setIsGenerating] = useState(false);
   const [downloadProgress, setDownloadProgress] = useState(0);
-  const model = useMemo(() => new VerticalOCRController({
-    modelDownloadProgressCallback: setDownloadProgress,
+  const controllerInstance = useMemo(() => new VerticalOCRController({
     isReadyCallback: setIsReady,
     isGeneratingCallback: setIsGenerating,
     errorCallback: setError
   }), []);
   useEffect(() => {
-    const loadModel = async () => {
-      await model.loadModel(detectorSources, recognizerSources, language, independentCharacters);
+    if (preventLoad) return;
+    (async () => {
+      await controllerInstance.load({
+        detectorLarge: model.detectorLarge,
+        detectorNarrow: model.detectorNarrow
+      }, {
+        recognizerLarge: model.recognizerLarge,
+        recognizerSmall: model.recognizerSmall
+      }, model.language, independentCharacters, setDownloadProgress);
+    })();
+    return () => {
+      controllerInstance.delete();
     };
-    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), preventLoad]);
+  }, [controllerInstance, model.detectorLarge, model.detectorNarrow, model.recognizerLarge, model.recognizerSmall, model.language, independentCharacters, preventLoad]);
   return {
     error,
     isReady,
     isGenerating,
-    forward: model.forward,
+    forward: controllerInstance.forward,
     downloadProgress
   };
 };

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

@@ -1 +1 @@
-{"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":[]}
+{"version":3,"names":["useEffect","useMemo","useState","VerticalOCRController","useVerticalOCR","model","independentCharacters","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","controllerInstance","isReadyCallback","isGeneratingCallback","errorCallback","load","detectorLarge","detectorNarrow","recognizerLarge","recognizerSmall","language","delete","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,KAAK;EACLC,qBAAqB,GAAG,KAAK;EAC7BC,WAAW,GAAG;AAWhB,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,MAAMc,kBAAkB,GAAGf,OAAO,CAChC,MACE,IAAIE,qBAAqB,CAAC;IACxBc,eAAe,EAAEN,UAAU;IAC3BO,oBAAoB,EAAEL,eAAe;IACrCM,aAAa,EAAEV;EACjB,CAAC,CAAC,EACJ,EACF,CAAC;EAEDT,SAAS,CAAC,MAAM;IACd,IAAIO,WAAW,EAAE;IAEjB,CAAC,YAAY;MACX,MAAMS,kBAAkB,CAACI,IAAI,CAC3B;QACEC,aAAa,EAAEhB,KAAK,CAACgB,aAAa;QAClCC,cAAc,EAAEjB,KAAK,CAACiB;MACxB,CAAC,EACD;QACEC,eAAe,EAAElB,KAAK,CAACkB,eAAe;QACtCC,eAAe,EAAEnB,KAAK,CAACmB;MACzB,CAAC,EACDnB,KAAK,CAACoB,QAAQ,EACdnB,qBAAqB,EACrBS,mBACF,CAAC;IACH,CAAC,EAAE,CAAC;IAEJ,OAAO,MAAM;MACXC,kBAAkB,CAACU,MAAM,CAAC,CAAC;IAC7B,CAAC;EACH,CAAC,EAAE,CACDV,kBAAkB,EAClBX,KAAK,CAACgB,aAAa,EACnBhB,KAAK,CAACiB,cAAc,EACpBjB,KAAK,CAACkB,eAAe,EACrBlB,KAAK,CAACmB,eAAe,EACrBnB,KAAK,CAACoB,QAAQ,EACdnB,qBAAqB,EACrBC,WAAW,CACZ,CAAC;EAEF,OAAO;IACLC,KAAK;IACLE,OAAO;IACPE,YAAY;IACZe,OAAO,EAAEX,kBAAkB,CAACW,OAAO;IACnCb;EACF,CAAC;AACH,CAAC","ignoreList":[]}

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

@@ -1,13 +1,15 @@
 "use strict";
 
 import { ExecutorchModule } from '../../modules/general/ExecutorchModule';
-import { useModule } from '../useModule';
+import { useNonStaticModule } from '../useNonStaticModule';
 export const useExecutorchModule = ({
   modelSource,
   preventLoad = false
-}) => useModule({
+}) => useNonStaticModule({
   module: ExecutorchModule,
-  loadArgs: [modelSource],
+  model: {
+    modelSource
+  },
   preventLoad
 });
 //# sourceMappingURL=useExecutorchModule.js.map

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

@@ -1 +1 @@
-{"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":[]}
+{"version":3,"names":["ExecutorchModule","useNonStaticModule","useExecutorchModule","modelSource","preventLoad","module","model"],"sourceRoot":"../../../../src","sources":["hooks/general/useExecutorchModule.ts"],"mappings":";;AAAA,SAASA,gBAAgB,QAAQ,wCAAwC;AAEzE,SAASC,kBAAkB,QAAQ,uBAAuB;AAO1D,OAAO,MAAMC,mBAAmB,GAAGA,CAAC;EAClCC,WAAW;EACXC,WAAW,GAAG;AACT,CAAC,KACNH,kBAAkB,CAAC;EACjBI,MAAM,EAAEL,gBAAgB;EACxBM,KAAK,EAAE;IAAEH;EAAY,CAAC;EACtBC;AACF,CAAC,CAAC","ignoreList":[]}

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

@@ -7,9 +7,7 @@ import { LLMController } from '../../controllers/LLMController';
 Hook version of LLMModule
 */
 export const useLLM = ({
-  modelSource,
-  tokenizerSource,
-  tokenizerConfigSource,
+  model,
   preventLoad = false
 }) => {
   const [token, setToken] = useState('');
@@ -23,52 +21,51 @@ export const useLLM = ({
     setToken(newToken);
     setResponse(prevResponse => prevResponse + newToken);
   }, []);
-  const model = useMemo(() => new LLMController({
+  const controllerInstance = useMemo(() => new LLMController({
     tokenCallback: tokenCallback,
     messageHistoryCallback: setMessageHistory,
     isReadyCallback: setIsReady,
-    isGeneratingCallback: setIsGenerating,
-    onDownloadProgressCallback: setDownloadProgress
+    isGeneratingCallback: setIsGenerating
   }), [tokenCallback]);
   useEffect(() => {
     setDownloadProgress(0);
     setError(null);
-    if (!preventLoad) {
+    if (preventLoad) return;
+    (async () => {
       try {
-        (async () => {
-          await model.load({
-            modelSource,
-            tokenizerSource,
-            tokenizerConfigSource
-          });
-        })();
+        await controllerInstance.load({
+          modelSource: model.modelSource,
+          tokenizerSource: model.tokenizerSource,
+          tokenizerConfigSource: model.tokenizerConfigSource,
+          onDownloadProgressCallback: setDownloadProgress
+        });
       } catch (e) {
         setError(e);
       }
-    }
+    })();
     return () => {
-      model.delete();
+      controllerInstance.delete();
     };
-  }, [modelSource, tokenizerSource, tokenizerConfigSource, preventLoad, model]);
+  }, [controllerInstance, model.modelSource, model.tokenizerSource, model.tokenizerConfigSource, preventLoad]);
 
   // memoization of returned functions
   const configure = useCallback(({
     chatConfig,
     toolsConfig
-  }) => model.configure({
+  }) => controllerInstance.configure({
     chatConfig,
     toolsConfig
-  }), [model]);
+  }), [controllerInstance]);
   const generate = useCallback((messages, tools) => {
     setResponse('');
-    return model.generate(messages, tools);
-  }, [model]);
+    return controllerInstance.generate(messages, tools);
+  }, [controllerInstance]);
   const sendMessage = useCallback(message => {
     setResponse('');
-    return model.sendMessage(message);
-  }, [model]);
-  const deleteMessage = useCallback(index => model.deleteMessage(index), [model]);
-  const interrupt = useCallback(() => model.interrupt(), [model]);
+    return controllerInstance.sendMessage(message);
+  }, [controllerInstance]);
+  const deleteMessage = useCallback(index => controllerInstance.deleteMessage(index), [controllerInstance]);
+  const interrupt = useCallback(() => controllerInstance.interrupt(), [controllerInstance]);
   return {
     messageHistory,
     response,

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

@@ -1 +1 @@
-{"version":3,"names":["useCallback","useEffect","useMemo","useState","LLMController","useLLM","modelSource","tokenizerSource","tokenizerConfigSource","preventLoad","token","setToken","response","setResponse","messageHistory","setMessageHistory","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","error","setError","tokenCallback","newToken","prevResponse","model","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,KAAK,EAAEC,QAAQ,CAAC,GAAGR,QAAQ,CAAS,EAAE,CAAC;EAC9C,MAAM,CAACS,QAAQ,EAAEC,WAAW,CAAC,GAAGV,QAAQ,CAAS,EAAE,CAAC;EACpD,MAAM,CAACW,cAAc,EAAEC,iBAAiB,CAAC,GAAGZ,QAAQ,CAAY,EAAE,CAAC;EACnE,MAAM,CAACa,OAAO,EAAEC,UAAU,CAAC,GAAGd,QAAQ,CAAC,KAAK,CAAC;EAC7C,MAAM,CAACe,YAAY,EAAEC,eAAe,CAAC,GAAGhB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACiB,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGlB,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAM,CAACmB,KAAK,EAAEC,QAAQ,CAAC,GAAGpB,QAAQ,CAAM,IAAI,CAAC;EAE7C,MAAMqB,aAAa,GAAGxB,WAAW,CAAEyB,QAAgB,IAAK;IACtDd,QAAQ,CAACc,QAAQ,CAAC;IAClBZ,WAAW,CAAEa,YAAY,IAAKA,YAAY,GAAGD,QAAQ,CAAC;EACxD,CAAC,EAAE,EAAE,CAAC;EAEN,MAAME,KAAK,GAAGzB,OAAO,CACnB,MACE,IAAIE,aAAa,CAAC;IAChBoB,aAAa,EAAEA,aAAa;IAC5BI,sBAAsB,EAAEb,iBAAiB;IACzCc,eAAe,EAAEZ,UAAU;IAC3Ba,oBAAoB,EAAEX,eAAe;IACrCY,0BAA0B,EAAEV;EAC9B,CAAC,CAAC,EACJ,CAACG,aAAa,CAChB,CAAC;EAEDvB,SAAS,CAAC,MAAM;IACdoB,mBAAmB,CAAC,CAAC,CAAC;IACtBE,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAI,CAACd,WAAW,EAAE;MAChB,IAAI;QACF,CAAC,YAAY;UACX,MAAMkB,KAAK,CAACK,IAAI,CAAC;YACf1B,WAAW;YACXC,eAAe;YACfC;UACF,CAAC,CAAC;QACJ,CAAC,EAAE,CAAC;MACN,CAAC,CAAC,OAAOyB,CAAC,EAAE;QACVV,QAAQ,CAACU,CAAC,CAAC;MACb;IACF;IAEA,OAAO,MAAM;MACXN,KAAK,CAACO,MAAM,CAAC,CAAC;IAChB,CAAC;EACH,CAAC,EAAE,CAAC5B,WAAW,EAAEC,eAAe,EAAEC,qBAAqB,EAAEC,WAAW,EAAEkB,KAAK,CAAC,CAAC;;EAE7E;EACA,MAAMQ,SAAS,GAAGnC,WAAW,CAC3B,CAAC;IACCoC,UAAU;IACVC;EAIF,CAAC,KAAKV,KAAK,CAACQ,SAAS,CAAC;IAAEC,UAAU;IAAEC;EAAY,CAAC,CAAC,EAClD,CAACV,KAAK,CACR,CAAC;EAED,MAAMW,QAAQ,GAAGtC,WAAW,CAC1B,CAACuC,QAAmB,EAAEC,KAAiB,KAAK;IAC1C3B,WAAW,CAAC,EAAE,CAAC;IACf,OAAOc,KAAK,CAACW,QAAQ,CAACC,QAAQ,EAAEC,KAAK,CAAC;EACxC,CAAC,EACD,CAACb,KAAK,CACR,CAAC;EAED,MAAMc,WAAW,GAAGzC,WAAW,CAC5B0C,OAAe,IAAK;IACnB7B,WAAW,CAAC,EAAE,CAAC;IACf,OAAOc,KAAK,CAACc,WAAW,CAACC,OAAO,CAAC;EACnC,CAAC,EACD,CAACf,KAAK,CACR,CAAC;EAED,MAAMgB,aAAa,GAAG3C,WAAW,CAC9B4C,KAAa,IAAKjB,KAAK,CAACgB,aAAa,CAACC,KAAK,CAAC,EAC7C,CAACjB,KAAK,CACR,CAAC;EACD,MAAMkB,SAAS,GAAG7C,WAAW,CAAC,MAAM2B,KAAK,CAACkB,SAAS,CAAC,CAAC,EAAE,CAAClB,KAAK,CAAC,CAAC;EAE/D,OAAO;IACLb,cAAc;IACdF,QAAQ;IACRF,KAAK;IACLM,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBE,KAAK;IACLa,SAAS,EAAEA,SAAS;IACpBG,QAAQ,EAAEA,QAAQ;IAClBG,WAAW,EAAEA,WAAW;IACxBE,aAAa,EAAEA,aAAa;IAC5BE,SAAS,EAAEA;EACb,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["useCallback","useEffect","useMemo","useState","LLMController","useLLM","model","preventLoad","token","setToken","response","setResponse","messageHistory","setMessageHistory","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","error","setError","tokenCallback","newToken","prevResponse","controllerInstance","messageHistoryCallback","isReadyCallback","isGeneratingCallback","load","modelSource","tokenizerSource","tokenizerConfigSource","onDownloadProgressCallback","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,KAAK;EACLC,WAAW,GAAG;AAQhB,CAAC,KAAc;EACb,MAAM,CAACC,KAAK,EAAEC,QAAQ,CAAC,GAAGN,QAAQ,CAAS,EAAE,CAAC;EAC9C,MAAM,CAACO,QAAQ,EAAEC,WAAW,CAAC,GAAGR,QAAQ,CAAS,EAAE,CAAC;EACpD,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,aAAa,GAAGtB,WAAW,CAAEuB,QAAgB,IAAK;IACtDd,QAAQ,CAACc,QAAQ,CAAC;IAClBZ,WAAW,CAAEa,YAAY,IAAKA,YAAY,GAAGD,QAAQ,CAAC;EACxD,CAAC,EAAE,EAAE,CAAC;EAEN,MAAME,kBAAkB,GAAGvB,OAAO,CAChC,MACE,IAAIE,aAAa,CAAC;IAChBkB,aAAa,EAAEA,aAAa;IAC5BI,sBAAsB,EAAEb,iBAAiB;IACzCc,eAAe,EAAEZ,UAAU;IAC3Ba,oBAAoB,EAAEX;EACxB,CAAC,CAAC,EACJ,CAACK,aAAa,CAChB,CAAC;EAEDrB,SAAS,CAAC,MAAM;IACdkB,mBAAmB,CAAC,CAAC,CAAC;IACtBE,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAId,WAAW,EAAE;IAEjB,CAAC,YAAY;MACX,IAAI;QACF,MAAMkB,kBAAkB,CAACI,IAAI,CAAC;UAC5BC,WAAW,EAAExB,KAAK,CAACwB,WAAW;UAC9BC,eAAe,EAAEzB,KAAK,CAACyB,eAAe;UACtCC,qBAAqB,EAAE1B,KAAK,CAAC0B,qBAAqB;UAClDC,0BAA0B,EAAEd;QAC9B,CAAC,CAAC;MACJ,CAAC,CAAC,OAAOe,CAAC,EAAE;QACVb,QAAQ,CAACa,CAAC,CAAC;MACb;IACF,CAAC,EAAE,CAAC;IAEJ,OAAO,MAAM;MACXT,kBAAkB,CAACU,MAAM,CAAC,CAAC;IAC7B,CAAC;EACH,CAAC,EAAE,CACDV,kBAAkB,EAClBnB,KAAK,CAACwB,WAAW,EACjBxB,KAAK,CAACyB,eAAe,EACrBzB,KAAK,CAAC0B,qBAAqB,EAC3BzB,WAAW,CACZ,CAAC;;EAEF;EACA,MAAM6B,SAAS,GAAGpC,WAAW,CAC3B,CAAC;IACCqC,UAAU;IACVC;EAIF,CAAC,KAAKb,kBAAkB,CAACW,SAAS,CAAC;IAAEC,UAAU;IAAEC;EAAY,CAAC,CAAC,EAC/D,CAACb,kBAAkB,CACrB,CAAC;EAED,MAAMc,QAAQ,GAAGvC,WAAW,CAC1B,CAACwC,QAAmB,EAAEC,KAAiB,KAAK;IAC1C9B,WAAW,CAAC,EAAE,CAAC;IACf,OAAOc,kBAAkB,CAACc,QAAQ,CAACC,QAAQ,EAAEC,KAAK,CAAC;EACrD,CAAC,EACD,CAAChB,kBAAkB,CACrB,CAAC;EAED,MAAMiB,WAAW,GAAG1C,WAAW,CAC5B2C,OAAe,IAAK;IACnBhC,WAAW,CAAC,EAAE,CAAC;IACf,OAAOc,kBAAkB,CAACiB,WAAW,CAACC,OAAO,CAAC;EAChD,CAAC,EACD,CAAClB,kBAAkB,CACrB,CAAC;EAED,MAAMmB,aAAa,GAAG5C,WAAW,CAC9B6C,KAAa,IAAKpB,kBAAkB,CAACmB,aAAa,CAACC,KAAK,CAAC,EAC1D,CAACpB,kBAAkB,CACrB,CAAC;EAED,MAAMqB,SAAS,GAAG9C,WAAW,CAC3B,MAAMyB,kBAAkB,CAACqB,SAAS,CAAC,CAAC,EACpC,CAACrB,kBAAkB,CACrB,CAAC;EAED,OAAO;IACLb,cAAc;IACdF,QAAQ;IACRF,KAAK;IACLM,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBE,KAAK;IACLgB,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,49 +1,88 @@
 "use strict";
 
-import { useEffect, useMemo, useState } from 'react';
-import { SpeechToTextController } from '../../controllers/SpeechToTextController';
+import { useEffect, useCallback, useState } from 'react';
+import { ETError, getError } from '../../Error';
+import { SpeechToTextModule } from '../../modules/natural_language_processing/SpeechToTextModule';
 export const useSpeechToText = ({
-  modelName,
-  encoderSource,
-  decoderSource,
-  tokenizerSource,
-  overlapSeconds,
-  windowSize,
-  streamingConfig,
+  model,
   preventLoad = false
 }) => {
-  const [sequence, setSequence] = useState('');
+  const [error, setError] = useState(null);
   const [isReady, setIsReady] = useState(false);
-  const [downloadProgress, setDownloadProgress] = useState(0);
   const [isGenerating, setIsGenerating] = useState(false);
-  const [error, setError] = useState();
-  const model = useMemo(() => new SpeechToTextController({
-    transcribeCallback: setSequence,
-    isReadyCallback: setIsReady,
-    isGeneratingCallback: setIsGenerating,
-    onErrorCallback: setError,
-    modelDownloadProgressCallback: setDownloadProgress
-  }), []);
-  useEffect(() => {
-    model.configureStreaming(overlapSeconds, windowSize, streamingConfig);
-  }, [model, overlapSeconds, windowSize, streamingConfig]);
+  const [downloadProgress, setDownloadProgress] = useState(0);
+  const [modelInstance] = useState(() => new SpeechToTextModule());
+  const [committedTranscription, setCommittedTranscription] = useState('');
+  const [nonCommittedTranscription, setNonCommittedTranscription] = useState('');
   useEffect(() => {
-    const loadModel = async () => {
-      await model.loadModel(modelName, encoderSource, decoderSource, tokenizerSource);
-    };
-    if (!preventLoad) {
-      loadModel();
+    if (preventLoad) return;
+    (async () => {
+      setDownloadProgress(0);
+      setError(null);
+      try {
+        setIsReady(false);
+        await modelInstance.load({
+          isMultilingual: model.isMultilingual,
+          encoderSource: model.encoderSource,
+          decoderSource: model.decoderSource,
+          tokenizerSource: model.tokenizerSource
+        }, setDownloadProgress);
+        setIsReady(true);
+      } catch (err) {
+        setError(err.message);
+      }
+    })();
+  }, [modelInstance, model.isMultilingual, model.encoderSource, model.decoderSource, model.tokenizerSource, preventLoad]);
+  const stateWrapper = useCallback(fn => async (...args) => {
+    if (!isReady) throw new Error(getError(ETError.ModuleNotLoaded));
+    if (isGenerating) throw new Error(getError(ETError.ModelGenerating));
+    setIsGenerating(true);
+    try {
+      return await fn.apply(modelInstance, args);
+    } finally {
+      setIsGenerating(false);
     }
-  }, [model, modelName, encoderSource, decoderSource, tokenizerSource, preventLoad]);
+  }, [isReady, isGenerating, modelInstance]);
+  const stream = useCallback(async () => {
+    if (!isReady) throw new Error(getError(ETError.ModuleNotLoaded));
+    if (isGenerating) throw new Error(getError(ETError.ModelGenerating));
+    setIsGenerating(true);
+    setCommittedTranscription('');
+    setNonCommittedTranscription('');
+    let transcription = '';
+    try {
+      for await (const {
+        committed,
+        nonCommitted
+      } of modelInstance.stream()) {
+        setCommittedTranscription(prev => prev + committed);
+        setNonCommittedTranscription(nonCommitted);
+        transcription += committed;
+      }
+    } finally {
+      setIsGenerating(false);
+    }
+    return transcription;
+  }, [isReady, isGenerating, modelInstance]);
+  const wrapper = useCallback(fn => {
+    return (...args) => {
+      if (!isReady) throw new Error(getError(ETError.ModuleNotLoaded));
+      return fn.apply(modelInstance, args);
+    };
+  }, [isReady, modelInstance]);
   return {
+    error,
     isReady,
     isGenerating,
     downloadProgress,
-    configureStreaming: model.configureStreaming,
-    sequence,
-    error,
-    transcribe: (waveform, audioLanguage) => model.transcribe(waveform, audioLanguage),
-    streamingTranscribe: (streamAction, waveform, audioLanguage) => model.streamingTranscribe(streamAction, waveform, audioLanguage)
+    committedTranscription,
+    nonCommittedTranscription,
+    encode: stateWrapper(SpeechToTextModule.prototype.encode),
+    decode: stateWrapper(SpeechToTextModule.prototype.decode),
+    transcribe: stateWrapper(SpeechToTextModule.prototype.transcribe),
+    stream,
+    streamStop: wrapper(SpeechToTextModule.prototype.streamStop),
+    streamInsert: wrapper(SpeechToTextModule.prototype.streamInsert)
   };
 };
 //# sourceMappingURL=useSpeechToText.js.map

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

@@ -1 +1 @@
-{"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":[]}
+{"version":3,"names":["useEffect","useCallback","useState","ETError","getError","SpeechToTextModule","useSpeechToText","model","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","modelInstance","committedTranscription","setCommittedTranscription","nonCommittedTranscription","setNonCommittedTranscription","load","isMultilingual","encoderSource","decoderSource","tokenizerSource","err","message","stateWrapper","fn","args","Error","ModuleNotLoaded","ModelGenerating","apply","stream","transcription","committed","nonCommitted","prev","wrapper","encode","prototype","decode","transcribe","streamStop","streamInsert"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useSpeechToText.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,WAAW,EAAEC,QAAQ,QAAQ,OAAO;AACxD,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAC/C,SAASC,kBAAkB,QAAQ,8DAA8D;AAGjG,OAAO,MAAMC,eAAe,GAAGA,CAAC;EAC9BC,KAAK;EACLC,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;EAE3D,MAAM,CAACe,aAAa,CAAC,GAAGf,QAAQ,CAAC,MAAM,IAAIG,kBAAkB,CAAC,CAAC,CAAC;EAChE,MAAM,CAACa,sBAAsB,EAAEC,yBAAyB,CAAC,GAAGjB,QAAQ,CAAC,EAAE,CAAC;EACxE,MAAM,CAACkB,yBAAyB,EAAEC,4BAA4B,CAAC,GAC7DnB,QAAQ,CAAC,EAAE,CAAC;EAEdF,SAAS,CAAC,MAAM;IACd,IAAIQ,WAAW,EAAE;IACjB,CAAC,YAAY;MACXQ,mBAAmB,CAAC,CAAC,CAAC;MACtBN,QAAQ,CAAC,IAAI,CAAC;MACd,IAAI;QACFE,UAAU,CAAC,KAAK,CAAC;QACjB,MAAMK,aAAa,CAACK,IAAI,CACtB;UACEC,cAAc,EAAEhB,KAAK,CAACgB,cAAc;UACpCC,aAAa,EAAEjB,KAAK,CAACiB,aAAa;UAClCC,aAAa,EAAElB,KAAK,CAACkB,aAAa;UAClCC,eAAe,EAAEnB,KAAK,CAACmB;QACzB,CAAC,EACDV,mBACF,CAAC;QACDJ,UAAU,CAAC,IAAI,CAAC;MAClB,CAAC,CAAC,OAAOe,GAAG,EAAE;QACZjB,QAAQ,CAAEiB,GAAG,CAAWC,OAAO,CAAC;MAClC;IACF,CAAC,EAAE,CAAC;EACN,CAAC,EAAE,CACDX,aAAa,EACbV,KAAK,CAACgB,cAAc,EACpBhB,KAAK,CAACiB,aAAa,EACnBjB,KAAK,CAACkB,aAAa,EACnBlB,KAAK,CAACmB,eAAe,EACrBlB,WAAW,CACZ,CAAC;EAEF,MAAMqB,YAAY,GAAG5B,WAAW,CACe6B,EAAK,IAChD,OAAO,GAAGC,IAAmB,KAAsC;IACjE,IAAI,CAACpB,OAAO,EAAE,MAAM,IAAIqB,KAAK,CAAC5B,QAAQ,CAACD,OAAO,CAAC8B,eAAe,CAAC,CAAC;IAChE,IAAIpB,YAAY,EAAE,MAAM,IAAImB,KAAK,CAAC5B,QAAQ,CAACD,OAAO,CAAC+B,eAAe,CAAC,CAAC;IACpEpB,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAMgB,EAAE,CAACK,KAAK,CAAClB,aAAa,EAAEc,IAAI,CAAC;IAC5C,CAAC,SAAS;MACRjB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACH,CAACH,OAAO,EAAEE,YAAY,EAAEI,aAAa,CACvC,CAAC;EAED,MAAMmB,MAAM,GAAGnC,WAAW,CAAC,YAAY;IACrC,IAAI,CAACU,OAAO,EAAE,MAAM,IAAIqB,KAAK,CAAC5B,QAAQ,CAACD,OAAO,CAAC8B,eAAe,CAAC,CAAC;IAChE,IAAIpB,YAAY,EAAE,MAAM,IAAImB,KAAK,CAAC5B,QAAQ,CAACD,OAAO,CAAC+B,eAAe,CAAC,CAAC;IACpEpB,eAAe,CAAC,IAAI,CAAC;IACrBK,yBAAyB,CAAC,EAAE,CAAC;IAC7BE,4BAA4B,CAAC,EAAE,CAAC;IAChC,IAAIgB,aAAa,GAAG,EAAE;IACtB,IAAI;MACF,WAAW,MAAM;QAAEC,SAAS;QAAEC;MAAa,CAAC,IAAItB,aAAa,CAACmB,MAAM,CAAC,CAAC,EAAE;QACtEjB,yBAAyB,CAAEqB,IAAI,IAAKA,IAAI,GAAGF,SAAS,CAAC;QACrDjB,4BAA4B,CAACkB,YAAY,CAAC;QAC1CF,aAAa,IAAIC,SAAS;MAC5B;IACF,CAAC,SAAS;MACRxB,eAAe,CAAC,KAAK,CAAC;IACxB;IACA,OAAOuB,aAAa;EACtB,CAAC,EAAE,CAAC1B,OAAO,EAAEE,YAAY,EAAEI,aAAa,CAAC,CAAC;EAE1C,MAAMwB,OAAO,GAAGxC,WAAW,CACW6B,EAAK,IAAK;IAC5C,OAAO,CAAC,GAAGC,IAAmB,KAAoB;MAChD,IAAI,CAACpB,OAAO,EAAE,MAAM,IAAIqB,KAAK,CAAC5B,QAAQ,CAACD,OAAO,CAAC8B,eAAe,CAAC,CAAC;MAChE,OAAOH,EAAE,CAACK,KAAK,CAAClB,aAAa,EAAEc,IAAI,CAAC;IACtC,CAAC;EACH,CAAC,EACD,CAACpB,OAAO,EAAEM,aAAa,CACzB,CAAC;EAED,OAAO;IACLR,KAAK;IACLE,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBG,sBAAsB;IACtBE,yBAAyB;IACzBsB,MAAM,EAAEb,YAAY,CAACxB,kBAAkB,CAACsC,SAAS,CAACD,MAAM,CAAC;IACzDE,MAAM,EAAEf,YAAY,CAACxB,kBAAkB,CAACsC,SAAS,CAACC,MAAM,CAAC;IACzDC,UAAU,EAAEhB,YAAY,CAACxB,kBAAkB,CAACsC,SAAS,CAACE,UAAU,CAAC;IACjET,MAAM;IACNU,UAAU,EAAEL,OAAO,CAACpC,kBAAkB,CAACsC,SAAS,CAACG,UAAU,CAAC;IAC5DC,YAAY,EAAEN,OAAO,CAACpC,kBAAkB,CAACsC,SAAS,CAACI,YAAY;EACjE,CAAC;AACH,CAAC","ignoreList":[]}

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

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

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

@@ -1 +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":[]}
+{"version":3,"names":["TextEmbeddingsModule","useNonStaticModule","useTextEmbeddings","model","preventLoad","module"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useTextEmbeddings.ts"],"mappings":";;AAAA,SAASA,oBAAoB,QAAQ,gEAAgE;AAErG,SAASC,kBAAkB,QAAQ,uBAAuB;AAU1D,OAAO,MAAMC,iBAAiB,GAAGA,CAAC;EAAEC,KAAK;EAAEC,WAAW,GAAG;AAAa,CAAC,KACrEH,kBAAkB,CAAC;EACjBI,MAAM,EAAEL,oBAAoB;EAC5BG,KAAK;EACLC;AACF,CAAC,CAAC","ignoreList":[]}

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

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

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

@@ -1 +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":[]}
+{"version":3,"names":["useEffect","useMemo","useState","TokenizerModule","ETError","getError","useTokenizer","tokenizer","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","_tokenizer","load","tokenizerSource","err","message","stateWrapper","fn","args","Error","ModuleNotLoaded","ModelGenerating","apply","decode","prototype","encode","getVocabSize","idToToken","tokenToId"],"sourceRoot":"../../../../src","sources":["hooks/natural_language_processing/useTokenizer.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,OAAO,EAAEC,QAAQ,QAAQ,OAAO;AACpD,SAASC,eAAe,QAAQ,2DAA2D;AAE3F,SAASC,OAAO,EAAEC,QAAQ,QAAQ,aAAa;AAE/C,OAAO,MAAMC,YAAY,GAAGA,CAAC;EAC3BC,SAAS;EACTC,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;EAC3D,MAAMe,UAAU,GAAGhB,OAAO,CAAC,MAAM,IAAIE,eAAe,CAAC,CAAC,EAAE,EAAE,CAAC;EAE3DH,SAAS,CAAC,MAAM;IACd,IAAIQ,WAAW,EAAE;IACjB,CAAC,YAAY;MACXQ,mBAAmB,CAAC,CAAC,CAAC;MACtBN,QAAQ,CAAC,IAAI,CAAC;MACd,IAAI;QACFE,UAAU,CAAC,KAAK,CAAC;QACjB,MAAMK,UAAU,CAACC,IAAI,CACnB;UAAEC,eAAe,EAAEZ,SAAS,CAACY;QAAgB,CAAC,EAC9CH,mBACF,CAAC;QACDJ,UAAU,CAAC,IAAI,CAAC;MAClB,CAAC,CAAC,OAAOQ,GAAG,EAAE;QACZV,QAAQ,CAAEU,GAAG,CAAWC,OAAO,CAAC;MAClC;IACF,CAAC,EAAE,CAAC;EACN,CAAC,EAAE,CAACJ,UAAU,EAAEV,SAAS,CAACY,eAAe,EAAEX,WAAW,CAAC,CAAC;EAExD,MAAMc,YAAY,GAAgDC,EAAK,IAAK;IAC1E,OAAO,CAAC,GAAGC,IAAmB,KAA6B;MACzD,IAAI,CAACb,OAAO,EAAE,MAAM,IAAIc,KAAK,CAACpB,QAAQ,CAACD,OAAO,CAACsB,eAAe,CAAC,CAAC;MAChE,IAAIb,YAAY,EAAE,MAAM,IAAIY,KAAK,CAACpB,QAAQ,CAACD,OAAO,CAACuB,eAAe,CAAC,CAAC;MACpE,IAAI;QACFb,eAAe,CAAC,IAAI,CAAC;QACrB,OAAOS,EAAE,CAACK,KAAK,CAACX,UAAU,EAAEO,IAAI,CAAC;MACnC,CAAC,SAAS;QACRV,eAAe,CAAC,KAAK,CAAC;MACxB;IACF,CAAC;EACH,CAAC;EAED,OAAO;IACLL,KAAK;IACLE,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBc,MAAM,EAAEP,YAAY,CAACnB,eAAe,CAAC2B,SAAS,CAACD,MAAM,CAAC;IACtDE,MAAM,EAAET,YAAY,CAACnB,eAAe,CAAC2B,SAAS,CAACC,MAAM,CAAC;IACtDC,YAAY,EAAEV,YAAY,CAACnB,eAAe,CAAC2B,SAAS,CAACE,YAAY,CAAC;IAClEC,SAAS,EAAEX,YAAY,CAACnB,eAAe,CAAC2B,SAAS,CAACG,SAAS,CAAC;IAC5DC,SAAS,EAAEZ,YAAY,CAACnB,eAAe,CAAC2B,SAAS,CAACI,SAAS;EAC7D,CAAC;AACH,CAAC","ignoreList":[]}

package/lib/module/hooks/useNonStaticModule.js

@@ -0,0 +1,52 @@
+"use strict";
+
+import { useEffect, useState, useMemo } from 'react';
+import { ETError, getError } from '../Error';
+export const useNonStaticModule = ({
+  module,
+  model,
+  preventLoad = false
+}) => {
+  const [error, setError] = useState(null);
+  const [isReady, setIsReady] = useState(false);
+  const [isGenerating, setIsGenerating] = useState(false);
+  const [downloadProgress, setDownloadProgress] = useState(0);
+  const moduleInstance = useMemo(() => new module(), [module]);
+  useEffect(() => {
+    if (preventLoad) return;
+    (async () => {
+      setDownloadProgress(0);
+      setError(null);
+      try {
+        setIsReady(false);
+        await moduleInstance.load(model, setDownloadProgress);
+        setIsReady(true);
+      } catch (err) {
+        setError(err.message);
+      }
+    })();
+    return () => {
+      moduleInstance.delete();
+    };
+
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [moduleInstance, ...Object.values(model), preventLoad]);
+  const forward = async (...input) => {
+    if (!isReady) throw new Error(getError(ETError.ModuleNotLoaded));
+    if (isGenerating) throw new Error(getError(ETError.ModelGenerating));
+    try {
+      setIsGenerating(true);
+      return await moduleInstance.forward(...input);
+    } finally {
+      setIsGenerating(false);
+    }
+  };
+  return {
+    error,
+    isReady,
+    isGenerating,
+    downloadProgress,
+    forward
+  };
+};
+//# sourceMappingURL=useNonStaticModule.js.map

package/lib/module/hooks/useNonStaticModule.js.map

@@ -0,0 +1 @@
+{"version":3,"names":["useEffect","useState","useMemo","ETError","getError","useNonStaticModule","module","model","preventLoad","error","setError","isReady","setIsReady","isGenerating","setIsGenerating","downloadProgress","setDownloadProgress","moduleInstance","load","err","message","delete","Object","values","forward","input","Error","ModuleNotLoaded","ModelGenerating"],"sourceRoot":"../../../src","sources":["hooks/useNonStaticModule.ts"],"mappings":";;AAAA,SAASA,SAAS,EAAEC,QAAQ,EAAEC,OAAO,QAAQ,OAAO;AACpD,SAASC,OAAO,EAAEC,QAAQ,QAAQ,UAAU;AAY5C,OAAO,MAAMC,kBAAkB,GAAGA,CAKhC;EACAC,MAAM;EACNC,KAAK;EACLC,WAAW,GAAG;AAKhB,CAAC,KAAK;EACJ,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;EAC3D,MAAMgB,cAAc,GAAGf,OAAO,CAAC,MAAM,IAAII,MAAM,CAAC,CAAC,EAAE,CAACA,MAAM,CAAC,CAAC;EAE5DN,SAAS,CAAC,MAAM;IACd,IAAIQ,WAAW,EAAE;IAEjB,CAAC,YAAY;MACXQ,mBAAmB,CAAC,CAAC,CAAC;MACtBN,QAAQ,CAAC,IAAI,CAAC;MACd,IAAI;QACFE,UAAU,CAAC,KAAK,CAAC;QACjB,MAAMK,cAAc,CAACC,IAAI,CAACX,KAAK,EAAES,mBAAmB,CAAC;QACrDJ,UAAU,CAAC,IAAI,CAAC;MAClB,CAAC,CAAC,OAAOO,GAAG,EAAE;QACZT,QAAQ,CAAES,GAAG,CAAWC,OAAO,CAAC;MAClC;IACF,CAAC,EAAE,CAAC;IAEJ,OAAO,MAAM;MACXH,cAAc,CAACI,MAAM,CAAC,CAAC;IACzB,CAAC;;IAED;EACF,CAAC,EAAE,CAACJ,cAAc,EAAE,GAAGK,MAAM,CAACC,MAAM,CAAChB,KAAK,CAAC,EAAEC,WAAW,CAAC,CAAC;EAE1D,MAAMgB,OAAO,GAAG,MAAAA,CAAO,GAAGC,KAAkB,KAA6B;IACvE,IAAI,CAACd,OAAO,EAAE,MAAM,IAAIe,KAAK,CAACtB,QAAQ,CAACD,OAAO,CAACwB,eAAe,CAAC,CAAC;IAChE,IAAId,YAAY,EAAE,MAAM,IAAIa,KAAK,CAACtB,QAAQ,CAACD,OAAO,CAACyB,eAAe,CAAC,CAAC;IACpE,IAAI;MACFd,eAAe,CAAC,IAAI,CAAC;MACrB,OAAO,MAAMG,cAAc,CAACO,OAAO,CAAC,GAAGC,KAAK,CAAC;IAC/C,CAAC,SAAS;MACRX,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC;EAED,OAAO;IACLL,KAAK;IACLE,OAAO;IACPE,YAAY;IACZE,gBAAgB;IAChBS;EACF,CAAC;AACH,CAAC","ignoreList":[]}