cactus-react-native 1.5.0 → 1.7.0

package/ios/cactus.xcframework/ios-arm64-simulator/cactus.framework/Headers/kernel_utils.h

@@ -4,6 +4,7 @@
 #include <arm_neon.h>
 #if defined(__APPLE__)
 #include <TargetConditionals.h>
+#include <sys/sysctl.h>
 #endif
 #if defined(__ANDROID__)
 #include <sys/auxv.h>
@@ -43,59 +44,6 @@ inline void stream_store_f16x8(__fp16* dst, float16x8_t val) {
 #endif
 }
 
-#if defined(__ARM_FEATURE_DOTPROD)
-inline int32x4_t accum_dot(int32x4_t acc, int8x16_t a, int8x16_t b) {
-    return vdotq_s32(acc, a, b);
-}
-#else
-inline int32x4_t accum_dot(int32x4_t acc, int8x16_t a, int8x16_t b) {
-    int16x8_t prod_low = vmull_s8(vget_low_s8(a), vget_low_s8(b));
-    int32x4_t acc_high = vpaddlq_s16(vmull_s8(vget_high_s8(a), vget_high_s8(b)));
-    return vaddq_s32(vaddq_s32(acc, vpaddlq_s16(prod_low)), acc_high);
-}
-#endif
-
-// I8MM support: runtime detection on Android, compile-time on Apple
-#if defined(__ANDROID__) && defined(__aarch64__)
-
-inline bool cactus_has_i8mm() {
-    static int8_t supported = -1;
-    if (supported == -1) {
-        unsigned long hwcaps = getauxval(AT_HWCAP2);
-        supported = (hwcaps & HWCAP2_I8MM) ? 1 : 0;
-    }
-    return supported;
-}
-
-__attribute__((target("arch=armv8.2-a+i8mm")))
-inline int32x4_t accum_matmul(int32x4_t acc, int8x16_t a, int8x16_t b) {
-    return vmmlaq_s32(acc, a, b);
-}
-
-#elif defined(__APPLE__) && defined(__aarch64__)
-
-inline bool cactus_has_i8mm() {
-    return true;
-}
-
-__attribute__((target("i8mm")))
-inline int32x4_t accum_matmul(int32x4_t acc, int8x16_t a, int8x16_t b) {
-    return vmmlaq_s32(acc, a, b);
-}
-
-#else
-
-inline bool cactus_has_i8mm() {
-    return false;
-}
-
-#endif
-
-inline float16x8_t accum_f16_dot(float16x8_t acc, float16x8_t a_low, float16x8_t a_high,
-                                 float16x8_t b_low, float16x8_t b_high) {
-    acc = vfmaq_f16(acc, a_low, b_low);
-    return vfmaq_f16(acc, a_high, b_high);
-}
 
 inline float32x4_t fast_exp_f32x4(float32x4_t x) {
     const float32x4_t log2e = vdupq_n_f32(1.4426950408889634f);
@@ -154,102 +102,6 @@ inline float32x4_t fast_tanh_f32x4(float32x4_t x) {
     return result;
 }
 
-inline int8x16_t unpack_int4_lo(uint8x16_t packed) {
-    uint8x16_t lo = vandq_u8(packed, vdupq_n_u8(0x0F));
-    uint8x16_t sign_mask = vcgtq_u8(lo, vdupq_n_u8(7));
-    uint8x16_t correction = vandq_u8(sign_mask, vdupq_n_u8(16));
-    return vreinterpretq_s8_u8(vsubq_u8(lo, correction));
-}
-
-inline int8x16_t unpack_int4_hi(uint8x16_t packed) {
-    uint8x16_t hi = vshrq_n_u8(packed, 4);
-    uint8x16_t sign_mask = vcgtq_u8(hi, vdupq_n_u8(7));
-    uint8x16_t correction = vandq_u8(sign_mask, vdupq_n_u8(16));
-    return vreinterpretq_s8_u8(vsubq_u8(hi, correction));
-}
-
-inline void unpack_int4_to_int8x32(uint8x16_t packed, int8x16_t& out_lo, int8x16_t& out_hi) {
-    int8x16_t lo_nibbles = unpack_int4_lo(packed);
-    int8x16_t hi_nibbles = unpack_int4_hi(packed);
-    int8x16x2_t interleaved = vzipq_s8(lo_nibbles, hi_nibbles);
-    out_lo = interleaved.val[0];
-    out_hi = interleaved.val[1];
-}
-
-inline int32x4_t int4_dot_asm(int32x4_t acc, uint8x16_t packed, int8x16_t a_lo, int8x16_t a_hi) {
-#if defined(__aarch64__)
-    int8x16_t b_lo, b_hi;
-
-    __asm__ __volatile__ (
-        "movi   v16.16b, #0x0F          \n"  // low nibble mask
-        "movi   v17.16b, #7             \n"  // sign threshold
-        "movi   v18.16b, #16            \n"  // sign correction
-
-        "and    %[b_lo].16b, %[packed].16b, v16.16b  \n"
-
-        "ushr   %[b_hi].16b, %[packed].16b, #4      \n"
-
-        "cmgt   v19.16b, %[b_lo].16b, v17.16b       \n"
-        "and    v19.16b, v19.16b, v18.16b           \n"
-        "sub    %[b_lo].16b, %[b_lo].16b, v19.16b   \n"
-
-        "cmgt   v20.16b, %[b_hi].16b, v17.16b       \n"
-        "and    v20.16b, v20.16b, v18.16b           \n"
-        "sub    %[b_hi].16b, %[b_hi].16b, v20.16b   \n"
-
-        "zip1   v21.16b, %[b_lo].16b, %[b_hi].16b   \n"
-        "zip2   v22.16b, %[b_lo].16b, %[b_hi].16b   \n"
-
-        ".arch armv8.2-a+dotprod                    \n"
-        "sdot   %[acc].4s, %[a_lo].16b, v21.16b     \n"
-        "sdot   %[acc].4s, %[a_hi].16b, v22.16b     \n"
-
-        : [acc] "+w"(acc), [b_lo] "=w"(b_lo), [b_hi] "=w"(b_hi)
-        : [packed] "w"(packed), [a_lo] "w"(a_lo), [a_hi] "w"(a_hi)
-        : "v16", "v17", "v18", "v19", "v20", "v21", "v22"
-    );
-
-    return acc;
-#else
-    int8x16_t b_lo, b_hi;
-    unpack_int4_to_int8x32(packed, b_lo, b_hi);
-    acc = accum_dot(acc, a_lo, b_lo);
-    acc = accum_dot(acc, a_hi, b_hi);
-    return acc;
-#endif
-}
-
-inline int32_t int4_dot_m1_asm(const int8_t* a_ptr, const uint8_t* b_packed, size_t group_size) {
-#if defined(__aarch64__)
-    int32x4_t acc = vdupq_n_s32(0);
-
-    for (size_t k = 0; k < group_size; k += 64) {
-        uint8x16_t p0 = vld1q_u8(b_packed + k/2);
-        uint8x16_t p1 = vld1q_u8(b_packed + k/2 + 16);
-
-        int8x16_t a0 = vld1q_s8(a_ptr + k);
-        int8x16_t a1 = vld1q_s8(a_ptr + k + 16);
-        int8x16_t a2 = vld1q_s8(a_ptr + k + 32);
-        int8x16_t a3 = vld1q_s8(a_ptr + k + 48);
-
-        acc = int4_dot_asm(acc, p0, a0, a1);
-        acc = int4_dot_asm(acc, p1, a2, a3);
-    }
-
-    return vaddvq_s32(acc);
-#else
-    int32x4_t acc = vdupq_n_s32(0);
-    for (size_t k = 0; k < group_size; k += 32) {
-        uint8x16_t packed = vld1q_u8(b_packed + k/2);
-        int8x16_t b_lo, b_hi;
-        unpack_int4_to_int8x32(packed, b_lo, b_hi);
-        acc = accum_dot(acc, vld1q_s8(a_ptr + k), b_lo);
-        acc = accum_dot(acc, vld1q_s8(a_ptr + k + 16), b_hi);
-    }
-    return vaddvq_s32(acc);
-#endif
-}
-
 namespace CactusThreading {
 
     class ThreadPool {
@@ -431,18 +283,32 @@ namespace CactusThreading {
     struct GemmThreading {
         #if defined(__ANDROID__)
         static size_t get_num_threads(size_t M, size_t pool_size) {
-            if (M <= 1) return 1; 
-            return pool_size; 
+            if (M <= 1) return 1;
+            return pool_size;
+        }
+        static size_t get_gemv_threads(size_t /*N_blocks*/, size_t /*pool_size*/) {
+            return 1; 
         }
         #elif defined(__APPLE__) && TARGET_OS_IPHONE
+        static constexpr size_t GEMV_MIN_N_BLOCKS = 512; 
         static size_t get_num_threads(size_t M, size_t pool_size) {
-            if (M <= 1) return std::min(pool_size, static_cast<size_t>(2)); 
-            return pool_size; 
+            if (M <= 1) return std::min(pool_size, static_cast<size_t>(2));
+            return pool_size;
+        }
+        static size_t get_gemv_threads(size_t N_blocks, size_t pool_size) {
+            if (N_blocks < GEMV_MIN_N_BLOCKS) return 1;
+            return std::min(pool_size, static_cast<size_t>(2));
         }
-        #else // Mac
+        #else 
+        static constexpr size_t GEMV_MIN_N_BLOCKS = 256;  
         static size_t get_num_threads(size_t M, size_t pool_size) {
             if (M <= 1) return std::min(pool_size, static_cast<size_t>(4));
-            return pool_size; 
+            return pool_size;
+        }
+        static size_t get_gemv_threads(size_t N_blocks, size_t pool_size) {
+            if (N_blocks < GEMV_MIN_N_BLOCKS) return 1;
+            if (N_blocks < 512) return std::min(pool_size, static_cast<size_t>(2));
+            return std::min(pool_size, static_cast<size_t>(4));
         }
         #endif
     };

package/lib/module/classes/CactusLM.js

@@ -1,20 +1,15 @@
 "use strict";
 
 import { Cactus, CactusFileSystem } from "../native/index.js";
-import { Telemetry } from "../telemetry/Telemetry.js";
-import { CactusConfig } from "../config/CactusConfig.js";
-import { getErrorMessage } from "../utils/error.js";
-import { RemoteLM } from "../api/RemoteLM.js";
-import models from "../models.js";
+import { getRegistry } from "../modelRegistry.js";
 export class CactusLM {
   cactus = new Cactus();
   isDownloading = false;
   isInitialized = false;
   isGenerating = false;
   static defaultModel = 'qwen3-0.6b';
-  static defaultContextSize = 2048;
   static defaultOptions = {
-    quantization: 'int4',
+    quantization: 'int8',
     pro: false
   };
   static quantizationExceptions = {
@@ -24,18 +19,16 @@ export class CactusLM {
   static defaultCompleteOptions = {
     maxTokens: 512
   };
-  static defaultCompleteMode = 'local';
   static defaultEmbedBufferSize = 2048;
   constructor({
     model,
-    contextSize,
     corpusDir,
+    cacheIndex,
     options
   } = {}) {
-    Telemetry.init(CactusConfig.telemetryToken);
     this.model = model ?? CactusLM.defaultModel;
-    this.contextSize = contextSize ?? CactusLM.defaultContextSize;
     this.corpusDir = corpusDir;
+    this.cacheIndex = cacheIndex ?? false;
     this.options = {
       quantization: options?.quantization ?? CactusLM.quantizationExceptions[this.model] ?? CactusLM.defaultOptions.quantization,
       pro: options?.pro ?? CactusLM.defaultOptions.pro
@@ -58,7 +51,8 @@ export class CactusLM {
     }
     this.isDownloading = true;
     try {
-      const modelConfig = models[this.model]?.quantization[this.options.quantization];
+      const registry = await getRegistry();
+      const modelConfig = registry[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`);
@@ -82,21 +76,16 @@ export class CactusLM {
       }
       modelPath = await CactusFileSystem.getModelPath(this.getModelName());
     }
-    try {
-      await this.cactus.init(modelPath, this.contextSize, this.corpusDir);
-      Telemetry.logInit(this.model, true);
-      this.isInitialized = true;
-    } catch (error) {
-      Telemetry.logInit(this.model, false, getErrorMessage(error));
-      throw error;
-    }
+    const cacheDir = await CactusFileSystem.getCactusDirectory();
+    await this.cactus.setTelemetryEnvironment(cacheDir);
+    await this.cactus.init(modelPath, this.corpusDir, this.cacheIndex);
+    this.isInitialized = true;
   }
   async complete({
     messages,
     options,
     tools,
-    onToken,
-    mode
+    onToken
   }) {
     if (this.isGenerating) {
       throw new Error('CactusLM is already generating');
@@ -109,25 +98,11 @@ export class CactusLM {
       type: 'function',
       function: tool
     }));
-    mode = mode ?? CactusLM.defaultCompleteMode;
     const responseBufferSize = 8 * (options.maxTokens ?? CactusLM.defaultCompleteOptions.maxTokens) + 256;
+    await this.init();
+    this.isGenerating = true;
     try {
-      await this.init();
-      this.isGenerating = true;
-      const result = await this.cactus.complete(messages, responseBufferSize, options, toolsInternal, onToken);
-      Telemetry.logCompletion(this.model, result.success, result.success ? undefined : result.response, result);
-      return result;
-    } catch (localError) {
-      if (mode === 'local') {
-        Telemetry.logCompletion(this.model, false, getErrorMessage(localError));
-        throw localError;
-      }
-      Telemetry.logCompletion(this.model, false, `Local completion error: ${getErrorMessage(localError)}. Falling back to remote completion.`);
-      try {
-        return RemoteLM.complete(messages, options, toolsInternal, onToken);
-      } catch (remoteError) {
-        throw new Error(`Remote completion error: ${getErrorMessage(remoteError)}`);
-      }
+      return await this.cactus.complete(messages, responseBufferSize, options, toolsInternal, onToken);
     } finally {
       this.isGenerating = false;
     }
@@ -160,13 +135,9 @@ export class CactusLM {
     this.isGenerating = true;
     try {
       const embedding = await this.cactus.embed(text, CactusLM.defaultEmbedBufferSize, normalize);
-      Telemetry.logEmbedding(this.model, true);
       return {
         embedding
       };
-    } catch (error) {
-      Telemetry.logEmbedding(this.model, false, getErrorMessage(error));
-      throw error;
     } finally {
       this.isGenerating = false;
     }
@@ -181,13 +152,9 @@ export class CactusLM {
     this.isGenerating = true;
     try {
       const embedding = await this.cactus.imageEmbed(imagePath, CactusLM.defaultEmbedBufferSize);
-      Telemetry.logImageEmbedding(this.model, true);
       return {
         embedding
       };
-    } catch (error) {
-      Telemetry.logImageEmbedding(this.model, false, getErrorMessage(error));
-      throw error;
     } finally {
       this.isGenerating = false;
     }
@@ -207,8 +174,8 @@ export class CactusLM {
     await this.cactus.destroy();
     this.isInitialized = false;
   }
-  getModels() {
-    return Object.values(models).filter(model => model.completion);
+  async getModels() {
+    return Object.values(await getRegistry());
   }
   isModelPath(model) {
     return model.startsWith('file://') || model.startsWith('/');

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

@@ -1 +1 @@
-{"version":3,"names":["Cactus","CactusFileSystem","Telemetry","CactusConfig","getErrorMessage","RemoteLM","models","CactusLM","cactus","isDownloading","isInitialized","isGenerating","defaultModel","defaultContextSize","defaultOptions","quantization","pro","quantizationExceptions","defaultCompleteOptions","maxTokens","defaultCompleteMode","defaultEmbedBufferSize","constructor","model","contextSize","corpusDir","options","init","telemetryToken","download","onProgress","isModelPath","Error","modelExists","getModelName","console","log","modelConfig","url","apple","downloadModel","modelPath","replace","JSON","stringify","getModelPath","logInit","error","complete","messages","tools","onToken","mode","toolsInternal","map","tool","type","function","responseBufferSize","result","logCompletion","success","undefined","response","localError","remoteError","tokenize","text","tokens","scoreWindow","start","end","context","score","embed","normalize","embedding","logEmbedding","imageEmbed","imagePath","logImageEmbedding","stop","reset","destroy","getModels","Object","values","filter","completion","startsWith"],"sourceRoot":"../../../src","sources":["classes/CactusLM.ts"],"mappings":";;AAAA,SAASA,MAAM,EAAEC,gBAAgB,QAAQ,oBAAW;AAepD,SAASC,SAAS,QAAQ,2BAAwB;AAClD,SAASC,YAAY,QAAQ,2BAAwB;AACrD,SAASC,eAAe,QAAQ,mBAAgB;AAChD,SAASC,QAAQ,QAAQ,oBAAiB;AAC1C,OAAOC,MAAM,MAAM,cAAW;AAG9B,OAAO,MAAMC,QAAQ,CAAC;EACHC,MAAM,GAAG,IAAIR,MAAM,CAAC,CAAC;EAU9BS,aAAa,GAAG,KAAK;EACrBC,aAAa,GAAG,KAAK;EACrBC,YAAY,GAAG,KAAK;EAE5B,OAAwBC,YAAY,GAAG,YAAY;EACnD,OAAwBC,kBAAkB,GAAG,IAAI;EACjD,OAAwBC,cAAc,GAAG;IACvCC,YAAY,EAAE,MAAe;IAC7BC,GAAG,EAAE;EACP,CAAC;EACD,OAAwBC,sBAAsB,GAE1C;IACF,iBAAiB,EAAE,MAAe;IAClC,uBAAuB,EAAE;EAC3B,CAAC;EACD,OAAwBC,sBAAsB,GAAG;IAC/CC,SAAS,EAAE;EACb,CAAC;EACD,OAAwBC,mBAAmB,GAAG,OAAO;EACrD,OAAwBC,sBAAsB,GAAG,IAAI;EAErDC,WAAWA,CAAC;IAAEC,KAAK;IAAEC,WAAW;IAAEC,SAAS;IAAEC;EAAwB,CAAC,GAAG,CAAC,CAAC,EAAE;IAC3ExB,SAAS,CAACyB,IAAI,CAACxB,YAAY,CAACyB,cAAc,CAAC;IAE3C,IAAI,CAACL,KAAK,GAAGA,KAAK,IAAIhB,QAAQ,CAACK,YAAY;IAC3C,IAAI,CAACY,WAAW,GAAGA,WAAW,IAAIjB,QAAQ,CAACM,kBAAkB;IAC7D,IAAI,CAACY,SAAS,GAAGA,SAAS;IAC1B,IAAI,CAACC,OAAO,GAAG;MACbX,YAAY,EACVW,OAAO,EAAEX,YAAY,IACrBR,QAAQ,CAACU,sBAAsB,CAAC,IAAI,CAACM,KAAK,CAAC,IAC3ChB,QAAQ,CAACO,cAAc,CAACC,YAAY;MACtCC,GAAG,EAAEU,OAAO,EAAEV,GAAG,IAAIT,QAAQ,CAACO,cAAc,CAACE;IAC/C,CAAC;EACH;EAEA,MAAaa,QAAQA,CAAC;IACpBC;EACsB,CAAC,GAAG,CAAC,CAAC,EAAiB;IAC7C,IAAI,IAAI,CAACC,WAAW,CAAC,IAAI,CAACR,KAAK,CAAC,EAAE;MAChCO,UAAU,GAAG,GAAG,CAAC;MACjB;IACF;IAEA,IAAI,IAAI,CAACrB,aAAa,EAAE;MACtB,MAAM,IAAIuB,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,IAAI,MAAM/B,gBAAgB,CAACgC,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,CAACrB,aAAa,GAAG,IAAI;IACzB,IAAI;MACF,MAAM4B,WAAW,GACf/B,MAAM,CAAC,IAAI,CAACiB,KAAK,CAAC,EAAER,YAAY,CAAC,IAAI,CAACW,OAAO,CAACX,YAAY,CAAC;MAC7D,MAAMuB,GAAG,GAAG,IAAI,CAACZ,OAAO,CAACV,GAAG,GAAGqB,WAAW,EAAErB,GAAG,EAAEuB,KAAK,GAAGF,WAAW,EAAEC,GAAG;MAEzE,IAAI,CAACA,GAAG,EAAE;QACR,MAAM,IAAIN,KAAK,CAAC,SAAS,IAAI,CAACT,KAAK,mCAAmC,CAAC;MACzE;MAEA,MAAMtB,gBAAgB,CAACuC,aAAa,CAClC,IAAI,CAACN,YAAY,CAAC,CAAC,EACnBI,GAAG,EACHR,UACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAACrB,aAAa,GAAG,KAAK;IAC5B;EACF;EAEA,MAAakB,IAAIA,CAAA,EAAkB;IACjC,IAAI,IAAI,CAACjB,aAAa,EAAE;MACtB;IACF;IAEA,IAAI+B,SAAiB;IACrB,IAAI,IAAI,CAACV,WAAW,CAAC,IAAI,CAACR,KAAK,CAAC,EAAE;MAChCkB,SAAS,GAAG,IAAI,CAAClB,KAAK,CAACmB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;IAC/C,CAAC,MAAM;MACL,IAAI,EAAE,MAAMzC,gBAAgB,CAACgC,WAAW,CAAC,IAAI,CAACC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE;QAC9DC,OAAO,CAACC,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAACF,YAAY,CAAC,CAAC,CAAC;QACpD,MAAM,IAAIF,KAAK,CACb,UAAU,IAAI,CAACT,KAAK,kBAAkBoB,IAAI,CAACC,SAAS,CAAC,IAAI,CAAClB,OAAO,CAAC,oBACpE,CAAC;MACH;MACAe,SAAS,GAAG,MAAMxC,gBAAgB,CAAC4C,YAAY,CAAC,IAAI,CAACX,YAAY,CAAC,CAAC,CAAC;IACtE;IAEA,IAAI;MACF,MAAM,IAAI,CAAC1B,MAAM,CAACmB,IAAI,CAACc,SAAS,EAAE,IAAI,CAACjB,WAAW,EAAE,IAAI,CAACC,SAAS,CAAC;MACnEvB,SAAS,CAAC4C,OAAO,CAAC,IAAI,CAACvB,KAAK,EAAE,IAAI,CAAC;MACnC,IAAI,CAACb,aAAa,GAAG,IAAI;IAC3B,CAAC,CAAC,OAAOqC,KAAK,EAAE;MACd7C,SAAS,CAAC4C,OAAO,CAAC,IAAI,CAACvB,KAAK,EAAE,KAAK,EAAEnB,eAAe,CAAC2C,KAAK,CAAC,CAAC;MAC5D,MAAMA,KAAK;IACb;EACF;EAEA,MAAaC,QAAQA,CAAC;IACpBC,QAAQ;IACRvB,OAAO;IACPwB,KAAK;IACLC,OAAO;IACPC;EACsB,CAAC,EAAmC;IAC1D,IAAI,IAAI,CAACzC,YAAY,EAAE;MACrB,MAAM,IAAIqB,KAAK,CAAC,gCAAgC,CAAC;IACnD;IAEAN,OAAO,GAAG;MAAE,GAAGnB,QAAQ,CAACW,sBAAsB;MAAE,GAAGQ;IAAQ,CAAC;IAC5D,MAAM2B,aAAa,GAAGH,KAAK,EAAEI,GAAG,CAAEC,IAAI,KAAM;MAC1CC,IAAI,EAAE,UAAmB;MACzBC,QAAQ,EAAEF;IACZ,CAAC,CAAC,CAAC;IACHH,IAAI,GAAGA,IAAI,IAAI7C,QAAQ,CAACa,mBAAmB;IAE3C,MAAMsC,kBAAkB,GACtB,CAAC,IAAIhC,OAAO,CAACP,SAAS,IAAIZ,QAAQ,CAACW,sBAAsB,CAACC,SAAS,CAAC,GACpE,GAAG;IAEL,IAAI;MACF,MAAM,IAAI,CAACQ,IAAI,CAAC,CAAC;MAEjB,IAAI,CAAChB,YAAY,GAAG,IAAI;MACxB,MAAMgD,MAAM,GAAG,MAAM,IAAI,CAACnD,MAAM,CAACwC,QAAQ,CACvCC,QAAQ,EACRS,kBAAkB,EAClBhC,OAAO,EACP2B,aAAa,EACbF,OACF,CAAC;MACDjD,SAAS,CAAC0D,aAAa,CACrB,IAAI,CAACrC,KAAK,EACVoC,MAAM,CAACE,OAAO,EACdF,MAAM,CAACE,OAAO,GAAGC,SAAS,GAAGH,MAAM,CAACI,QAAQ,EAC5CJ,MACF,CAAC;MACD,OAAOA,MAAM;IACf,CAAC,CAAC,OAAOK,UAAU,EAAE;MACnB,IAAIZ,IAAI,KAAK,OAAO,EAAE;QACpBlD,SAAS,CAAC0D,aAAa,CAAC,IAAI,CAACrC,KAAK,EAAE,KAAK,EAAEnB,eAAe,CAAC4D,UAAU,CAAC,CAAC;QACvE,MAAMA,UAAU;MAClB;MAEA9D,SAAS,CAAC0D,aAAa,CACrB,IAAI,CAACrC,KAAK,EACV,KAAK,EACL,2BAA2BnB,eAAe,CAAC4D,UAAU,CAAC,sCACxD,CAAC;MAED,IAAI;QACF,OAAO3D,QAAQ,CAAC2C,QAAQ,CAACC,QAAQ,EAAEvB,OAAO,EAAE2B,aAAa,EAAEF,OAAO,CAAC;MACrE,CAAC,CAAC,OAAOc,WAAW,EAAE;QACpB,MAAM,IAAIjC,KAAK,CACb,4BAA4B5B,eAAe,CAAC6D,WAAW,CAAC,EAC1D,CAAC;MACH;IACF,CAAC,SAAS;MACR,IAAI,CAACtD,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAauD,QAAQA,CAAC;IACpBC;EACsB,CAAC,EAAmC;IAC1D,OAAO;MAAEC,MAAM,EAAE,MAAM,IAAI,CAAC5D,MAAM,CAAC0D,QAAQ,CAACC,IAAI;IAAE,CAAC;EACrD;EAEA,MAAaE,WAAWA,CAAC;IACvBD,MAAM;IACNE,KAAK;IACLC,GAAG;IACHC;EACyB,CAAC,EAAsC;IAChE,OAAO;MACLC,KAAK,EAAE,MAAM,IAAI,CAACjE,MAAM,CAAC6D,WAAW,CAACD,MAAM,EAAEE,KAAK,EAAEC,GAAG,EAAEC,OAAO;IAClE,CAAC;EACH;EAEA,MAAaE,KAAKA,CAAC;IACjBP,IAAI;IACJQ,SAAS,GAAG;EACO,CAAC,EAAgC;IACpD,IAAI,IAAI,CAAChE,YAAY,EAAE;MACrB,MAAM,IAAIqB,KAAK,CAAC,gCAAgC,CAAC;IACnD;IAEA,MAAM,IAAI,CAACL,IAAI,CAAC,CAAC;IAEjB,IAAI,CAAChB,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMiE,SAAS,GAAG,MAAM,IAAI,CAACpE,MAAM,CAACkE,KAAK,CACvCP,IAAI,EACJ5D,QAAQ,CAACc,sBAAsB,EAC/BsD,SACF,CAAC;MACDzE,SAAS,CAAC2E,YAAY,CAAC,IAAI,CAACtD,KAAK,EAAE,IAAI,CAAC;MACxC,OAAO;QAAEqD;MAAU,CAAC;IACtB,CAAC,CAAC,OAAO7B,KAAK,EAAE;MACd7C,SAAS,CAAC2E,YAAY,CAAC,IAAI,CAACtD,KAAK,EAAE,KAAK,EAAEnB,eAAe,CAAC2C,KAAK,CAAC,CAAC;MACjE,MAAMA,KAAK;IACb,CAAC,SAAS;MACR,IAAI,CAACpC,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAamE,UAAUA,CAAC;IACtBC;EACwB,CAAC,EAAqC;IAC9D,IAAI,IAAI,CAACpE,YAAY,EAAE;MACrB,MAAM,IAAIqB,KAAK,CAAC,gCAAgC,CAAC;IACnD;IAEA,MAAM,IAAI,CAACL,IAAI,CAAC,CAAC;IAEjB,IAAI,CAAChB,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMiE,SAAS,GAAG,MAAM,IAAI,CAACpE,MAAM,CAACsE,UAAU,CAC5CC,SAAS,EACTxE,QAAQ,CAACc,sBACX,CAAC;MACDnB,SAAS,CAAC8E,iBAAiB,CAAC,IAAI,CAACzD,KAAK,EAAE,IAAI,CAAC;MAC7C,OAAO;QAAEqD;MAAU,CAAC;IACtB,CAAC,CAAC,OAAO7B,KAAK,EAAE;MACd7C,SAAS,CAAC8E,iBAAiB,CAAC,IAAI,CAACzD,KAAK,EAAE,KAAK,EAAEnB,eAAe,CAAC2C,KAAK,CAAC,CAAC;MACtE,MAAMA,KAAK;IACb,CAAC,SAAS;MACR,IAAI,CAACpC,YAAY,GAAG,KAAK;IAC3B;EACF;EAEOsE,IAAIA,CAAA,EAAkB;IAC3B,OAAO,IAAI,CAACzE,MAAM,CAACyE,IAAI,CAAC,CAAC;EAC3B;EAEA,MAAaC,KAAKA,CAAA,EAAkB;IAClC,MAAM,IAAI,CAACD,IAAI,CAAC,CAAC;IACjB,OAAO,IAAI,CAACzE,MAAM,CAAC0E,KAAK,CAAC,CAAC;EAC5B;EAEA,MAAaC,OAAOA,CAAA,EAAkB;IACpC,IAAI,CAAC,IAAI,CAACzE,aAAa,EAAE;MACvB;IACF;IAEA,MAAM,IAAI,CAACuE,IAAI,CAAC,CAAC;IACjB,MAAM,IAAI,CAACzE,MAAM,CAAC2E,OAAO,CAAC,CAAC;IAE3B,IAAI,CAACzE,aAAa,GAAG,KAAK;EAC5B;EAEO0E,SAASA,CAAA,EAAkB;IAChC,OAAOC,MAAM,CAACC,MAAM,CAAChF,MAAM,CAAC,CAACiF,MAAM,CAAEhE,KAAK,IAAKA,KAAK,CAACiE,UAAU,CAAC;EAClE;EAEQzD,WAAWA,CAACR,KAAa,EAAW;IAC1C,OAAOA,KAAK,CAACkE,UAAU,CAAC,SAAS,CAAC,IAAIlE,KAAK,CAACkE,UAAU,CAAC,GAAG,CAAC;EAC7D;EAEQvD,YAAYA,CAAA,EAAW;IAC7B,OAAO,GAAG,IAAI,CAACX,KAAK,IAAI,IAAI,CAACG,OAAO,CAACX,YAAY,GAAG,IAAI,CAACW,OAAO,CAACV,GAAG,GAAG,MAAM,GAAG,EAAE,EAAE;EACtF;AACF","ignoreList":[]}
+{"version":3,"names":["Cactus","CactusFileSystem","getRegistry","CactusLM","cactus","isDownloading","isInitialized","isGenerating","defaultModel","defaultOptions","quantization","pro","quantizationExceptions","defaultCompleteOptions","maxTokens","defaultEmbedBufferSize","constructor","model","corpusDir","cacheIndex","options","download","onProgress","isModelPath","Error","modelExists","getModelName","console","log","registry","modelConfig","url","apple","downloadModel","init","modelPath","replace","JSON","stringify","getModelPath","cacheDir","getCactusDirectory","setTelemetryEnvironment","complete","messages","tools","onToken","toolsInternal","map","tool","type","function","responseBufferSize","tokenize","text","tokens","scoreWindow","start","end","context","score","embed","normalize","embedding","imageEmbed","imagePath","stop","reset","destroy","getModels","Object","values","startsWith"],"sourceRoot":"../../../src","sources":["classes/CactusLM.ts"],"mappings":";;AAAA,SAASA,MAAM,EAAEC,gBAAgB,QAAQ,oBAAW;AAepD,SAASC,WAAW,QAAQ,qBAAkB;AAG9C,OAAO,MAAMC,QAAQ,CAAC;EACHC,MAAM,GAAG,IAAIJ,MAAM,CAAC,CAAC;EAU9BK,aAAa,GAAG,KAAK;EACrBC,aAAa,GAAG,KAAK;EACrBC,YAAY,GAAG,KAAK;EAE5B,OAAwBC,YAAY,GAAG,YAAY;EACnD,OAAwBC,cAAc,GAAG;IACvCC,YAAY,EAAE,MAAe;IAC7BC,GAAG,EAAE;EACP,CAAC;EACD,OAAwBC,sBAAsB,GAE1C;IACF,iBAAiB,EAAE,MAAe;IAClC,uBAAuB,EAAE;EAC3B,CAAC;EACD,OAAwBC,sBAAsB,GAAG;IAC/CC,SAAS,EAAE;EACb,CAAC;EACD,OAAwBC,sBAAsB,GAAG,IAAI;EAErDC,WAAWA,CAAC;IAAEC,KAAK;IAAEC,SAAS;IAAEC,UAAU;IAAEC;EAAwB,CAAC,GAAG,CAAC,CAAC,EAAE;IAC1E,IAAI,CAACH,KAAK,GAAGA,KAAK,IAAId,QAAQ,CAACK,YAAY;IAC3C,IAAI,CAACU,SAAS,GAAGA,SAAS;IAC1B,IAAI,CAACC,UAAU,GAAGA,UAAU,IAAI,KAAK;IACrC,IAAI,CAACC,OAAO,GAAG;MACbV,YAAY,EACVU,OAAO,EAAEV,YAAY,IACrBP,QAAQ,CAACS,sBAAsB,CAAC,IAAI,CAACK,KAAK,CAAC,IAC3Cd,QAAQ,CAACM,cAAc,CAACC,YAAY;MACtCC,GAAG,EAAES,OAAO,EAAET,GAAG,IAAIR,QAAQ,CAACM,cAAc,CAACE;IAC/C,CAAC;EACH;EAEA,MAAaU,QAAQA,CAAC;IACpBC;EACsB,CAAC,GAAG,CAAC,CAAC,EAAiB;IAC7C,IAAI,IAAI,CAACC,WAAW,CAAC,IAAI,CAACN,KAAK,CAAC,EAAE;MAChCK,UAAU,GAAG,GAAG,CAAC;MACjB;IACF;IAEA,IAAI,IAAI,CAACjB,aAAa,EAAE;MACtB,MAAM,IAAImB,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,IAAI,MAAMvB,gBAAgB,CAACwB,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,CAACjB,aAAa,GAAG,IAAI;IACzB,IAAI;MACF,MAAMwB,QAAQ,GAAG,MAAM3B,WAAW,CAAC,CAAC;MACpC,MAAM4B,WAAW,GACfD,QAAQ,CAAC,IAAI,CAACZ,KAAK,CAAC,EAAEP,YAAY,CAAC,IAAI,CAACU,OAAO,CAACV,YAAY,CAAC;MAC/D,MAAMqB,GAAG,GAAG,IAAI,CAACX,OAAO,CAACT,GAAG,GAAGmB,WAAW,EAAEnB,GAAG,EAAEqB,KAAK,GAAGF,WAAW,EAAEC,GAAG;MAEzE,IAAI,CAACA,GAAG,EAAE;QACR,MAAM,IAAIP,KAAK,CAAC,SAAS,IAAI,CAACP,KAAK,mCAAmC,CAAC;MACzE;MAEA,MAAMhB,gBAAgB,CAACgC,aAAa,CAClC,IAAI,CAACP,YAAY,CAAC,CAAC,EACnBK,GAAG,EACHT,UACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAACjB,aAAa,GAAG,KAAK;IAC5B;EACF;EAEA,MAAa6B,IAAIA,CAAA,EAAkB;IACjC,IAAI,IAAI,CAAC5B,aAAa,EAAE;MACtB;IACF;IAEA,IAAI6B,SAAiB;IACrB,IAAI,IAAI,CAACZ,WAAW,CAAC,IAAI,CAACN,KAAK,CAAC,EAAE;MAChCkB,SAAS,GAAG,IAAI,CAAClB,KAAK,CAACmB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;IAC/C,CAAC,MAAM;MACL,IAAI,EAAE,MAAMnC,gBAAgB,CAACwB,WAAW,CAAC,IAAI,CAACC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE;QAC9DC,OAAO,CAACC,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAACF,YAAY,CAAC,CAAC,CAAC;QACpD,MAAM,IAAIF,KAAK,CACb,UAAU,IAAI,CAACP,KAAK,kBAAkBoB,IAAI,CAACC,SAAS,CAAC,IAAI,CAAClB,OAAO,CAAC,oBACpE,CAAC;MACH;MACAe,SAAS,GAAG,MAAMlC,gBAAgB,CAACsC,YAAY,CAAC,IAAI,CAACb,YAAY,CAAC,CAAC,CAAC;IACtE;IAEA,MAAMc,QAAQ,GAAG,MAAMvC,gBAAgB,CAACwC,kBAAkB,CAAC,CAAC;IAC5D,MAAM,IAAI,CAACrC,MAAM,CAACsC,uBAAuB,CAACF,QAAQ,CAAC;IACnD,MAAM,IAAI,CAACpC,MAAM,CAAC8B,IAAI,CAACC,SAAS,EAAE,IAAI,CAACjB,SAAS,EAAE,IAAI,CAACC,UAAU,CAAC;IAClE,IAAI,CAACb,aAAa,GAAG,IAAI;EAC3B;EAEA,MAAaqC,QAAQA,CAAC;IACpBC,QAAQ;IACRxB,OAAO;IACPyB,KAAK;IACLC;EACsB,CAAC,EAAmC;IAC1D,IAAI,IAAI,CAACvC,YAAY,EAAE;MACrB,MAAM,IAAIiB,KAAK,CAAC,gCAAgC,CAAC;IACnD;IAEAJ,OAAO,GAAG;MAAE,GAAGjB,QAAQ,CAACU,sBAAsB;MAAE,GAAGO;IAAQ,CAAC;IAC5D,MAAM2B,aAAa,GAAGF,KAAK,EAAEG,GAAG,CAAEC,IAAI,KAAM;MAC1CC,IAAI,EAAE,UAAmB;MACzBC,QAAQ,EAAEF;IACZ,CAAC,CAAC,CAAC;IAEH,MAAMG,kBAAkB,GACtB,CAAC,IAAIhC,OAAO,CAACN,SAAS,IAAIX,QAAQ,CAACU,sBAAsB,CAACC,SAAS,CAAC,GACpE,GAAG;IAEL,MAAM,IAAI,CAACoB,IAAI,CAAC,CAAC;IAEjB,IAAI,CAAC3B,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,OAAO,MAAM,IAAI,CAACH,MAAM,CAACuC,QAAQ,CAC/BC,QAAQ,EACRQ,kBAAkB,EAClBhC,OAAO,EACP2B,aAAa,EACbD,OACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAACvC,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAa8C,QAAQA,CAAC;IACpBC;EACsB,CAAC,EAAmC;IAC1D,OAAO;MAAEC,MAAM,EAAE,MAAM,IAAI,CAACnD,MAAM,CAACiD,QAAQ,CAACC,IAAI;IAAE,CAAC;EACrD;EAEA,MAAaE,WAAWA,CAAC;IACvBD,MAAM;IACNE,KAAK;IACLC,GAAG;IACHC;EACyB,CAAC,EAAsC;IAChE,OAAO;MACLC,KAAK,EAAE,MAAM,IAAI,CAACxD,MAAM,CAACoD,WAAW,CAACD,MAAM,EAAEE,KAAK,EAAEC,GAAG,EAAEC,OAAO;IAClE,CAAC;EACH;EAEA,MAAaE,KAAKA,CAAC;IACjBP,IAAI;IACJQ,SAAS,GAAG;EACO,CAAC,EAAgC;IACpD,IAAI,IAAI,CAACvD,YAAY,EAAE;MACrB,MAAM,IAAIiB,KAAK,CAAC,gCAAgC,CAAC;IACnD;IAEA,MAAM,IAAI,CAACU,IAAI,CAAC,CAAC;IAEjB,IAAI,CAAC3B,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMwD,SAAS,GAAG,MAAM,IAAI,CAAC3D,MAAM,CAACyD,KAAK,CACvCP,IAAI,EACJnD,QAAQ,CAACY,sBAAsB,EAC/B+C,SACF,CAAC;MACD,OAAO;QAAEC;MAAU,CAAC;IACtB,CAAC,SAAS;MACR,IAAI,CAACxD,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAayD,UAAUA,CAAC;IACtBC;EACwB,CAAC,EAAqC;IAC9D,IAAI,IAAI,CAAC1D,YAAY,EAAE;MACrB,MAAM,IAAIiB,KAAK,CAAC,gCAAgC,CAAC;IACnD;IAEA,MAAM,IAAI,CAACU,IAAI,CAAC,CAAC;IAEjB,IAAI,CAAC3B,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAMwD,SAAS,GAAG,MAAM,IAAI,CAAC3D,MAAM,CAAC4D,UAAU,CAC5CC,SAAS,EACT9D,QAAQ,CAACY,sBACX,CAAC;MACD,OAAO;QAAEgD;MAAU,CAAC;IACtB,CAAC,SAAS;MACR,IAAI,CAACxD,YAAY,GAAG,KAAK;IAC3B;EACF;EAEO2D,IAAIA,CAAA,EAAkB;IAC3B,OAAO,IAAI,CAAC9D,MAAM,CAAC8D,IAAI,CAAC,CAAC;EAC3B;EAEA,MAAaC,KAAKA,CAAA,EAAkB;IAClC,MAAM,IAAI,CAACD,IAAI,CAAC,CAAC;IACjB,OAAO,IAAI,CAAC9D,MAAM,CAAC+D,KAAK,CAAC,CAAC;EAC5B;EAEA,MAAaC,OAAOA,CAAA,EAAkB;IACpC,IAAI,CAAC,IAAI,CAAC9D,aAAa,EAAE;MACvB;IACF;IAEA,MAAM,IAAI,CAAC4D,IAAI,CAAC,CAAC;IACjB,MAAM,IAAI,CAAC9D,MAAM,CAACgE,OAAO,CAAC,CAAC;IAE3B,IAAI,CAAC9D,aAAa,GAAG,KAAK;EAC5B;EAEA,MAAa+D,SAASA,CAAA,EAA2B;IAC/C,OAAOC,MAAM,CAACC,MAAM,CAAC,MAAMrE,WAAW,CAAC,CAAC,CAAC;EAC3C;EAEQqB,WAAWA,CAACN,KAAa,EAAW;IAC1C,OAAOA,KAAK,CAACuD,UAAU,CAAC,SAAS,CAAC,IAAIvD,KAAK,CAACuD,UAAU,CAAC,GAAG,CAAC;EAC7D;EAEO9C,YAAYA,CAAA,EAAW;IAC5B,OAAO,GAAG,IAAI,CAACT,KAAK,IAAI,IAAI,CAACG,OAAO,CAACV,YAAY,GAAG,IAAI,CAACU,OAAO,CAACT,GAAG,GAAG,MAAM,GAAG,EAAE,EAAE;EACtF;AACF","ignoreList":[]}

package/lib/module/classes/CactusSTT.js

@@ -1,20 +1,16 @@
 "use strict";
 
 import { Cactus, CactusFileSystem } from "../native/index.js";
-import { Telemetry } from "../telemetry/Telemetry.js";
-import { CactusConfig } from "../config/CactusConfig.js";
-import { getErrorMessage } from "../utils/error.js";
-import models from "../models.js";
+import { getRegistry } from "../modelRegistry.js";
 export class CactusSTT {
   cactus = new Cactus();
   isDownloading = false;
   isInitialized = false;
   isGenerating = false;
-  isStreamTranscribeInitialized = false;
+  isStreamTranscribing = false;
   static defaultModel = 'whisper-small';
-  static defaultContextSize = 2048;
   static defaultOptions = {
-    quantization: 'int4',
+    quantization: 'int8',
     pro: false
   };
   static defaultPrompt = '<|startoftranscript|><|en|><|transcribe|><|notimestamps|>';
@@ -24,12 +20,9 @@ export class CactusSTT {
   static defaultEmbedBufferSize = 4096;
   constructor({
     model,
-    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
@@ -52,7 +45,8 @@ export class CactusSTT {
     }
     this.isDownloading = true;
     try {
-      const modelConfig = models[this.model]?.quantization[this.options.quantization];
+      const registry = await getRegistry();
+      const modelConfig = registry[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`);
@@ -76,14 +70,10 @@ export class CactusSTT {
       }
       modelPath = await CactusFileSystem.getModelPath(this.getModelName());
     }
-    try {
-      await this.cactus.init(modelPath, this.contextSize);
-      Telemetry.logInit(this.model, true);
-      this.isInitialized = true;
-    } catch (error) {
-      Telemetry.logInit(this.model, false, getErrorMessage(error));
-      throw error;
-    }
+    const cacheDir = await CactusFileSystem.getCactusDirectory();
+    await this.cactus.setTelemetryEnvironment(cacheDir);
+    await this.cactus.init(modelPath);
+    this.isInitialized = true;
   }
   async transcribe({
     audio,
@@ -103,73 +93,35 @@ export class CactusSTT {
     const responseBufferSize = 8 * (options.maxTokens ?? CactusSTT.defaultTranscribeOptions.maxTokens) + 256;
     this.isGenerating = true;
     try {
-      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) {
-      Telemetry.logTranscribe(this.model, false, getErrorMessage(error));
-      throw error;
+      return await this.cactus.transcribe(audio, prompt, responseBufferSize, options, onToken);
     } finally {
       this.isGenerating = false;
     }
   }
-  async streamTranscribeInit() {
-    if (this.isStreamTranscribeInitialized) {
+  async streamTranscribeStart(options) {
+    if (this.isStreamTranscribing) {
       return;
     }
     await this.init();
-    try {
-      await this.cactus.streamTranscribeInit();
-      this.isStreamTranscribeInitialized = true;
-    } catch (error) {
-      throw error;
-    }
+    await this.cactus.streamTranscribeStart(options);
+    this.isStreamTranscribing = true;
   }
-  async streamTranscribeInsert({
+  async streamTranscribeProcess({
     audio
   }) {
-    if (!this.isStreamTranscribeInitialized) {
-      throw new Error('CactusSTT stream transcribe is not initialized');
-    }
-    try {
-      await this.cactus.streamTranscribeInsert(audio);
-    } catch (error) {
-      throw error;
+    if (!this.isStreamTranscribing) {
+      throw new Error('CactusSTT stream transcribe is not started');
     }
+    return this.cactus.streamTranscribeProcess(audio);
   }
-  async streamTranscribeProcess({
-    options
-  } = {}) {
-    if (!this.isStreamTranscribeInitialized) {
-      throw new Error('CactusSTT stream transcribe is not initialized');
+  async streamTranscribeStop() {
+    if (!this.isStreamTranscribing) {
+      throw new Error('CactusSTT stream transcribe is not started');
     }
     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;
+      return await this.cactus.streamTranscribeStop();
+    } finally {
+      this.isStreamTranscribing = false;
     }
   }
   async audioEmbed({
@@ -182,13 +134,9 @@ export class CactusSTT {
     this.isGenerating = true;
     try {
       const embedding = await this.cactus.audioEmbed(audioPath, CactusSTT.defaultEmbedBufferSize);
-      Telemetry.logAudioEmbedding(this.model, true);
       return {
         embedding
       };
-    } catch (error) {
-      Telemetry.logAudioEmbedding(this.model, false, getErrorMessage(error));
-      throw error;
     } finally {
       this.isGenerating = false;
     }
@@ -205,12 +153,15 @@ export class CactusSTT {
       return;
     }
     await this.stop();
-    await this.streamTranscribeDestroy();
+    if (this.isStreamTranscribing) {
+      await this.cactus.streamTranscribeStop().catch(() => {});
+      this.isStreamTranscribing = false;
+    }
     await this.cactus.destroy();
     this.isInitialized = false;
   }
-  getModels() {
-    return Object.values(models).filter(model => model.speech);
+  async getModels() {
+    return Object.values(await getRegistry());
   }
   isModelPath(model) {
     return model.startsWith('file://') || model.startsWith('/');

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

@@ -1 +1 @@
-{"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":[]}
+{"version":3,"names":["Cactus","CactusFileSystem","getRegistry","CactusSTT","cactus","isDownloading","isInitialized","isGenerating","isStreamTranscribing","defaultModel","defaultOptions","quantization","pro","defaultPrompt","defaultTranscribeOptions","maxTokens","defaultEmbedBufferSize","constructor","model","options","download","onProgress","isModelPath","Error","modelExists","getModelName","console","log","registry","modelConfig","url","apple","downloadModel","init","modelPath","replace","JSON","stringify","getModelPath","cacheDir","getCactusDirectory","setTelemetryEnvironment","transcribe","audio","prompt","onToken","responseBufferSize","streamTranscribeStart","streamTranscribeProcess","streamTranscribeStop","audioEmbed","audioPath","embedding","stop","reset","destroy","catch","getModels","Object","values","startsWith"],"sourceRoot":"../../../src","sources":["classes/CactusSTT.ts"],"mappings":";;AAAA,SAASA,MAAM,EAAEC,gBAAgB,QAAQ,oBAAW;AAapD,SAASC,WAAW,QAAQ,qBAAkB;AAG9C,OAAO,MAAMC,SAAS,CAAC;EACJC,MAAM,GAAG,IAAIJ,MAAM,CAAC,CAAC;EAQ9BK,aAAa,GAAG,KAAK;EACrBC,aAAa,GAAG,KAAK;EACrBC,YAAY,GAAG,KAAK;EACpBC,oBAAoB,GAAG,KAAK;EAEpC,OAAwBC,YAAY,GAAG,eAAe;EACtD,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;EAAyB,CAAC,GAAG,CAAC,CAAC,EAAE;IACpD,IAAI,CAACD,KAAK,GAAGA,KAAK,IAAIf,SAAS,CAACM,YAAY;IAC5C,IAAI,CAACU,OAAO,GAAG;MACbR,YAAY,EACVQ,OAAO,EAAER,YAAY,IAAIR,SAAS,CAACO,cAAc,CAACC,YAAY;MAChEC,GAAG,EAAEO,OAAO,EAAEP,GAAG,IAAIT,SAAS,CAACO,cAAc,CAACE;IAChD,CAAC;EACH;EAEA,MAAaQ,QAAQA,CAAC;IACpBC;EACuB,CAAC,GAAG,CAAC,CAAC,EAAiB;IAC9C,IAAI,IAAI,CAACC,WAAW,CAAC,IAAI,CAACJ,KAAK,CAAC,EAAE;MAChCG,UAAU,GAAG,GAAG,CAAC;MACjB;IACF;IAEA,IAAI,IAAI,CAAChB,aAAa,EAAE;MACtB,MAAM,IAAIkB,KAAK,CAAC,kCAAkC,CAAC;IACrD;IAEA,IAAI,MAAMtB,gBAAgB,CAACuB,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,CAAChB,aAAa,GAAG,IAAI;IACzB,IAAI;MACF,MAAMuB,QAAQ,GAAG,MAAM1B,WAAW,CAAC,CAAC;MACpC,MAAM2B,WAAW,GACfD,QAAQ,CAAC,IAAI,CAACV,KAAK,CAAC,EAAEP,YAAY,CAAC,IAAI,CAACQ,OAAO,CAACR,YAAY,CAAC;MAC/D,MAAMmB,GAAG,GAAG,IAAI,CAACX,OAAO,CAACP,GAAG,GAAGiB,WAAW,EAAEjB,GAAG,EAAEmB,KAAK,GAAGF,WAAW,EAAEC,GAAG;MAEzE,IAAI,CAACA,GAAG,EAAE;QACR,MAAM,IAAIP,KAAK,CAAC,SAAS,IAAI,CAACL,KAAK,mCAAmC,CAAC;MACzE;MAEA,MAAMjB,gBAAgB,CAAC+B,aAAa,CAClC,IAAI,CAACP,YAAY,CAAC,CAAC,EACnBK,GAAG,EACHT,UACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAAChB,aAAa,GAAG,KAAK;IAC5B;EACF;EAEA,MAAa4B,IAAIA,CAAA,EAAkB;IACjC,IAAI,IAAI,CAAC3B,aAAa,EAAE;MACtB;IACF;IAEA,IAAI4B,SAAiB;IACrB,IAAI,IAAI,CAACZ,WAAW,CAAC,IAAI,CAACJ,KAAK,CAAC,EAAE;MAChCgB,SAAS,GAAG,IAAI,CAAChB,KAAK,CAACiB,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC;IAC/C,CAAC,MAAM;MACL,IAAI,EAAE,MAAMlC,gBAAgB,CAACuB,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,CAACL,KAAK,kBAAkBkB,IAAI,CAACC,SAAS,CAAC,IAAI,CAAClB,OAAO,CAAC,oBACpE,CAAC;MACH;MACAe,SAAS,GAAG,MAAMjC,gBAAgB,CAACqC,YAAY,CAAC,IAAI,CAACb,YAAY,CAAC,CAAC,CAAC;IACtE;IAEA,MAAMc,QAAQ,GAAG,MAAMtC,gBAAgB,CAACuC,kBAAkB,CAAC,CAAC;IAC5D,MAAM,IAAI,CAACpC,MAAM,CAACqC,uBAAuB,CAACF,QAAQ,CAAC;IACnD,MAAM,IAAI,CAACnC,MAAM,CAAC6B,IAAI,CAACC,SAAS,CAAC;IACjC,IAAI,CAAC5B,aAAa,GAAG,IAAI;EAC3B;EAEA,MAAaoC,UAAUA,CAAC;IACtBC,KAAK;IACLC,MAAM;IACNzB,OAAO;IACP0B;EACyB,CAAC,EAAsC;IAChE,IAAI,IAAI,CAACtC,YAAY,EAAE;MACrB,MAAM,IAAIgB,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,MAAM,IAAI,CAACU,IAAI,CAAC,CAAC;IAEjBW,MAAM,GAAGA,MAAM,IAAIzC,SAAS,CAACU,aAAa;IAC1CM,OAAO,GAAG;MAAE,GAAGhB,SAAS,CAACW,wBAAwB;MAAE,GAAGK;IAAQ,CAAC;IAE/D,MAAM2B,kBAAkB,GACtB,CAAC,IAAI3B,OAAO,CAACJ,SAAS,IAAIZ,SAAS,CAACW,wBAAwB,CAACC,SAAS,CAAC,GACvE,GAAG;IAEL,IAAI,CAACR,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,OAAO,MAAM,IAAI,CAACH,MAAM,CAACsC,UAAU,CACjCC,KAAK,EACLC,MAAM,EACNE,kBAAkB,EAClB3B,OAAO,EACP0B,OACF,CAAC;IACH,CAAC,SAAS;MACR,IAAI,CAACtC,YAAY,GAAG,KAAK;IAC3B;EACF;EAEA,MAAawC,qBAAqBA,CAChC5B,OAA+C,EAChC;IACf,IAAI,IAAI,CAACX,oBAAoB,EAAE;MAC7B;IACF;IAEA,MAAM,IAAI,CAACyB,IAAI,CAAC,CAAC;IACjB,MAAM,IAAI,CAAC7B,MAAM,CAAC2C,qBAAqB,CAAC5B,OAAO,CAAC;IAChD,IAAI,CAACX,oBAAoB,GAAG,IAAI;EAClC;EAEA,MAAawC,uBAAuBA,CAAC;IACnCL;EACsC,CAAC,EAAmD;IAC1F,IAAI,CAAC,IAAI,CAACnC,oBAAoB,EAAE;MAC9B,MAAM,IAAIe,KAAK,CAAC,4CAA4C,CAAC;IAC/D;IAEA,OAAO,IAAI,CAACnB,MAAM,CAAC4C,uBAAuB,CAACL,KAAK,CAAC;EACnD;EAEA,MAAaM,oBAAoBA,CAAA,EAAiD;IAChF,IAAI,CAAC,IAAI,CAACzC,oBAAoB,EAAE;MAC9B,MAAM,IAAIe,KAAK,CAAC,4CAA4C,CAAC;IAC/D;IAEA,IAAI;MACF,OAAO,MAAM,IAAI,CAACnB,MAAM,CAAC6C,oBAAoB,CAAC,CAAC;IACjD,CAAC,SAAS;MACR,IAAI,CAACzC,oBAAoB,GAAG,KAAK;IACnC;EACF;EAEA,MAAa0C,UAAUA,CAAC;IACtBC;EACyB,CAAC,EAAsC;IAChE,IAAI,IAAI,CAAC5C,YAAY,EAAE;MACrB,MAAM,IAAIgB,KAAK,CAAC,iCAAiC,CAAC;IACpD;IAEA,MAAM,IAAI,CAACU,IAAI,CAAC,CAAC;IAEjB,IAAI,CAAC1B,YAAY,GAAG,IAAI;IACxB,IAAI;MACF,MAAM6C,SAAS,GAAG,MAAM,IAAI,CAAChD,MAAM,CAAC8C,UAAU,CAC5CC,SAAS,EACThD,SAAS,CAACa,sBACZ,CAAC;MACD,OAAO;QAAEoC;MAAU,CAAC;IACtB,CAAC,SAAS;MACR,IAAI,CAAC7C,YAAY,GAAG,KAAK;IAC3B;EACF;EAEO8C,IAAIA,CAAA,EAAkB;IAC3B,OAAO,IAAI,CAACjD,MAAM,CAACiD,IAAI,CAAC,CAAC;EAC3B;EAEA,MAAaC,KAAKA,CAAA,EAAkB;IAClC,MAAM,IAAI,CAACD,IAAI,CAAC,CAAC;IACjB,OAAO,IAAI,CAACjD,MAAM,CAACkD,KAAK,CAAC,CAAC;EAC5B;EAEA,MAAaC,OAAOA,CAAA,EAAkB;IACpC,IAAI,CAAC,IAAI,CAACjD,aAAa,EAAE;MACvB;IACF;IAEA,MAAM,IAAI,CAAC+C,IAAI,CAAC,CAAC;IAEjB,IAAI,IAAI,CAAC7C,oBAAoB,EAAE;MAC7B,MAAM,IAAI,CAACJ,MAAM,CAAC6C,oBAAoB,CAAC,CAAC,CAACO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;MACxD,IAAI,CAAChD,oBAAoB,GAAG,KAAK;IACnC;IAEA,MAAM,IAAI,CAACJ,MAAM,CAACmD,OAAO,CAAC,CAAC;IAC3B,IAAI,CAACjD,aAAa,GAAG,KAAK;EAC5B;EAEA,MAAamD,SAASA,CAAA,EAA2B;IAC/C,OAAOC,MAAM,CAACC,MAAM,CAAC,MAAMzD,WAAW,CAAC,CAAC,CAAC;EAC3C;EAEQoB,WAAWA,CAACJ,KAAa,EAAW;IAC1C,OAAOA,KAAK,CAAC0C,UAAU,CAAC,SAAS,CAAC,IAAI1C,KAAK,CAAC0C,UAAU,CAAC,GAAG,CAAC;EAC7D;EAEOnC,YAAYA,CAAA,EAAW;IAC5B,OAAO,GAAG,IAAI,CAACP,KAAK,IAAI,IAAI,CAACC,OAAO,CAACR,YAAY,GAAG,IAAI,CAACQ,OAAO,CAACP,GAAG,GAAG,MAAM,GAAG,EAAE,EAAE;EACtF;AACF","ignoreList":[]}