@henrychong-ai/mcp-neo4j-knowledge-graph 1.0.0

package/dist/embeddings/EmbeddingJobManager.js

@@ -0,0 +1,545 @@
+import { v4 as uuidv4 } from 'uuid';
+import { LRUCache } from 'lru-cache';
+import crypto from 'crypto';
+/**
+ * Default logger implementation
+ */
+const nullLogger = {
+    debug: () => { },
+    info: () => { },
+    warn: () => { },
+    error: () => { },
+};
+/**
+ * Manages embedding jobs for semantic search
+ */
+export class EmbeddingJobManager {
+    /**
+     * Creates a new embedding job manager
+     *
+     * @param storageProvider - Provider for entity storage
+     * @param embeddingService - Service to generate embeddings
+     * @param rateLimiterOptions - Optional configuration for rate limiting
+     * @param cacheOptions - Optional configuration for caching
+     * @param logger - Optional logger for operation logging
+     */
+    constructor(storageProvider, embeddingService, rateLimiterOptions, cacheOptions, logger) {
+        this.cacheOptions = { size: 1000, ttl: 3600000 };
+        this.storageProvider = storageProvider;
+        this.embeddingService = embeddingService;
+        this.logger = logger || nullLogger;
+        // Setup rate limiter with defaults
+        const defaultRateLimiter = {
+            tokensPerInterval: 60,
+            interval: 60 * 1000,
+        };
+        const rateOptions = rateLimiterOptions || defaultRateLimiter;
+        this.rateLimiter = {
+            tokens: rateOptions.tokensPerInterval,
+            lastRefill: Date.now(),
+            tokensPerInterval: rateOptions.tokensPerInterval,
+            interval: rateOptions.interval,
+        };
+        // Setup LRU cache
+        if (cacheOptions) {
+            // Support both API styles (tests use maxItems/ttlHours)
+            this.cacheOptions = {
+                size: cacheOptions.size || cacheOptions.maxItems || 1000,
+                ttl: cacheOptions.ttl ||
+                    (cacheOptions.ttlHours ? Math.round(cacheOptions.ttlHours * 60 * 60 * 1000) : 3600000),
+            };
+        }
+        this.cache = new LRUCache({
+            max: this.cacheOptions.size,
+            ttl: Math.max(1, Math.round(this.cacheOptions.ttl)),
+            updateAgeOnGet: true,
+            allowStale: false,
+            // Use a ttlAutopurge option to ensure items are purged when TTL expires
+            ttlAutopurge: true,
+        });
+        // Initialize database schema
+        this._initializeDatabase();
+        this.logger.info('EmbeddingJobManager initialized', {
+            cacheSize: this.cacheOptions.size,
+            cacheTtl: this.cacheOptions.ttl,
+            rateLimit: `${this.rateLimiter.tokensPerInterval} per ${this.rateLimiter.interval}ms`,
+        });
+    }
+    /**
+     * Initialize the database schema for embedding jobs
+     *
+     * @private
+     */
+    _initializeDatabase() {
+        const createTableSql = `
+      CREATE TABLE IF NOT EXISTS embedding_jobs (
+        id TEXT PRIMARY KEY,
+        entity_name TEXT NOT NULL,
+        status TEXT NOT NULL,
+        priority INTEGER NOT NULL DEFAULT 1,
+        created_at INTEGER NOT NULL,
+        processed_at INTEGER,
+        error TEXT,
+        attempts INTEGER NOT NULL DEFAULT 0,
+        max_attempts INTEGER NOT NULL DEFAULT 3
+      )
+    `;
+        // Create an index for efficient job retrieval
+        const createIndexSql = `
+      CREATE INDEX IF NOT EXISTS idx_embedding_jobs_status_priority
+      ON embedding_jobs (status, priority DESC)
+    `;
+        try {
+            this.storageProvider.db.exec(createTableSql);
+            this.storageProvider.db.exec(createIndexSql);
+            this.logger.debug('Database schema initialized for embedding jobs');
+        }
+        catch (error) {
+            this.logger.error('Failed to initialize database schema', { error });
+            throw error;
+        }
+    }
+    /**
+     * Schedule an entity for embedding generation
+     *
+     * @param entityName - Name of the entity to generate embedding for
+     * @param priority - Optional priority (higher priority jobs are processed first)
+     * @returns Job ID
+     */
+    async scheduleEntityEmbedding(entityName, priority = 1) {
+        // Verify entity exists
+        const entity = await this.storageProvider.getEntity(entityName);
+        if (!entity) {
+            const error = `Entity ${entityName} not found`;
+            this.logger.error('Failed to schedule embedding', { entityName, error });
+            throw new Error(error);
+        }
+        // Create a job ID
+        const jobId = uuidv4();
+        // Insert a new job record
+        const stmt = this.storageProvider.db.prepare(`
+      INSERT INTO embedding_jobs (
+        id, entity_name, status, priority, created_at, attempts, max_attempts
+      ) VALUES (?, ?, ?, ?, ?, ?, ?)
+    `);
+        stmt.run(jobId, entityName, 'pending', priority, Date.now(), 0, 3);
+        this.logger.info('Scheduled embedding job', {
+            jobId,
+            entityName,
+            priority,
+        });
+        return jobId;
+    }
+    /**
+     * Process a batch of pending embedding jobs
+     *
+     * @param batchSize - Maximum number of jobs to process
+     * @returns Result statistics
+     */
+    async processJobs(batchSize = 10) {
+        this.logger.info('Starting job processing', { batchSize });
+        // Get pending jobs, ordered by priority (highest first)
+        const stmt = this.storageProvider.db.prepare(`
+      SELECT * FROM embedding_jobs
+      WHERE status = 'pending'
+      ORDER BY priority DESC, created_at ASC
+      LIMIT ?
+    `);
+        const jobs = stmt.all(batchSize);
+        this.logger.debug('Found pending jobs', { count: jobs.length });
+        // Initialize counters
+        const result = {
+            processed: 0,
+            successful: 0,
+            failed: 0,
+        };
+        // Process each job
+        for (const job of jobs) {
+            // Check rate limiter before processing
+            const rateLimitCheck = this._checkRateLimiter();
+            if (!rateLimitCheck.success) {
+                this.logger.warn('Rate limit reached, pausing job processing', {
+                    remaining: jobs.length - result.processed,
+                });
+                break; // Stop processing jobs if rate limit is reached
+            }
+            this.logger.info('Processing embedding job', {
+                jobId: job.id,
+                entityName: job.entity_name,
+                attempt: job.attempts + 1,
+                maxAttempts: job.max_attempts,
+            });
+            // Update job status to processing
+            this._updateJobStatus(job.id, 'processing', job.attempts + 1);
+            try {
+                // Get the entity
+                const entity = await this.storageProvider.getEntity(job.entity_name);
+                if (!entity) {
+                    throw new Error(`Entity ${job.entity_name} not found`);
+                }
+                // Log entity details for debugging
+                this.logger.debug('Retrieved entity for embedding', {
+                    entityName: job.entity_name,
+                    entityType: entity.entityType,
+                    hasObservations: entity.observations ? 'yes' : 'no',
+                    observationsType: entity.observations ? typeof entity.observations : 'undefined',
+                    observationsLength: entity.observations && Array.isArray(entity.observations)
+                        ? entity.observations.length
+                        : 'n/a',
+                });
+                // Prepare text for embedding
+                const text = this._prepareEntityText(entity);
+                // Try to get from cache or generate new embedding
+                this.logger.debug('Generating embedding for entity', { entityName: job.entity_name });
+                const embedding = await this._getCachedEmbeddingOrGenerate(text);
+                // Get model info for embedding metadata
+                const modelInfo = this.embeddingService.getModelInfo();
+                // Store the embedding with the entity
+                this.logger.debug('Storing entity vector', {
+                    entityName: job.entity_name,
+                    vectorLength: embedding.length,
+                    model: modelInfo.name,
+                });
+                await this.storageProvider.storeEntityVector(job.entity_name, {
+                    vector: embedding,
+                    model: modelInfo.name,
+                    lastUpdated: Date.now(),
+                });
+                // Update job status to completed
+                this._updateJobStatus(job.id, 'completed');
+                this.logger.info('Successfully processed embedding job', {
+                    jobId: job.id,
+                    entityName: job.entity_name,
+                    model: modelInfo.name,
+                    dimensions: embedding.length,
+                });
+                result.successful++;
+            }
+            catch (error) {
+                // Handle failures
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                const errorStack = error instanceof Error ? error.stack : undefined;
+                this.logger.error('Failed to process embedding job', {
+                    jobId: job.id,
+                    entityName: job.entity_name,
+                    error: errorMessage,
+                    errorStack,
+                    attempt: job.attempts + 1,
+                    maxAttempts: job.max_attempts,
+                });
+                // Determine if we should mark as failed or keep for retry
+                if (job.attempts + 1 >= job.max_attempts) {
+                    this._updateJobStatus(job.id, 'failed', job.attempts + 1, errorMessage);
+                }
+                else {
+                    this._updateJobStatus(job.id, 'pending', job.attempts + 1, errorMessage);
+                }
+                result.failed++;
+            }
+            result.processed++;
+        }
+        // Log job processing results
+        const queueStatus = await this.getQueueStatus();
+        this.logger.info('Job processing complete', {
+            processed: result.processed,
+            successful: result.successful,
+            failed: result.failed,
+            remaining: queueStatus.pending,
+        });
+        return result;
+    }
+    /**
+     * Get the current status of the job queue
+     *
+     * @returns Queue statistics
+     */
+    async getQueueStatus() {
+        const getCountForStatus = (status) => {
+            let sql = 'SELECT COUNT(*) as count FROM embedding_jobs';
+            const params = [];
+            if (status) {
+                sql += ' WHERE status = ?';
+                params.push(status);
+            }
+            const stmt = this.storageProvider.db.prepare(sql);
+            const result = stmt.get(...params);
+            return result?.count || 0;
+        };
+        const pending = getCountForStatus('pending');
+        const processing = getCountForStatus('processing');
+        const completed = getCountForStatus('completed');
+        const failed = getCountForStatus('failed');
+        const total = getCountForStatus();
+        const result = {
+            pending,
+            processing,
+            completed,
+            failed,
+            totalJobs: total,
+        };
+        this.logger.debug('Retrieved queue status', result);
+        return result;
+    }
+    /**
+     * Retry failed embedding jobs
+     *
+     * @returns Number of jobs reset for retry
+     */
+    async retryFailedJobs() {
+        const stmt = this.storageProvider.db.prepare(`
+      UPDATE embedding_jobs
+      SET status = 'pending', attempts = 0
+      WHERE status = 'failed'
+    `);
+        const result = stmt.run();
+        const resetCount = result.changes || 0;
+        this.logger.info('Reset failed jobs for retry', { count: resetCount });
+        return resetCount;
+    }
+    /**
+     * Clean up old completed jobs
+     *
+     * @param threshold - Age in milliseconds after which to delete completed jobs, defaults to 7 days
+     * @returns Number of jobs cleaned up
+     */
+    async cleanupJobs(threshold) {
+        const cleanupThreshold = threshold || 7 * 24 * 60 * 60 * 1000; // Default: 7 days
+        const cutoffTime = Date.now() - cleanupThreshold;
+        const stmt = this.storageProvider.db.prepare(`
+      DELETE FROM embedding_jobs
+      WHERE status = 'completed'
+      AND processed_at < ?
+    `);
+        const result = stmt.run(cutoffTime);
+        const deletedCount = result.changes || 0;
+        this.logger.info('Cleaned up old completed jobs', {
+            count: deletedCount,
+            threshold: cleanupThreshold,
+            olderThan: new Date(cutoffTime).toISOString(),
+        });
+        return deletedCount;
+    }
+    /**
+     * Update a job's status in the database
+     *
+     * @private
+     * @param jobId - ID of the job to update
+     * @param status - New status
+     * @param attempts - Optional attempts count update
+     * @param error - Optional error message
+     * @returns Database result
+     */
+    _updateJobStatus(jobId, status, attempts, error) {
+        let sql = `
+      UPDATE embedding_jobs
+      SET status = ?
+    `;
+        const params = [status];
+        // Add processed_at timestamp for completed/failed statuses
+        if (status === 'completed' || status === 'failed') {
+            sql += ', processed_at = ?';
+            params.push(Date.now());
+        }
+        // Update attempts if provided
+        if (attempts !== undefined) {
+            sql += ', attempts = ?';
+            params.push(attempts);
+        }
+        // Include error message if provided
+        if (error) {
+            sql += ', error = ?';
+            params.push(error);
+        }
+        sql += ' WHERE id = ?';
+        params.push(jobId);
+        const stmt = this.storageProvider.db.prepare(sql);
+        return stmt.run(...params);
+    }
+    /**
+     * Check rate limiter and consume a token if available
+     *
+     * @private
+     * @returns Object with success flag
+     */
+    _checkRateLimiter() {
+        // For testing purposes, make it public by removing 'private'
+        const now = Date.now();
+        const elapsed = now - this.rateLimiter.lastRefill;
+        // If enough time has passed, refill tokens
+        if (elapsed >= this.rateLimiter.interval) {
+            // Calculate how many full intervals have passed
+            const intervals = Math.floor(elapsed / this.rateLimiter.interval);
+            // Completely refill tokens (don't accumulate beyond max)
+            this.rateLimiter.tokens = this.rateLimiter.tokensPerInterval;
+            // Update last refill time, keeping track of remaining time
+            this.rateLimiter.lastRefill = now;
+            this.logger.debug('Refilled rate limiter tokens', {
+                current: this.rateLimiter.tokens,
+                max: this.rateLimiter.tokensPerInterval,
+                intervals,
+            });
+        }
+        // If we have tokens, consume one and return success
+        if (this.rateLimiter.tokens > 0) {
+            this.rateLimiter.tokens--;
+            this.logger.debug('Consumed rate limiter token', {
+                remaining: this.rateLimiter.tokens,
+                max: this.rateLimiter.tokensPerInterval,
+            });
+            return { success: true };
+        }
+        // No tokens available
+        this.logger.warn('Rate limit exceeded', {
+            availableTokens: 0,
+            maxTokens: this.rateLimiter.tokensPerInterval,
+            nextRefillIn: this.rateLimiter.interval - (now - this.rateLimiter.lastRefill),
+        });
+        return { success: false };
+    }
+    /**
+     * Get the current status of the rate limiter
+     *
+     * @returns Rate limiter status information
+     */
+    getRateLimiterStatus() {
+        const now = Date.now();
+        const elapsed = now - this.rateLimiter.lastRefill;
+        // If enough time has passed for a complete refill
+        if (elapsed >= this.rateLimiter.interval) {
+            return {
+                availableTokens: this.rateLimiter.tokensPerInterval,
+                maxTokens: this.rateLimiter.tokensPerInterval,
+                resetInMs: this.rateLimiter.interval,
+            };
+        }
+        // Otherwise return current state
+        return {
+            availableTokens: this.rateLimiter.tokens,
+            maxTokens: this.rateLimiter.tokensPerInterval,
+            resetInMs: this.rateLimiter.interval - elapsed,
+        };
+    }
+    /**
+     * Retrieve a cached embedding or generate a new one
+     *
+     * @param text - Text to generate embedding for
+     * @returns Embedding vector
+     */
+    async _getCachedEmbeddingOrGenerate(text) {
+        const cacheKey = this._generateCacheKey(text);
+        // Try to get from cache first
+        const cachedValue = this.cache.get(cacheKey);
+        if (cachedValue) {
+            this.logger.debug('Cache hit', {
+                textHash: cacheKey.substring(0, 8),
+                age: Date.now() - cachedValue.timestamp,
+            });
+            return cachedValue.embedding;
+        }
+        this.logger.debug('Cache miss', { textHash: cacheKey.substring(0, 8) });
+        try {
+            // Generate new embedding
+            const embedding = await this.embeddingService.generateEmbedding(text);
+            // Store in cache
+            this._cacheEmbedding(text, embedding);
+            return embedding;
+        }
+        catch (error) {
+            this.logger.error('Failed to generate embedding', {
+                error,
+                textLength: text.length,
+            });
+            throw error;
+        }
+    }
+    /**
+     * Store an embedding in the cache
+     *
+     * @private
+     * @param text - Original text
+     * @param embedding - Embedding vector
+     */
+    _cacheEmbedding(text, embedding) {
+        const cacheKey = this._generateCacheKey(text);
+        const modelInfo = this.embeddingService.getModelInfo();
+        this.cache.set(cacheKey, {
+            embedding,
+            timestamp: Date.now(),
+            model: modelInfo.name,
+        });
+        this.logger.debug('Cached embedding', {
+            textHash: cacheKey.substring(0, 8),
+            model: modelInfo.name,
+            dimensions: embedding.length,
+        });
+    }
+    /**
+     * Generate a deterministic cache key for text
+     *
+     * @private
+     * @param text - Text to hash
+     * @returns Cache key
+     */
+    _generateCacheKey(text) {
+        return crypto.createHash('md5').update(text).digest('hex');
+    }
+    /**
+     * Prepare text for embedding from an entity
+     *
+     * @private
+     * @param entity - Entity to prepare text from
+     * @returns Processed text ready for embedding
+     */
+    _prepareEntityText(entity) {
+        // Create a descriptive text from entity data
+        const lines = [`Name: ${entity.name}`, `Type: ${entity.entityType}`, 'Observations:'];
+        // Add observations, ensuring we handle both string arrays and other formats
+        if (entity.observations) {
+            // Handle case where observations might be stored as JSON string in some providers
+            let observationsArray = entity.observations;
+            // If observations is a string, try to parse it as JSON
+            if (typeof entity.observations === 'string') {
+                try {
+                    observationsArray = JSON.parse(entity.observations);
+                }
+                catch {
+                    // If parsing fails, treat it as a single observation
+                    observationsArray = [entity.observations];
+                }
+            }
+            // Ensure it's an array at this point
+            if (!Array.isArray(observationsArray)) {
+                observationsArray = [String(observationsArray)];
+            }
+            // Add each observation to the text
+            if (observationsArray.length > 0) {
+                lines.push(...observationsArray.map((obs) => `- ${obs}`));
+            }
+            else {
+                lines.push('  (No observations)');
+            }
+        }
+        else {
+            lines.push('  (No observations)');
+        }
+        const text = lines.join('\n');
+        // Log the prepared text for debugging
+        this.logger.debug('Prepared entity text for embedding', {
+            entityName: entity.name,
+            entityType: entity.entityType,
+            observationCount: Array.isArray(entity.observations) ? entity.observations.length : 0,
+            textLength: text.length,
+        });
+        return text;
+    }
+    /**
+     * Get a cached embedding entry (used for testing)
+     *
+     * @param key - Cache key
+     * @returns Cached embedding or undefined
+     */
+    getCacheEntry(key) {
+        return this.cache.get(key);
+    }
+}
+//# sourceMappingURL=EmbeddingJobManager.js.map

