grepmind-core 0.1.0-alpha

package/dist/indexer/embeddings.d.ts

@@ -0,0 +1,113 @@
+import type { EmbeddingConfig } from '../config/types.js';
+import { type NotesError } from '../types.js';
+/**
+ * Get dimensions for a model, with fallback to config or default.
+ */
+export declare function getModelDimensions(model: string, configDimensions?: number): number;
+/**
+ * Embedding provider interface.
+ * All embedding providers implement this interface.
+ */
+export interface EmbeddingProvider {
+    /**
+     * Initialize the provider (load models, connect to APIs, etc.)
+     */
+    initialize(): Promise<void>;
+    /**
+     * Get embedding for a single text.
+     * @param text - Text to embed
+     * @param inputType - 'document' for indexing, 'query' for search
+     */
+    getEmbedding(text: string, inputType?: 'document' | 'query'): Promise<number[] | NotesError>;
+    /**
+     * Get embeddings for multiple texts in a batch.
+     * @param texts - Array of texts to embed
+     * @param batchSize - Batch size for processing
+     * @param inputType - 'document' for indexing, 'query' for search
+     * @param onProgress - Optional callback for batch progress
+     */
+    getEmbeddingsBatch(texts: string[], batchSize?: number, inputType?: 'document' | 'query', onProgress?: (current: number, total: number) => void): Promise<number[][] | NotesError>;
+    /**
+     * Get the model name for this provider.
+     */
+    getModelName(): string;
+    /**
+     * Get the embedding dimensions for this provider.
+     */
+    getDimensions(): number;
+}
+/**
+ * WASM embedding provider using Xenova Transformers.js
+ * Runs models in Node.js via WebAssembly
+ */
+export declare class WasmEmbeddingProvider implements EmbeddingProvider {
+    private config;
+    private dimensions;
+    private extractor;
+    private initPromise;
+    constructor(config: EmbeddingConfig);
+    initialize(): Promise<void>;
+    private getExtractor;
+    getEmbedding(text: string, inputType?: 'document' | 'query'): Promise<number[] | NotesError>;
+    getEmbeddingsBatch(texts: string[], batchSize?: number, inputType?: 'document' | 'query', onProgress?: (current: number, total: number) => void): Promise<number[][] | NotesError>;
+    getModelName(): string;
+    getDimensions(): number;
+}
+/**
+ * Voyage AI embedding provider.
+ * Uses the Voyage AI API for embeddings (recommended for code).
+ */
+export declare class VoyageEmbeddingProvider implements EmbeddingProvider {
+    private config;
+    private dimensions;
+    private client;
+    private queue;
+    private dimWarningShown;
+    constructor(config: EmbeddingConfig);
+    initialize(): Promise<void>;
+    private getClient;
+    private getQueue;
+    private embedBatch;
+    getEmbedding(text: string, inputType?: 'document' | 'query'): Promise<number[] | NotesError>;
+    getEmbeddingsBatch(texts: string[], batchSize?: number, inputType?: 'document' | 'query', onProgress?: (current: number, total: number) => void): Promise<number[][] | NotesError>;
+    getModelName(): string;
+    getDimensions(): number;
+}
+/**
+ * Ollama embedding provider.
+ * Uses local Ollama server for embeddings.
+ */
+export declare class OllamaEmbeddingProvider implements EmbeddingProvider {
+    private config;
+    private dimensions;
+    private baseUrl;
+    constructor(config: EmbeddingConfig);
+    initialize(): Promise<void>;
+    private embedBatch;
+    getEmbedding(text: string, inputType?: 'document' | 'query'): Promise<number[] | NotesError>;
+    getEmbeddingsBatch(texts: string[], batchSize?: number, // Ollama handles large batches well with GPU
+    inputType?: 'document' | 'query', onProgress?: (current: number, total: number) => void): Promise<number[][] | NotesError>;
+    getModelName(): string;
+    getDimensions(): number;
+}
+/**
+ * LocalAI embedding provider.
+ * Uses OpenAI-compatible `/v1/embeddings` endpoint.
+ */
+export declare class LocalAIEmbeddingProvider implements EmbeddingProvider {
+    private config;
+    private dimensions;
+    private baseUrl;
+    constructor(config: EmbeddingConfig);
+    initialize(): Promise<void>;
+    private embedBatch;
+    getEmbedding(text: string, inputType?: 'document' | 'query'): Promise<number[] | NotesError>;
+    getEmbeddingsBatch(texts: string[], batchSize?: number, inputType?: 'document' | 'query', onProgress?: (current: number, total: number) => void): Promise<number[][] | NotesError>;
+    getModelName(): string;
+    getDimensions(): number;
+}
+/**
+ * Factory function to create an embedding provider based on config.
+ */
+export declare function createEmbeddingProvider(config: EmbeddingConfig): EmbeddingProvider;
+//# sourceMappingURL=embeddings.d.ts.map

