cactus-react-native 1.2.1 → 1.5.0

package/lib/module/classes/CactusSTT.js

@@ -3,42 +3,61 @@
 import { Cactus, CactusFileSystem } from "../native/index.js";
 import { Telemetry } from "../telemetry/Telemetry.js";
 import { CactusConfig } from "../config/CactusConfig.js";
-import { Database } from "../api/Database.js";
 import { getErrorMessage } from "../utils/error.js";
+import models from "../models.js";
 export class CactusSTT {
   cactus = new Cactus();
   isDownloading = false;
   isInitialized = false;
   isGenerating = false;
+  isStreamTranscribeInitialized = false;
   static defaultModel = 'whisper-small';
   static defaultContextSize = 2048;
+  static defaultOptions = {
+    quantization: 'int4',
+    pro: false
+  };
   static defaultPrompt = '<|startoftranscript|><|en|><|transcribe|><|notimestamps|>';
   static defaultTranscribeOptions = {
-    maxTokens: 512
+    maxTokens: 384
   };
   static defaultEmbedBufferSize = 4096;
   constructor({
     model,
-    contextSize
+    contextSize,
+    options
   } = {}) {
     Telemetry.init(CactusConfig.telemetryToken);
     this.model = model ?? CactusSTT.defaultModel;
     this.contextSize = contextSize ?? CactusSTT.defaultContextSize;
+    this.options = {
+      quantization: options?.quantization ?? CactusSTT.defaultOptions.quantization,
+      pro: options?.pro ?? CactusSTT.defaultOptions.pro
+    };
   }
   async download({
     onProgress
   } = {}) {
+    if (this.isModelPath(this.model)) {
+      onProgress?.(1.0);
+      return;
+    }
     if (this.isDownloading) {
       throw new Error('CactusSTT is already downloading');
     }
-    if (await CactusFileSystem.modelExists(this.model)) {
+    if (await CactusFileSystem.modelExists(this.getModelName())) {
+      console.log('Model already exists', this.getModelName());
       onProgress?.(1.0);
       return;
     }
     this.isDownloading = true;
     try {
-      const model = await Database.getSTTModel(this.model);
-      await CactusFileSystem.downloadModel(this.model, model.downloadUrl, onProgress);
+      const modelConfig = models[this.model]?.quantization[this.options.quantization];
+      const url = this.options.pro ? modelConfig?.pro?.apple : modelConfig?.url;
+      if (!url) {
+        throw new Error(`Model ${this.model} with specified options not found`);
+      }
+      await CactusFileSystem.downloadModel(this.getModelName(), url, onProgress);
     } finally {
       this.isDownloading = false;
     }
@@ -47,10 +66,16 @@ export class CactusSTT {
     if (this.isInitialized) {
       return;
     }
-    if (!(await CactusFileSystem.modelExists(this.model))) {
-      throw new Error(`Model "${this.model}" is not downloaded`);
+    let modelPath;
+    if (this.isModelPath(this.model)) {
+      modelPath = this.model.replace('file://', '');
+    } else {
+      if (!(await CactusFileSystem.modelExists(this.getModelName()))) {
+        console.log('Model does not exist', this.getModelName());
+        throw new Error(`Model "${this.model}" with options ${JSON.stringify(this.options)} is not downloaded`);
+      }
+      modelPath = await CactusFileSystem.getModelPath(this.getModelName());
     }
-    const modelPath = await CactusFileSystem.getModelPath(this.model);
     try {
       await this.cactus.init(modelPath, this.contextSize);
       Telemetry.logInit(this.model, true);
@@ -61,7 +86,7 @@ export class CactusSTT {
     }
   }
   async transcribe({
-    audioFilePath,
+    audio,
     prompt,
     options,
     onToken
@@ -78,7 +103,7 @@ export class CactusSTT {
     const responseBufferSize = 8 * (options.maxTokens ?? CactusSTT.defaultTranscribeOptions.maxTokens) + 256;
     this.isGenerating = true;
     try {
-      const result = await this.cactus.transcribe(audioFilePath, prompt, responseBufferSize, options, onToken);
+      const result = await this.cactus.transcribe(audio, prompt, responseBufferSize, options, onToken);
       Telemetry.logTranscribe(this.model, result.success, result.success ? undefined : result.response, result);
       return result;
     } catch (error) {
@@ -88,6 +113,65 @@ export class CactusSTT {
       this.isGenerating = false;
     }
   }
+  async streamTranscribeInit() {
+    if (this.isStreamTranscribeInitialized) {
+      return;
+    }
+    await this.init();
+    try {
+      await this.cactus.streamTranscribeInit();
+      this.isStreamTranscribeInitialized = true;
+    } catch (error) {
+      throw error;
+    }
+  }
+  async streamTranscribeInsert({
+    audio
+  }) {
+    if (!this.isStreamTranscribeInitialized) {
+      throw new Error('CactusSTT stream transcribe is not initialized');
+    }
+    try {
+      await this.cactus.streamTranscribeInsert(audio);
+    } catch (error) {
+      throw error;
+    }
+  }
+  async streamTranscribeProcess({
+    options
+  } = {}) {
+    if (!this.isStreamTranscribeInitialized) {
+      throw new Error('CactusSTT stream transcribe is not initialized');
+    }
+    try {
+      const result = await this.cactus.streamTranscribeProcess(options);
+      return result;
+    } catch (error) {
+      throw error;
+    }
+  }
+  async streamTranscribeFinalize() {
+    if (!this.isStreamTranscribeInitialized) {
+      throw new Error('CactusSTT stream transcribe is not initialized');
+    }
+    try {
+      const result = await this.cactus.streamTranscribeFinalize();
+      return result;
+    } catch (error) {
+      throw error;
+    }
+  }
+  async streamTranscribeDestroy() {
+    if (!this.isStreamTranscribeInitialized) {
+      return;
+    }
+    try {
+      await this.cactus.streamTranscribeDestroy();
+      this.isStreamTranscribeInitialized = false;
+    } catch (error) {
+      throw error;
+    }
+  }
   async audioEmbed({
     audioPath
   }) {
@@ -121,15 +205,18 @@ export class CactusSTT {
       return;
     }
     await this.stop();
+    await this.streamTranscribeDestroy();
     await this.cactus.destroy();
     this.isInitialized = false;
   }
-  async getModels() {
-    const models = await Database.getSTTModels();
-    for (const model of models) {
-      model.isDownloaded = await CactusFileSystem.modelExists(model.slug);
-    }
-    return models;
+  getModels() {
+    return Object.values(models).filter(model => model.speech);
+  }
+  isModelPath(model) {
+    return model.startsWith('file://') || model.startsWith('/');
+  }
+  getModelName() {
+    return `${this.model}-${this.options.quantization}${this.options.pro ? '-pro' : ''}`;
   }
 }
 //# sourceMappingURL=CactusSTT.js.map

package/lib/module/classes/CactusSTT.js.map

@@ -1 +1 @@
-{"version":3,"names":["Cactus","CactusFileSystem","Telemetry","CactusConfig","Database","getErrorMessage","CactusSTT","cactus","isDownloading","isInitialized","isGenerating","defaultModel","defaultContextSize","defaultPrompt","defaultTranscribeOptions","maxTokens","defaultEmbedBufferSize","constructor","model","contextSize","init","telemetryToken","download","onProgress","Error","modelExists","getSTTModel","downloadModel","downloadUrl","modelPath","getModelPath","logInit","error","transcribe","audioFilePath","prompt","options","onToken","responseBufferSize","result","logTranscribe","success","undefined","response","audioEmbed","audioPath","embedding","logAudioEmbedding","stop","reset","destroy","getModels","models","getSTTModels","isDownloaded","slug"],"sourceRoot":"../../../src","sources":["classes/CactusSTT.ts"],"mappings":";;AAAA,SAASA,MAAM,EAAEC,gBAAgB,QAAQ,oBAAW;AASpD,SAASC,SAAS,QAAQ,2BAAwB;AAClD,SAASC,YAAY,QAAQ,2BAAwB;AACrD,SAASC,QAAQ,QAAQ,oBAAiB;AAC1C,SAASC,eAAe,QAAQ,mBAAgB;AAGhD,OAAO,MAAMC,SAAS,CAAC;EACJC,MAAM,GAAG,IAAIP,MAAM,CAAC,CAAC;EAK9BQ,aAAa,GAAG,KAAK;EACrBC,aAAa,GAAG,KAAK;EACrBC,YAAY,GAAG,KAAK;EAE5B,OAAwBC,YAAY,GAAG,eAAe;EACtD,OAAwBC,kBAAkB,GAAG,IAAI;EACjD,OAAwBC,aAAa,GACnC,2DAA2D;EAC7D,OAAwBC,wBAAwB,GAAG;IACjDC,SAAS,EAAE;EACb,CAAC;EACD,OAAwBC,sBAAsB,GAAG,IAAI;EAErDC,WAAWA,CAAC;IAAEC,KAAK;IAAEC;EAA6B,CAAC,GAAG,CAAC,CAAC,EAAE;IACxDjB,SAAS,CAACkB,IAAI,CAACjB,YAAY,CAACkB,cAAc,CAAC;IAE3C,IAAI,CAACH,KAAK,GAAGA,KAAK,IAAIZ,SAAS,CAACK,YAAY;IAC5C,IAAI,CAACQ,WAAW,GAAGA,WAAW,IAAIb,SAAS,CAACM,kBAAkB;EAChE;EAEA,MAAaU,QAAQA,CAAC;IACpBC;EACuB,CAAC,GAAG,CAAC,CAAC,EAAiB;IAC9C,IAAI,IAAI,CAACf,aAAa,EAAE;MACtB,MAAM,IAAIgB,KAAK,CAAC,kCAAkC,CAAC;IACrD;IAEA,IAAI,MAAMvB,gBAAgB,CAACwB,WAAW,CAAC,IAAI,CAACP,KAAK,CAAC,EAAE;MAClDK,UAAU,GAAG,GAAG,CAAC;MACjB;IACF;IAEA,IAAI,CAACf,aAAa,GAAG,IAAI;IACzB,IAAI;MACF,MAAMU,KAAK,GAAG,MAAMd,QAAQ,CAACsB,WAAW,CAAC,IAAI,CAACR,KAAK,CAAC;MACpD,MAAMjB,gBAAgB,CAAC0B,aAAa,CAClC,IAAI,CAACT,KAAK,EACVA,KAAK,CAACU,WAAW,EACjBL,UACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAACf,aAAa,GAAG,KAAK;IAC5B;EACF;EAEA,MAAaY,IAAIA,CAAA,EAAkB;IACjC,IAAI,IAAI,CAACX,aAAa,EAAE;MACtB;IACF;IAEA,IAAI,EAAE,MAAMR,gBAAgB,CAACwB,WAAW,CAAC,IAAI,CAACP,KAAK,CAAC,CAAC,EAAE;MACrD,MAAM,IAAIM,KAAK,CAAC,UAAU,IAAI,CAACN,KAAK,qBAAqB,CAAC;IAC5D;IAEA,MAAMW,SAAS,GAAG,MAAM5B,gBAAgB,CAAC6B,YAAY,CAAC,IAAI,CAACZ,KAAK,CAAC;IAEjE,IAAI;MACF,MAAM,IAAI,CAACX,MAAM,CAACa,IAAI,CAACS,SAAS,EAAE,IAAI,CAACV,WAAW,CAAC;MACnDjB,SAAS,CAAC6B,OAAO,CAAC,IAAI,CAACb,KAAK,EAAE,IAAI,CAAC;MACnC,IAAI,CAACT,aAAa,GAAG,IAAI;IAC3B,CAAC,CAAC,OAAOuB,KAAK,EAAE;MACd9B,SAAS,CAAC6B,OAAO,CAAC,IAAI,CAACb,KAAK,EAAE,KAAK,EAAEb,eAAe,CAAC2B,KAAK,CAAC,CAAC;MAC5D,MAAMA,KAAK;IACb;EACF;EAEA,MAAaC,UAAUA,CAAC;IACtBC,aAAa;IACbC,MAAM;IACNC,OAAO;IACPC;EACyB,CAAC,EAAsC;IAChE,IAAI,IAAI,CAAC3B,YAAY,EAAE;MACrB,MAAM,IAAIc,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,MAAM,IAAI,CAACJ,IAAI,CAAC,CAAC;IAEjBe,MAAM,GAAGA,MAAM,IAAI7B,SAAS,CAACO,aAAa;IAC1CuB,OAAO,GAAG;MAAE,GAAG9B,SAAS,CAACQ,wBAAwB;MAAE,GAAGsB;IAAQ,CAAC;IAE/D,MAAME,kBAAkB,GACtB,CAAC,IAAIF,OAAO,CAACrB,SAAS,IAAIT,SAAS,CAACQ,wBAAwB,CAACC,SAAS,CAAC,GACvE,GAAG;IAEL,IAAI,CAACL,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAM6B,MAAM,GAAG,MAAM,IAAI,CAAChC,MAAM,CAAC0B,UAAU,CACzCC,aAAa,EACbC,MAAM,EACNG,kBAAkB,EAClBF,OAAO,EACPC,OACF,CAAC;MACDnC,SAAS,CAACsC,aAAa,CACrB,IAAI,CAACtB,KAAK,EACVqB,MAAM,CAACE,OAAO,EACdF,MAAM,CAACE,OAAO,GAAGC,SAAS,GAAGH,MAAM,CAACI,QAAQ,EAC5CJ,MACF,CAAC;MACD,OAAOA,MAAM;IACf,CAAC,CAAC,OAAOP,KAAK,EAAE;MACd9B,SAAS,CAACsC,aAAa,CAAC,IAAI,CAACtB,KAAK,EAAE,KAAK,EAAEb,eAAe,CAAC2B,KAAK,CAAC,CAAC;MAClE,MAAMA,KAAK;IACb,CAAC,SAAS;MACR,IAAI,CAACtB,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAakC,UAAUA,CAAC;IACtBC;EACyB,CAAC,EAAsC;IAChE,IAAI,IAAI,CAACnC,YAAY,EAAE;MACrB,MAAM,IAAIc,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,MAAM,IAAI,CAACJ,IAAI,CAAC,CAAC;IAEjB,IAAI,CAACV,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMoC,SAAS,GAAG,MAAM,IAAI,CAACvC,MAAM,CAACqC,UAAU,CAC5CC,SAAS,EACTvC,SAAS,CAACU,sBACZ,CAAC;MACDd,SAAS,CAAC6C,iBAAiB,CAAC,IAAI,CAAC7B,KAAK,EAAE,IAAI,CAAC;MAC7C,OAAO;QAAE4B;MAAU,CAAC;IACtB,CAAC,CAAC,OAAOd,KAAK,EAAE;MACd9B,SAAS,CAAC6C,iBAAiB,CAAC,IAAI,CAAC7B,KAAK,EAAE,KAAK,EAAEb,eAAe,CAAC2B,KAAK,CAAC,CAAC;MACtE,MAAMA,KAAK;IACb,CAAC,SAAS;MACR,IAAI,CAACtB,YAAY,GAAG,KAAK;IAC3B;EACF;EAEOsC,IAAIA,CAAA,EAAkB;IAC3B,OAAO,IAAI,CAACzC,MAAM,CAACyC,IAAI,CAAC,CAAC;EAC3B;EAEA,MAAaC,KAAKA,CAAA,EAAkB;IAClC,MAAM,IAAI,CAACD,IAAI,CAAC,CAAC;IACjB,OAAO,IAAI,CAACzC,MAAM,CAAC0C,KAAK,CAAC,CAAC;EAC5B;EAEA,MAAaC,OAAOA,CAAA,EAAkB;IACpC,IAAI,CAAC,IAAI,CAACzC,aAAa,EAAE;MACvB;IACF;IAEA,MAAM,IAAI,CAACuC,IAAI,CAAC,CAAC;IACjB,MAAM,IAAI,CAACzC,MAAM,CAAC2C,OAAO,CAAC,CAAC;IAE3B,IAAI,CAACzC,aAAa,GAAG,KAAK;EAC5B;EAEA,MAAa0C,SAASA,CAAA,EAA8B;IAClD,MAAMC,MAAM,GAAG,MAAMhD,QAAQ,CAACiD,YAAY,CAAC,CAAC;IAC5C,KAAK,MAAMnC,KAAK,IAAIkC,MAAM,EAAE;MAC1BlC,KAAK,CAACoC,YAAY,GAAG,MAAMrD,gBAAgB,CAACwB,WAAW,CAACP,KAAK,CAACqC,IAAI,CAAC;IACrE;IACA,OAAOH,MAAM;EACf;AACF","ignoreList":[]}
+{"version":3,"names":["Cactus","CactusFileSystem","Telemetry","CactusConfig","getErrorMessage","models","CactusSTT","cactus","isDownloading","isInitialized","isGenerating","isStreamTranscribeInitialized","defaultModel","defaultContextSize","defaultOptions","quantization","pro","defaultPrompt","defaultTranscribeOptions","maxTokens","defaultEmbedBufferSize","constructor","model","contextSize","options","init","telemetryToken","download","onProgress","isModelPath","Error","modelExists","getModelName","console","log","modelConfig","url","apple","downloadModel","modelPath","replace","JSON","stringify","getModelPath","logInit","error","transcribe","audio","prompt","onToken","responseBufferSize","result","logTranscribe","success","undefined","response","streamTranscribeInit","streamTranscribeInsert","streamTranscribeProcess","streamTranscribeFinalize","streamTranscribeDestroy","audioEmbed","audioPath","embedding","logAudioEmbedding","stop","reset","destroy","getModels","Object","values","filter","speech","startsWith"],"sourceRoot":"../../../src","sources":["classes/CactusSTT.ts"],"mappings":";;AAAA,SAASA,MAAM,EAAEC,gBAAgB,QAAQ,oBAAW;AAapD,SAASC,SAAS,QAAQ,2BAAwB;AAClD,SAASC,YAAY,QAAQ,2BAAwB;AACrD,SAASC,eAAe,QAAQ,mBAAgB;AAChD,OAAOC,MAAM,MAAM,cAAW;AAG9B,OAAO,MAAMC,SAAS,CAAC;EACJC,MAAM,GAAG,IAAIP,MAAM,CAAC,CAAC;EAS9BQ,aAAa,GAAG,KAAK;EACrBC,aAAa,GAAG,KAAK;EACrBC,YAAY,GAAG,KAAK;EAEpBC,6BAA6B,GAAG,KAAK;EAE7C,OAAwBC,YAAY,GAAG,eAAe;EACtD,OAAwBC,kBAAkB,GAAG,IAAI;EACjD,OAAwBC,cAAc,GAAG;IACvCC,YAAY,EAAE,MAAe;IAC7BC,GAAG,EAAE;EACP,CAAC;EACD,OAAwBC,aAAa,GACnC,2DAA2D;EAC7D,OAAwBC,wBAAwB,GAAG;IACjDC,SAAS,EAAE;EACb,CAAC;EACD,OAAwBC,sBAAsB,GAAG,IAAI;EAErDC,WAAWA,CAAC;IAAEC,KAAK;IAAEC,WAAW;IAAEC;EAAyB,CAAC,GAAG,CAAC,CAAC,EAAE;IACjEtB,SAAS,CAACuB,IAAI,CAACtB,YAAY,CAACuB,cAAc,CAAC;IAE3C,IAAI,CAACJ,KAAK,GAAGA,KAAK,IAAIhB,SAAS,CAACM,YAAY;IAC5C,IAAI,CAACW,WAAW,GAAGA,WAAW,IAAIjB,SAAS,CAACO,kBAAkB;IAC9D,IAAI,CAACW,OAAO,GAAG;MACbT,YAAY,EACVS,OAAO,EAAET,YAAY,IAAIT,SAAS,CAACQ,cAAc,CAACC,YAAY;MAChEC,GAAG,EAAEQ,OAAO,EAAER,GAAG,IAAIV,SAAS,CAACQ,cAAc,CAACE;IAChD,CAAC;EACH;EAEA,MAAaW,QAAQA,CAAC;IACpBC;EACuB,CAAC,GAAG,CAAC,CAAC,EAAiB;IAC9C,IAAI,IAAI,CAACC,WAAW,CAAC,IAAI,CAACP,KAAK,CAAC,EAAE;MAChCM,UAAU,GAAG,GAAG,CAAC;MACjB;IACF;IAEA,IAAI,IAAI,CAACpB,aAAa,EAAE;MACtB,MAAM,IAAIsB,KAAK,CAAC,kCAAkC,CAAC;IACrD;IAEA,IAAI,MAAM7B,gBAAgB,CAAC8B,WAAW,CAAC,IAAI,CAACC,YAAY,CAAC,CAAC,CAAC,EAAE;MAC3DC,OAAO,CAACC,GAAG,CAAC,sBAAsB,EAAE,IAAI,CAACF,YAAY,CAAC,CAAC,CAAC;MACxDJ,UAAU,GAAG,GAAG,CAAC;MACjB;IACF;IAEA,IAAI,CAACpB,aAAa,GAAG,IAAI;IACzB,IAAI;MACF,MAAM2B,WAAW,GACf9B,MAAM,CAAC,IAAI,CAACiB,KAAK,CAAC,EAAEP,YAAY,CAAC,IAAI,CAACS,OAAO,CAACT,YAAY,CAAC;MAC7D,MAAMqB,GAAG,GAAG,IAAI,CAACZ,OAAO,CAACR,GAAG,GAAGmB,WAAW,EAAEnB,GAAG,EAAEqB,KAAK,GAAGF,WAAW,EAAEC,GAAG;MAEzE,IAAI,CAACA,GAAG,EAAE;QACR,MAAM,IAAIN,KAAK,CAAC,SAAS,IAAI,CAACR,KAAK,mCAAmC,CAAC;MACzE;MAEA,MAAMrB,gBAAgB,CAACqC,aAAa,CAClC,IAAI,CAACN,YAAY,CAAC,CAAC,EACnBI,GAAG,EACHR,UACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAACpB,aAAa,GAAG,KAAK;IAC5B;EACF;EAEA,MAAaiB,IAAIA,CAAA,EAAkB;IACjC,IAAI,IAAI,CAAChB,aAAa,EAAE;MACtB;IACF;IAEA,IAAI8B,SAAiB;IACrB,IAAI,IAAI,CAACV,WAAW,CAAC,IAAI,CAACP,KAAK,CAAC,EAAE;MAChCiB,SAAS,GAAG,IAAI,CAACjB,KAAK,CAACkB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;IAC/C,CAAC,MAAM;MACL,IAAI,EAAE,MAAMvC,gBAAgB,CAAC8B,WAAW,CAAC,IAAI,CAACC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE;QAC9DC,OAAO,CAACC,GAAG,CAAC,sBAAsB,EAAE,IAAI,CAACF,YAAY,CAAC,CAAC,CAAC;QACxD,MAAM,IAAIF,KAAK,CACb,UAAU,IAAI,CAACR,KAAK,kBAAkBmB,IAAI,CAACC,SAAS,CAAC,IAAI,CAAClB,OAAO,CAAC,oBACpE,CAAC;MACH;MACAe,SAAS,GAAG,MAAMtC,gBAAgB,CAAC0C,YAAY,CAAC,IAAI,CAACX,YAAY,CAAC,CAAC,CAAC;IACtE;IAEA,IAAI;MACF,MAAM,IAAI,CAACzB,MAAM,CAACkB,IAAI,CAACc,SAAS,EAAE,IAAI,CAAChB,WAAW,CAAC;MACnDrB,SAAS,CAAC0C,OAAO,CAAC,IAAI,CAACtB,KAAK,EAAE,IAAI,CAAC;MACnC,IAAI,CAACb,aAAa,GAAG,IAAI;IAC3B,CAAC,CAAC,OAAOoC,KAAK,EAAE;MACd3C,SAAS,CAAC0C,OAAO,CAAC,IAAI,CAACtB,KAAK,EAAE,KAAK,EAAElB,eAAe,CAACyC,KAAK,CAAC,CAAC;MAC5D,MAAMA,KAAK;IACb;EACF;EAEA,MAAaC,UAAUA,CAAC;IACtBC,KAAK;IACLC,MAAM;IACNxB,OAAO;IACPyB;EACyB,CAAC,EAAsC;IAChE,IAAI,IAAI,CAACvC,YAAY,EAAE;MACrB,MAAM,IAAIoB,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,MAAM,IAAI,CAACL,IAAI,CAAC,CAAC;IAEjBuB,MAAM,GAAGA,MAAM,IAAI1C,SAAS,CAACW,aAAa;IAC1CO,OAAO,GAAG;MAAE,GAAGlB,SAAS,CAACY,wBAAwB;MAAE,GAAGM;IAAQ,CAAC;IAE/D,MAAM0B,kBAAkB,GACtB,CAAC,IAAI1B,OAAO,CAACL,SAAS,IAAIb,SAAS,CAACY,wBAAwB,CAACC,SAAS,CAAC,GACvE,GAAG;IAEL,IAAI,CAACT,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMyC,MAAM,GAAG,MAAM,IAAI,CAAC5C,MAAM,CAACuC,UAAU,CACzCC,KAAK,EACLC,MAAM,EACNE,kBAAkB,EAClB1B,OAAO,EACPyB,OACF,CAAC;MACD/C,SAAS,CAACkD,aAAa,CACrB,IAAI,CAAC9B,KAAK,EACV6B,MAAM,CAACE,OAAO,EACdF,MAAM,CAACE,OAAO,GAAGC,SAAS,GAAGH,MAAM,CAACI,QAAQ,EAC5CJ,MACF,CAAC;MACD,OAAOA,MAAM;IACf,CAAC,CAAC,OAAON,KAAK,EAAE;MACd3C,SAAS,CAACkD,aAAa,CAAC,IAAI,CAAC9B,KAAK,EAAE,KAAK,EAAElB,eAAe,CAACyC,KAAK,CAAC,CAAC;MAClE,MAAMA,KAAK;IACb,CAAC,SAAS;MACR,IAAI,CAACnC,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAa8C,oBAAoBA,CAAA,EAAkB;IACjD,IAAI,IAAI,CAAC7C,6BAA6B,EAAE;MACtC;IACF;IAEA,MAAM,IAAI,CAACc,IAAI,CAAC,CAAC;IAEjB,IAAI;MACF,MAAM,IAAI,CAAClB,MAAM,CAACiD,oBAAoB,CAAC,CAAC;MACxC,IAAI,CAAC7C,6BAA6B,GAAG,IAAI;IAC3C,CAAC,CAAC,OAAOkC,KAAK,EAAE;MACd,MAAMA,KAAK;IACb;EACF;EAEA,MAAaY,sBAAsBA,CAAC;IAClCV;EACqC,CAAC,EAAiB;IACvD,IAAI,CAAC,IAAI,CAACpC,6BAA6B,EAAE;MACvC,MAAM,IAAImB,KAAK,CAAC,gDAAgD,CAAC;IACnE;IAEA,IAAI;MACF,MAAM,IAAI,CAACvB,MAAM,CAACkD,sBAAsB,CAACV,KAAK,CAAC;IACjD,CAAC,CAAC,OAAOF,KAAK,EAAE;MACd,MAAMA,KAAK;IACb;EACF;EAEA,MAAaa,uBAAuBA,CAAC;IACnClC;EACsC,CAAC,GAAG,CAAC,CAAC,EAAmD;IAC/F,IAAI,CAAC,IAAI,CAACb,6BAA6B,EAAE;MACvC,MAAM,IAAImB,KAAK,CAAC,gDAAgD,CAAC;IACnE;IAEA,IAAI;MACF,MAAMqB,MAAM,GAAG,MAAM,IAAI,CAAC5C,MAAM,CAACmD,uBAAuB,CAAClC,OAAO,CAAC;MACjE,OAAO2B,MAAM;IACf,CAAC,CAAC,OAAON,KAAK,EAAE;MACd,MAAMA,KAAK;IACb;EACF;EAEA,MAAac,wBAAwBA,CAAA,EAAqD;IACxF,IAAI,CAAC,IAAI,CAAChD,6BAA6B,EAAE;MACvC,MAAM,IAAImB,KAAK,CAAC,gDAAgD,CAAC;IACnE;IAEA,IAAI;MACF,MAAMqB,MAAM,GAAG,MAAM,IAAI,CAAC5C,MAAM,CAACoD,wBAAwB,CAAC,CAAC;MAC3D,OAAOR,MAAM;IACf,CAAC,CAAC,OAAON,KAAK,EAAE;MACd,MAAMA,KAAK;IACb;EACF;EAEA,MAAae,uBAAuBA,CAAA,EAAkB;IACpD,IAAI,CAAC,IAAI,CAACjD,6BAA6B,EAAE;MACvC;IACF;IAEA,IAAI;MACF,MAAM,IAAI,CAACJ,MAAM,CAACqD,uBAAuB,CAAC,CAAC;MAC3C,IAAI,CAACjD,6BAA6B,GAAG,KAAK;IAC5C,CAAC,CAAC,OAAOkC,KAAK,EAAE;MACd,MAAMA,KAAK;IACb;EACF;EAEA,MAAagB,UAAUA,CAAC;IACtBC;EACyB,CAAC,EAAsC;IAChE,IAAI,IAAI,CAACpD,YAAY,EAAE;MACrB,MAAM,IAAIoB,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,MAAM,IAAI,CAACL,IAAI,CAAC,CAAC;IAEjB,IAAI,CAACf,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMqD,SAAS,GAAG,MAAM,IAAI,CAACxD,MAAM,CAACsD,UAAU,CAC5CC,SAAS,EACTxD,SAAS,CAACc,sBACZ,CAAC;MACDlB,SAAS,CAAC8D,iBAAiB,CAAC,IAAI,CAAC1C,KAAK,EAAE,IAAI,CAAC;MAC7C,OAAO;QAAEyC;MAAU,CAAC;IACtB,CAAC,CAAC,OAAOlB,KAAK,EAAE;MACd3C,SAAS,CAAC8D,iBAAiB,CAAC,IAAI,CAAC1C,KAAK,EAAE,KAAK,EAAElB,eAAe,CAACyC,KAAK,CAAC,CAAC;MACtE,MAAMA,KAAK;IACb,CAAC,SAAS;MACR,IAAI,CAACnC,YAAY,GAAG,KAAK;IAC3B;EACF;EAEOuD,IAAIA,CAAA,EAAkB;IAC3B,OAAO,IAAI,CAAC1D,MAAM,CAAC0D,IAAI,CAAC,CAAC;EAC3B;EAEA,MAAaC,KAAKA,CAAA,EAAkB;IAClC,MAAM,IAAI,CAACD,IAAI,CAAC,CAAC;IACjB,OAAO,IAAI,CAAC1D,MAAM,CAAC2D,KAAK,CAAC,CAAC;EAC5B;EAEA,MAAaC,OAAOA,CAAA,EAAkB;IACpC,IAAI,CAAC,IAAI,CAAC1D,aAAa,EAAE;MACvB;IACF;IAEA,MAAM,IAAI,CAACwD,IAAI,CAAC,CAAC;IACjB,MAAM,IAAI,CAACL,uBAAuB,CAAC,CAAC;IACpC,MAAM,IAAI,CAACrD,MAAM,CAAC4D,OAAO,CAAC,CAAC;IAE3B,IAAI,CAAC1D,aAAa,GAAG,KAAK;EAC5B;EAEO2D,SAASA,CAAA,EAAkB;IAChC,OAAOC,MAAM,CAACC,MAAM,CAACjE,MAAM,CAAC,CAACkE,MAAM,CAAEjD,KAAK,IAAKA,KAAK,CAACkD,MAAM,CAAC;EAC9D;EAEQ3C,WAAWA,CAACP,KAAa,EAAW;IAC1C,OAAOA,KAAK,CAACmD,UAAU,CAAC,SAAS,CAAC,IAAInD,KAAK,CAACmD,UAAU,CAAC,GAAG,CAAC;EAC7D;EAEQzC,YAAYA,CAAA,EAAW;IAC7B,OAAO,GAAG,IAAI,CAACV,KAAK,IAAI,IAAI,CAACE,OAAO,CAACT,YAAY,GAAG,IAAI,CAACS,OAAO,CAACR,GAAG,GAAG,MAAM,GAAG,EAAE,EAAE;EACtF;AACF","ignoreList":[]}

package/lib/module/config/CactusConfig.js

@@ -6,5 +6,7 @@ export class CactusConfig {
   static isTelemetryEnabled = true;
 
   // Hybrid mode
+
+  // Pro features
 }
 //# sourceMappingURL=CactusConfig.js.map

package/lib/module/config/CactusConfig.js.map

@@ -1 +1 @@
-{"version":3,"names":["CactusConfig","isTelemetryEnabled"],"sourceRoot":"../../../src","sources":["config/CactusConfig.ts"],"mappings":";;AAAA,OAAO,MAAMA,YAAY,CAAC;EACxB;;EAEA,OAAcC,kBAAkB,GAAY,IAAI;;EAEhD;AAEF","ignoreList":[]}
+{"version":3,"names":["CactusConfig","isTelemetryEnabled"],"sourceRoot":"../../../src","sources":["config/CactusConfig.ts"],"mappings":";;AAAA,OAAO,MAAMA,YAAY,CAAC;EACxB;;EAEA,OAAcC,kBAAkB,GAAY,IAAI;;EAEhD;;EAGA;AAEF","ignoreList":[]}

package/lib/module/constants/packageVersion.js

@@ -1,4 +1,4 @@
 "use strict";
 
-export const packageVersion = '1.2.1';
+export const packageVersion = '1.4.0';
 //# sourceMappingURL=packageVersion.js.map

package/lib/module/hooks/useCactusIndex.js

@@ -0,0 +1,175 @@
+"use strict";
+
+import { useCallback, useEffect, useState } from 'react';
+import { CactusIndex } from "../classes/CactusIndex.js";
+import { getErrorMessage } from "../utils/error.js";
+export const useCactusIndex = ({
+  name,
+  embeddingDim
+}) => {
+  const [cactusIndex, setCactusIndex] = useState(() => new CactusIndex(name, embeddingDim));
+
+  // State
+  const [isInitializing, setIsInitializing] = useState(false);
+  const [isProcessing, setIsProcessing] = useState(false);
+  const [error, setError] = useState(null);
+  useEffect(() => {
+    setCactusIndex(new CactusIndex(name, embeddingDim));
+    setIsInitializing(false);
+    setIsProcessing(false);
+    setError(null);
+  }, [name, embeddingDim]);
+  useEffect(() => {
+    return () => {
+      cactusIndex.destroy().catch(() => {});
+    };
+  }, [cactusIndex]);
+  const init = useCallback(async () => {
+    if (isInitializing) {
+      const message = 'CactusIndex is already initializing';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsInitializing(true);
+    try {
+      await cactusIndex.init();
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsInitializing(false);
+    }
+  }, [cactusIndex, isInitializing]);
+  const add = useCallback(async ({
+    ids,
+    documents,
+    metadatas,
+    embeddings
+  }) => {
+    if (isProcessing) {
+      const message = 'CactusIndex is already processing';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsProcessing(true);
+    try {
+      await cactusIndex.add({
+        ids,
+        documents,
+        metadatas,
+        embeddings
+      });
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsProcessing(false);
+    }
+  }, [cactusIndex, isProcessing]);
+  const _delete = useCallback(async ({
+    ids
+  }) => {
+    if (isProcessing) {
+      const message = 'CactusIndex is already processing';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsProcessing(true);
+    try {
+      await cactusIndex.delete({
+        ids
+      });
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsProcessing(false);
+    }
+  }, [cactusIndex, isProcessing]);
+  const get = useCallback(async ({
+    ids
+  }) => {
+    if (isProcessing) {
+      const message = 'CactusIndex is already processing';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsProcessing(true);
+    try {
+      return await cactusIndex.get({
+        ids
+      });
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsProcessing(false);
+    }
+  }, [cactusIndex, isProcessing]);
+  const query = useCallback(async ({
+    embeddings,
+    options
+  }) => {
+    if (isProcessing) {
+      const message = 'CactusIndex is already processing';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsProcessing(true);
+    try {
+      return await cactusIndex.query({
+        embeddings,
+        options
+      });
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsProcessing(false);
+    }
+  }, [cactusIndex, isProcessing]);
+  const compact = useCallback(async () => {
+    if (isProcessing) {
+      const message = 'CactusIndex is already processing';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsProcessing(true);
+    try {
+      await cactusIndex.compact();
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsProcessing(false);
+    }
+  }, [cactusIndex, isProcessing]);
+  const destroy = useCallback(async () => {
+    setError(null);
+    try {
+      await cactusIndex.destroy();
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    }
+  }, [cactusIndex]);
+  return {
+    isInitializing,
+    isProcessing,
+    error,
+    init,
+    add,
+    delete: _delete,
+    get,
+    query,
+    compact,
+    destroy
+  };
+};
+//# sourceMappingURL=useCactusIndex.js.map

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

@@ -0,0 +1 @@
+{"version":3,"names":["useCallback","useEffect","useState","CactusIndex","getErrorMessage","useCactusIndex","name","embeddingDim","cactusIndex","setCactusIndex","isInitializing","setIsInitializing","isProcessing","setIsProcessing","error","setError","destroy","catch","init","message","Error","e","add","ids","documents","metadatas","embeddings","_delete","delete","get","query","options","compact"],"sourceRoot":"../../../src","sources":["hooks/useCactusIndex.ts"],"mappings":";;AAAA,SAASA,WAAW,EAAEC,SAAS,EAAEC,QAAQ,QAAQ,OAAO;AACxD,SAASC,WAAW,QAAQ,2BAAwB;AACpD,SAASC,eAAe,QAAQ,mBAAgB;AAWhD,OAAO,MAAMC,cAAc,GAAGA,CAAC;EAAEC,IAAI;EAAEC;AAAgC,CAAC,KAAK;EAC3E,MAAM,CAACC,WAAW,EAAEC,cAAc,CAAC,GAAGP,QAAQ,CAC5C,MAAM,IAAIC,WAAW,CAACG,IAAI,EAAEC,YAAY,CAC1C,CAAC;;EAED;EACA,MAAM,CAACG,cAAc,EAAEC,iBAAiB,CAAC,GAAGT,QAAQ,CAAC,KAAK,CAAC;EAC3D,MAAM,CAACU,YAAY,EAAEC,eAAe,CAAC,GAAGX,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACY,KAAK,EAAEC,QAAQ,CAAC,GAAGb,QAAQ,CAAgB,IAAI,CAAC;EAEvDD,SAAS,CAAC,MAAM;IACdQ,cAAc,CAAC,IAAIN,WAAW,CAACG,IAAI,EAAEC,YAAY,CAAC,CAAC;IAEnDI,iBAAiB,CAAC,KAAK,CAAC;IACxBE,eAAe,CAAC,KAAK,CAAC;IACtBE,QAAQ,CAAC,IAAI,CAAC;EAChB,CAAC,EAAE,CAACT,IAAI,EAAEC,YAAY,CAAC,CAAC;EAExBN,SAAS,CAAC,MAAM;IACd,OAAO,MAAM;MACXO,WAAW,CAACQ,OAAO,CAAC,CAAC,CAACC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IACvC,CAAC;EACH,CAAC,EAAE,CAACT,WAAW,CAAC,CAAC;EAEjB,MAAMU,IAAI,GAAGlB,WAAW,CAAC,YAAY;IACnC,IAAIU,cAAc,EAAE;MAClB,MAAMS,OAAO,GAAG,qCAAqC;MACrDJ,QAAQ,CAACI,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAJ,QAAQ,CAAC,IAAI,CAAC;IACdJ,iBAAiB,CAAC,IAAI,CAAC;IACvB,IAAI;MACF,MAAMH,WAAW,CAACU,IAAI,CAAC,CAAC;IAC1B,CAAC,CAAC,OAAOG,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRV,iBAAiB,CAAC,KAAK,CAAC;IAC1B;EACF,CAAC,EAAE,CAACH,WAAW,EAAEE,cAAc,CAAC,CAAC;EAEjC,MAAMY,GAAG,GAAGtB,WAAW,CACrB,OAAO;IACLuB,GAAG;IACHC,SAAS;IACTC,SAAS;IACTC;EACoB,CAAC,KAAoB;IACzC,IAAId,YAAY,EAAE;MAChB,MAAMO,OAAO,GAAG,mCAAmC;MACnDJ,QAAQ,CAACI,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAJ,QAAQ,CAAC,IAAI,CAAC;IACdF,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,MAAML,WAAW,CAACc,GAAG,CAAC;QAAEC,GAAG;QAAEC,SAAS;QAAEC,SAAS;QAAEC;MAAW,CAAC,CAAC;IAClE,CAAC,CAAC,OAAOL,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRR,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,WAAW,EAAEI,YAAY,CAC5B,CAAC;EAED,MAAMe,OAAO,GAAG3B,WAAW,CACzB,OAAO;IAAEuB;EAA6B,CAAC,KAAoB;IACzD,IAAIX,YAAY,EAAE;MAChB,MAAMO,OAAO,GAAG,mCAAmC;MACnDJ,QAAQ,CAACI,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAJ,QAAQ,CAAC,IAAI,CAAC;IACdF,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,MAAML,WAAW,CAACoB,MAAM,CAAC;QAAEL;MAAI,CAAC,CAAC;IACnC,CAAC,CAAC,OAAOF,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRR,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,WAAW,EAAEI,YAAY,CAC5B,CAAC;EAED,MAAMiB,GAAG,GAAG7B,WAAW,CACrB,OAAO;IAAEuB;EAA0B,CAAC,KAAoC;IACtE,IAAIX,YAAY,EAAE;MAChB,MAAMO,OAAO,GAAG,mCAAmC;MACnDJ,QAAQ,CAACI,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAJ,QAAQ,CAAC,IAAI,CAAC;IACdF,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,WAAW,CAACqB,GAAG,CAAC;QAAEN;MAAI,CAAC,CAAC;IACvC,CAAC,CAAC,OAAOF,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRR,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,WAAW,EAAEI,YAAY,CAC5B,CAAC;EAED,MAAMkB,KAAK,GAAG9B,WAAW,CACvB,OAAO;IACL0B,UAAU;IACVK;EACsB,CAAC,KAAsC;IAC7D,IAAInB,YAAY,EAAE;MAChB,MAAMO,OAAO,GAAG,mCAAmC;MACnDJ,QAAQ,CAACI,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAJ,QAAQ,CAAC,IAAI,CAAC;IACdF,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,WAAW,CAACsB,KAAK,CAAC;QAAEJ,UAAU;QAAEK;MAAQ,CAAC,CAAC;IACzD,CAAC,CAAC,OAAOV,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRR,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,WAAW,EAAEI,YAAY,CAC5B,CAAC;EAED,MAAMoB,OAAO,GAAGhC,WAAW,CAAC,YAAY;IACtC,IAAIY,YAAY,EAAE;MAChB,MAAMO,OAAO,GAAG,mCAAmC;MACnDJ,QAAQ,CAACI,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAJ,QAAQ,CAAC,IAAI,CAAC;IACdF,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,MAAML,WAAW,CAACwB,OAAO,CAAC,CAAC;IAC7B,CAAC,CAAC,OAAOX,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRR,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EAAE,CAACL,WAAW,EAAEI,YAAY,CAAC,CAAC;EAE/B,MAAMI,OAAO,GAAGhB,WAAW,CAAC,YAAY;IACtCe,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMP,WAAW,CAACQ,OAAO,CAAC,CAAC;IAC7B,CAAC,CAAC,OAAOK,CAAC,EAAE;MACVN,QAAQ,CAACX,eAAe,CAACiB,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT;EACF,CAAC,EAAE,CAACb,WAAW,CAAC,CAAC;EAEjB,OAAO;IACLE,cAAc;IACdE,YAAY;IACZE,KAAK;IAELI,IAAI;IACJI,GAAG;IACHM,MAAM,EAAED,OAAO;IACfE,GAAG;IACHC,KAAK;IACLE,OAAO;IACPhB;EACF,CAAC;AACH,CAAC","ignoreList":[]}

package/lib/module/hooks/useCactusLM.js

@@ -5,14 +5,19 @@ import { CactusLM } from "../classes/CactusLM.js";
 import { CactusFileSystem } from "../native/index.js";
 import { getErrorMessage } from "../utils/error.js";
 export const useCactusLM = ({
-  model = 'qwen3-0.6',
+  model = 'qwen3-0.6b',
   contextSize = 2048,
-  corpusDir = undefined
+  corpusDir = undefined,
+  options: modelOptions = {
+    quantization: undefined,
+    pro: false
+  }
 } = {}) => {
   const [cactusLM, setCactusLM] = useState(() => new CactusLM({
     model,
     contextSize,
-    corpusDir
+    corpusDir,
+    options: modelOptions
   }));
 
   // State
@@ -32,7 +37,11 @@ export const useCactusLM = ({
     setCactusLM(new CactusLM({
       model,
       contextSize,
-      corpusDir
+      corpusDir,
+      options: {
+        quantization: modelOptions.quantization,
+        pro: modelOptions.pro
+      }
     }));
     setCompletion('');
     setIsGenerating(false);
@@ -57,7 +66,7 @@ export const useCactusLM = ({
     return () => {
       mounted = false;
     };
-  }, [model, contextSize, corpusDir]);
+  }, [model, contextSize, corpusDir, modelOptions.quantization, modelOptions.pro]);
   useEffect(() => {
     return () => {
       cactusLM.destroy().catch(() => {});
@@ -157,7 +166,7 @@ export const useCactusLM = ({
       setIsGenerating(false);
     }
   }, [cactusLM, isGenerating]);
-  const embed = useCallback(async ({
+  const tokenize = useCallback(async ({
     text
   }) => {
     if (isGenerating) {
@@ -168,7 +177,7 @@ export const useCactusLM = ({
     setError(null);
     setIsGenerating(true);
     try {
-      return await cactusLM.embed({
+      return await cactusLM.tokenize({
         text
       });
     } catch (e) {
@@ -178,6 +187,56 @@ export const useCactusLM = ({
       setIsGenerating(false);
     }
   }, [cactusLM, isGenerating]);
+  const scoreWindow = useCallback(async ({
+    tokens,
+    start,
+    end,
+    context
+  }) => {
+    if (isGenerating) {
+      const message = 'CactusLM is already generating';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsGenerating(true);
+    try {
+      return await cactusLM.scoreWindow({
+        tokens,
+        start,
+        end,
+        context
+      });
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsGenerating(false);
+    }
+  }, [cactusLM, isGenerating]);
+  const embed = useCallback(async ({
+    text,
+    normalize = false
+  }) => {
+    if (isGenerating) {
+      const message = 'CactusLM is already generating';
+      setError(message);
+      throw new Error(message);
+    }
+    setError(null);
+    setIsGenerating(true);
+    try {
+      return await cactusLM.embed({
+        text,
+        normalize
+      });
+    } catch (e) {
+      setError(getErrorMessage(e));
+      throw e;
+    } finally {
+      setIsGenerating(false);
+    }
+  }, [cactusLM, isGenerating]);
   const imageEmbed = useCallback(async ({
     imagePath
   }) => {
@@ -250,6 +309,8 @@ export const useCactusLM = ({
     download,
     init,
     complete,
+    tokenize,
+    scoreWindow,
     embed,
     imageEmbed,
     reset,

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

@@ -1 +1 @@
-{"version":3,"names":["useCallback","useEffect","useState","useRef","CactusLM","CactusFileSystem","getErrorMessage","useCactusLM","model","contextSize","corpusDir","undefined","cactusLM","setCactusLM","completion","setCompletion","isGenerating","setIsGenerating","isInitializing","setIsInitializing","isDownloaded","setIsDownloaded","isDownloading","setIsDownloading","downloadProgress","setDownloadProgress","error","setError","currentModelRef","currentDownloadIdRef","current","mounted","modelExists","then","exists","catch","e","destroy","download","onProgress","message","Error","thisModel","thisDownloadId","progress","init","complete","messages","options","tools","onToken","mode","token","prev","embed","text","imageEmbed","imagePath","stop","reset","getModels"],"sourceRoot":"../../../src","sources":["hooks/useCactusLM.ts"],"mappings":";;AAAA,SAASA,WAAW,EAAEC,SAAS,EAAEC,QAAQ,EAAEC,MAAM,QAAQ,OAAO;AAChE,SAASC,QAAQ,QAAQ,wBAAqB;AAC9C,SAASC,gBAAgB,QAAQ,oBAAW;AAC5C,SAASC,eAAe,QAAQ,mBAAgB;AAahD,OAAO,MAAMC,WAAW,GAAGA,CAAC;EAC1BC,KAAK,GAAG,WAAW;EACnBC,WAAW,GAAG,IAAI;EAClBC,SAAS,GAAGC;AACE,CAAC,GAAG,CAAC,CAAC,KAAK;EACzB,MAAM,CAACC,QAAQ,EAAEC,WAAW,CAAC,GAAGX,QAAQ,CACtC,MAAM,IAAIE,QAAQ,CAAC;IAAEI,KAAK;IAAEC,WAAW;IAAEC;EAAU,CAAC,CACtD,CAAC;;EAED;EACA,MAAM,CAACI,UAAU,EAAEC,aAAa,CAAC,GAAGb,QAAQ,CAAC,EAAE,CAAC;EAChD,MAAM,CAACc,YAAY,EAAEC,eAAe,CAAC,GAAGf,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACgB,cAAc,EAAEC,iBAAiB,CAAC,GAAGjB,QAAQ,CAAC,KAAK,CAAC;EAC3D,MAAM,CAACkB,YAAY,EAAEC,eAAe,CAAC,GAAGnB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACoB,aAAa,EAAEC,gBAAgB,CAAC,GAAGrB,QAAQ,CAAC,KAAK,CAAC;EACzD,MAAM,CAACsB,gBAAgB,EAAEC,mBAAmB,CAAC,GAAGvB,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAM,CAACwB,KAAK,EAAEC,QAAQ,CAAC,GAAGzB,QAAQ,CAAgB,IAAI,CAAC;EAEvD,MAAM0B,eAAe,GAAGzB,MAAM,CAACK,KAAK,CAAC;EACrC,MAAMqB,oBAAoB,GAAG1B,MAAM,CAAC,CAAC,CAAC;EAEtCF,SAAS,CAAC,MAAM;IACd2B,eAAe,CAACE,OAAO,GAAGtB,KAAK;EACjC,CAAC,EAAE,CAACA,KAAK,CAAC,CAAC;EAEXP,SAAS,CAAC,MAAM;IACdY,WAAW,CAAC,IAAIT,QAAQ,CAAC;MAAEI,KAAK;MAAEC,WAAW;MAAEC;IAAU,CAAC,CAAC,CAAC;IAE5DK,aAAa,CAAC,EAAE,CAAC;IACjBE,eAAe,CAAC,KAAK,CAAC;IACtBE,iBAAiB,CAAC,KAAK,CAAC;IACxBE,eAAe,CAAC,KAAK,CAAC;IACtBE,gBAAgB,CAAC,KAAK,CAAC;IACvBE,mBAAmB,CAAC,CAAC,CAAC;IACtBE,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAII,OAAO,GAAG,IAAI;IAClB1B,gBAAgB,CAAC2B,WAAW,CAACxB,KAAK,CAAC,CAChCyB,IAAI,CAAEC,MAAM,IAAK;MAChB,IAAI,CAACH,OAAO,EAAE;QACZ;MACF;MACAV,eAAe,CAACa,MAAM,CAAC;IACzB,CAAC,CAAC,CACDC,KAAK,CAAEC,CAAC,IAAK;MACZ,IAAI,CAACL,OAAO,EAAE;QACZ;MACF;MACAV,eAAe,CAAC,KAAK,CAAC;MACtBM,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;IAC9B,CAAC,CAAC;IAEJ,OAAO,MAAM;MACXL,OAAO,GAAG,KAAK;IACjB,CAAC;EACH,CAAC,EAAE,CAACvB,KAAK,EAAEC,WAAW,EAAEC,SAAS,CAAC,CAAC;EAEnCT,SAAS,CAAC,MAAM;IACd,OAAO,MAAM;MACXW,QAAQ,CAACyB,OAAO,CAAC,CAAC,CAACF,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IACpC,CAAC;EACH,CAAC,EAAE,CAACvB,QAAQ,CAAC,CAAC;EAEd,MAAM0B,QAAQ,GAAGtC,WAAW,CAC1B,OAAO;IAAEuC;EAAmC,CAAC,GAAG,CAAC,CAAC,KAAK;IACrD,IAAIjB,aAAa,EAAE;MACjB,MAAMkB,OAAO,GAAG,iCAAiC;MACjDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAIP,YAAY,EAAE;MAChB;IACF;IAEA,MAAMsB,SAAS,GAAGd,eAAe,CAACE,OAAO;IACzC,MAAMa,cAAc,GAAG,EAAEd,oBAAoB,CAACC,OAAO;IAErDL,mBAAmB,CAAC,CAAC,CAAC;IACtBF,gBAAgB,CAAC,IAAI,CAAC;IACtB,IAAI;MACF,MAAMX,QAAQ,CAAC0B,QAAQ,CAAC;QACtBC,UAAU,EAAGK,QAAQ,IAAK;UACxB,IACEhB,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;YACA;UACF;UAEAlB,mBAAmB,CAACmB,QAAQ,CAAC;UAC7BL,UAAU,GAAGK,QAAQ,CAAC;QACxB;MACF,CAAC,CAAC;MAEF,IACEhB,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;QACA;MACF;MAEAtB,eAAe,CAAC,IAAI,CAAC;IACvB,CAAC,CAAC,OAAOe,CAAC,EAAE;MACV,IACER,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;QACA;MACF;MAEAhB,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACR,IACER,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;QACA;MACF;MAEApB,gBAAgB,CAAC,KAAK,CAAC;MACvBE,mBAAmB,CAAC,CAAC,CAAC;IACxB;EACF,CAAC,EACD,CAACb,QAAQ,EAAEU,aAAa,EAAEF,YAAY,CACxC,CAAC;EAED,MAAMyB,IAAI,GAAG7C,WAAW,CAAC,YAAY;IACnC,IAAIkB,cAAc,EAAE;MAClB,MAAMsB,OAAO,GAAG,kCAAkC;MAClDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdR,iBAAiB,CAAC,IAAI,CAAC;IACvB,IAAI;MACF,MAAMP,QAAQ,CAACiC,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,OAAOT,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRjB,iBAAiB,CAAC,KAAK,CAAC;IAC1B;EACF,CAAC,EAAE,CAACP,QAAQ,EAAEM,cAAc,CAAC,CAAC;EAE9B,MAAM4B,QAAQ,GAAG9C,WAAW,CAC1B,OAAO;IACL+C,QAAQ;IACRC,OAAO;IACPC,KAAK;IACLC,OAAO;IACPC;EACsB,CAAC,KAAsC;IAC7D,IAAInC,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdZ,aAAa,CAAC,EAAE,CAAC;IACjBE,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAACkC,QAAQ,CAAC;QAC7BC,QAAQ;QACRC,OAAO;QACPC,KAAK;QACLC,OAAO,EAAGE,KAAK,IAAK;UAClBrC,aAAa,CAAEsC,IAAI,IAAKA,IAAI,GAAGD,KAAK,CAAC;UACrCF,OAAO,GAAGE,KAAK,CAAC;QAClB,CAAC;QACDD;MACF,CAAC,CAAC;IACJ,CAAC,CAAC,OAAOf,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAMsC,KAAK,GAAGtD,WAAW,CACvB,OAAO;IAAEuD;EAA0B,CAAC,KAAmC;IACrE,IAAIvC,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdV,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAAC0C,KAAK,CAAC;QAAEC;MAAK,CAAC,CAAC;IACvC,CAAC,CAAC,OAAOnB,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAMwC,UAAU,GAAGxD,WAAW,CAC5B,OAAO;IACLyD;EACwB,CAAC,KAAwC;IACjE,IAAIzC,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdV,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAAC4C,UAAU,CAAC;QAAEC;MAAU,CAAC,CAAC;IACjD,CAAC,CAAC,OAAOrB,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAM0C,IAAI,GAAG1D,WAAW,CAAC,YAAY;IACnC2B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMf,QAAQ,CAAC8C,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,OAAOtB,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT;EACF,CAAC,EAAE,CAACxB,QAAQ,CAAC,CAAC;EAEd,MAAM+C,KAAK,GAAG3D,WAAW,CAAC,YAAY;IACpC2B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMf,QAAQ,CAAC+C,KAAK,CAAC,CAAC;IACxB,CAAC,CAAC,OAAOvB,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRrB,aAAa,CAAC,EAAE,CAAC;IACnB;EACF,CAAC,EAAE,CAACH,QAAQ,CAAC,CAAC;EAEd,MAAMyB,OAAO,GAAGrC,WAAW,CAAC,YAAY;IACtC2B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMf,QAAQ,CAACyB,OAAO,CAAC,CAAC;IAC1B,CAAC,CAAC,OAAOD,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRrB,aAAa,CAAC,EAAE,CAAC;IACnB;EACF,CAAC,EAAE,CAACH,QAAQ,CAAC,CAAC;EAEd,MAAMgD,SAAS,GAAG5D,WAAW,CAAC,YAAoC;IAChE2B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,OAAO,MAAMf,QAAQ,CAACgD,SAAS,CAAC,CAAC;IACnC,CAAC,CAAC,OAAOxB,CAAC,EAAE;MACVT,QAAQ,CAACrB,eAAe,CAAC8B,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT;EACF,CAAC,EAAE,CAACxB,QAAQ,CAAC,CAAC;EAEd,OAAO;IACLE,UAAU;IACVE,YAAY;IACZE,cAAc;IACdE,YAAY;IACZE,aAAa;IACbE,gBAAgB;IAChBE,KAAK;IAELY,QAAQ;IACRO,IAAI;IACJC,QAAQ;IACRQ,KAAK;IACLE,UAAU;IACVG,KAAK;IACLD,IAAI;IACJrB,OAAO;IACPuB;EACF,CAAC;AACH,CAAC","ignoreList":[]}
+{"version":3,"names":["useCallback","useEffect","useState","useRef","CactusLM","CactusFileSystem","getErrorMessage","useCactusLM","model","contextSize","corpusDir","undefined","options","modelOptions","quantization","pro","cactusLM","setCactusLM","completion","setCompletion","isGenerating","setIsGenerating","isInitializing","setIsInitializing","isDownloaded","setIsDownloaded","isDownloading","setIsDownloading","downloadProgress","setDownloadProgress","error","setError","currentModelRef","currentDownloadIdRef","current","mounted","modelExists","then","exists","catch","e","destroy","download","onProgress","message","Error","thisModel","thisDownloadId","progress","init","complete","messages","tools","onToken","mode","token","prev","tokenize","text","scoreWindow","tokens","start","end","context","embed","normalize","imageEmbed","imagePath","stop","reset","getModels"],"sourceRoot":"../../../src","sources":["hooks/useCactusLM.ts"],"mappings":";;AAAA,SAASA,WAAW,EAAEC,SAAS,EAAEC,QAAQ,EAAEC,MAAM,QAAQ,OAAO;AAChE,SAASC,QAAQ,QAAQ,wBAAqB;AAC9C,SAASC,gBAAgB,QAAQ,oBAAW;AAC5C,SAASC,eAAe,QAAQ,mBAAgB;AAiBhD,OAAO,MAAMC,WAAW,GAAGA,CAAC;EAC1BC,KAAK,GAAG,YAAY;EACpBC,WAAW,GAAG,IAAI;EAClBC,SAAS,GAAGC,SAAS;EACrBC,OAAO,EAAEC,YAAY,GAAG;IACtBC,YAAY,EAAEH,SAAS;IACvBI,GAAG,EAAE;EACP;AACc,CAAC,GAAG,CAAC,CAAC,KAAK;EACzB,MAAM,CAACC,QAAQ,EAAEC,WAAW,CAAC,GAAGf,QAAQ,CACtC,MAAM,IAAIE,QAAQ,CAAC;IAAEI,KAAK;IAAEC,WAAW;IAAEC,SAAS;IAAEE,OAAO,EAAEC;EAAa,CAAC,CAC7E,CAAC;;EAED;EACA,MAAM,CAACK,UAAU,EAAEC,aAAa,CAAC,GAAGjB,QAAQ,CAAC,EAAE,CAAC;EAChD,MAAM,CAACkB,YAAY,EAAEC,eAAe,CAAC,GAAGnB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACoB,cAAc,EAAEC,iBAAiB,CAAC,GAAGrB,QAAQ,CAAC,KAAK,CAAC;EAC3D,MAAM,CAACsB,YAAY,EAAEC,eAAe,CAAC,GAAGvB,QAAQ,CAAC,KAAK,CAAC;EACvD,MAAM,CAACwB,aAAa,EAAEC,gBAAgB,CAAC,GAAGzB,QAAQ,CAAC,KAAK,CAAC;EACzD,MAAM,CAAC0B,gBAAgB,EAAEC,mBAAmB,CAAC,GAAG3B,QAAQ,CAAC,CAAC,CAAC;EAC3D,MAAM,CAAC4B,KAAK,EAAEC,QAAQ,CAAC,GAAG7B,QAAQ,CAAgB,IAAI,CAAC;EAEvD,MAAM8B,eAAe,GAAG7B,MAAM,CAACK,KAAK,CAAC;EACrC,MAAMyB,oBAAoB,GAAG9B,MAAM,CAAC,CAAC,CAAC;EAEtCF,SAAS,CAAC,MAAM;IACd+B,eAAe,CAACE,OAAO,GAAG1B,KAAK;EACjC,CAAC,EAAE,CAACA,KAAK,CAAC,CAAC;EAEXP,SAAS,CAAC,MAAM;IACdgB,WAAW,CACT,IAAIb,QAAQ,CAAC;MACXI,KAAK;MACLC,WAAW;MACXC,SAAS;MACTE,OAAO,EAAE;QACPE,YAAY,EAAED,YAAY,CAACC,YAAY;QACvCC,GAAG,EAAEF,YAAY,CAACE;MACpB;IACF,CAAC,CACH,CAAC;IAEDI,aAAa,CAAC,EAAE,CAAC;IACjBE,eAAe,CAAC,KAAK,CAAC;IACtBE,iBAAiB,CAAC,KAAK,CAAC;IACxBE,eAAe,CAAC,KAAK,CAAC;IACtBE,gBAAgB,CAAC,KAAK,CAAC;IACvBE,mBAAmB,CAAC,CAAC,CAAC;IACtBE,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAII,OAAO,GAAG,IAAI;IAClB9B,gBAAgB,CAAC+B,WAAW,CAAC5B,KAAK,CAAC,CAChC6B,IAAI,CAAEC,MAAM,IAAK;MAChB,IAAI,CAACH,OAAO,EAAE;QACZ;MACF;MACAV,eAAe,CAACa,MAAM,CAAC;IACzB,CAAC,CAAC,CACDC,KAAK,CAAEC,CAAC,IAAK;MACZ,IAAI,CAACL,OAAO,EAAE;QACZ;MACF;MACAV,eAAe,CAAC,KAAK,CAAC;MACtBM,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;IAC9B,CAAC,CAAC;IAEJ,OAAO,MAAM;MACXL,OAAO,GAAG,KAAK;IACjB,CAAC;EACH,CAAC,EAAE,CACD3B,KAAK,EACLC,WAAW,EACXC,SAAS,EACTG,YAAY,CAACC,YAAY,EACzBD,YAAY,CAACE,GAAG,CACjB,CAAC;EAEFd,SAAS,CAAC,MAAM;IACd,OAAO,MAAM;MACXe,QAAQ,CAACyB,OAAO,CAAC,CAAC,CAACF,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IACpC,CAAC;EACH,CAAC,EAAE,CAACvB,QAAQ,CAAC,CAAC;EAEd,MAAM0B,QAAQ,GAAG1C,WAAW,CAC1B,OAAO;IAAE2C;EAAmC,CAAC,GAAG,CAAC,CAAC,KAAK;IACrD,IAAIjB,aAAa,EAAE;MACjB,MAAMkB,OAAO,GAAG,iCAAiC;MACjDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IAEd,IAAIP,YAAY,EAAE;MAChB;IACF;IAEA,MAAMsB,SAAS,GAAGd,eAAe,CAACE,OAAO;IACzC,MAAMa,cAAc,GAAG,EAAEd,oBAAoB,CAACC,OAAO;IAErDL,mBAAmB,CAAC,CAAC,CAAC;IACtBF,gBAAgB,CAAC,IAAI,CAAC;IACtB,IAAI;MACF,MAAMX,QAAQ,CAAC0B,QAAQ,CAAC;QACtBC,UAAU,EAAGK,QAAQ,IAAK;UACxB,IACEhB,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;YACA;UACF;UAEAlB,mBAAmB,CAACmB,QAAQ,CAAC;UAC7BL,UAAU,GAAGK,QAAQ,CAAC;QACxB;MACF,CAAC,CAAC;MAEF,IACEhB,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;QACA;MACF;MAEAtB,eAAe,CAAC,IAAI,CAAC;IACvB,CAAC,CAAC,OAAOe,CAAC,EAAE;MACV,IACER,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;QACA;MACF;MAEAhB,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACR,IACER,eAAe,CAACE,OAAO,KAAKY,SAAS,IACrCb,oBAAoB,CAACC,OAAO,KAAKa,cAAc,EAC/C;QACA;MACF;MAEApB,gBAAgB,CAAC,KAAK,CAAC;MACvBE,mBAAmB,CAAC,CAAC,CAAC;IACxB;EACF,CAAC,EACD,CAACb,QAAQ,EAAEU,aAAa,EAAEF,YAAY,CACxC,CAAC;EAED,MAAMyB,IAAI,GAAGjD,WAAW,CAAC,YAAY;IACnC,IAAIsB,cAAc,EAAE;MAClB,MAAMsB,OAAO,GAAG,kCAAkC;MAClDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdR,iBAAiB,CAAC,IAAI,CAAC;IACvB,IAAI;MACF,MAAMP,QAAQ,CAACiC,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,OAAOT,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRjB,iBAAiB,CAAC,KAAK,CAAC;IAC1B;EACF,CAAC,EAAE,CAACP,QAAQ,EAAEM,cAAc,CAAC,CAAC;EAE9B,MAAM4B,QAAQ,GAAGlD,WAAW,CAC1B,OAAO;IACLmD,QAAQ;IACRvC,OAAO;IACPwC,KAAK;IACLC,OAAO;IACPC;EACsB,CAAC,KAAsC;IAC7D,IAAIlC,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdZ,aAAa,CAAC,EAAE,CAAC;IACjBE,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAACkC,QAAQ,CAAC;QAC7BC,QAAQ;QACRvC,OAAO;QACPwC,KAAK;QACLC,OAAO,EAAGE,KAAK,IAAK;UAClBpC,aAAa,CAAEqC,IAAI,IAAKA,IAAI,GAAGD,KAAK,CAAC;UACrCF,OAAO,GAAGE,KAAK,CAAC;QAClB,CAAC;QACDD;MACF,CAAC,CAAC;IACJ,CAAC,CAAC,OAAOd,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAMqC,QAAQ,GAAGzD,WAAW,CAC1B,OAAO;IACL0D;EACsB,CAAC,KAAsC;IAC7D,IAAItC,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdV,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAACyC,QAAQ,CAAC;QAAEC;MAAK,CAAC,CAAC;IAC1C,CAAC,CAAC,OAAOlB,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAMuC,WAAW,GAAG3D,WAAW,CAC7B,OAAO;IACL4D,MAAM;IACNC,KAAK;IACLC,GAAG;IACHC;EACyB,CAAC,KAAyC;IACnE,IAAI3C,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdV,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAAC2C,WAAW,CAAC;QAAEC,MAAM;QAAEC,KAAK;QAAEC,GAAG;QAAEC;MAAQ,CAAC,CAAC;IACpE,CAAC,CAAC,OAAOvB,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAM4C,KAAK,GAAGhE,WAAW,CACvB,OAAO;IACL0D,IAAI;IACJO,SAAS,GAAG;EACO,CAAC,KAAmC;IACvD,IAAI7C,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdV,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAACgD,KAAK,CAAC;QAAEN,IAAI;QAAEO;MAAU,CAAC,CAAC;IAClD,CAAC,CAAC,OAAOzB,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAM8C,UAAU,GAAGlE,WAAW,CAC5B,OAAO;IACLmE;EACwB,CAAC,KAAwC;IACjE,IAAI/C,YAAY,EAAE;MAChB,MAAMwB,OAAO,GAAG,gCAAgC;MAChDb,QAAQ,CAACa,OAAO,CAAC;MACjB,MAAM,IAAIC,KAAK,CAACD,OAAO,CAAC;IAC1B;IAEAb,QAAQ,CAAC,IAAI,CAAC;IACdV,eAAe,CAAC,IAAI,CAAC;IACrB,IAAI;MACF,OAAO,MAAML,QAAQ,CAACkD,UAAU,CAAC;QAAEC;MAAU,CAAC,CAAC;IACjD,CAAC,CAAC,OAAO3B,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRnB,eAAe,CAAC,KAAK,CAAC;IACxB;EACF,CAAC,EACD,CAACL,QAAQ,EAAEI,YAAY,CACzB,CAAC;EAED,MAAMgD,IAAI,GAAGpE,WAAW,CAAC,YAAY;IACnC+B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMf,QAAQ,CAACoD,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,OAAO5B,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT;EACF,CAAC,EAAE,CAACxB,QAAQ,CAAC,CAAC;EAEd,MAAMqD,KAAK,GAAGrE,WAAW,CAAC,YAAY;IACpC+B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMf,QAAQ,CAACqD,KAAK,CAAC,CAAC;IACxB,CAAC,CAAC,OAAO7B,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRrB,aAAa,CAAC,EAAE,CAAC;IACnB;EACF,CAAC,EAAE,CAACH,QAAQ,CAAC,CAAC;EAEd,MAAMyB,OAAO,GAAGzC,WAAW,CAAC,YAAY;IACtC+B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,MAAMf,QAAQ,CAACyB,OAAO,CAAC,CAAC;IAC1B,CAAC,CAAC,OAAOD,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT,CAAC,SAAS;MACRrB,aAAa,CAAC,EAAE,CAAC;IACnB;EACF,CAAC,EAAE,CAACH,QAAQ,CAAC,CAAC;EAEd,MAAMsD,SAAS,GAAGtE,WAAW,CAAC,YAAoC;IAChE+B,QAAQ,CAAC,IAAI,CAAC;IACd,IAAI;MACF,OAAO,MAAMf,QAAQ,CAACsD,SAAS,CAAC,CAAC;IACnC,CAAC,CAAC,OAAO9B,CAAC,EAAE;MACVT,QAAQ,CAACzB,eAAe,CAACkC,CAAC,CAAC,CAAC;MAC5B,MAAMA,CAAC;IACT;EACF,CAAC,EAAE,CAACxB,QAAQ,CAAC,CAAC;EAEd,OAAO;IACLE,UAAU;IACVE,YAAY;IACZE,cAAc;IACdE,YAAY;IACZE,aAAa;IACbE,gBAAgB;IAChBE,KAAK;IAELY,QAAQ;IACRO,IAAI;IACJC,QAAQ;IACRO,QAAQ;IACRE,WAAW;IACXK,KAAK;IACLE,UAAU;IACVG,KAAK;IACLD,IAAI;IACJ3B,OAAO;IACP6B;EACF,CAAC;AACH,CAAC","ignoreList":[]}