package/dist/embeddings/EmbeddingJobManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"EmbeddingJobManager.js","sourceRoot":"","sources":["../../src/embeddings/EmbeddingJobManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,IAAI,MAAM,EAAE,MAAM,MAAM,CAAC;AACpC,OAAO,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAKrC,OAAO,MAAM,MAAM,QAAQ,CAAC;AAqH5B;;GAEG;AACH,MAAM,UAAU,GAAW;IACzB,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;IACf,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,IAAI,EAAE,GAAG,EAAE,GAAE,CAAC;IACd,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;CAChB,CAAC;AAEF;;GAEG;AACH,MAAM,OAAO,mBAAmB;IAa9B;;;;;;;;OAQG;IACH,YACE,eAAyC,EACzC,gBAAkC,EAClC,kBAA8C,EAC9C,YAAkC,EAClC,MAAsB;QAjBhB,iBAAY,GAAiB,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC;QAmBhE,IAAI,CAAC,eAAe,GAAG,eAAe,CAAC;QACvC,IAAI,CAAC,gBAAgB,GAAG,gBAAgB,CAAC;QACzC,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,UAAU,CAAC;QAEnC,mCAAmC;QACnC,MAAM,kBAAkB,GAAG;YACzB,iBAAiB,EAAE,EAAE;YACrB,QAAQ,EAAE,EAAE,GAAG,IAAI;SACpB,CAAC;QAEF,MAAM,WAAW,GAAG,kBAAkB,IAAI,kBAAkB,CAAC;QAE7D,IAAI,CAAC,WAAW,GAAG;YACjB,MAAM,EAAE,WAAW,CAAC,iBAAiB;YACrC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;YACtB,iBAAiB,EAAE,WAAW,CAAC,iBAAiB;YAChD,QAAQ,EAAE,WAAW,CAAC,QAAQ;SAC/B,CAAC;QAEF,kBAAkB;QAClB,IAAI,YAAY,EAAE,CAAC;YACjB,wDAAwD;YACxD,IAAI,CAAC,YAAY,GAAG;gBAClB,IAAI,EAAE,YAAY,CAAC,IAAI,IAAI,YAAY,CAAC,QAAQ,IAAI,IAAI;gBACxD,GAAG,EACD,YAAY,CAAC,GAAG;oBAChB,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,QAAQ,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;aACzF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,KAAK,GAAG,IAAI,QAAQ,CAAC;YACxB,GAAG,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;YAC3B,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;YACnD,cAAc,EAAE,IAAI;YACpB,UAAU,EAAE,KAAK;YACjB,wEAAwE;YACxE,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;QAEH,6BAA6B;QAC7B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QAE3B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;YAClD,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;YACjC,QAAQ,EAAE,IAAI,CAAC,YAAY,CAAC,GAAG;YAC/B,SAAS,EAAE,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,QAAQ,IAAI,CAAC,WAAW,CAAC,QAAQ,IAAI;SACtF,CAAC,CAAC;IACL,CAAC;IAED;;;;OAIG;IACK,mBAAmB;QACzB,MAAM,cAAc,GAAG;;;;;;;;;;;;KAYtB,CAAC;QAEF,8CAA8C;QAC9C,MAAM,cAAc,GAAG;;;KAGtB,CAAC;QAEF,IAAI,CAAC;YACH,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC7C,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gDAAgD,CAAC,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,uBAAuB,CAAC,UAAkB,EAAE,QAAQ,GAAG,CAAC;QAC5D,uBAAuB;QACvB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAChE,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,UAAU,UAAU,YAAY,CAAC;YAC/C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;YACzE,MAAM,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;QAED,kBAAkB;QAClB,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC;QAEvB,0BAA0B;QAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC;;;;KAI5C,CAAC,CAAC;QAEH,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAEnE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE;YAC1C,KAAK;YACL,UAAU;YACV,QAAQ;SACT,CAAC,CAAC;QAEH,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CAAC,SAAS,GAAG,EAAE;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,SAAS,EAAE,CAAC,CAAC;QAE3D,wDAAwD;QACxD,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC;;;;;KAK5C,CAAC,CAAC;QAEH,MAAM,IAAI,GAAmB,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAEhE,sBAAsB;QACtB,MAAM,MAAM,GAAsB;YAChC,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,MAAM,EAAE,CAAC;SACV,CAAC;QAEF,mBAAmB;QACnB,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACvB,uCAAuC;YACvC,MAAM,cAAc,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAChD,IAAI,CAAC,cAAc,CAAC,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4CAA4C,EAAE;oBAC7D,SAAS,EAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,SAAS;iBAC1C,CAAC,CAAC;gBACH,MAAM,CAAC,gDAAgD;YACzD,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE;gBAC3C,KAAK,EAAE,GAAG,CAAC,EAAE;gBACb,UAAU,EAAE,GAAG,CAAC,WAAW;gBAC3B,OAAO,EAAE,GAAG,CAAC,QAAQ,GAAG,CAAC;gBACzB,WAAW,EAAE,GAAG,CAAC,YAAY;aAC9B,CAAC,CAAC;YAEH,kCAAkC;YAClC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,EAAE,YAAY,EAAE,GAAG,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC;YAE9D,IAAI,CAAC;gBACH,iBAAiB;gBACjB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAErE,IAAI,CAAC,MAAM,EAAE,CAAC;oBACZ,MAAM,IAAI,KAAK,CAAC,UAAU,GAAG,CAAC,WAAW,YAAY,CAAC,CAAC;gBACzD,CAAC;gBAED,mCAAmC;gBACnC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE;oBAClD,UAAU,EAAE,GAAG,CAAC,WAAW;oBAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;oBAC7B,eAAe,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI;oBACnD,gBAAgB,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,WAAW;oBAChF,kBAAkB,EAChB,MAAM,CAAC,YAAY,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC;wBACvD,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,MAAM;wBAC5B,CAAC,CAAC,KAAK;iBACZ,CAAC,CAAC;gBAEH,6BAA6B;gBAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;gBAE7C,kDAAkD;gBAClD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,UAAU,EAAE,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC;gBACtF,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,6BAA6B,CAAC,IAAI,CAAC,CAAC;gBAEjE,wCAAwC;gBACxC,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC;gBAEvD,sCAAsC;gBACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE;oBACzC,UAAU,EAAE,GAAG,CAAC,WAAW;oBAC3B,YAAY,EAAE,SAAS,CAAC,MAAM;oBAC9B,KAAK,EAAE,SAAS,CAAC,IAAI;iBACtB,CAAC,CAAC;gBAEH,MAAM,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,GAAG,CAAC,WAAW,EAAE;oBAC5D,MAAM,EAAE,SAAS;oBACjB,KAAK,EAAE,SAAS,CAAC,IAAI;oBACrB,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE;iBACxB,CAAC,CAAC;gBAEH,iCAAiC;gBACjC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;gBAE3C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sCAAsC,EAAE;oBACvD,KAAK,EAAE,GAAG,CAAC,EAAE;oBACb,UAAU,EAAE,GAAG,CAAC,WAAW;oBAC3B,KAAK,EAAE,SAAS,CAAC,IAAI;oBACrB,UAAU,EAAE,SAAS,CAAC,MAAM;iBAC7B,CAAC,CAAC;gBAEH,MAAM,CAAC,UAAU,EAAE,CAAC;YACtB,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,kBAAkB;gBAClB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC5E,MAAM,UAAU,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;gBAEpE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE;oBACnD,KAAK,EAAE,GAAG,CAAC,EAAE;oBACb,UAAU,EAAE,GAAG,CAAC,WAAW;oBAC3B,KAAK,EAAE,YAAY;oBACnB,UAAU;oBACV,OAAO,EAAE,GAAG,CAAC,QAAQ,GAAG,CAAC;oBACzB,WAAW,EAAE,GAAG,CAAC,YAAY;iBAC9B,CAAC,CAAC;gBAEH,0DAA0D;gBAC1D,IAAI,GAAG,CAAC,QAAQ,GAAG,CAAC,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;oBACzC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,GAAG,CAAC,EAAE,YAAY,CAAC,CAAC;gBAC1E,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,EAAE,GAAG,CAAC,QAAQ,GAAG,CAAC,EAAE,YAAY,CAAC,CAAC;gBAC3E,CAAC;gBAED,MAAM,CAAC,MAAM,EAAE,CAAC;YAClB,CAAC;YAED,MAAM,CAAC,SAAS,EAAE,CAAC;QACrB,CAAC;QAED,6BAA6B;QAC7B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE;YAC1C,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,SAAS,EAAE,WAAW,CAAC,OAAO;SAC/B,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,iBAAiB,GAAG,CAAC,MAAe,EAAU,EAAE;YACpD,IAAI,GAAG,GAAG,8CAA8C,CAAC;YACzD,MAAM,MAAM,GAAa,EAAE,CAAC;YAE5B,IAAI,MAAM,EAAE,CAAC;gBACX,GAAG,IAAI,mBAAmB,CAAC;gBAC3B,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,CAAC;YAED,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClD,MAAM,MAAM,GAAgB,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;YAEhD,OAAO,MAAM,EAAE,KAAK,IAAI,CAAC,CAAC;QAC5B,CAAC,CAAC;QAEF,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7C,MAAM,UAAU,GAAG,iBAAiB,CAAC,YAAY,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,iBAAiB,CAAC,WAAW,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,iBAAiB,CAAC,QAAQ,CAAC,CAAC;QAC3C,MAAM,KAAK,GAAG,iBAAiB,EAAE,CAAC;QAElC,MAAM,MAAM,GAAG;YACb,OAAO;YACP,UAAU;YACV,SAAS;YACT,MAAM;YACN,SAAS,EAAE,KAAK;SACjB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,MAAM,CAAC,CAAC;QAEpD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,eAAe;QACnB,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC;;;;KAI5C,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC1B,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC;QAEvC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6BAA6B,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAEvE,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CAAC,SAAkB;QAClC,MAAM,gBAAgB,GAAG,SAAS,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,kBAAkB;QACjF,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,gBAAgB,CAAC;QAEjD,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC;;;;KAI5C,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACpC,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC;QAEzC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE;YAChD,KAAK,EAAE,YAAY;YACnB,SAAS,EAAE,gBAAgB;YAC3B,SAAS,EAAE,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE;SAC9C,CAAC,CAAC;QAEH,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;;;;;;;;OASG;IACK,gBAAgB,CACtB,KAAa,EACb,MAAiB,EACjB,QAAiB,EACjB,KAAc;QAEd,IAAI,GAAG,GAAG;;;KAGT,CAAC;QAEF,MAAM,MAAM,GAAwB,CAAC,MAAM,CAAC,CAAC;QAE7C,2DAA2D;QAC3D,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,QAAQ,EAAE,CAAC;YAClD,GAAG,IAAI,oBAAoB,CAAC;YAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC1B,CAAC;QAED,8BAA8B;QAC9B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,GAAG,IAAI,gBAAgB,CAAC;YACxB,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxB,CAAC;QAED,oCAAoC;QACpC,IAAI,KAAK,EAAE,CAAC;YACV,GAAG,IAAI,aAAa,CAAC;YACrB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;QAED,GAAG,IAAI,eAAe,CAAC;QACvB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAEnB,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAClD,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC;IAC7B,CAAC;IAED;;;;;OAKG;IACH,iBAAiB;QACf,6DAA6D;QAC7D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;QAElD,2CAA2C;QAC3C,IAAI,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;YACzC,gDAAgD;YAChD,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAElE,yDAAyD;YACzD,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC;YAE7D,2DAA2D;YAC3D,IAAI,CAAC,WAAW,CAAC,UAAU,GAAG,GAAG,CAAC;YAElC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;gBAChD,OAAO,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;gBAChC,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;gBACvC,SAAS;aACV,CAAC,CAAC;QACL,CAAC;QAED,oDAAoD;QACpD,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC;YAE1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE;gBAC/C,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;gBAClC,GAAG,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;aACxC,CAAC,CAAC;YAEH,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC3B,CAAC;QAED,sBAAsB;QACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,qBAAqB,EAAE;YACtC,eAAe,EAAE,CAAC;YAClB,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;YAC7C,YAAY,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;SAC9E,CAAC,CAAC;QAEH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED;;;;OAIG;IACH,oBAAoB;QAClB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC;QAElD,kDAAkD;QAClD,IAAI,OAAO,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC;YACzC,OAAO;gBACL,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;gBACnD,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;gBAC7C,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ;aACrC,CAAC;QACJ,CAAC;QAED,iCAAiC;QACjC,OAAO;YACL,eAAe,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM;YACxC,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,iBAAiB;YAC7C,SAAS,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,GAAG,OAAO;SAC/C,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,6BAA6B,CAAC,IAAY;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAE9C,8BAA8B;QAC9B,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAE7C,IAAI,WAAW,EAAE,CAAC;YAChB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE;gBAC7B,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC;gBAClC,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,SAAS;aACxC,CAAC,CAAC;YACH,OAAO,WAAW,CAAC,SAAS,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,YAAY,EAAE,EAAE,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;QAExE,IAAI,CAAC;YACH,yBAAyB;YACzB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAEtE,iBAAiB;YACjB,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YAEtC,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;gBAChD,KAAK;gBACL,UAAU,EAAE,IAAI,CAAC,MAAM;aACxB,CAAC,CAAC;YACH,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACK,eAAe,CAAC,IAAY,EAAE,SAAmB;QACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,CAAC;QAEvD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE;YACvB,SAAS;YACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,KAAK,EAAE,SAAS,CAAC,IAAI;SACtB,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE;YACpC,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC;YAClC,KAAK,EAAE,SAAS,CAAC,IAAI;YACrB,UAAU,EAAE,SAAS,CAAC,MAAM;SAC7B,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,iBAAiB,CAAC,IAAY;QAC5B,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7D,CAAC;IAED;;;;;;OAMG;IACK,kBAAkB,CAAC,MAAc;QACvC,6CAA6C;QAC7C,MAAM,KAAK,GAAG,CAAC,SAAS,MAAM,CAAC,IAAI,EAAE,EAAE,SAAS,MAAM,CAAC,UAAU,EAAE,EAAE,eAAe,CAAC,CAAC;QAEtF,4EAA4E;QAC5E,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,kFAAkF;YAClF,IAAI,iBAAiB,GAAG,MAAM,CAAC,YAAY,CAAC;YAE5C,uDAAuD;YACvD,IAAI,OAAO,MAAM,CAAC,YAAY,KAAK,QAAQ,EAAE,CAAC;gBAC5C,IAAI,CAAC;oBACH,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;gBACtD,CAAC;gBAAC,MAAM,CAAC;oBACP,qDAAqD;oBACrD,iBAAiB,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,qCAAqC;YACrC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACtC,iBAAiB,GAAG,CAAC,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC;YAClD,CAAC;YAED,mCAAmC;YACnC,IAAI,iBAAiB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACjC,KAAK,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACpC,CAAC;QAED,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE9B,sCAAsC;QACtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE;YACtD,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,gBAAgB,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACrF,UAAU,EAAE,IAAI,CAAC,MAAM;SACxB,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;;OAKG;IACH,aAAa,CAAC,GAAW;QACvB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC7B,CAAC;CACF"}