package/dist/indexer/embeddings.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.d.ts","sourceRoot":"","sources":["../../src/indexer/embeddings.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAC1D,OAAO,EAAe,KAAK,UAAU,EAAE,MAAM,aAAa,CAAC;AA2B3D;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,gBAAgB,CAAC,EAAE,MAAM,GAAG,MAAM,CAEnF;AAED;;;GAGG;AACH,MAAM,WAAW,iBAAiB;IAChC;;OAEG;IACH,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;IAE5B;;;;OAIG;IACH,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,UAAU,GAAG,OAAO,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,UAAU,CAAC,CAAC;IAE7F;;;;;;OAMG;IACH,kBAAkB,CAChB,KAAK,EAAE,MAAM,EAAE,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,CAAC,EAAE,UAAU,GAAG,OAAO,EAChC,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACpD,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,UAAU,CAAC,CAAC;IAEpC;;OAEG;IACH,YAAY,IAAI,MAAM,CAAC;IAEvB;;OAEG;IACH,aAAa,IAAI,MAAM,CAAC;CACzB;AAED;;;GAGG;AACH,qBAAa,qBAAsB,YAAW,iBAAiB;IAC7D,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,SAAS,CAA0C;IAC3D,OAAO,CAAC,WAAW,CAAmD;gBAE1D,MAAM,EAAE,eAAe;IAU7B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YAInB,YAAY;IAapB,YAAY,CAChB,IAAI,EAAE,MAAM,EACZ,SAAS,GAAE,UAAU,GAAG,OAAoB,GAC3C,OAAO,CAAC,MAAM,EAAE,GAAG,UAAU,CAAC;IAqB3B,kBAAkB,CACtB,KAAK,EAAE,MAAM,EAAE,EACf,SAAS,GAAE,MAAW,EACtB,SAAS,GAAE,UAAU,GAAG,OAAoB,EAC5C,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACpD,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,UAAU,CAAC;IAqDnC,YAAY,IAAI,MAAM;IAItB,aAAa,IAAI,MAAM;CAGxB;AAMD;;;GAGG;AACH,qBAAa,uBAAwB,YAAW,iBAAiB;IAC/D,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,MAAM,CAA+B;IAC7C,OAAO,CAAC,KAAK,CAAuB;IACpC,OAAO,CAAC,eAAe,CAAS;gBAEpB,MAAM,EAAE,eAAe;IAM7B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAIjC,OAAO,CAAC,SAAS;IAUjB,OAAO,CAAC,QAAQ;YAOF,UAAU;IAmDlB,YAAY,CAChB,IAAI,EAAE,MAAM,EACZ,SAAS,GAAE,UAAU,GAAG,OAAoB,GAC3C,OAAO,CAAC,MAAM,EAAE,GAAG,UAAU,CAAC;IAM3B,kBAAkB,CACtB,KAAK,EAAE,MAAM,EAAE,EACf,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,GAAE,UAAU,GAAG,OAAoB,EAC5C,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACpD,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,UAAU,CAAC;IA2DnC,YAAY,IAAI,MAAM;IAItB,aAAa,IAAI,MAAM;CAGxB;AAED;;;GAGG;AACH,qBAAa,uBAAwB,YAAW,iBAAiB;IAC/D,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,eAAe;IAM7B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YAenB,UAAU;IAyDlB,YAAY,CAChB,IAAI,EAAE,MAAM,EACZ,SAAS,GAAE,UAAU,GAAG,OAAoB,GAC3C,OAAO,CAAC,MAAM,EAAE,GAAG,UAAU,CAAC;IAM3B,kBAAkB,CACtB,KAAK,EAAE,MAAM,EAAE,EACf,SAAS,GAAE,MAAY,EAAE,6CAA6C;IACtE,SAAS,GAAE,UAAU,GAAG,OAAoB,EAC5C,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACpD,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,UAAU,CAAC;IAsCnC,YAAY,IAAI,MAAM;IAItB,aAAa,IAAI,MAAM;CAGxB;AAED;;;GAGG;AACH,qBAAa,wBAAyB,YAAW,iBAAiB;IAChE,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,UAAU,CAAS;IAC3B,OAAO,CAAC,OAAO,CAAS;gBAEZ,MAAM,EAAE,eAAe;IAM7B,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;YAcnB,UAAU;IAgDlB,YAAY,CAChB,IAAI,EAAE,MAAM,EACZ,SAAS,GAAE,UAAU,GAAG,OAAoB,GAC3C,OAAO,CAAC,MAAM,EAAE,GAAG,UAAU,CAAC;IAM3B,kBAAkB,CACtB,KAAK,EAAE,MAAM,EAAE,EACf,SAAS,GAAE,MAAY,EACvB,SAAS,GAAE,UAAU,GAAG,OAAoB,EAC5C,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,GACpD,OAAO,CAAC,MAAM,EAAE,EAAE,GAAG,UAAU,CAAC;IAqCnC,YAAY,IAAI,MAAM;IAItB,aAAa,IAAI,MAAM;CAGxB;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CAAC,MAAM,EAAE,eAAe,GAAG,iBAAiB,CAWlF"}

package/dist/indexer/embeddings.js

