@zvec/zvec-grep 0.1.4 → 0.1.6

package/dist/engine/models/{providers/qwen/embedding.js → backends/qwen.js}

@@ -1,65 +1,94 @@
-import { EngineError } from "../../../errors/index.js";
-import { EmbeddingModel, } from "../../embeddings.js";
+import { globalConfigPath } from "../../config.js";
+import { EngineError } from "../../errors.js";
+import { BaseEmbeddingModel, } from "../embeddings.js";
+import { traceHeaders } from "../../../observability/trace-context.js";
 // -----------------------------------------------------------------------------
-// text-embedding-v4
+// Text embedding models (OpenAI-compatible API)
 // -----------------------------------------------------------------------------
-const DEFAULT_QWEN_TEXT_EMBEDDING_V4_ENDPOINT = "https://dashscope.aliyuncs.com/compatible-mode/v1/embeddings";
-export class QwenTextEmbeddingV4Model extends EmbeddingModel {
-    ref = {
-        provider: "qwen",
-        model: "text-embedding-v4",
-    };
-    dimension = 1024;
-    metric = "cosine";
-    supportedContentKinds = ["text"];
-    limits = {
-        maxBatchSize: 10,
-        maxInputTokens: 8192,
-    };
-    recommendedIndexConcurrency = 8;
-    maxIndexConcurrency = 12;
+const DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS = 60_000;
+const defaultDependencies = {
+    fetch: (...args) => globalThis.fetch(...args),
+};
+const QWEN_TEXT_EMBEDDING_V4_SPEC = {
+    displayName: "Qwen text-embedding-v4",
+    errorCodePrefix: "QWEN_TEXT_EMBEDDING_V4",
+};
+const QWEN37_TEXT_EMBEDDING_SPEC = {
+    displayName: "Qwen3.7 text embedding",
+    errorCodePrefix: "QWEN37_TEXT_EMBEDDING",
+};
+class QwenTextEmbeddingModel extends BaseEmbeddingModel {
+    info;
+    entry;
     apiKey;
     endpoint;
-    constructor(options) {
+    displayName;
+    errorCodePrefix;
+    dependencies;
+    constructor(entry, spec, options, dependencies) {
         super();
-        if (options.apiKey.trim().length === 0) {
-            throw new EngineError("Qwen text-embedding-v4 model requires an API key", {
-                code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_MISSING_API_KEY",
-                context: `model=${this.ref.model}`,
+        this.entry = entry;
+        const endpoint = options.endpoint === undefined
+            ? entry.defaultEndpoint
+            : options.endpoint.trim();
+        this.info = {
+            reference: entry.reference,
+            provider: entry.provider,
+            name: entry.model,
+            dimension: entry.dimension,
+            metric: entry.metric,
+            endpoint,
+            inputKinds: ["text"],
+            limits: {
+                maxBatchSize: entry.maxBatchSize,
+                maxInputTokens: entry.maxInputTokens,
+            },
+        };
+        this.displayName = spec.displayName;
+        this.errorCodePrefix = spec.errorCodePrefix;
+        this.dependencies = { ...defaultDependencies, ...dependencies };
+        const apiKey = options.apiKey?.trim() ?? "";
+        if (apiKey.length === 0) {
+            throw new EngineError(`${this.displayName} model requires an API key`, {
+                code: this.errorCode("MISSING_API_KEY"),
+                context: `model=${this.info.reference}\nhint=Pass --api-key, set ZVEC_GREP_API_KEY, or configure providers.qwen.apiKey in ${globalConfigPath()}.`,
             });
         }
-        const endpoint = normalizeEndpoint(options.endpoint ?? DEFAULT_QWEN_TEXT_EMBEDDING_V4_ENDPOINT);
         if (endpoint.length === 0) {
-            throw new EngineError("Qwen text-embedding-v4 model requires an endpoint", {
-                code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_MISSING_ENDPOINT",
-                context: `model=${this.ref.model}`,
+            throw new EngineError(`${this.displayName} model requires an endpoint`, {
+                code: this.errorCode("MISSING_ENDPOINT"),
+                context: `model=${this.info.reference}`,
             });
         }
-        this.apiKey = options.apiKey;
+        this.apiKey = apiKey;
         this.endpoint = endpoint;
     }
-    async doEmbed(contents, _options) {
+    async doEmbed(contents, options) {
         const texts = contents.map((content) => content.text);
         let response;
+        const signal = remoteEmbeddingSignal(options.signal);
         try {
-            response = await fetch(this.endpoint, {
+            response = await this.dependencies.fetch(this.endpoint, {
                 method: "POST",
                 headers: {
+                    ...traceHeaders(),
                     Authorization: `Bearer ${this.apiKey}`,
                     "Content-Type": "application/json",
                 },
                 body: JSON.stringify({
-                    model: this.ref.model,
+                    model: this.entry.model,
                     input: texts,
-                    dimensions: this.dimension,
+                    dimensions: this.info.dimension,
                     encoding_format: "float",
                 }),
+                signal,
             });
         }
         catch (cause) {
-            throw new EngineError("Qwen text-embedding-v4 request failed", {
-                code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_REQUEST_FAILED",
-                context: `model=${this.ref.model} endpoint=${this.endpoint}`,
+            throwIfEmbeddingCancelled(options.signal);
+            throw new EngineError(`${this.displayName} request failed`, {
+                code: this.errorCode("REQUEST_FAILED"),
+                context: `model=${this.info.reference} endpoint=${this.endpoint} timeoutMs=${DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS}`,
                 cause,
             });
         }
@@ -68,97 +97,119 @@ export class QwenTextEmbeddingV4Model extends EmbeddingModel {
             body = await response.json();
         }
         catch (cause) {
-            throw new EngineError("Qwen text-embedding-v4 response was not valid JSON", {
-                code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_INVALID_JSON",
-                context: `model=${this.ref.model} status=${response.status}`,
+            throw new EngineError(`${this.displayName} response was not valid JSON`, {
+                code: this.errorCode("INVALID_JSON"),
+                context: `model=${this.info.reference} status=${response.status}`,
                 cause,
             });
         }
         if (!response.ok) {
             const error = readProviderError(body);
-            throw new EngineError("Qwen text-embedding-v4 request returned an error", {
-                code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_API_ERROR",
-                context: providerErrorContext(this.ref.model, response, error),
+            throw new EngineError(`${this.displayName} request returned an error`, {
+                code: this.errorCode("API_ERROR"),
+                context: providerErrorContext(this.entry.model, response, error),
             });
         }
         if (!isRecord(body) || !Array.isArray(body.data)) {
-            throw new EngineError("Qwen text-embedding-v4 response did not include data", {
-                code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_MISSING_DATA",
-                context: `model=${this.ref.model}`,
+            throw new EngineError(`${this.displayName} response did not include data`, {
+                code: this.errorCode("MISSING_DATA"),
+                context: `model=${this.info.reference}`,
             });
         }
         const vectors = new Array(texts.length);
         for (const item of body.data) {
-            if (!isRecord(item) || typeof item.index !== "number" || !Number.isInteger(item.index)) {
-                throw new EngineError("Qwen text-embedding-v4 response included an invalid index", {
-                    code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_INVALID_INDEX",
-                    context: `model=${this.ref.model} index=${isRecord(item) ? String(item.index) : "unknown"}`,
+            if (!isRecord(item) ||
+                typeof item.index !== "number" ||
+                !Number.isInteger(item.index)) {
+                throw new EngineError(`${this.displayName} response included an invalid index`, {
+                    code: this.errorCode("INVALID_INDEX"),
+                    context: `model=${this.info.reference} index=${isRecord(item) ? String(item.index) : "unknown"}`,
                 });
             }
             if (item.index < 0 || item.index >= texts.length) {
-                throw new EngineError("Qwen text-embedding-v4 response index was out of range", {
-                    code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_INDEX_OUT_OF_RANGE",
-                    context: `model=${this.ref.model} index=${item.index} inputCount=${texts.length}`,
+                throw new EngineError(`${this.displayName} response index was out of range`, {
+                    code: this.errorCode("INDEX_OUT_OF_RANGE"),
+                    context: `model=${this.info.reference} index=${item.index} inputCount=${texts.length}`,
                 });
             }
             if (!Array.isArray(item.embedding)) {
-                throw new EngineError("Qwen text-embedding-v4 response included an invalid embedding", {
-                    code: "ZVEC_GREP.ENGINE.MODELS.QWEN_TEXT_EMBEDDING_V4_INVALID_VECTOR",
-                    context: `model=${this.ref.model} index=${item.index}`,
+                throw new EngineError(`${this.displayName} response included an invalid embedding`, {
+                    code: this.errorCode("INVALID_VECTOR"),
+                    context: `model=${this.info.reference} index=${item.index}`,
                 });
             }
             vectors[item.index] = item.embedding;
         }
-        return vectors;
+        return { vectors, truncated: [] };
+    }
+    errorCode(suffix) {
+        return `ZVEC_GREP.ENGINE.MODELS.${this.errorCodePrefix}_${suffix}`;
+    }
+}
+export class QwenTextEmbeddingV4Model extends QwenTextEmbeddingModel {
+    constructor(entry, options, dependencies = {}) {
+        super(entry, QWEN_TEXT_EMBEDDING_V4_SPEC, options, dependencies);
+    }
+}
+export class Qwen37TextEmbeddingModel extends QwenTextEmbeddingModel {
+    constructor(entry, options, dependencies = {}) {
+        super(entry, QWEN37_TEXT_EMBEDDING_SPEC, options, dependencies);
     }
 }
 // -----------------------------------------------------------------------------
 // qwen3-vl-embedding
 // -----------------------------------------------------------------------------
-const DEFAULT_QWEN3_VL_EMBEDDING_ENDPOINT = "https://dashscope.aliyuncs.com/api/v1/services/embeddings/multimodal-embedding/multimodal-embedding";
 const QWEN3_VL_EMBEDDING_MAX_IMAGE_COUNT = 10;
 const QWEN3_VL_EMBEDDING_SUPPORTED_IMAGE_FORMATS = [
     "jpeg",
     "png",
     "webp",
 ];
-export class Qwen3VlEmbeddingModel extends EmbeddingModel {
-    ref = {
-        provider: "qwen",
-        model: "qwen3-vl-embedding",
-    };
-    dimension = 2560;
-    metric = "cosine";
-    supportedContentKinds = ["text", "image"];
-    limits = {
-        maxBatchSize: 20,
-        maxInputTokens: 32000,
-        maxImageBytes: 10 * 1024 * 1024,
-    };
-    recommendedIndexConcurrency = 4;
-    maxIndexConcurrency = 8;
+export class Qwen3VlEmbeddingModel extends BaseEmbeddingModel {
+    info;
+    entry;
     apiKey;
     endpoint;
-    constructor(options) {
+    dependencies;
+    constructor(entry, options, dependencies = {}) {
         super();
-        if (options.apiKey.trim().length === 0) {
+        this.entry = entry;
+        this.dependencies = { ...defaultDependencies, ...dependencies };
+        const endpoint = options.endpoint === undefined
+            ? entry.defaultEndpoint
+            : options.endpoint.trim();
+        this.info = {
+            reference: entry.reference,
+            provider: entry.provider,
+            name: entry.model,
+            dimension: entry.dimension,
+            metric: entry.metric,
+            endpoint,
+            inputKinds: ["text", "image"],
+            limits: {
+                maxBatchSize: entry.maxBatchSize,
+                maxInputTokens: entry.maxInputTokens,
+                maxImageBytes: entry.maxImageBytes,
+            },
+        };
+        const apiKey = options.apiKey?.trim() ?? "";
+        if (apiKey.length === 0) {
             throw new EngineError("Qwen3 VL embedding model requires an API key", {
                 code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_MISSING_API_KEY",
-                context: `model=${this.ref.model}`,
+                context: `model=${this.info.reference}\nhint=Pass --api-key, set ZVEC_GREP_API_KEY, or configure providers.qwen.apiKey in ${globalConfigPath()}.`,
             });
         }
-        const endpoint = normalizeEndpoint(options.endpoint ?? DEFAULT_QWEN3_VL_EMBEDDING_ENDPOINT);
         if (endpoint.length === 0) {
             throw new EngineError("Qwen3 VL embedding model requires an endpoint", {
                 code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_MISSING_ENDPOINT",
-                context: `model=${this.ref.model}`,
+                context: `model=${this.info.reference}`,
             });
         }
-        this.apiKey = options.apiKey;
+        this.apiKey = apiKey;
         this.endpoint = endpoint;
     }
-    async doEmbed(contents, _options) {
-        validateQwen3VlContents(this.ref.model, contents);
+    async doEmbed(contents, options) {
+        validateQwen3VlContents(this.entry.model, contents);
         const requestContents = contents.map((content) => {
             if (content.kind === "text") {
                 return {
@@ -170,28 +221,32 @@ export class Qwen3VlEmbeddingModel extends EmbeddingModel {
             };
         });
         let response;
+        const signal = remoteEmbeddingSignal(options.signal);
         try {
-            response = await fetch(this.endpoint, {
+            response = await this.dependencies.fetch(this.endpoint, {
                 method: "POST",
                 headers: {
+                    ...traceHeaders(),
                     Authorization: `Bearer ${this.apiKey}`,
                     "Content-Type": "application/json",
                 },
                 body: JSON.stringify({
-                    model: this.ref.model,
+                    model: this.entry.model,
                     input: {
                         contents: requestContents,
                     },
                     parameters: {
-                        dimension: this.dimension,
+                        dimension: this.info.dimension,
                     },
                 }),
+                signal,
             });
         }
         catch (cause) {
+            throwIfEmbeddingCancelled(options.signal);
             throw new EngineError("Qwen3 VL embedding request failed", {
                 code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_REQUEST_FAILED",
-                context: `model=${this.ref.model} endpoint=${this.endpoint}`,
+                context: `model=${this.info.reference} endpoint=${this.endpoint} timeoutMs=${DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS}`,
                 cause,
             });
         }
@@ -202,7 +257,7 @@ export class Qwen3VlEmbeddingModel extends EmbeddingModel {
         catch (cause) {
             throw new EngineError("Qwen3 VL embedding response was not valid JSON", {
                 code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INVALID_JSON",
-                context: `model=${this.ref.model} status=${response.status}`,
+                context: `model=${this.info.reference} status=${response.status}`,
                 cause,
             });
         }
@@ -210,13 +265,15 @@ export class Qwen3VlEmbeddingModel extends EmbeddingModel {
             const error = readProviderError(body);
             throw new EngineError("Qwen3 VL embedding request returned an error", {
                 code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_API_ERROR",
-                context: providerErrorContext(this.ref.model, response, error),
+                context: providerErrorContext(this.entry.model, response, error),
             });
         }
-        if (!isRecord(body) || !isRecord(body.output) || !Array.isArray(body.output.embeddings)) {
+        if (!isRecord(body) ||
+            !isRecord(body.output) ||
+            !Array.isArray(body.output.embeddings)) {
             throw new EngineError("Qwen3 VL embedding response did not include embeddings", {
                 code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_MISSING_EMBEDDINGS",
-                context: `model=${this.ref.model}`,
+                context: `model=${this.info.reference}`,
             });
         }
         const vectors = new Array(contents.length);
@@ -224,34 +281,31 @@ export class Qwen3VlEmbeddingModel extends EmbeddingModel {
             if (!isRecord(item)) {
                 throw new EngineError("Qwen3 VL embedding response included an invalid embedding item", {
                     code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INVALID_ITEM",
-                    context: `model=${this.ref.model} index=${fallbackIndex}`,
+                    context: `model=${this.info.reference} index=${fallbackIndex}`,
                 });
             }
             const index = readEmbeddingIndex(item, fallbackIndex);
             if (index < 0 || index >= contents.length) {
                 throw new EngineError("Qwen3 VL embedding response index was out of range", {
                     code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INDEX_OUT_OF_RANGE",
-                    context: `model=${this.ref.model} index=${index} inputCount=${contents.length}`,
+                    context: `model=${this.info.reference} index=${index} inputCount=${contents.length}`,
                 });
             }
             if (!Array.isArray(item.embedding)) {
                 throw new EngineError("Qwen3 VL embedding response included an invalid embedding", {
                     code: "ZVEC_GREP.ENGINE.MODELS.QWEN3_VL_EMBEDDING_INVALID_VECTOR",
-                    context: `model=${this.ref.model} index=${index}`,
+                    context: `model=${this.info.reference} index=${index}`,
                 });
             }
             vectors[index] = item.embedding;
         }
-        return vectors;
+        return { vectors, truncated: [] };
     }
 }
 // -----------------------------------------------------------------------------
 // Shared helpers
 // -----------------------------------------------------------------------------
 const BASE64_CHARS = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
-function normalizeEndpoint(endpoint) {
-    return endpoint.trim();
-}
 function providerErrorContext(model, response, error) {
     const retryAfter = retryAfterHeaderMs(response.headers.get("retry-after"));
     const retryAfterDetail = typeof retryAfter === "number" ? ` retryAfterMs=${retryAfter}` : "";
@@ -296,7 +350,8 @@ function readEmbeddingIndex(item, fallbackIndex) {
     if (typeof item.index === "number" && Number.isInteger(item.index)) {
         return item.index;
     }
-    if (typeof item.text_index === "number" && Number.isInteger(item.text_index)) {
+    if (typeof item.text_index === "number" &&
+        Number.isInteger(item.text_index)) {
         return item.text_index;
     }
     return fallbackIndex;
@@ -333,13 +388,24 @@ function bytesToBase64(bytes) {
         const third = bytes[index + 2];
         output += BASE64_CHARS.charAt(first >> 2);
         output += BASE64_CHARS.charAt(((first & 3) << 4) | ((second ?? 0) >> 4));
-        output += second === undefined
-            ? "="
-            : BASE64_CHARS.charAt(((second & 15) << 2) | ((third ?? 0) >> 6));
-        output += third === undefined
-            ? "="
-            : BASE64_CHARS.charAt(third & 63);
+        output +=
+            second === undefined
+                ? "="
+                : BASE64_CHARS.charAt(((second & 15) << 2) | ((third ?? 0) >> 6));
+        output += third === undefined ? "=" : BASE64_CHARS.charAt(third & 63);
     }
     return output;
 }
-//# sourceMappingURL=embedding.js.map
+function remoteEmbeddingSignal(signal) {
+    const timeout = AbortSignal.timeout(DEFAULT_REMOTE_EMBEDDING_TIMEOUT_MS);
+    return signal ? AbortSignal.any([signal, timeout]) : timeout;
+}
+function throwIfEmbeddingCancelled(signal) {
+    if (!signal?.aborted) {
+        return;
+    }
+    throw signal.reason instanceof Error
+        ? signal.reason
+        : new Error("Embedding request was cancelled.");
+}
+//# sourceMappingURL=qwen.js.map

package/dist/engine/models/backends/qwen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"qwen.js","sourceRoot":"","sources":["../../../../src/engine/models/backends/qwen.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,WAAW,EAAwB,MAAM,iBAAiB,CAAC;AAEpE,OAAO,EACL,kBAAkB,GAKnB,MAAM,kBAAkB,CAAC;AAK1B,OAAO,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAEvE,gFAAgF;AAChF,gDAAgD;AAChD,gFAAgF;AAEhF,MAAM,mCAAmC,GAAG,MAAM,CAAC;AAMnD,MAAM,mBAAmB,GAAqB;IAC5C,KAAK,EAAE,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;CAC9C,CAAC;AAOF,MAAM,2BAA2B,GAAG;IAClC,WAAW,EAAE,wBAAwB;IACrC,eAAe,EAAE,wBAAwB;CACD,CAAC;AAE3C,MAAM,0BAA0B,GAAG;IACjC,WAAW,EAAE,wBAAwB;IACrC,eAAe,EAAE,uBAAuB;CACA,CAAC;AAY3C,MAAe,sBAAuB,SAAQ,kBAAkB;IACrD,IAAI,CAAqB;IAEjB,KAAK,CAAgC;IACrC,MAAM,CAAS;IACf,QAAQ,CAAS;IACjB,WAAW,CAAS;IACpB,eAAe,CAAS;IACxB,YAAY,CAAmB;IAEhD,YACE,KAAoC,EACpC,IAA2B,EAC3B,OAAoC,EACpC,YAAuC;QAEvC,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,KAAK,SAAS;YAC5B,CAAC,CAAC,KAAK,CAAC,eAAe;YACvB,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG;YACV,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,KAAK;YACjB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ;YACR,UAAU,EAAE,CAAC,MAAM,CAAC;YACpB,MAAM,EAAE;gBACN,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,cAAc,EAAE,KAAK,CAAC,cAAc;aACrC;SACF,CAAC;QACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;QACpC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;QAC5C,IAAI,CAAC,YAAY,GAAG,EAAE,GAAG,mBAAmB,EAAE,GAAG,YAAY,EAAE,CAAC;QAEhE,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC5C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,4BAA4B,EAAE;gBACrE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC;gBACvC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,uFAAuF,gBAAgB,EAAE,GAAG;aAClJ,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,6BAA6B,EAAE;gBACtE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,kBAAkB,CAAC;gBACxC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAES,KAAK,CAAC,OAAO,CACrB,QAA4B,EAC5B,OAAmC;QAEnC,MAAM,KAAK,GAAI,QAAmC,CAAC,GAAG,CACpD,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAC1B,CAAC;QAEF,IAAI,QAAkB,CAAC;QACvB,MAAM,MAAM,GAAG,qBAAqB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAErD,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,YAAY,EAAE;oBACjB,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACtC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;oBACvB,KAAK,EAAE,KAAK;oBACZ,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;oBAC/B,eAAe,EAAE,OAAO;iBACzB,CAAC;gBACF,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,iBAAiB,EAAE;gBAC1D,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC;gBACtC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,aAAa,IAAI,CAAC,QAAQ,cAAc,mCAAmC,EAAE;gBAClH,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAa,CAAC;QAElB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,8BAA8B,EAAE;gBACvE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;gBACpC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,WAAW,QAAQ,CAAC,MAAM,EAAE;gBACjE,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEtC,MAAM,IAAI,WAAW,CAAC,GAAG,IAAI,CAAC,WAAW,4BAA4B,EAAE;gBACrE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;gBACjC,OAAO,EAAE,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC;aACjE,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACjD,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,gCAAgC,EACnD;gBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC;gBACpC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CACF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,KAAK,CAAW,KAAK,CAAC,MAAM,CAAC,CAAC;QAElD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;YAC7B,IACE,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACf,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ;gBAC9B,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAC7B,CAAC;gBACD,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,qCAAqC,EACxD;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC;oBACrC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE;iBACjG,CACF,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,CAAC,KAAK,GAAG,CAAC,IAAI,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;gBACjD,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,kCAAkC,EACrD;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC;oBAC1C,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,IAAI,CAAC,KAAK,eAAe,KAAK,CAAC,MAAM,EAAE;iBACvF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,WAAW,CACnB,GAAG,IAAI,CAAC,WAAW,yCAAyC,EAC5D;oBACE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC;oBACtC,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,IAAI,CAAC,KAAK,EAAE;iBAC5D,CACF,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAqB,CAAC;QACnD,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IACpC,CAAC;IAEO,SAAS,CAAC,MAAc;QAC9B,OAAO,2BAA2B,IAAI,CAAC,eAAe,IAAI,MAAM,EAAE,CAAC;IACrE,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,sBAAsB;IAClE,YACE,KAAsC,EACtC,OAAoC,EACpC,eAA0C,EAAE;QAE5C,KAAK,CAAC,KAAK,EAAE,2BAA2B,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;IACnE,CAAC;CACF;AAED,MAAM,OAAO,wBAAyB,SAAQ,sBAAsB;IAClE,YACE,KAAsC,EACtC,OAAoC,EACpC,eAA0C,EAAE;QAE5C,KAAK,CAAC,KAAK,EAAE,0BAA0B,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;IAClE,CAAC;CACF;AAED,gFAAgF;AAChF,qBAAqB;AACrB,gFAAgF;AAEhF,MAAM,kCAAkC,GAAG,EAAE,CAAC;AAE9C,MAAM,0CAA0C,GAA2B;IACzE,MAAM;IACN,KAAK;IACL,MAAM;CACP,CAAC;AAEF,MAAM,OAAO,qBAAsB,SAAQ,kBAAkB;IAClD,IAAI,CAAqB;IAEjB,KAAK,CAAsC;IAC3C,MAAM,CAAS;IACf,QAAQ,CAAS;IACjB,YAAY,CAAmB;IAEhD,YACE,KAA0C,EAC1C,OAAoC,EACpC,eAA0C,EAAE;QAE5C,KAAK,EAAE,CAAC;QAER,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,YAAY,GAAG,EAAE,GAAG,mBAAmB,EAAE,GAAG,YAAY,EAAE,CAAC;QAChE,MAAM,QAAQ,GACZ,OAAO,CAAC,QAAQ,KAAK,SAAS;YAC5B,CAAC,CAAC,KAAK,CAAC,eAAe;YACvB,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC9B,IAAI,CAAC,IAAI,GAAG;YACV,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,QAAQ,EAAE,KAAK,CAAC,QAAQ;YACxB,IAAI,EAAE,KAAK,CAAC,KAAK;YACjB,SAAS,EAAE,KAAK,CAAC,SAAS;YAC1B,MAAM,EAAE,KAAK,CAAC,MAAM;YACpB,QAAQ;YACR,UAAU,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;YAC7B,MAAM,EAAE;gBACN,YAAY,EAAE,KAAK,CAAC,YAAY;gBAChC,cAAc,EAAE,KAAK,CAAC,cAAc;gBACpC,aAAa,EAAE,KAAK,CAAC,aAAa;aACnC;SACF,CAAC;QAEF,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC5C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,WAAW,CAAC,8CAA8C,EAAE;gBACpE,IAAI,EAAE,4DAA4D;gBAClE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,uFAAuF,gBAAgB,EAAE,GAAG;aAClJ,CAAC,CAAC;QACL,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,WAAW,CAAC,+CAA+C,EAAE;gBACrE,IAAI,EAAE,6DAA6D;gBACnE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAES,KAAK,CAAC,OAAO,CACrB,QAA4B,EAC5B,OAAmC;QAEnC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAEpD,MAAM,eAAe,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;YAC/C,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC5B,OAAO;oBACL,IAAI,EAAE,OAAO,CAAC,IAAI;iBACnB,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,KAAK,EAAE,aAAa,CAAC,OAAO,CAAC,IAAI,CAAC;aACnC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,IAAI,QAAkB,CAAC;QACvB,MAAM,MAAM,GAAG,qBAAqB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAErD,IAAI,CAAC;YACH,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACtD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,GAAG,YAAY,EAAE;oBACjB,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;oBACtC,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;oBACvB,KAAK,EAAE;wBACL,QAAQ,EAAE,eAAe;qBAC1B;oBACD,UAAU,EAAE;wBACV,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,SAAS;qBAC/B;iBACF,CAAC;gBACF,MAAM;aACP,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,yBAAyB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,IAAI,WAAW,CAAC,mCAAmC,EAAE;gBACzD,IAAI,EAAE,2DAA2D;gBACjE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,aAAa,IAAI,CAAC,QAAQ,cAAc,mCAAmC,EAAE;gBAClH,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,IAAa,CAAC;QAElB,IAAI,CAAC;YACH,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,WAAW,CAAC,gDAAgD,EAAE;gBACtE,IAAI,EAAE,yDAAyD;gBAC/D,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,WAAW,QAAQ,CAAC,MAAM,EAAE;gBACjE,KAAK;aACN,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEtC,MAAM,IAAI,WAAW,CAAC,8CAA8C,EAAE;gBACpE,IAAI,EAAE,sDAAsD;gBAC5D,OAAO,EAAE,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,QAAQ,EAAE,KAAK,CAAC;aACjE,CAAC,CAAC;QACL,CAAC;QAED,IACE,CAAC,QAAQ,CAAC,IAAI,CAAC;YACf,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC;YACtB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,EACtC,CAAC;YACD,MAAM,IAAI,WAAW,CACnB,wDAAwD,EACxD;gBACE,IAAI,EAAE,+DAA+D;gBACrE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;aACxC,CACF,CAAC;QACJ,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,KAAK,CAAW,QAAQ,CAAC,MAAM,CAAC,CAAC;QAErD,KAAK,MAAM,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACrE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBACpB,MAAM,IAAI,WAAW,CACnB,gEAAgE,EAChE;oBACE,IAAI,EAAE,yDAAyD;oBAC/D,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,aAAa,EAAE;iBAC/D,CACF,CAAC;YACJ,CAAC;YAED,MAAM,KAAK,GAAG,kBAAkB,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;YAEtD,IAAI,KAAK,GAAG,CAAC,IAAI,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBAC1C,MAAM,IAAI,WAAW,CACnB,oDAAoD,EACpD;oBACE,IAAI,EAAE,+DAA+D;oBACrE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,KAAK,eAAe,QAAQ,CAAC,MAAM,EAAE;iBACrF,CACF,CAAC;YACJ,CAAC;YAED,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC;gBACnC,MAAM,IAAI,WAAW,CACnB,2DAA2D,EAC3D;oBACE,IAAI,EAAE,2DAA2D;oBACjE,OAAO,EAAE,SAAS,IAAI,CAAC,IAAI,CAAC,SAAS,UAAU,KAAK,EAAE;iBACvD,CACF,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,SAAqB,CAAC;QAC9C,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC;IACpC,CAAC;CACF;AAED,gFAAgF;AAChF,iBAAiB;AACjB,gFAAgF;AAEhF,MAAM,YAAY,GAChB,kEAAkE,CAAC;AAErE,SAAS,oBAAoB,CAC3B,KAAa,EACb,QAAkB,EAClB,KAAsD;IAEtD,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;IAC3E,MAAM,gBAAgB,GACpB,OAAO,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,iBAAiB,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAEtE,OAAO,SAAS,KAAK,WAAW,QAAQ,CAAC,MAAM,GAAG,gBAAgB,iBAAiB,KAAK,CAAC,IAAI,iBAAiB,KAAK,CAAC,IAAI,oBAAoB,KAAK,CAAC,OAAO,EAAE,CAAC;AAC9J,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAoB;IAC9C,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC9B,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;QAC7C,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAa;IAKtC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QAC7C,IAAI,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACnB,OAAO;gBACL,IAAI,EAAE,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;gBAC3D,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,OAAO,IAAI,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;aACrE,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI,EAAE,SAAS;YACf,IAAI,EAAE,SAAS;YACf,OAAO,EAAE,SAAS;SACnB,CAAC;IACJ,CAAC;IAED,OAAO;QACL,IAAI,EAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;QACvE,IAAI,EAAE,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;QACvE,OAAO,EACL,OAAO,IAAI,CAAC,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;KAC1E,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CACzB,IAA6B,EAC7B,aAAqB;IAErB,IAAI,OAAO,IAAI,CAAC,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACnE,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED,IACE,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ;QACnC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC,EACjC,CAAC;QACD,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;IAED,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAS,uBAAuB,CAC9B,KAAa,EACb,QAA4B;IAE5B,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;QAClD,IAAI,OAAO,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC7B,SAAS;QACX,CAAC;QAED,UAAU,IAAI,CAAC,CAAC;QAEhB,IAAI,CAAC,0CAA0C,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACzE,MAAM,IAAI,WAAW,CACnB,wDAAwD,EACxD;gBACE,IAAI,EAAE,qEAAqE;gBAC3E,OAAO,EAAE,SAAS,KAAK,UAAU,KAAK,WAAW,OAAO,CAAC,MAAM,EAAE;aAClE,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,UAAU,GAAG,kCAAkC,EAAE,CAAC;QACpD,MAAM,IAAI,WAAW,CACnB,oDAAoD,EACpD;YACE,IAAI,EAAE,4DAA4D;YAClE,OAAO,EAAE,SAAS,KAAK,eAAe,UAAU,kBAAkB,kCAAkC,EAAE;SACvG,CACF,CAAC;IACJ,CAAC;AACH,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,aAAa,CAAC,KAAiB;IACtC,IAAI,MAAM,GAAG,EAAE,CAAC;IAEhB,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QACrD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAChC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QAChC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;QAE/B,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;QAC1C,MAAM,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACzE,MAAM;YACJ,MAAM,KAAK,SAAS;gBAClB,CAAC,CAAC,GAAG;gBACL,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtE,MAAM,IAAI,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC;IACxE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,qBAAqB,CAAC,MAA+B;IAC5D,MAAM,OAAO,GAAG,WAAW,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAC;IACzE,OAAO,MAAM,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;AAC/D,CAAC;AAED,SAAS,yBAAyB,CAAC,MAA+B;IAChE,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;QACrB,OAAO;IACT,CAAC;IACD,MAAM,MAAM,CAAC,MAAM,YAAY,KAAK;QAClC,CAAC,CAAC,MAAM,CAAC,MAAM;QACf,CAAC,CAAC,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;AACpD,CAAC"}

package/dist/engine/models/backends/transformers-js.d.ts

@@ -0,0 +1,87 @@
+import type { Content } from "../../types.js";
+import { BaseEmbeddingModel, type CreateEmbeddingModelOptions, type EmbeddingModelInfo, type EmbeddingResult, type NormalizedEmbeddingOptions } from "../embeddings.js";
+import type { TransformersJsEmbeddingCatalogEntry } from "../catalog.js";
+type TensorLike = {
+    data: ArrayLike<number>;
+    dims: readonly number[];
+};
+type TokenizerLike = {
+    (text: string | string[], options: {
+        truncation: true;
+        max_length: number;
+        padding: true;
+    }): {
+        input_ids: {
+            data: ArrayLike<number | bigint>;
+            dims: readonly number[];
+        };
+        attention_mask: {
+            data: ArrayLike<number | bigint>;
+            dims: readonly number[];
+        };
+    } | Promise<{
+        input_ids: {
+            data: ArrayLike<number | bigint>;
+            dims: readonly number[];
+        };
+        attention_mask: {
+            data: ArrayLike<number | bigint>;
+            dims: readonly number[];
+        };
+    }>;
+    model_max_length: number;
+};
+type FeatureExtractionPipeline = {
+    (texts: string[], options: {
+        pooling: "mean" | "cls";
+        normalize: boolean;
+        truncation: true;
+        max_length: number;
+    }): Promise<TensorLike>;
+    tokenizer: TokenizerLike;
+    dispose(): Promise<void>;
+};
+type TransformersJsModule = {
+    pipeline(task: "feature-extraction", repo: string, options: {
+        cache_dir: string;
+        revision: string;
+        dtype: "fp32" | "q8" | "q4";
+        progress_callback?: (progress: TransformersJsProgressInfo) => void;
+        session_options?: {
+            executionProviders: TransformersJsExecutionProvider[];
+        };
+    }): Promise<FeatureExtractionPipeline>;
+};
+type TransformersJsProgressInfo = {
+    status: "initiate" | "download" | "progress" | "done" | "ready";
+    file?: string;
+    loaded?: number;
+    total?: number;
+};
+type TransformersJsLoader = () => Promise<TransformersJsModule>;
+type TransformersJsExecutionProvider = "cpu" | "webgpu" | "cuda" | "dml";
+type TransformersJsDependencies = {
+    loadRuntime: TransformersJsLoader;
+};
+export declare class TransformersJsEmbeddingModel extends BaseEmbeddingModel {
+    private readonly entry;
+    readonly info: EmbeddingModelInfo;
+    private readonly modelCacheDir;
+    private readonly executionProvider;
+    private readonly dependencies;
+    private pipeline;
+    private pipelineLoadPromise;
+    private usingCpuFallback;
+    private disposed;
+    constructor(entry: TransformersJsEmbeddingCatalogEntry, options: CreateEmbeddingModelOptions, dependencies?: Partial<TransformersJsDependencies>);
+    protected doEmbed(contents: readonly Content[], options: NormalizedEmbeddingOptions): Promise<EmbeddingResult>;
+    private embedBatch;
+    dispose(): Promise<void>;
+    private ensurePipeline;
+    private loadPipeline;
+    private embedTexts;
+    private fallbackToCpu;
+    private createPipeline;
+    private ensureNotDisposed;
+}
+export {};