package/dist/embeddings/EmbeddingService.d.ts

@@ -0,0 +1,96 @@
+/**
+ * Provider information for embedding services
+ */
+export interface EmbeddingProviderInfo {
+    /**
+     * Name of the embedding provider
+     */
+    provider: string;
+    /**
+     * Name of the embedding model
+     */
+    model: string;
+    /**
+     * Number of dimensions in the embedding vectors
+     */
+    dimensions: number;
+}
+/**
+ * Model information for embedding models
+ */
+export interface EmbeddingModelInfo {
+    /**
+     * Name of the embedding model
+     */
+    name: string;
+    /**
+     * Number of dimensions in the embedding vectors
+     */
+    dimensions: number;
+    /**
+     * Version of the model
+     */
+    version: string;
+}
+/**
+ * Interface for text embedding services
+ */
+export interface IEmbeddingService {
+    /**
+     * Generate embedding vector for text
+     *
+     * @param text - Text to embed
+     * @returns Embedding vector
+     */
+    generateEmbedding(text: string): Promise<number[]>;
+    /**
+     * Generate embeddings for multiple texts
+     *
+     * @param texts - Array of texts to embed
+     * @returns Array of embedding vectors
+     */
+    generateEmbeddings(texts: string[]): Promise<number[][]>;
+    /**
+     * Get information about the embedding model
+     *
+     * @returns Model information
+     */
+    getModelInfo(): EmbeddingModelInfo;
+    /**
+     * Get information about the embedding provider
+     *
+     * @returns Provider information
+     */
+    getProviderInfo(): EmbeddingProviderInfo;
+}
+/**
+ * Abstract class for embedding services
+ */
+export declare class EmbeddingService implements IEmbeddingService {
+    /**
+     * Generate embedding vector for text
+     *
+     * @param text - Text to embed
+     * @returns Embedding vector
+     */
+    generateEmbedding(_text: string): Promise<number[]>;
+    /**
+     * Generate embeddings for multiple texts
+     *
+     * @param texts - Array of texts to embed
+     * @returns Array of embedding vectors
+     */
+    generateEmbeddings(_texts: string[]): Promise<number[][]>;
+    /**
+     * Get information about the embedding model
+     *
+     * @returns Model information
+     */
+    getModelInfo(): EmbeddingModelInfo;
+    /**
+     * Get information about the embedding provider
+     *
+     * @returns Provider information
+     */
+    getProviderInfo(): EmbeddingProviderInfo;
+}