@@ -0,0 +1,477 @@
+// Embedding providers for semantic search
+// Supports WASM (Xenova Transformers.js), Voyage AI, Ollama, and LocalAI providers
+import { pipeline, env } from '@xenova/transformers';
+import { VoyageAIClient, VoyageAIError } from 'voyageai';
+import PQueue from 'p-queue';
+import { createError } from '../types.js';
+/**
+ * Known model dimensions.
+ * Used to auto-detect dimensions when not explicitly configured.
+ */
+const MODEL_DIMENSIONS = {
+    // Ollama models
+    'nomic-embed-text': 768,
+    'mxbai-embed-large': 1024,
+    'all-minilm': 384,
+    'snowflake-arctic-embed': 1024,
+    'bge-m3': 1024,
+    'bge-large': 1024,
+    // WASM / Transformers.js models
+    'nomic-ai/nomic-embed-text-v1.5': 768,
+    'Xenova/all-MiniLM-L6-v2': 384,
+    'Xenova/bge-small-en-v1.5': 384,
+    'Xenova/bge-base-en-v1.5': 768,
+    // Voyage models
+    'voyage-3': 1024,
+    'voyage-3-lite': 512,
+    'voyage-code-3': 1024,
+    'voyage-finance-2': 1024,
+    'voyage-law-2': 1024,
+};
+/**
+ * Get dimensions for a model, with fallback to config or default.
+ */
+export function getModelDimensions(model, configDimensions) {
+    return MODEL_DIMENSIONS[model] ?? configDimensions ?? 768;
+}
+/**
+ * WASM embedding provider using Xenova Transformers.js
+ * Runs models in Node.js via WebAssembly
+ */
+export class WasmEmbeddingProvider {
+    config;
+    dimensions;
+    extractor = null;
+    initPromise = null;
+    constructor(config) {
+        this.config = config;
+        this.dimensions = getModelDimensions(config.model, config.dimensions);
+        // Configure transformers.js cache directory
+        if (config.cacheDir) {
+            env.cacheDir = config.cacheDir;
+        }
+    }
+    async initialize() {
+        await this.getExtractor();
+    }
+    async getExtractor() {
+        if (this.extractor)
+            return this.extractor;
+        if (!this.initPromise) {
+            this.initPromise = pipeline('feature-extraction', this.config.model, {
+                quantized: true,
+            });
+        }
+        this.extractor = await this.initPromise;
+        return this.extractor;
+    }
+    async getEmbedding(text, inputType = 'document') {
+        try {
+            const ext = await this.getExtractor();
+            const output = await ext(text, { pooling: 'mean', normalize: true });
+            const fullVector = Array.from(output.data);
+            // Apply Matryoshka dimension truncation if needed
+            if (this.dimensions < fullVector.length) {
+                return fullVector.slice(0, this.dimensions);
+            }
+            return fullVector;
+        }
+        catch (error) {
+            return createError('EMBEDDING_FAILED', `Failed to get embedding: ${error.message}`, { error: error.message });
+        }
+    }
+    async getEmbeddingsBatch(texts, batchSize = 32, inputType = 'document', onProgress) {
+        if (texts.length === 0)
+            return [];
+        try {
+            const ext = await this.getExtractor();
+            const allEmbeddings = [];
+            const totalBatches = Math.ceil(texts.length / batchSize);
+            for (let i = 0; i < texts.length; i += batchSize) {
+                const batchNum = Math.floor(i / batchSize) + 1;
+                const batch = texts.slice(i, i + batchSize);
+                if (texts.length > 50) {
+                    process.stderr.write(`\r  Embedding batch ${batchNum}/${totalBatches}...`);
+                }
+                // Call progress callback if provided
+                if (onProgress) {
+                    onProgress(batchNum, totalBatches);
+                }
+                const output = await ext(batch, { pooling: 'mean', normalize: true });
+                const data = output.data;
+                const vectorSize = output.dims[1];
+                for (let j = 0; j < batch.length; j++) {
+                    const start = j * vectorSize;
+                    const end = start + vectorSize;
+                    let vector = Array.from(data.slice(start, end));
+                    if (this.dimensions < vector.length) {
+                        vector = vector.slice(0, this.dimensions);
+                    }
+                    allEmbeddings.push(vector);
+                }
+            }
+            if (texts.length > 50) {
+                process.stderr.write('\r                                      \r');
+            }
+            return allEmbeddings;
+        }
+        catch (error) {
+            return createError('EMBEDDING_FAILED', `Failed to get batch embeddings: ${error.message}`, { error: error.message });
+        }
+    }
+    getModelName() {
+        return this.config.model;
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+}
+// Voyage-supported output dimensions (voyage-code-3)
+const VOYAGE_SUPPORTED_DIMS = [256, 512, 1024, 2048];
+const VOYAGE_CODE_3_DEFAULT_DIM = 1024;
+/**
+ * Voyage AI embedding provider.
+ * Uses the Voyage AI API for embeddings (recommended for code).
+ */
+export class VoyageEmbeddingProvider {
+    config;
+    dimensions;
+    client = null;
+    queue = null;
+    dimWarningShown = false;
+    constructor(config) {
+        this.config = config;
+        this.dimensions = getModelDimensions(config.voyageModel, config.dimensions);
+        // Note: API key is validated lazily in getClient() to allow project opening without secrets
+    }
+    async initialize() {
+        // Voyage doesn't need initialization - client is created lazily on first use
+    }
+    getClient() {
+        if (!this.client) {
+            if (!this.config.voyageApiKey) {
+                throw new Error('VOYAGE_API_KEY is required for Voyage provider. Please add it in Settings.');
+            }
+            this.client = new VoyageAIClient({ apiKey: this.config.voyageApiKey });
+        }
+        return this.client;
+    }
+    getQueue() {
+        if (!this.queue) {
+            this.queue = new PQueue({ concurrency: this.config.voyageConcurrency });
+        }
+        return this.queue;
+    }
+    async embedBatch(texts, inputType = 'document') {
+        try {
+            const client = this.getClient();
+            // Determine output dimension (Voyage only supports specific values)
+            let outputDimension;
+            if (VOYAGE_SUPPORTED_DIMS.includes(this.dimensions)) {
+                outputDimension = this.dimensions;
+            }
+            else if (!this.dimWarningShown) {
+                this.dimWarningShown = true;
+                process.stderr.write(`Warning: dimensions=${this.dimensions} not supported by Voyage. ` +
+                    `Using default ${VOYAGE_CODE_3_DEFAULT_DIM}. Supported: ${VOYAGE_SUPPORTED_DIMS.join(', ')}\n`);
+            }
+            const response = await client.embed({
+                input: texts,
+                model: this.config.voyageModel,
+                inputType: inputType,
+                truncation: true,
+                outputDimension,
+            });
+            if (!response.data) {
+                return createError('EMBEDDING_FAILED', 'Voyage API returned no data');
+            }
+            const sorted = [...response.data].sort((a, b) => (a.index ?? 0) - (b.index ?? 0));
+            const embeddings = sorted.map((item) => item.embedding ?? []);
+            return embeddings;
+        }
+        catch (error) {
+            if (error instanceof VoyageAIError) {
+                return createError('EMBEDDING_FAILED', `Voyage API error ${error.statusCode}: ${error.message}`);
+            }
+            return createError('EMBEDDING_FAILED', `Voyage API request failed: ${error.message}`, { error: error.message });
+        }
+    }
+    async getEmbedding(text, inputType = 'document') {
+        const result = await this.embedBatch([text], inputType);
+        if ('code' in result)
+            return result;
+        return result[0];
+    }
+    async getEmbeddingsBatch(texts, batchSize, inputType = 'document', onProgress) {
+        if (texts.length === 0)
+            return [];
+        const effectiveBatchSize = batchSize ?? this.config.voyageBatchSize;
+        const totalBatches = Math.ceil(texts.length / effectiveBatchSize);
+        const queue = this.getQueue();
+        // Create batches with their indices
+        const batches = [];
+        for (let i = 0; i < texts.length; i += effectiveBatchSize) {
+            batches.push({
+                index: batches.length,
+                texts: texts.slice(i, i + effectiveBatchSize),
+            });
+        }
+        // Track progress
+        let completed = 0;
+        const showProgress = texts.length > 50;
+        // Process batches concurrently with queue
+        const results = await Promise.all(batches.map((batch) => queue.add(async () => {
+            const result = await this.embedBatch(batch.texts, inputType);
+            completed++;
+            if (showProgress) {
+                process.stderr.write(`\r  Voyage: ${completed}/${totalBatches} batches (${this.config.voyageConcurrency} concurrent)...`);
+            }
+            // Call progress callback if provided
+            if (onProgress) {
+                onProgress(completed, totalBatches);
+            }
+            return { index: batch.index, result };
+        })));
+        // Clear progress line
+        if (showProgress) {
+            process.stderr.write('\r                                                            \r');
+        }
+        // Check for errors and sort by original index
+        const sortedResults = results.sort((a, b) => a.index - b.index);
+        const allEmbeddings = [];
+        for (const item of sortedResults) {
+            if (!item || 'code' in item.result) {
+                return item?.result ?? createError('EMBEDDING_FAILED', 'Unknown error');
+            }
+            allEmbeddings.push(...item.result);
+        }
+        return allEmbeddings;
+    }
+    getModelName() {
+        return `voyage:${this.config.voyageModel}`;
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+}
+/**
+ * Ollama embedding provider.
+ * Uses local Ollama server for embeddings.
+ */
+export class OllamaEmbeddingProvider {
+    config;
+    dimensions;
+    baseUrl;
+    constructor(config) {
+        this.config = config;
+        this.dimensions = getModelDimensions(config.ollamaModel, config.dimensions);
+        this.baseUrl = config.ollamaBaseUrl.replace(/\/$/, '');
+    }
+    async initialize() {
+        // Check server availability
+        try {
+            const response = await fetch(`${this.baseUrl}/api/tags`);
+            if (!response.ok) {
+                throw new Error(`Ollama server returned ${response.status}`);
+            }
+        }
+        catch (error) {
+            throw new Error(`Cannot connect to Ollama at ${this.baseUrl}. ` +
+                `Ensure Ollama is running: ${error.message}`);
+        }
+    }
+    async embedBatch(texts) {
+        try {
+            const startTime = Date.now();
+            const totalChars = texts.reduce((sum, t) => sum + t.length, 0);
+            const response = await fetch(`${this.baseUrl}/api/embed`, {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({
+                    model: this.config.ollamaModel,
+                    input: texts,
+                    keep_alive: '30m', // Keep model in GPU memory
+                }),
+            });
+            const fetchTime = Date.now() - startTime;
+            if (!response.ok) {
+                const errorText = await response.text();
+                return createError('EMBEDDING_FAILED', `Ollama API error ${response.status}: ${errorText}`);
+            }
+            const data = (await response.json());
+            if (!data.embeddings || !Array.isArray(data.embeddings)) {
+                return createError('EMBEDDING_FAILED', 'Ollama API returned no embeddings');
+            }
+            // Log timing info
+            const loadMs = data.load_duration ? Math.round(data.load_duration / 1_000_000) : 0;
+            const totalMs = data.total_duration ? Math.round(data.total_duration / 1_000_000) : 0;
+            process.stderr.write(`\n  [ollama] ${texts.length} texts, ${totalChars} chars, ` +
+                `fetch=${fetchTime}ms, ollama_total=${totalMs}ms, load=${loadMs}ms\n`);
+            // Apply Matryoshka dimension truncation if needed
+            return data.embeddings.map((vec) => this.dimensions < vec.length ? vec.slice(0, this.dimensions) : vec);
+        }
+        catch (error) {
+            return createError('EMBEDDING_FAILED', `Ollama API request failed: ${error.message}`, { error: error.message });
+        }
+    }
+    async getEmbedding(text, inputType = 'document') {
+        const result = await this.embedBatch([text]);
+        if ('code' in result)
+            return result;
+        return result[0];
+    }
+    async getEmbeddingsBatch(texts, batchSize = 128, // Ollama handles large batches well with GPU
+    inputType = 'document', onProgress) {
+        if (texts.length === 0)
+            return [];
+        const allEmbeddings = [];
+        const totalBatches = Math.ceil(texts.length / batchSize);
+        const showProgress = texts.length > 50;
+        for (let i = 0; i < texts.length; i += batchSize) {
+            const batchNum = Math.floor(i / batchSize) + 1;
+            const batch = texts.slice(i, i + batchSize);
+            if (showProgress) {
+                process.stderr.write(`\r  Ollama: ${batchNum}/${totalBatches} batches...`);
+            }
+            // Call progress callback if provided
+            if (onProgress) {
+                onProgress(batchNum, totalBatches);
+            }
+            const result = await this.embedBatch(batch);
+            if ('code' in result) {
+                if (showProgress) {
+                    process.stderr.write('\r                                      \r');
+                }
+                return result;
+            }
+            allEmbeddings.push(...result);
+        }
+        if (showProgress) {
+            process.stderr.write('\r                                      \r');
+        }
+        return allEmbeddings;
+    }
+    getModelName() {
+        return `ollama:${this.config.ollamaModel}`;
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+}
+/**
+ * LocalAI embedding provider.
+ * Uses OpenAI-compatible `/v1/embeddings` endpoint.
+ */
+export class LocalAIEmbeddingProvider {
+    config;
+    dimensions;
+    baseUrl;
+    constructor(config) {
+        this.config = config;
+        this.dimensions = getModelDimensions(config.localaiModel, config.dimensions);
+        this.baseUrl = config.localaiBaseUrl.replace(/\/$/, '');
+    }
+    async initialize() {
+        try {
+            const response = await fetch(`${this.baseUrl}/v1/models`);
+            if (!response.ok) {
+                throw new Error(`LocalAI server returned ${response.status}`);
+            }
+        }
+        catch (error) {
+            throw new Error(`Cannot connect to LocalAI at ${this.baseUrl}. ` +
+                `Ensure LocalAI is running: ${error.message}`);
+        }
+    }
+    async embedBatch(texts) {
+        try {
+            const headers = {
+                'Content-Type': 'application/json',
+            };
+            if (this.config.localaiApiKey) {
+                headers.Authorization = `Bearer ${this.config.localaiApiKey}`;
+            }
+            const response = await fetch(`${this.baseUrl}/v1/embeddings`, {
+                method: 'POST',
+                headers,
+                body: JSON.stringify({
+                    model: this.config.localaiModel,
+                    input: texts,
+                }),
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                return createError('EMBEDDING_FAILED', `LocalAI API error ${response.status}: ${errorText}`);
+            }
+            const data = (await response.json());
+            if (!data.data || !Array.isArray(data.data)) {
+                return createError('EMBEDDING_FAILED', 'LocalAI API returned no embeddings');
+            }
+            const sorted = [...data.data].sort((a, b) => (a.index ?? 0) - (b.index ?? 0));
+            return sorted.map((item) => {
+                const vector = item.embedding ?? [];
+                return this.dimensions < vector.length ? vector.slice(0, this.dimensions) : vector;
+            });
+        }
+        catch (error) {
+            return createError('EMBEDDING_FAILED', `LocalAI API request failed: ${error.message}`, { error: error.message });
+        }
+    }
+    async getEmbedding(text, inputType = 'document') {
+        const result = await this.embedBatch([text]);
+        if ('code' in result)
+            return result;
+        return result[0];
+    }
+    async getEmbeddingsBatch(texts, batchSize = 128, inputType = 'document', onProgress) {
+        if (texts.length === 0)
+            return [];
+        const allEmbeddings = [];
+        const totalBatches = Math.ceil(texts.length / batchSize);
+        const showProgress = texts.length > 50;
+        for (let i = 0; i < texts.length; i += batchSize) {
+            const batchNum = Math.floor(i / batchSize) + 1;
+            const batch = texts.slice(i, i + batchSize);
+            if (showProgress) {
+                process.stderr.write(`\r  LocalAI: ${batchNum}/${totalBatches} batches...`);
+            }
+            if (onProgress) {
+                onProgress(batchNum, totalBatches);
+            }
+            const result = await this.embedBatch(batch);
+            if ('code' in result) {
+                if (showProgress) {
+                    process.stderr.write('\r                                      \r');
+                }
+                return result;
+            }
+            allEmbeddings.push(...result);
+        }
+        if (showProgress) {
+            process.stderr.write('\r                                      \r');
+        }
+        return allEmbeddings;
+    }
+    getModelName() {
+        return `localai:${this.config.localaiModel}`;
+    }
+    getDimensions() {
+        return this.dimensions;
+    }
+}
+/**
+ * Factory function to create an embedding provider based on config.
+ */
+export function createEmbeddingProvider(config) {
+    if (config.provider === 'voyage') {
+        return new VoyageEmbeddingProvider(config);
+    }
+    if (config.provider === 'ollama') {
+        return new OllamaEmbeddingProvider(config);
+    }
+    if (config.provider === 'localai') {
+        return new LocalAIEmbeddingProvider(config);
+    }
+    return new WasmEmbeddingProvider(config);
+}
+//# sourceMappingURL=embeddings.js.map

package/dist/indexer/embeddings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embeddings.js","sourceRoot":"","sources":["../../src/indexer/embeddings.ts"],"names":[],"mappings":"AAAA,0CAA0C;AAC1C,mFAAmF;AAEnF,OAAO,EAAE,QAAQ,EAAkC,GAAG,EAAE,MAAM,sBAAsB,CAAC;AACrF,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AACzD,OAAO,MAAM,MAAM,SAAS,CAAC;AAE7B,OAAO,EAAE,WAAW,EAAmB,MAAM,aAAa,CAAC;AAE3D;;;GAGG;AACH,MAAM,gBAAgB,GAA2B;IAC/C,gBAAgB;IAChB,kBAAkB,EAAE,GAAG;IACvB,mBAAmB,EAAE,IAAI;IACzB,YAAY,EAAE,GAAG;IACjB,wBAAwB,EAAE,IAAI;IAC9B,QAAQ,EAAE,IAAI;IACd,WAAW,EAAE,IAAI;IACjB,gCAAgC;IAChC,gCAAgC,EAAE,GAAG;IACrC,yBAAyB,EAAE,GAAG;IAC9B,0BAA0B,EAAE,GAAG;IAC/B,yBAAyB,EAAE,GAAG;IAC9B,gBAAgB;IAChB,UAAU,EAAE,IAAI;IAChB,eAAe,EAAE,GAAG;IACpB,eAAe,EAAE,IAAI;IACrB,kBAAkB,EAAE,IAAI;IACxB,cAAc,EAAE,IAAI;CACrB,CAAC;AAEF;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAa,EAAE,gBAAyB;IACzE,OAAO,gBAAgB,CAAC,KAAK,CAAC,IAAI,gBAAgB,IAAI,GAAG,CAAC;AAC5D,CAAC;AA4CD;;;GAGG;AACH,MAAM,OAAO,qBAAqB;IACxB,MAAM,CAAkB;IACxB,UAAU,CAAS;IACnB,SAAS,GAAqC,IAAI,CAAC;IACnD,WAAW,GAA8C,IAAI,CAAC;IAEtE,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QAEtE,4CAA4C;QAC5C,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACpB,GAAG,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO,IAAI,CAAC,SAAS,CAAC;QAE1C,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,GAAG,QAAQ,CAAC,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE;gBACnE,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC;QACxC,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,IAAY,EACZ,YAAkC,UAAU;QAE5C,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACrE,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,IAAoB,CAAC,CAAC;YAE3D,kDAAkD;YAClD,IAAI,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC;gBACxC,OAAO,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;YAC9C,CAAC;YAED,OAAO,UAAU,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,WAAW,CAChB,kBAAkB,EAClB,4BAA6B,KAAe,CAAC,OAAO,EAAE,EACtD,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CACpC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,KAAe,EACf,YAAoB,EAAE,EACtB,YAAkC,UAAU,EAC5C,UAAqD;QAErD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YACtC,MAAM,aAAa,GAAe,EAAE,CAAC;YAErC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;YACzD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;gBACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;gBAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;gBAE5C,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;oBACtB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,QAAQ,IAAI,YAAY,KAAK,CAAC,CAAC;gBAC7E,CAAC;gBAED,qCAAqC;gBACrC,IAAI,UAAU,EAAE,CAAC;oBACf,UAAU,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;gBACrC,CAAC;gBAED,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAEtE,MAAM,IAAI,GAAG,MAAM,CAAC,IAAoB,CAAC;gBACzC,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;gBAElC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACtC,MAAM,KAAK,GAAG,CAAC,GAAG,UAAU,CAAC;oBAC7B,MAAM,GAAG,GAAG,KAAK,GAAG,UAAU,CAAC;oBAC/B,IAAI,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC;oBAEhD,IAAI,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC;wBACpC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;oBAC5C,CAAC;oBAED,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7B,CAAC;YACH,CAAC;YAED,IAAI,KAAK,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;gBACtB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;YACrE,CAAC;YAED,OAAO,aAAa,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,WAAW,CAChB,kBAAkB,EAClB,mCAAoC,KAAe,CAAC,OAAO,EAAE,EAC7D,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CACpC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;IAC3B,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF;AAED,qDAAqD;AACrD,MAAM,qBAAqB,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;AACrD,MAAM,yBAAyB,GAAG,IAAI,CAAC;AAEvC;;;GAGG;AACH,MAAM,OAAO,uBAAuB;IAC1B,MAAM,CAAkB;IACxB,UAAU,CAAS;IACnB,MAAM,GAA0B,IAAI,CAAC;IACrC,KAAK,GAAkB,IAAI,CAAC;IAC5B,eAAe,GAAG,KAAK,CAAC;IAEhC,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QAC5E,4FAA4F;IAC9F,CAAC;IAED,KAAK,CAAC,UAAU;QACd,6EAA6E;IAC/E,CAAC;IAEO,SAAS;QACf,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,4EAA4E,CAAC,CAAC;YAChG,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC;QACzE,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,UAAU,CACtB,KAAe,EACf,YAAkC,UAAU;QAE5C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YAEhC,oEAAoE;YACpE,IAAI,eAAmC,CAAC;YAExC,IAAI,qBAAqB,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpD,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC;YACpC,CAAC;iBAAM,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC;gBACjC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC;gBAC5B,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,uBAAuB,IAAI,CAAC,UAAU,4BAA4B;oBAChE,iBAAiB,yBAAyB,gBAAgB,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CACjG,CAAC;YACJ,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC;gBAClC,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;gBAC9B,SAAS,EAAE,SAAS;gBACpB,UAAU,EAAE,IAAI;gBAChB,eAAe;aAChB,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACnB,OAAO,WAAW,CAAC,kBAAkB,EAAE,6BAA6B,CAAC,CAAC;YACxE,CAAC;YAED,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC;YAClF,MAAM,UAAU,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;YAE9D,OAAO,UAAU,CAAC;QACpB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,aAAa,EAAE,CAAC;gBACnC,OAAO,WAAW,CAChB,kBAAkB,EAClB,oBAAoB,KAAK,CAAC,UAAU,KAAK,KAAK,CAAC,OAAO,EAAE,CACzD,CAAC;YACJ,CAAC;YACD,OAAO,WAAW,CAChB,kBAAkB,EAClB,8BAA+B,KAAe,CAAC,OAAO,EAAE,EACxD,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CACpC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,IAAY,EACZ,YAAkC,UAAU;QAE5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,EAAE,SAAS,CAAC,CAAC;QACxD,IAAI,MAAM,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QACpC,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,KAAe,EACf,SAAkB,EAClB,YAAkC,UAAU,EAC5C,UAAqD;QAErD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,MAAM,kBAAkB,GAAG,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC;QACpE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,kBAAkB,CAAC,CAAC;QAClE,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAE9B,oCAAoC;QACpC,MAAM,OAAO,GAAyC,EAAE,CAAC;QACzD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,kBAAkB,EAAE,CAAC;YAC1D,OAAO,CAAC,IAAI,CAAC;gBACX,KAAK,EAAE,OAAO,CAAC,MAAM;gBACrB,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,kBAAkB,CAAC;aAC9C,CAAC,CAAC;QACL,CAAC;QAED,iBAAiB;QACjB,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,GAAG,EAAE,CAAC;QAEvC,0CAA0C;QAC1C,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CACpB,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE;YACnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YAC7D,SAAS,EAAE,CAAC;YACZ,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,eAAe,SAAS,IAAI,YAAY,aAAa,IAAI,CAAC,MAAM,CAAC,iBAAiB,iBAAiB,CACpG,CAAC;YACJ,CAAC;YACD,qCAAqC;YACrC,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YACtC,CAAC;YACD,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;QACxC,CAAC,CAAC,CACH,CACF,CAAC;QAEF,sBAAsB;QACtB,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,kEAAkE,CAAC,CAAC;QAC3F,CAAC;QAED,8CAA8C;QAC9C,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAE,CAAC,KAAK,GAAG,CAAE,CAAC,KAAK,CAAC,CAAC;QAClE,MAAM,aAAa,GAAe,EAAE,CAAC;QAErC,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;YACjC,IAAI,CAAC,IAAI,IAAI,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;gBACnC,OAAO,IAAI,EAAE,MAAoB,IAAI,WAAW,CAAC,kBAAkB,EAAE,eAAe,CAAC,CAAC;YACxF,CAAC;YACD,aAAa,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,YAAY;QACV,OAAO,UAAU,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC7C,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,uBAAuB;IAC1B,MAAM,CAAkB;IACxB,UAAU,CAAS;IACnB,OAAO,CAAS;IAExB,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QAC5E,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,KAAK,CAAC,UAAU;QACd,4BAA4B;QAC5B,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,WAAW,CAAC,CAAC;YACzD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,+BAA+B,IAAI,CAAC,OAAO,IAAI;gBAC7C,6BAA8B,KAAe,CAAC,OAAO,EAAE,CAC1D,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAe;QACtC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC7B,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAE/D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,YAAY,EAAE;gBACxD,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC/C,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;oBAC9B,KAAK,EAAE,KAAK;oBACZ,UAAU,EAAE,KAAK,EAAE,2BAA2B;iBAC/C,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAEzC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,WAAW,CAChB,kBAAkB,EAClB,oBAAoB,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CACpD,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAKlC,CAAC;YAEF,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;gBACxD,OAAO,WAAW,CAAC,kBAAkB,EAAE,mCAAmC,CAAC,CAAC;YAC9E,CAAC;YAED,kBAAkB;YAClB,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnF,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtF,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,gBAAgB,KAAK,CAAC,MAAM,WAAW,UAAU,UAAU;gBAC3D,SAAS,SAAS,oBAAoB,OAAO,YAAY,MAAM,MAAM,CACtE,CAAC;YAEF,kDAAkD;YAClD,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACjC,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,CACnE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,WAAW,CAChB,kBAAkB,EAClB,8BAA+B,KAAe,CAAC,OAAO,EAAE,EACxD,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CACpC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,IAAY,EACZ,YAAkC,UAAU;QAE5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7C,IAAI,MAAM,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QACpC,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,KAAe,EACf,YAAoB,GAAG,EAAE,6CAA6C;IACtE,YAAkC,UAAU,EAC5C,UAAqD;QAErD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,MAAM,aAAa,GAAe,EAAE,CAAC;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;QACzD,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,GAAG,EAAE,CAAC;QAEvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAE5C,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,QAAQ,IAAI,YAAY,aAAa,CAAC,CAAC;YAC7E,CAAC;YAED,qCAAqC;YACrC,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YACrC,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;gBACrB,IAAI,YAAY,EAAE,CAAC;oBACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBACrE,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,aAAa,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,YAAY;QACV,OAAO,UAAU,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;IAC7C,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF;AAED;;;GAGG;AACH,MAAM,OAAO,wBAAwB;IAC3B,MAAM,CAAkB;IACxB,UAAU,CAAS;IACnB,OAAO,CAAS;IAExB,YAAY,MAAuB;QACjC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC,MAAM,CAAC,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;QAC7E,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IAC1D,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,YAAY,CAAC,CAAC;YAC1D,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAChE,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CACb,gCAAgC,IAAI,CAAC,OAAO,IAAI;gBAC9C,8BAA+B,KAAe,CAAC,OAAO,EAAE,CAC3D,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,KAAe;QACtC,IAAI,CAAC;YACH,MAAM,OAAO,GAA2B;gBACtC,cAAc,EAAE,kBAAkB;aACnC,CAAC;YAEF,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC9B,OAAO,CAAC,aAAa,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAChE,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,IAAI,CAAC,OAAO,gBAAgB,EAAE;gBAC5D,MAAM,EAAE,MAAM;gBACd,OAAO;gBACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;oBACnB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;oBAC/B,KAAK,EAAE,KAAK;iBACb,CAAC;aACH,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;gBACxC,OAAO,WAAW,CAChB,kBAAkB,EAClB,qBAAqB,QAAQ,CAAC,MAAM,KAAK,SAAS,EAAE,CACrD,CAAC;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAElC,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5C,OAAO,WAAW,CAAC,kBAAkB,EAAE,oCAAoC,CAAC,CAAC;YAC/E,CAAC;YAED,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9E,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACzB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;gBACpC,OAAO,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;YACrF,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,WAAW,CAChB,kBAAkB,EAClB,+BAAgC,KAAe,CAAC,OAAO,EAAE,EACzD,EAAE,KAAK,EAAG,KAAe,CAAC,OAAO,EAAE,CACpC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,YAAY,CAChB,IAAY,EACZ,YAAkC,UAAU;QAE5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7C,IAAI,MAAM,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QACpC,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IACnB,CAAC;IAED,KAAK,CAAC,kBAAkB,CACtB,KAAe,EACf,YAAoB,GAAG,EACvB,YAAkC,UAAU,EAC5C,UAAqD;QAErD,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,MAAM,aAAa,GAAe,EAAE,CAAC;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC;QACzD,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,GAAG,EAAE,CAAC;QAEvC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACjD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC;YAC/C,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAE5C,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,QAAQ,IAAI,YAAY,aAAa,CAAC,CAAC;YAC9E,CAAC;YAED,IAAI,UAAU,EAAE,CAAC;gBACf,UAAU,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YACrC,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;gBACrB,IAAI,YAAY,EAAE,CAAC;oBACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;gBACrE,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;YAED,aAAa,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,4CAA4C,CAAC,CAAC;QACrE,CAAC;QAED,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,YAAY;QACV,OAAO,WAAW,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;IAC/C,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,UAAU,CAAC;IACzB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,MAAuB;IAC7D,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,QAAQ,EAAE,CAAC;QACjC,OAAO,IAAI,uBAAuB,CAAC,MAAM,CAAC,CAAC;IAC7C,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;QAClC,OAAO,IAAI,wBAAwB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;IACD,OAAO,IAAI,qBAAqB,CAAC,MAAM,CAAC,CAAC;AAC3C,CAAC"}