@sochdb/sochdb 0.4.0 → 0.4.1

package/dist/cjs/semantic-cache.js

@@ -0,0 +1,220 @@
+"use strict";
+/**
+ * Semantic Cache for LLM responses
+ *
+ * Cache LLM responses with similarity-based retrieval for cost savings.
+ * Uses database prefix scanning to store and retrieve cached responses.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SemanticCache = void 0;
+/**
+ * Calculate cosine similarity between two vectors
+ */
+function cosineSimilarity(a, b) {
+    if (a.length !== b.length) {
+        throw new Error('Vectors must have same length');
+    }
+    let dotProduct = 0;
+    let normA = 0;
+    let normB = 0;
+    for (let i = 0; i < a.length; i++) {
+        dotProduct += a[i] * b[i];
+        normA += a[i] * a[i];
+        normB += b[i] * b[i];
+    }
+    if (normA === 0 || normB === 0) {
+        return 0;
+    }
+    return dotProduct / (Math.sqrt(normA) * Math.sqrt(normB));
+}
+/**
+ * Semantic Cache with vector similarity matching
+ *
+ * @example
+ * ```typescript
+ * const cache = new SemanticCache(db, 'llm_responses');
+ *
+ * // Store response
+ * await cache.put(
+ *   'What is Python?',
+ *   'Python is a high-level programming language...',
+ *   embedding,
+ *   3600  // TTL in seconds
+ * );
+ *
+ * // Check cache
+ * const hit = await cache.get(queryEmbedding, 0.85);
+ * if (hit) {
+ *   console.log(`Cache HIT: ${hit.value} (similarity: ${hit.score})`);
+ * }
+ * ```
+ */
+class SemanticCache {
+    constructor(db, cacheName) {
+        this.hits = 0;
+        this.misses = 0;
+        this.db = db;
+        this.cacheName = cacheName;
+        this.prefix = Buffer.from(`cache:${cacheName}:`);
+    }
+    /**
+     * Store a cached response
+     */
+    async put(key, value, embedding, ttlSeconds = 0, metadata) {
+        const entry = {
+            key,
+            value,
+            embedding,
+            timestamp: Date.now(),
+            ttl: ttlSeconds > 0 ? ttlSeconds : undefined,
+            metadata,
+        };
+        const entryKey = Buffer.concat([
+            this.prefix,
+            Buffer.from(key),
+        ]);
+        await this.db.put(entryKey, Buffer.from(JSON.stringify(entry)));
+    }
+    /**
+     * Retrieve cached response by similarity
+     *
+     * @param queryEmbedding - Query embedding vector
+     * @param threshold - Minimum cosine similarity (0-1)
+     * @returns Best matching cache entry or null
+     */
+    async get(queryEmbedding, threshold = 0.85) {
+        const now = Date.now();
+        let bestMatch = null;
+        let bestScore = threshold;
+        // Scan all cache entries with this prefix
+        try {
+            for await (const [_, valueBuffer] of this.db.scanPrefix(this.prefix)) {
+                const entry = JSON.parse(valueBuffer.toString());
+                // Check TTL expiration
+                if (entry.ttl && entry.timestamp) {
+                    const expiresAt = entry.timestamp + entry.ttl * 1000;
+                    if (now > expiresAt) {
+                        continue; // Skip expired entries
+                    }
+                }
+                // Calculate similarity
+                const score = cosineSimilarity(queryEmbedding, entry.embedding);
+                // Update best match
+                if (score > bestScore) {
+                    bestScore = score;
+                    bestMatch = { ...entry, score };
+                }
+            }
+        }
+        catch (error) {
+            // If scan fails, return null
+            this.misses++;
+            return null;
+        }
+        if (bestMatch) {
+            this.hits++;
+        }
+        else {
+            this.misses++;
+        }
+        return bestMatch;
+    }
+    /**
+     * Delete a specific cache entry
+     */
+    async delete(key) {
+        const entryKey = Buffer.concat([
+            this.prefix,
+            Buffer.from(key),
+        ]);
+        await this.db.delete(entryKey);
+    }
+    /**
+     * Clear all entries in this cache
+     */
+    async clear() {
+        let deleted = 0;
+        try {
+            const toDelete = [];
+            for await (const [key] of this.db.scanPrefix(this.prefix)) {
+                toDelete.push(key);
+            }
+            for (const key of toDelete) {
+                await this.db.delete(key);
+                deleted++;
+            }
+        }
+        catch (error) {
+            // If operation fails, return count so far
+            return deleted;
+        }
+        // Reset stats
+        this.hits = 0;
+        this.misses = 0;
+        return deleted;
+    }
+    /**
+     * Get cache statistics
+     */
+    async stats() {
+        const now = Date.now();
+        let count = 0;
+        let memoryUsage = 0;
+        try {
+            for await (const [key, value] of this.db.scanPrefix(this.prefix)) {
+                const entry = JSON.parse(value.toString());
+                // Skip expired entries
+                if (entry.ttl && entry.timestamp) {
+                    const expiresAt = entry.timestamp + entry.ttl * 1000;
+                    if (now > expiresAt) {
+                        continue;
+                    }
+                }
+                count++;
+                memoryUsage += key.length + value.length;
+            }
+        }
+        catch (error) {
+            // Return partial stats if operation fails
+        }
+        const total = this.hits + this.misses;
+        const hitRate = total > 0 ? this.hits / total : 0;
+        return {
+            count,
+            hits: this.hits,
+            misses: this.misses,
+            hitRate,
+            memoryUsage,
+        };
+    }
+    /**
+     * Purge expired entries
+     */
+    async purgeExpired() {
+        const now = Date.now();
+        let purged = 0;
+        try {
+            const toDelete = [];
+            for await (const [key, value] of this.db.scanPrefix(this.prefix)) {
+                const entry = JSON.parse(value.toString());
+                if (entry.ttl && entry.timestamp) {
+                    const expiresAt = entry.timestamp + entry.ttl * 1000;
+                    if (now > expiresAt) {
+                        toDelete.push(key);
+                    }
+                }
+            }
+            for (const key of toDelete) {
+                await this.db.delete(key);
+                purged++;
+            }
+        }
+        catch (error) {
+            // Return count so far
+            return purged;
+        }
+        return purged;
+    }
+}
+exports.SemanticCache = SemanticCache;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoic2VtYW50aWMtY2FjaGUuanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvc2VtYW50aWMtY2FjaGUudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IjtBQUFBOzs7OztHQUtHOzs7QUFJSDs7R0FFRztBQUNILFNBQVMsZ0JBQWdCLENBQUMsQ0FBVyxFQUFFLENBQVc7SUFDaEQsSUFBSSxDQUFDLENBQUMsTUFBTSxLQUFLLENBQUMsQ0FBQyxNQUFNLEVBQUUsQ0FBQztRQUMxQixNQUFNLElBQUksS0FBSyxDQUFDLCtCQUErQixDQUFDLENBQUM7SUFDbkQsQ0FBQztJQUVELElBQUksVUFBVSxHQUFHLENBQUMsQ0FBQztJQUNuQixJQUFJLEtBQUssR0FBRyxDQUFDLENBQUM7SUFDZCxJQUFJLEtBQUssR0FBRyxDQUFDLENBQUM7SUFFZCxLQUFLLElBQUksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLEdBQUcsQ0FBQyxDQUFDLE1BQU0sRUFBRSxDQUFDLEVBQUUsRUFBRSxDQUFDO1FBQ2xDLFVBQVUsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO1FBQzFCLEtBQUssSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO1FBQ3JCLEtBQUssSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO0lBQ3ZCLENBQUM7SUFFRCxJQUFJLEtBQUssS0FBSyxDQUFDLElBQUksS0FBSyxLQUFLLENBQUMsRUFBRSxDQUFDO1FBQy9CLE9BQU8sQ0FBQyxDQUFDO0lBQ1gsQ0FBQztJQUVELE9BQU8sVUFBVSxHQUFHLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsR0FBRyxJQUFJLENBQUMsSUFBSSxDQUFDLEtBQUssQ0FBQyxDQUFDLENBQUM7QUFDNUQsQ0FBQztBQXVCRDs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7O0dBcUJHO0FBQ0gsTUFBYSxhQUFhO0lBT3hCLFlBQVksRUFBb0IsRUFBRSxTQUFpQjtRQUgzQyxTQUFJLEdBQUcsQ0FBQyxDQUFDO1FBQ1QsV0FBTSxHQUFHLENBQUMsQ0FBQztRQUdqQixJQUFJLENBQUMsRUFBRSxHQUFHLEVBQUUsQ0FBQztRQUNiLElBQUksQ0FBQyxTQUFTLEdBQUcsU0FBUyxDQUFDO1FBQzNCLElBQUksQ0FBQyxNQUFNLEdBQUcsTUFBTSxDQUFDLElBQUksQ0FBQyxTQUFTLFNBQVMsR0FBRyxDQUFDLENBQUM7SUFDbkQsQ0FBQztJQUVEOztPQUVHO0lBQ0gsS0FBSyxDQUFDLEdBQUcsQ0FDUCxHQUFXLEVBQ1gsS0FBYSxFQUNiLFNBQW1CLEVBQ25CLFVBQVUsR0FBRyxDQUFDLEVBQ2QsUUFBOEI7UUFFOUIsTUFBTSxLQUFLLEdBQWU7WUFDeEIsR0FBRztZQUNILEtBQUs7WUFDTCxTQUFTO1lBQ1QsU0FBUyxFQUFFLElBQUksQ0FBQyxHQUFHLEVBQUU7WUFDckIsR0FBRyxFQUFFLFVBQVUsR0FBRyxDQUFDLENBQUMsQ0FBQyxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUMsU0FBUztZQUM1QyxRQUFRO1NBQ1QsQ0FBQztRQUVGLE1BQU0sUUFBUSxHQUFHLE1BQU0sQ0FBQyxNQUFNLENBQUM7WUFDN0IsSUFBSSxDQUFDLE1BQU07WUFDWCxNQUFNLENBQUMsSUFBSSxDQUFDLEdBQUcsQ0FBQztTQUNqQixDQUFDLENBQUM7UUFFSCxNQUFNLElBQUksQ0FBQyxFQUFFLENBQUMsR0FBRyxDQUFDLFFBQVEsRUFBRSxNQUFNLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxTQUFTLENBQUMsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDO0lBQ2xFLENBQUM7SUFFRDs7Ozs7O09BTUc7SUFDSCxLQUFLLENBQUMsR0FBRyxDQUNQLGNBQXdCLEVBQ3hCLFNBQVMsR0FBRyxJQUFJO1FBRWhCLE1BQU0sR0FBRyxHQUFHLElBQUksQ0FBQyxHQUFHLEVBQUUsQ0FBQztRQUN2QixJQUFJLFNBQVMsR0FBb0IsSUFBSSxDQUFDO1FBQ3RDLElBQUksU0FBUyxHQUFHLFNBQVMsQ0FBQztRQUUxQiwwQ0FBMEM7UUFDMUMsSUFBSSxDQUFDO1lBQ0gsSUFBSSxLQUFLLEVBQUUsTUFBTSxDQUFDLENBQUMsRUFBRSxXQUFXLENBQUMsSUFBSSxJQUFJLENBQUMsRUFBRSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLEVBQUUsQ0FBQztnQkFDckUsTUFBTSxLQUFLLEdBQWUsSUFBSSxDQUFDLEtBQUssQ0FBQyxXQUFXLENBQUMsUUFBUSxFQUFFLENBQUMsQ0FBQztnQkFFN0QsdUJBQXVCO2dCQUN2QixJQUFJLEtBQUssQ0FBQyxHQUFHLElBQUksS0FBSyxDQUFDLFNBQVMsRUFBRSxDQUFDO29CQUNqQyxNQUFNLFNBQVMsR0FBRyxLQUFLLENBQUMsU0FBUyxHQUFHLEtBQUssQ0FBQyxHQUFHLEdBQUcsSUFBSSxDQUFDO29CQUNyRCxJQUFJLEdBQUcsR0FBRyxTQUFTLEVBQUUsQ0FBQzt3QkFDcEIsU0FBUyxDQUFDLHVCQUF1QjtvQkFDbkMsQ0FBQztnQkFDSCxDQUFDO2dCQUVELHVCQUF1QjtnQkFDdkIsTUFBTSxLQUFLLEdBQUcsZ0JBQWdCLENBQUMsY0FBYyxFQUFFLEtBQUssQ0FBQyxTQUFTLENBQUMsQ0FBQztnQkFFaEUsb0JBQW9CO2dCQUNwQixJQUFJLEtBQUssR0FBRyxTQUFTLEVBQUUsQ0FBQztvQkFDdEIsU0FBUyxHQUFHLEtBQUssQ0FBQztvQkFDbEIsU0FBUyxHQUFHLEVBQUUsR0FBRyxLQUFLLEVBQUUsS0FBSyxFQUFFLENBQUM7Z0JBQ2xDLENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztRQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7WUFDZiw2QkFBNkI7WUFDN0IsSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDO1lBQ2QsT0FBTyxJQUFJLENBQUM7UUFDZCxDQUFDO1FBRUQsSUFBSSxTQUFTLEVBQUUsQ0FBQztZQUNkLElBQUksQ0FBQyxJQUFJLEVBQUUsQ0FBQztRQUNkLENBQUM7YUFBTSxDQUFDO1lBQ04sSUFBSSxDQUFDLE1BQU0sRUFBRSxDQUFDO1FBQ2hCLENBQUM7UUFFRCxPQUFPLFNBQVMsQ0FBQztJQUNuQixDQUFDO0lBRUQ7O09BRUc7SUFDSCxLQUFLLENBQUMsTUFBTSxDQUFDLEdBQVc7UUFDdEIsTUFBTSxRQUFRLEdBQUcsTUFBTSxDQUFDLE1BQU0sQ0FBQztZQUM3QixJQUFJLENBQUMsTUFBTTtZQUNYLE1BQU0sQ0FBQyxJQUFJLENBQUMsR0FBRyxDQUFDO1NBQ2pCLENBQUMsQ0FBQztRQUNILE1BQU0sSUFBSSxDQUFDLEVBQUUsQ0FBQyxNQUFNLENBQUMsUUFBUSxDQUFDLENBQUM7SUFDakMsQ0FBQztJQUVEOztPQUVHO0lBQ0gsS0FBSyxDQUFDLEtBQUs7UUFDVCxJQUFJLE9BQU8sR0FBRyxDQUFDLENBQUM7UUFFaEIsSUFBSSxDQUFDO1lBQ0gsTUFBTSxRQUFRLEdBQWEsRUFBRSxDQUFDO1lBQzlCLElBQUksS0FBSyxFQUFFLE1BQU0sQ0FBQyxHQUFHLENBQUMsSUFBSSxJQUFJLENBQUMsRUFBRSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLEVBQUUsQ0FBQztnQkFDMUQsUUFBUSxDQUFDLElBQUksQ0FBQyxHQUFHLENBQUMsQ0FBQztZQUNyQixDQUFDO1lBRUQsS0FBSyxNQUFNLEdBQUcsSUFBSSxRQUFRLEVBQUUsQ0FBQztnQkFDM0IsTUFBTSxJQUFJLENBQUMsRUFBRSxDQUFDLE1BQU0sQ0FBQyxHQUFHLENBQUMsQ0FBQztnQkFDMUIsT0FBTyxFQUFFLENBQUM7WUFDWixDQUFDO1FBQ0gsQ0FBQztRQUFDLE9BQU8sS0FBSyxFQUFFLENBQUM7WUFDZiwwQ0FBMEM7WUFDMUMsT0FBTyxPQUFPLENBQUM7UUFDakIsQ0FBQztRQUVELGNBQWM7UUFDZCxJQUFJLENBQUMsSUFBSSxHQUFHLENBQUMsQ0FBQztRQUNkLElBQUksQ0FBQyxNQUFNLEdBQUcsQ0FBQyxDQUFDO1FBRWhCLE9BQU8sT0FBTyxDQUFDO0lBQ2pCLENBQUM7SUFFRDs7T0FFRztJQUNILEtBQUssQ0FBQyxLQUFLO1FBQ1QsTUFBTSxHQUFHLEdBQUcsSUFBSSxDQUFDLEdBQUcsRUFBRSxDQUFDO1FBQ3ZCLElBQUksS0FBSyxHQUFHLENBQUMsQ0FBQztRQUNkLElBQUksV0FBVyxHQUFHLENBQUMsQ0FBQztRQUVwQixJQUFJLENBQUM7WUFDSCxJQUFJLEtBQUssRUFBRSxNQUFNLENBQUMsR0FBRyxFQUFFLEtBQUssQ0FBQyxJQUFJLElBQUksQ0FBQyxFQUFFLENBQUMsVUFBVSxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsRUFBRSxDQUFDO2dCQUNqRSxNQUFNLEtBQUssR0FBZSxJQUFJLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxRQUFRLEVBQUUsQ0FBQyxDQUFDO2dCQUV2RCx1QkFBdUI7Z0JBQ3ZCLElBQUksS0FBSyxDQUFDLEdBQUcsSUFBSSxLQUFLLENBQUMsU0FBUyxFQUFFLENBQUM7b0JBQ2pDLE1BQU0sU0FBUyxHQUFHLEtBQUssQ0FBQyxTQUFTLEdBQUcsS0FBSyxDQUFDLEdBQUcsR0FBRyxJQUFJLENBQUM7b0JBQ3JELElBQUksR0FBRyxHQUFHLFNBQVMsRUFBRSxDQUFDO3dCQUNwQixTQUFTO29CQUNYLENBQUM7Z0JBQ0gsQ0FBQztnQkFFRCxLQUFLLEVBQUUsQ0FBQztnQkFDUixXQUFXLElBQUksR0FBRyxDQUFDLE1BQU0sR0FBRyxLQUFLLENBQUMsTUFBTSxDQUFDO1lBQzNDLENBQUM7UUFDSCxDQUFDO1FBQUMsT0FBTyxLQUFLLEVBQUUsQ0FBQztZQUNmLDBDQUEwQztRQUM1QyxDQUFDO1FBRUQsTUFBTSxLQUFLLEdBQUcsSUFBSSxDQUFDLElBQUksR0FBRyxJQUFJLENBQUMsTUFBTSxDQUFDO1FBQ3RDLE1BQU0sT0FBTyxHQUFHLEtBQUssR0FBRyxDQUFDLENBQUMsQ0FBQyxDQUFDLElBQUksQ0FBQyxJQUFJLEdBQUcsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQUM7UUFFbEQsT0FBTztZQUNMLEtBQUs7WUFDTCxJQUFJLEVBQUUsSUFBSSxDQUFDLElBQUk7WUFDZixNQUFNLEVBQUUsSUFBSSxDQUFDLE1BQU07WUFDbkIsT0FBTztZQUNQLFdBQVc7U0FDWixDQUFDO0lBQ0osQ0FBQztJQUVEOztPQUVHO0lBQ0gsS0FBSyxDQUFDLFlBQVk7UUFDaEIsTUFBTSxHQUFHLEdBQUcsSUFBSSxDQUFDLEdBQUcsRUFBRSxDQUFDO1FBQ3ZCLElBQUksTUFBTSxHQUFHLENBQUMsQ0FBQztRQUVmLElBQUksQ0FBQztZQUNILE1BQU0sUUFBUSxHQUFhLEVBQUUsQ0FBQztZQUU5QixJQUFJLEtBQUssRUFBRSxNQUFNLENBQUMsR0FBRyxFQUFFLEtBQUssQ0FBQyxJQUFJLElBQUksQ0FBQyxFQUFFLENBQUMsVUFBVSxDQUFDLElBQUksQ0FBQyxNQUFNLENBQUMsRUFBRSxDQUFDO2dCQUNqRSxNQUFNLEtBQUssR0FBZSxJQUFJLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxRQUFRLEVBQUUsQ0FBQyxDQUFDO2dCQUV2RCxJQUFJLEtBQUssQ0FBQyxHQUFHLElBQUksS0FBSyxDQUFDLFNBQVMsRUFBRSxDQUFDO29CQUNqQyxNQUFNLFNBQVMsR0FBRyxLQUFLLENBQUMsU0FBUyxHQUFHLEtBQUssQ0FBQyxHQUFHLEdBQUcsSUFBSSxDQUFDO29CQUNyRCxJQUFJLEdBQUcsR0FBRyxTQUFTLEVBQUUsQ0FBQzt3QkFDcEIsUUFBUSxDQUFDLElBQUksQ0FBQyxHQUFHLENBQUMsQ0FBQztvQkFDckIsQ0FBQztnQkFDSCxDQUFDO1lBQ0gsQ0FBQztZQUVELEtBQUssTUFBTSxHQUFHLElBQUksUUFBUSxFQUFFLENBQUM7Z0JBQzNCLE1BQU0sSUFBSSxDQUFDLEVBQUUsQ0FBQyxNQUFNLENBQUMsR0FBRyxDQUFDLENBQUM7Z0JBQzFCLE1BQU0sRUFBRSxDQUFDO1lBQ1gsQ0FBQztRQUNILENBQUM7UUFBQyxPQUFPLEtBQUssRUFBRSxDQUFDO1lBQ2Ysc0JBQXNCO1lBQ3RCLE9BQU8sTUFBTSxDQUFDO1FBQ2hCLENBQUM7UUFFRCxPQUFPLE1BQU0sQ0FBQztJQUNoQixDQUFDO0NBQ0Y7QUExTUQsc0NBME1DIiwic291cmNlc0NvbnRlbnQiOlsiLyoqXG4gKiBTZW1hbnRpYyBDYWNoZSBmb3IgTExNIHJlc3BvbnNlc1xuICogXG4gKiBDYWNoZSBMTE0gcmVzcG9uc2VzIHdpdGggc2ltaWxhcml0eS1iYXNlZCByZXRyaWV2YWwgZm9yIGNvc3Qgc2F2aW5ncy5cbiAqIFVzZXMgZGF0YWJhc2UgcHJlZml4IHNjYW5uaW5nIHRvIHN0b3JlIGFuZCByZXRyaWV2ZSBjYWNoZWQgcmVzcG9uc2VzLlxuICovXG5cbmltcG9ydCB7IEVtYmVkZGVkRGF0YWJhc2UgfSBmcm9tICcuL2VtYmVkZGVkJztcblxuLyoqXG4gKiBDYWxjdWxhdGUgY29zaW5lIHNpbWlsYXJpdHkgYmV0d2VlbiB0d28gdmVjdG9yc1xuICovXG5mdW5jdGlvbiBjb3NpbmVTaW1pbGFyaXR5KGE6IG51bWJlcltdLCBiOiBudW1iZXJbXSk6IG51bWJlciB7XG4gIGlmIChhLmxlbmd0aCAhPT0gYi5sZW5ndGgpIHtcbiAgICB0aHJvdyBuZXcgRXJyb3IoJ1ZlY3RvcnMgbXVzdCBoYXZlIHNhbWUgbGVuZ3RoJyk7XG4gIH1cblxuICBsZXQgZG90UHJvZHVjdCA9IDA7XG4gIGxldCBub3JtQSA9IDA7XG4gIGxldCBub3JtQiA9IDA7XG5cbiAgZm9yIChsZXQgaSA9IDA7IGkgPCBhLmxlbmd0aDsgaSsrKSB7XG4gICAgZG90UHJvZHVjdCArPSBhW2ldICogYltpXTtcbiAgICBub3JtQSArPSBhW2ldICogYVtpXTtcbiAgICBub3JtQiArPSBiW2ldICogYltpXTtcbiAgfVxuXG4gIGlmIChub3JtQSA9PT0gMCB8fCBub3JtQiA9PT0gMCkge1xuICAgIHJldHVybiAwO1xuICB9XG5cbiAgcmV0dXJuIGRvdFByb2R1Y3QgLyAoTWF0aC5zcXJ0KG5vcm1BKSAqIE1hdGguc3FydChub3JtQikpO1xufVxuXG5leHBvcnQgaW50ZXJmYWNlIENhY2hlRW50cnkge1xuICBrZXk6IHN0cmluZztcbiAgdmFsdWU6IHN0cmluZztcbiAgZW1iZWRkaW5nOiBudW1iZXJbXTtcbiAgdGltZXN0YW1wOiBudW1iZXI7XG4gIHR0bD86IG51bWJlcjtcbiAgbWV0YWRhdGE/OiBSZWNvcmQ8c3RyaW5nLCBhbnk+O1xufVxuXG5leHBvcnQgaW50ZXJmYWNlIENhY2hlSGl0IGV4dGVuZHMgQ2FjaGVFbnRyeSB7XG4gIHNjb3JlOiBudW1iZXI7XG59XG5cbmV4cG9ydCBpbnRlcmZhY2UgQ2FjaGVTdGF0cyB7XG4gIGNvdW50OiBudW1iZXI7XG4gIGhpdHM6IG51bWJlcjtcbiAgbWlzc2VzOiBudW1iZXI7XG4gIGhpdFJhdGU6IG51bWJlcjtcbiAgbWVtb3J5VXNhZ2U6IG51bWJlcjtcbn1cblxuLyoqXG4gKiBTZW1hbnRpYyBDYWNoZSB3aXRoIHZlY3RvciBzaW1pbGFyaXR5IG1hdGNoaW5nXG4gKiBcbiAqIEBleGFtcGxlXG4gKiBgYGB0eXBlc2NyaXB0XG4gKiBjb25zdCBjYWNoZSA9IG5ldyBTZW1hbnRpY0NhY2hlKGRiLCAnbGxtX3Jlc3BvbnNlcycpO1xuICogXG4gKiAvLyBTdG9yZSByZXNwb25zZVxuICogYXdhaXQgY2FjaGUucHV0KFxuICogICAnV2hhdCBpcyBQeXRob24/JyxcbiAqICAgJ1B5dGhvbiBpcyBhIGhpZ2gtbGV2ZWwgcHJvZ3JhbW1pbmcgbGFuZ3VhZ2UuLi4nLFxuICogICBlbWJlZGRpbmcsXG4gKiAgIDM2MDAgIC8vIFRUTCBpbiBzZWNvbmRzXG4gKiApO1xuICogXG4gKiAvLyBDaGVjayBjYWNoZVxuICogY29uc3QgaGl0ID0gYXdhaXQgY2FjaGUuZ2V0KHF1ZXJ5RW1iZWRkaW5nLCAwLjg1KTtcbiAqIGlmIChoaXQpIHtcbiAqICAgY29uc29sZS5sb2coYENhY2hlIEhJVDogJHtoaXQudmFsdWV9IChzaW1pbGFyaXR5OiAke2hpdC5zY29yZX0pYCk7XG4gKiB9XG4gKiBgYGBcbiAqL1xuZXhwb3J0IGNsYXNzIFNlbWFudGljQ2FjaGUge1xuICBwcml2YXRlIGRiOiBFbWJlZGRlZERhdGFiYXNlO1xuICBwcml2YXRlIGNhY2hlTmFtZTogc3RyaW5nO1xuICBwcml2YXRlIHByZWZpeDogQnVmZmVyO1xuICBwcml2YXRlIGhpdHMgPSAwO1xuICBwcml2YXRlIG1pc3NlcyA9IDA7XG5cbiAgY29uc3RydWN0b3IoZGI6IEVtYmVkZGVkRGF0YWJhc2UsIGNhY2hlTmFtZTogc3RyaW5nKSB7XG4gICAgdGhpcy5kYiA9IGRiO1xuICAgIHRoaXMuY2FjaGVOYW1lID0gY2FjaGVOYW1lO1xuICAgIHRoaXMucHJlZml4ID0gQnVmZmVyLmZyb20oYGNhY2hlOiR7Y2FjaGVOYW1lfTpgKTtcbiAgfVxuXG4gIC8qKlxuICAgKiBTdG9yZSBhIGNhY2hlZCByZXNwb25zZVxuICAgKi9cbiAgYXN5bmMgcHV0KFxuICAgIGtleTogc3RyaW5nLFxuICAgIHZhbHVlOiBzdHJpbmcsXG4gICAgZW1iZWRkaW5nOiBudW1iZXJbXSxcbiAgICB0dGxTZWNvbmRzID0gMCxcbiAgICBtZXRhZGF0YT86IFJlY29yZDxzdHJpbmcsIGFueT5cbiAgKTogUHJvbWlzZTx2b2lkPiB7XG4gICAgY29uc3QgZW50cnk6IENhY2hlRW50cnkgPSB7XG4gICAgICBrZXksXG4gICAgICB2YWx1ZSxcbiAgICAgIGVtYmVkZGluZyxcbiAgICAgIHRpbWVzdGFtcDogRGF0ZS5ub3coKSxcbiAgICAgIHR0bDogdHRsU2Vjb25kcyA+IDAgPyB0dGxTZWNvbmRzIDogdW5kZWZpbmVkLFxuICAgICAgbWV0YWRhdGEsXG4gICAgfTtcblxuICAgIGNvbnN0IGVudHJ5S2V5ID0gQnVmZmVyLmNvbmNhdChbXG4gICAgICB0aGlzLnByZWZpeCxcbiAgICAgIEJ1ZmZlci5mcm9tKGtleSksXG4gICAgXSk7XG5cbiAgICBhd2FpdCB0aGlzLmRiLnB1dChlbnRyeUtleSwgQnVmZmVyLmZyb20oSlNPTi5zdHJpbmdpZnkoZW50cnkpKSk7XG4gIH1cblxuICAvKipcbiAgICogUmV0cmlldmUgY2FjaGVkIHJlc3BvbnNlIGJ5IHNpbWlsYXJpdHlcbiAgICogXG4gICAqIEBwYXJhbSBxdWVyeUVtYmVkZGluZyAtIFF1ZXJ5IGVtYmVkZGluZyB2ZWN0b3JcbiAgICogQHBhcmFtIHRocmVzaG9sZCAtIE1pbmltdW0gY29zaW5lIHNpbWlsYXJpdHkgKDAtMSlcbiAgICogQHJldHVybnMgQmVzdCBtYXRjaGluZyBjYWNoZSBlbnRyeSBvciBudWxsXG4gICAqL1xuICBhc3luYyBnZXQoXG4gICAgcXVlcnlFbWJlZGRpbmc6IG51bWJlcltdLFxuICAgIHRocmVzaG9sZCA9IDAuODVcbiAgKTogUHJvbWlzZTxDYWNoZUhpdCB8IG51bGw+IHtcbiAgICBjb25zdCBub3cgPSBEYXRlLm5vdygpO1xuICAgIGxldCBiZXN0TWF0Y2g6IENhY2hlSGl0IHwgbnVsbCA9IG51bGw7XG4gICAgbGV0IGJlc3RTY29yZSA9IHRocmVzaG9sZDtcblxuICAgIC8vIFNjYW4gYWxsIGNhY2hlIGVudHJpZXMgd2l0aCB0aGlzIHByZWZpeFxuICAgIHRyeSB7XG4gICAgICBmb3IgYXdhaXQgKGNvbnN0IFtfLCB2YWx1ZUJ1ZmZlcl0gb2YgdGhpcy5kYi5zY2FuUHJlZml4KHRoaXMucHJlZml4KSkge1xuICAgICAgICBjb25zdCBlbnRyeTogQ2FjaGVFbnRyeSA9IEpTT04ucGFyc2UodmFsdWVCdWZmZXIudG9TdHJpbmcoKSk7XG5cbiAgICAgICAgLy8gQ2hlY2sgVFRMIGV4cGlyYXRpb25cbiAgICAgICAgaWYgKGVudHJ5LnR0bCAmJiBlbnRyeS50aW1lc3RhbXApIHtcbiAgICAgICAgICBjb25zdCBleHBpcmVzQXQgPSBlbnRyeS50aW1lc3RhbXAgKyBlbnRyeS50dGwgKiAxMDAwO1xuICAgICAgICAgIGlmIChub3cgPiBleHBpcmVzQXQpIHtcbiAgICAgICAgICAgIGNvbnRpbnVlOyAvLyBTa2lwIGV4cGlyZWQgZW50cmllc1xuICAgICAgICAgIH1cbiAgICAgICAgfVxuXG4gICAgICAgIC8vIENhbGN1bGF0ZSBzaW1pbGFyaXR5XG4gICAgICAgIGNvbnN0IHNjb3JlID0gY29zaW5lU2ltaWxhcml0eShxdWVyeUVtYmVkZGluZywgZW50cnkuZW1iZWRkaW5nKTtcblxuICAgICAgICAvLyBVcGRhdGUgYmVzdCBtYXRjaFxuICAgICAgICBpZiAoc2NvcmUgPiBiZXN0U2NvcmUpIHtcbiAgICAgICAgICBiZXN0U2NvcmUgPSBzY29yZTtcbiAgICAgICAgICBiZXN0TWF0Y2ggPSB7IC4uLmVudHJ5LCBzY29yZSB9O1xuICAgICAgICB9XG4gICAgICB9XG4gICAgfSBjYXRjaCAoZXJyb3IpIHtcbiAgICAgIC8vIElmIHNjYW4gZmFpbHMsIHJldHVybiBudWxsXG4gICAgICB0aGlzLm1pc3NlcysrO1xuICAgICAgcmV0dXJuIG51bGw7XG4gICAgfVxuXG4gICAgaWYgKGJlc3RNYXRjaCkge1xuICAgICAgdGhpcy5oaXRzKys7XG4gICAgfSBlbHNlIHtcbiAgICAgIHRoaXMubWlzc2VzKys7XG4gICAgfVxuXG4gICAgcmV0dXJuIGJlc3RNYXRjaDtcbiAgfVxuXG4gIC8qKlxuICAgKiBEZWxldGUgYSBzcGVjaWZpYyBjYWNoZSBlbnRyeVxuICAgKi9cbiAgYXN5bmMgZGVsZXRlKGtleTogc3RyaW5nKTogUHJvbWlzZTx2b2lkPiB7XG4gICAgY29uc3QgZW50cnlLZXkgPSBCdWZmZXIuY29uY2F0KFtcbiAgICAgIHRoaXMucHJlZml4LFxuICAgICAgQnVmZmVyLmZyb20oa2V5KSxcbiAgICBdKTtcbiAgICBhd2FpdCB0aGlzLmRiLmRlbGV0ZShlbnRyeUtleSk7XG4gIH1cblxuICAvKipcbiAgICogQ2xlYXIgYWxsIGVudHJpZXMgaW4gdGhpcyBjYWNoZVxuICAgKi9cbiAgYXN5bmMgY2xlYXIoKTogUHJvbWlzZTxudW1iZXI+IHtcbiAgICBsZXQgZGVsZXRlZCA9IDA7XG5cbiAgICB0cnkge1xuICAgICAgY29uc3QgdG9EZWxldGU6IEJ1ZmZlcltdID0gW107XG4gICAgICBmb3IgYXdhaXQgKGNvbnN0IFtrZXldIG9mIHRoaXMuZGIuc2NhblByZWZpeCh0aGlzLnByZWZpeCkpIHtcbiAgICAgICAgdG9EZWxldGUucHVzaChrZXkpO1xuICAgICAgfVxuXG4gICAgICBmb3IgKGNvbnN0IGtleSBvZiB0b0RlbGV0ZSkge1xuICAgICAgICBhd2FpdCB0aGlzLmRiLmRlbGV0ZShrZXkpO1xuICAgICAgICBkZWxldGVkKys7XG4gICAgICB9XG4gICAgfSBjYXRjaCAoZXJyb3IpIHtcbiAgICAgIC8vIElmIG9wZXJhdGlvbiBmYWlscywgcmV0dXJuIGNvdW50IHNvIGZhclxuICAgICAgcmV0dXJuIGRlbGV0ZWQ7XG4gICAgfVxuXG4gICAgLy8gUmVzZXQgc3RhdHNcbiAgICB0aGlzLmhpdHMgPSAwO1xuICAgIHRoaXMubWlzc2VzID0gMDtcblxuICAgIHJldHVybiBkZWxldGVkO1xuICB9XG5cbiAgLyoqXG4gICAqIEdldCBjYWNoZSBzdGF0aXN0aWNzXG4gICAqL1xuICBhc3luYyBzdGF0cygpOiBQcm9taXNlPENhY2hlU3RhdHM+IHtcbiAgICBjb25zdCBub3cgPSBEYXRlLm5vdygpO1xuICAgIGxldCBjb3VudCA9IDA7XG4gICAgbGV0IG1lbW9yeVVzYWdlID0gMDtcblxuICAgIHRyeSB7XG4gICAgICBmb3IgYXdhaXQgKGNvbnN0IFtrZXksIHZhbHVlXSBvZiB0aGlzLmRiLnNjYW5QcmVmaXgodGhpcy5wcmVmaXgpKSB7XG4gICAgICAgIGNvbnN0IGVudHJ5OiBDYWNoZUVudHJ5ID0gSlNPTi5wYXJzZSh2YWx1ZS50b1N0cmluZygpKTtcbiAgICAgICAgXG4gICAgICAgIC8vIFNraXAgZXhwaXJlZCBlbnRyaWVzXG4gICAgICAgIGlmIChlbnRyeS50dGwgJiYgZW50cnkudGltZXN0YW1wKSB7XG4gICAgICAgICAgY29uc3QgZXhwaXJlc0F0ID0gZW50cnkudGltZXN0YW1wICsgZW50cnkudHRsICogMTAwMDtcbiAgICAgICAgICBpZiAobm93ID4gZXhwaXJlc0F0KSB7XG4gICAgICAgICAgICBjb250aW51ZTtcbiAgICAgICAgICB9XG4gICAgICAgIH1cblxuICAgICAgICBjb3VudCsrO1xuICAgICAgICBtZW1vcnlVc2FnZSArPSBrZXkubGVuZ3RoICsgdmFsdWUubGVuZ3RoO1xuICAgICAgfVxuICAgIH0gY2F0Y2ggKGVycm9yKSB7XG4gICAgICAvLyBSZXR1cm4gcGFydGlhbCBzdGF0cyBpZiBvcGVyYXRpb24gZmFpbHNcbiAgICB9XG5cbiAgICBjb25zdCB0b3RhbCA9IHRoaXMuaGl0cyArIHRoaXMubWlzc2VzO1xuICAgIGNvbnN0IGhpdFJhdGUgPSB0b3RhbCA+IDAgPyB0aGlzLmhpdHMgLyB0b3RhbCA6IDA7XG5cbiAgICByZXR1cm4ge1xuICAgICAgY291bnQsXG4gICAgICBoaXRzOiB0aGlzLmhpdHMsXG4gICAgICBtaXNzZXM6IHRoaXMubWlzc2VzLFxuICAgICAgaGl0UmF0ZSxcbiAgICAgIG1lbW9yeVVzYWdlLFxuICAgIH07XG4gIH1cblxuICAvKipcbiAgICogUHVyZ2UgZXhwaXJlZCBlbnRyaWVzXG4gICAqL1xuICBhc3luYyBwdXJnZUV4cGlyZWQoKTogUHJvbWlzZTxudW1iZXI+IHtcbiAgICBjb25zdCBub3cgPSBEYXRlLm5vdygpO1xuICAgIGxldCBwdXJnZWQgPSAwO1xuXG4gICAgdHJ5IHtcbiAgICAgIGNvbnN0IHRvRGVsZXRlOiBCdWZmZXJbXSA9IFtdO1xuXG4gICAgICBmb3IgYXdhaXQgKGNvbnN0IFtrZXksIHZhbHVlXSBvZiB0aGlzLmRiLnNjYW5QcmVmaXgodGhpcy5wcmVmaXgpKSB7XG4gICAgICAgIGNvbnN0IGVudHJ5OiBDYWNoZUVudHJ5ID0gSlNPTi5wYXJzZSh2YWx1ZS50b1N0cmluZygpKTtcbiAgICAgICAgXG4gICAgICAgIGlmIChlbnRyeS50dGwgJiYgZW50cnkudGltZXN0YW1wKSB7XG4gICAgICAgICAgY29uc3QgZXhwaXJlc0F0ID0gZW50cnkudGltZXN0YW1wICsgZW50cnkudHRsICogMTAwMDtcbiAgICAgICAgICBpZiAobm93ID4gZXhwaXJlc0F0KSB7XG4gICAgICAgICAgICB0b0RlbGV0ZS5wdXNoKGtleSk7XG4gICAgICAgICAgfVxuICAgICAgICB9XG4gICAgICB9XG5cbiAgICAgIGZvciAoY29uc3Qga2V5IG9mIHRvRGVsZXRlKSB7XG4gICAgICAgIGF3YWl0IHRoaXMuZGIuZGVsZXRlKGtleSk7XG4gICAgICAgIHB1cmdlZCsrO1xuICAgICAgfVxuICAgIH0gY2F0Y2ggKGVycm9yKSB7XG4gICAgICAvLyBSZXR1cm4gY291bnQgc28gZmFyXG4gICAgICByZXR1cm4gcHVyZ2VkO1xuICAgIH1cblxuICAgIHJldHVybiBwdXJnZWQ7XG4gIH1cbn1cbiJdfQ==

package/dist/esm/context-builder.js

@@ -0,0 +1,280 @@
+"use strict";
+/**
+ * Context Query Builder for LLM Context Assembly
+ *
+ * Token-aware context assembly with priority-based truncation.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ContextQueryBuilder = exports.TruncationStrategy = exports.ContextOutputFormat = void 0;
+exports.createContextBuilder = createContextBuilder;
+var ContextOutputFormat;
+(function (ContextOutputFormat) {
+    ContextOutputFormat["TOON"] = "toon";
+    ContextOutputFormat["JSON"] = "json";
+    ContextOutputFormat["MARKDOWN"] = "markdown";
+})(ContextOutputFormat || (exports.ContextOutputFormat = ContextOutputFormat = {}));
+var TruncationStrategy;
+(function (TruncationStrategy) {
+    TruncationStrategy["TAIL_DROP"] = "tail_drop";
+    TruncationStrategy["HEAD_DROP"] = "head_drop";
+    TruncationStrategy["PROPORTIONAL"] = "proportional";
+})(TruncationStrategy || (exports.TruncationStrategy = TruncationStrategy = {}));
+/**
+ * Context Query Builder for assembling LLM context
+ */
+class ContextQueryBuilder {
+    sessionId;
+    tokenBudget = 4096;
+    format = ContextOutputFormat.TOON;
+    truncation = TruncationStrategy.TAIL_DROP;
+    sections = [];
+    currentSection;
+    /**
+     * Set session ID for context
+     */
+    forSession(sessionId) {
+        this.sessionId = sessionId;
+        return this;
+    }
+    /**
+     * Set token budget
+     */
+    withBudget(tokens) {
+        this.tokenBudget = tokens;
+        return this;
+    }
+    /**
+     * Set output format
+     */
+    setFormat(format) {
+        this.format = format;
+        return this;
+    }
+    /**
+     * Set truncation strategy
+     */
+    setTruncation(strategy) {
+        this.truncation = strategy;
+        return this;
+    }
+    /**
+     * Add literal text section
+     */
+    literal(name, priority, text) {
+        const tokenCount = this.estimateTokens(text);
+        this.sections.push({
+            name,
+            priority,
+            content: text,
+            tokenCount,
+        });
+        return this;
+    }
+    /**
+     * Start a new section
+     */
+    section(name, priority) {
+        this.currentSection = {
+            name,
+            priority,
+            content: '',
+            tokenCount: 0,
+        };
+        return this;
+    }
+    /**
+     * Add content to current section
+     */
+    get(path) {
+        if (!this.currentSection) {
+            throw new Error('No active section. Call section() first.');
+        }
+        this.currentSection.content += `GET ${path}\n`;
+        return this;
+    }
+    /**
+     * Add last N records query
+     */
+    last(n, table) {
+        if (!this.currentSection) {
+            throw new Error('No active section. Call section() first.');
+        }
+        this.currentSection.content += `LAST ${n} FROM ${table}\n`;
+        return this;
+    }
+    /**
+     * Add where equals condition
+     */
+    whereEq(field, value) {
+        if (!this.currentSection) {
+            throw new Error('No active section. Call section() first.');
+        }
+        this.currentSection.content += `WHERE ${field} = ${JSON.stringify(value)}\n`;
+        return this;
+    }
+    /**
+     * Add vector search
+     */
+    search(collection, embedding, k) {
+        if (!this.currentSection) {
+            throw new Error('No active section. Call section() first.');
+        }
+        this.currentSection.content += `SEARCH ${collection} WITH ${embedding} LIMIT ${k}\n`;
+        return this;
+    }
+    /**
+     * Add SQL query
+     */
+    sql(query) {
+        if (!this.currentSection) {
+            throw new Error('No active section. Call section() first.');
+        }
+        this.currentSection.content += `SQL: ${query}\n`;
+        return this;
+    }
+    /**
+     * Finish current section
+     */
+    done() {
+        if (this.currentSection) {
+            this.currentSection.tokenCount = this.estimateTokens(this.currentSection.content);
+            this.sections.push(this.currentSection);
+            this.currentSection = undefined;
+        }
+        return this;
+    }
+    /**
+     * Execute and build context
+     */
+    execute() {
+        // Finish any pending section
+        if (this.currentSection) {
+            this.done();
+        }
+        // Sort sections by priority (lower = higher priority)
+        const sortedSections = [...this.sections].sort((a, b) => a.priority - b.priority);
+        // Calculate total tokens
+        let totalTokens = sortedSections.reduce((sum, s) => sum + s.tokenCount, 0);
+        // Truncate if needed
+        const truncatedSections = [];
+        const includedSections = [];
+        if (totalTokens <= this.tokenBudget) {
+            // No truncation needed
+            for (const section of sortedSections) {
+                includedSections.push(section);
+                truncatedSections.push({
+                    name: section.name,
+                    tokenCount: section.tokenCount,
+                    truncated: false,
+                });
+            }
+        }
+        else {
+            // Apply truncation strategy
+            let remainingBudget = this.tokenBudget;
+            if (this.truncation === TruncationStrategy.TAIL_DROP) {
+                // Include sections in priority order until budget exhausted
+                for (const section of sortedSections) {
+                    if (section.tokenCount <= remainingBudget) {
+                        includedSections.push(section);
+                        remainingBudget -= section.tokenCount;
+                        truncatedSections.push({
+                            name: section.name,
+                            tokenCount: section.tokenCount,
+                            truncated: false,
+                        });
+                    }
+                    else {
+                        truncatedSections.push({
+                            name: section.name,
+                            tokenCount: 0,
+                            truncated: true,
+                        });
+                    }
+                }
+            }
+            else if (this.truncation === TruncationStrategy.PROPORTIONAL) {
+                // Proportionally reduce all sections
+                const ratio = this.tokenBudget / totalTokens;
+                for (const section of sortedSections) {
+                    const allocatedTokens = Math.floor(section.tokenCount * ratio);
+                    const truncatedContent = this.truncateText(section.content, allocatedTokens);
+                    includedSections.push({
+                        ...section,
+                        content: truncatedContent,
+                        tokenCount: allocatedTokens,
+                    });
+                    truncatedSections.push({
+                        name: section.name,
+                        tokenCount: allocatedTokens,
+                        truncated: allocatedTokens < section.tokenCount,
+                    });
+                }
+            }
+        }
+        // Build final context based on format
+        let text = '';
+        let actualTokens = 0;
+        if (this.format === ContextOutputFormat.TOON) {
+            text = this.buildToonFormat(includedSections);
+        }
+        else if (this.format === ContextOutputFormat.JSON) {
+            text = this.buildJsonFormat(includedSections);
+        }
+        else {
+            text = this.buildMarkdownFormat(includedSections);
+        }
+        actualTokens = this.estimateTokens(text);
+        return {
+            text,
+            tokenCount: actualTokens,
+            sections: truncatedSections,
+        };
+    }
+    // Helper methods
+    estimateTokens(text) {
+        // Rough estimate: ~4 characters per token for English
+        return Math.ceil(text.length / 4);
+    }
+    truncateText(text, maxTokens) {
+        const maxChars = maxTokens * 4;
+        if (text.length <= maxChars) {
+            return text;
+        }
+        return text.substring(0, maxChars) + '...';
+    }
+    buildToonFormat(sections) {
+        const lines = [];
+        for (const section of sections) {
+            lines.push(`[${section.name}]`);
+            lines.push(section.content);
+            lines.push('');
+        }
+        return lines.join('\n');
+    }
+    buildJsonFormat(sections) {
+        const obj = {};
+        for (const section of sections) {
+            obj[section.name] = section.content;
+        }
+        return JSON.stringify(obj, null, 2);
+    }
+    buildMarkdownFormat(sections) {
+        const lines = [];
+        for (const section of sections) {
+            lines.push(`## ${section.name}`);
+            lines.push('');
+            lines.push(section.content);
+            lines.push('');
+        }
+        return lines.join('\n');
+    }
+}
+exports.ContextQueryBuilder = ContextQueryBuilder;
+/**
+ * Create a context query builder
+ */
+function createContextBuilder() {
+    return new ContextQueryBuilder();
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiY29udGV4dC1idWlsZGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vc3JjL2NvbnRleHQtYnVpbGRlci50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiO0FBQUE7Ozs7R0FJRzs7O0FBc1RILG9EQUVDO0FBdFRELElBQVksbUJBSVg7QUFKRCxXQUFZLG1CQUFtQjtJQUM3QixvQ0FBYSxDQUFBO0lBQ2Isb0NBQWEsQ0FBQTtJQUNiLDRDQUFxQixDQUFBO0FBQ3ZCLENBQUMsRUFKVyxtQkFBbUIsbUNBQW5CLG1CQUFtQixRQUk5QjtBQUVELElBQVksa0JBSVg7QUFKRCxXQUFZLGtCQUFrQjtJQUM1Qiw2Q0FBdUIsQ0FBQTtJQUN2Qiw2Q0FBdUIsQ0FBQTtJQUN2QixtREFBNkIsQ0FBQTtBQUMvQixDQUFDLEVBSlcsa0JBQWtCLGtDQUFsQixrQkFBa0IsUUFJN0I7QUFlRDs7R0FFRztBQUNILE1BQWEsbUJBQW1CO0lBQ3RCLFNBQVMsQ0FBVTtJQUNuQixXQUFXLEdBQVcsSUFBSSxDQUFDO0lBQzNCLE1BQU0sR0FBd0IsbUJBQW1CLENBQUMsSUFBSSxDQUFDO0lBQ3ZELFVBQVUsR0FBdUIsa0JBQWtCLENBQUMsU0FBUyxDQUFDO0lBQzlELFFBQVEsR0FBYyxFQUFFLENBQUM7SUFDekIsY0FBYyxDQUFXO0lBRWpDOztPQUVHO0lBQ0gsVUFBVSxDQUFDLFNBQWlCO1FBQzFCLElBQUksQ0FBQyxTQUFTLEdBQUcsU0FBUyxDQUFDO1FBQzNCLE9BQU8sSUFBSSxDQUFDO0lBQ2QsQ0FBQztJQUVEOztPQUVHO0lBQ0gsVUFBVSxDQUFDLE1BQWM7UUFDdkIsSUFBSSxDQUFDLFdBQVcsR0FBRyxNQUFNLENBQUM7UUFDMUIsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQ7O09BRUc7SUFDSCxTQUFTLENBQUMsTUFBMkI7UUFDbkMsSUFBSSxDQUFDLE1BQU0sR0FBRyxNQUFNLENBQUM7UUFDckIsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQ7O09BRUc7SUFDSCxhQUFhLENBQUMsUUFBNEI7UUFDeEMsSUFBSSxDQUFDLFVBQVUsR0FBRyxRQUFRLENBQUM7UUFDM0IsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQ7O09BRUc7SUFDSCxPQUFPLENBQUMsSUFBWSxFQUFFLFFBQWdCLEVBQUUsSUFBWTtRQUNsRCxNQUFNLFVBQVUsR0FBRyxJQUFJLENBQUMsY0FBYyxDQUFDLElBQUksQ0FBQyxDQUFDO1FBQzdDLElBQUksQ0FBQyxRQUFRLENBQUMsSUFBSSxDQUFDO1lBQ2pCLElBQUk7WUFDSixRQUFRO1lBQ1IsT0FBTyxFQUFFLElBQUk7WUFDYixVQUFVO1NBQ1gsQ0FBQyxDQUFDO1FBQ0gsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQ7O09BRUc7SUFDSCxPQUFPLENBQUMsSUFBWSxFQUFFLFFBQWdCO1FBQ3BDLElBQUksQ0FBQyxjQUFjLEdBQUc7WUFDcEIsSUFBSTtZQUNKLFFBQVE7WUFDUixPQUFPLEVBQUUsRUFBRTtZQUNYLFVBQVUsRUFBRSxDQUFDO1NBQ2QsQ0FBQztRQUNGLE9BQU8sSUFBSSxDQUFDO0lBQ2QsQ0FBQztJQUVEOztPQUVHO0lBQ0gsR0FBRyxDQUFDLElBQVk7UUFDZCxJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO1lBQ3pCLE1BQU0sSUFBSSxLQUFLLENBQUMsMENBQTBDLENBQUMsQ0FBQztRQUM5RCxDQUFDO1FBQ0QsSUFBSSxDQUFDLGNBQWMsQ0FBQyxPQUFPLElBQUksT0FBTyxJQUFJLElBQUksQ0FBQztRQUMvQyxPQUFPLElBQUksQ0FBQztJQUNkLENBQUM7SUFFRDs7T0FFRztJQUNILElBQUksQ0FBQyxDQUFTLEVBQUUsS0FBYTtRQUMzQixJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO1lBQ3pCLE1BQU0sSUFBSSxLQUFLLENBQUMsMENBQTBDLENBQUMsQ0FBQztRQUM5RCxDQUFDO1FBQ0QsSUFBSSxDQUFDLGNBQWMsQ0FBQyxPQUFPLElBQUksUUFBUSxDQUFDLFNBQVMsS0FBSyxJQUFJLENBQUM7UUFDM0QsT0FBTyxJQUFJLENBQUM7SUFDZCxDQUFDO0lBRUQ7O09BRUc7SUFDSCxPQUFPLENBQUMsS0FBYSxFQUFFLEtBQVU7UUFDL0IsSUFBSSxDQUFDLElBQUksQ0FBQyxjQUFjLEVBQUUsQ0FBQztZQUN6QixNQUFNLElBQUksS0FBSyxDQUFDLDBDQUEwQyxDQUFDLENBQUM7UUFDOUQsQ0FBQztRQUNELElBQUksQ0FBQyxjQUFjLENBQUMsT0FBTyxJQUFJLFNBQVMsS0FBSyxNQUFNLElBQUksQ0FBQyxTQUFTLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQztRQUM3RSxPQUFPLElBQUksQ0FBQztJQUNkLENBQUM7SUFFRDs7T0FFRztJQUNILE1BQU0sQ0FBQyxVQUFrQixFQUFFLFNBQWlCLEVBQUUsQ0FBUztRQUNyRCxJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO1lBQ3pCLE1BQU0sSUFBSSxLQUFLLENBQUMsMENBQTBDLENBQUMsQ0FBQztRQUM5RCxDQUFDO1FBQ0QsSUFBSSxDQUFDLGNBQWMsQ0FBQyxPQUFPLElBQUksVUFBVSxVQUFVLFNBQVMsU0FBUyxVQUFVLENBQUMsSUFBSSxDQUFDO1FBQ3JGLE9BQU8sSUFBSSxDQUFDO0lBQ2QsQ0FBQztJQUVEOztPQUVHO0lBQ0gsR0FBRyxDQUFDLEtBQWE7UUFDZixJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO1lBQ3pCLE1BQU0sSUFBSSxLQUFLLENBQUMsMENBQTBDLENBQUMsQ0FBQztRQUM5RCxDQUFDO1FBQ0QsSUFBSSxDQUFDLGNBQWMsQ0FBQyxPQUFPLElBQUksUUFBUSxLQUFLLElBQUksQ0FBQztRQUNqRCxPQUFPLElBQUksQ0FBQztJQUNkLENBQUM7SUFFRDs7T0FFRztJQUNILElBQUk7UUFDRixJQUFJLElBQUksQ0FBQyxjQUFjLEVBQUUsQ0FBQztZQUN4QixJQUFJLENBQUMsY0FBYyxDQUFDLFVBQVUsR0FBRyxJQUFJLENBQUMsY0FBYyxDQUFDLElBQUksQ0FBQyxjQUFjLENBQUMsT0FBTyxDQUFDLENBQUM7WUFDbEYsSUFBSSxDQUFDLFFBQVEsQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLGNBQWMsQ0FBQyxDQUFDO1lBQ3hDLElBQUksQ0FBQyxjQUFjLEdBQUcsU0FBUyxDQUFDO1FBQ2xDLENBQUM7UUFDRCxPQUFPLElBQUksQ0FBQztJQUNkLENBQUM7SUFFRDs7T0FFRztJQUNILE9BQU87UUFDTCw2QkFBNkI7UUFDN0IsSUFBSSxJQUFJLENBQUMsY0FBYyxFQUFFLENBQUM7WUFDeEIsSUFBSSxDQUFDLElBQUksRUFBRSxDQUFDO1FBQ2QsQ0FBQztRQUVELHNEQUFzRDtRQUN0RCxNQUFNLGNBQWMsR0FBRyxDQUFDLEdBQUcsSUFBSSxDQUFDLFFBQVEsQ0FBQyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsRUFBRSxDQUFDLEVBQUUsRUFBRSxDQUFDLENBQUMsQ0FBQyxRQUFRLEdBQUcsQ0FBQyxDQUFDLFFBQVEsQ0FBQyxDQUFDO1FBRWxGLHlCQUF5QjtRQUN6QixJQUFJLFdBQVcsR0FBRyxjQUFjLENBQUMsTUFBTSxDQUFDLENBQUMsR0FBRyxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUMsR0FBRyxHQUFHLENBQUMsQ0FBQyxVQUFVLEVBQUUsQ0FBQyxDQUFDLENBQUM7UUFFM0UscUJBQXFCO1FBQ3JCLE1BQU0saUJBQWlCLEdBQW9FLEVBQUUsQ0FBQztRQUM5RixNQUFNLGdCQUFnQixHQUFjLEVBQUUsQ0FBQztRQUV2QyxJQUFJLFdBQVcsSUFBSSxJQUFJLENBQUMsV0FBVyxFQUFFLENBQUM7WUFDcEMsdUJBQXVCO1lBQ3ZCLEtBQUssTUFBTSxPQUFPLElBQUksY0FBYyxFQUFFLENBQUM7Z0JBQ3JDLGdCQUFnQixDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsQ0FBQztnQkFDL0IsaUJBQWlCLENBQUMsSUFBSSxDQUFDO29CQUNyQixJQUFJLEVBQUUsT0FBTyxDQUFDLElBQUk7b0JBQ2xCLFVBQVUsRUFBRSxPQUFPLENBQUMsVUFBVTtvQkFDOUIsU0FBUyxFQUFFLEtBQUs7aUJBQ2pCLENBQUMsQ0FBQztZQUNMLENBQUM7UUFDSCxDQUFDO2FBQU0sQ0FBQztZQUNOLDRCQUE0QjtZQUM1QixJQUFJLGVBQWUsR0FBRyxJQUFJLENBQUMsV0FBVyxDQUFDO1lBRXZDLElBQUksSUFBSSxDQUFDLFVBQVUsS0FBSyxrQkFBa0IsQ0FBQyxTQUFTLEVBQUUsQ0FBQztnQkFDckQsNERBQTREO2dCQUM1RCxLQUFLLE1BQU0sT0FBTyxJQUFJLGNBQWMsRUFBRSxDQUFDO29CQUNyQyxJQUFJLE9BQU8sQ0FBQyxVQUFVLElBQUksZUFBZSxFQUFFLENBQUM7d0JBQzFDLGdCQUFnQixDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsQ0FBQzt3QkFDL0IsZUFBZSxJQUFJLE9BQU8sQ0FBQyxVQUFVLENBQUM7d0JBQ3RDLGlCQUFpQixDQUFDLElBQUksQ0FBQzs0QkFDckIsSUFBSSxFQUFFLE9BQU8sQ0FBQyxJQUFJOzRCQUNsQixVQUFVLEVBQUUsT0FBTyxDQUFDLFVBQVU7NEJBQzlCLFNBQVMsRUFBRSxLQUFLO3lCQUNqQixDQUFDLENBQUM7b0JBQ0wsQ0FBQzt5QkFBTSxDQUFDO3dCQUNOLGlCQUFpQixDQUFDLElBQUksQ0FBQzs0QkFDckIsSUFBSSxFQUFFLE9BQU8sQ0FBQyxJQUFJOzRCQUNsQixVQUFVLEVBQUUsQ0FBQzs0QkFDYixTQUFTLEVBQUUsSUFBSTt5QkFDaEIsQ0FBQyxDQUFDO29CQUNMLENBQUM7Z0JBQ0gsQ0FBQztZQUNILENBQUM7aUJBQU0sSUFBSSxJQUFJLENBQUMsVUFBVSxLQUFLLGtCQUFrQixDQUFDLFlBQVksRUFBRSxDQUFDO2dCQUMvRCxxQ0FBcUM7Z0JBQ3JDLE1BQU0sS0FBSyxHQUFHLElBQUksQ0FBQyxXQUFXLEdBQUcsV0FBVyxDQUFDO2dCQUM3QyxLQUFLLE1BQU0sT0FBTyxJQUFJLGNBQWMsRUFBRSxDQUFDO29CQUNyQyxNQUFNLGVBQWUsR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxVQUFVLEdBQUcsS0FBSyxDQUFDLENBQUM7b0JBQy9ELE1BQU0sZ0JBQWdCLEdBQUcsSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLENBQUMsT0FBTyxFQUFFLGVBQWUsQ0FBQyxDQUFDO29CQUM3RSxnQkFBZ0IsQ0FBQyxJQUFJLENBQUM7d0JBQ3BCLEdBQUcsT0FBTzt3QkFDVixPQUFPLEVBQUUsZ0JBQWdCO3dCQUN6QixVQUFVLEVBQUUsZUFBZTtxQkFDNUIsQ0FBQyxDQUFDO29CQUNILGlCQUFpQixDQUFDLElBQUksQ0FBQzt3QkFDckIsSUFBSSxFQUFFLE9BQU8sQ0FBQyxJQUFJO3dCQUNsQixVQUFVLEVBQUUsZUFBZTt3QkFDM0IsU0FBUyxFQUFFLGVBQWUsR0FBRyxPQUFPLENBQUMsVUFBVTtxQkFDaEQsQ0FBQyxDQUFDO2dCQUNMLENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztRQUVELHNDQUFzQztRQUN0QyxJQUFJLElBQUksR0FBRyxFQUFFLENBQUM7UUFDZCxJQUFJLFlBQVksR0FBRyxDQUFDLENBQUM7UUFFckIsSUFBSSxJQUFJLENBQUMsTUFBTSxLQUFLLG1CQUFtQixDQUFDLElBQUksRUFBRSxDQUFDO1lBQzdDLElBQUksR0FBRyxJQUFJLENBQUMsZUFBZSxDQUFDLGdCQUFnQixDQUFDLENBQUM7UUFDaEQsQ0FBQzthQUFNLElBQUksSUFBSSxDQUFDLE1BQU0sS0FBSyxtQkFBbUIsQ0FBQyxJQUFJLEVBQUUsQ0FBQztZQUNwRCxJQUFJLEdBQUcsSUFBSSxDQUFDLGVBQWUsQ0FBQyxnQkFBZ0IsQ0FBQyxDQUFDO1FBQ2hELENBQUM7YUFBTSxDQUFDO1lBQ04sSUFBSSxHQUFHLElBQUksQ0FBQyxtQkFBbUIsQ0FBQyxnQkFBZ0IsQ0FBQyxDQUFDO1FBQ3BELENBQUM7UUFFRCxZQUFZLEdBQUcsSUFBSSxDQUFDLGNBQWMsQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUV6QyxPQUFPO1lBQ0wsSUFBSTtZQUNKLFVBQVUsRUFBRSxZQUFZO1lBQ3hCLFFBQVEsRUFBRSxpQkFBaUI7U0FDNUIsQ0FBQztJQUNKLENBQUM7SUFFRCxpQkFBaUI7SUFDVCxjQUFjLENBQUMsSUFBWTtRQUNqQyxzREFBc0Q7UUFDdEQsT0FBTyxJQUFJLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxNQUFNLEdBQUcsQ0FBQyxDQUFDLENBQUM7SUFDcEMsQ0FBQztJQUVPLFlBQVksQ0FBQyxJQUFZLEVBQUUsU0FBaUI7UUFDbEQsTUFBTSxRQUFRLEdBQUcsU0FBUyxHQUFHLENBQUMsQ0FBQztRQUMvQixJQUFJLElBQUksQ0FBQyxNQUFNLElBQUksUUFBUSxFQUFFLENBQUM7WUFDNUIsT0FBTyxJQUFJLENBQUM7UUFDZCxDQUFDO1FBQ0QsT0FBTyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUMsRUFBRSxRQUFRLENBQUMsR0FBRyxLQUFLLENBQUM7SUFDN0MsQ0FBQztJQUVPLGVBQWUsQ0FBQyxRQUFtQjtRQUN6QyxNQUFNLEtBQUssR0FBYSxFQUFFLENBQUM7UUFFM0IsS0FBSyxNQUFNLE9BQU8sSUFBSSxRQUFRLEVBQUUsQ0FBQztZQUMvQixLQUFLLENBQUMsSUFBSSxDQUFDLElBQUksT0FBTyxDQUFDLElBQUksR0FBRyxDQUFDLENBQUM7WUFDaEMsS0FBSyxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFDLENBQUM7WUFDNUIsS0FBSyxDQUFDLElBQUksQ0FBQyxFQUFFLENBQUMsQ0FBQztRQUNqQixDQUFDO1FBRUQsT0FBTyxLQUFLLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxDQUFDO0lBQzFCLENBQUM7SUFFTyxlQUFlLENBQUMsUUFBbUI7UUFDekMsTUFBTSxHQUFHLEdBQTJCLEVBQUUsQ0FBQztRQUV2QyxLQUFLLE1BQU0sT0FBTyxJQUFJLFFBQVEsRUFBRSxDQUFDO1lBQy9CLEdBQUcsQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLEdBQUcsT0FBTyxDQUFDLE9BQU8sQ0FBQztRQUN0QyxDQUFDO1FBRUQsT0FBTyxJQUFJLENBQUMsU0FBUyxDQUFDLEdBQUcsRUFBRSxJQUFJLEVBQUUsQ0FBQyxDQUFDLENBQUM7SUFDdEMsQ0FBQztJQUVPLG1CQUFtQixDQUFDLFFBQW1CO1FBQzdDLE1BQU0sS0FBSyxHQUFhLEVBQUUsQ0FBQztRQUUzQixLQUFLLE1BQU0sT0FBTyxJQUFJLFFBQVEsRUFBRSxDQUFDO1lBQy9CLEtBQUssQ0FBQyxJQUFJLENBQUMsTUFBTSxPQUFPLENBQUMsSUFBSSxFQUFFLENBQUMsQ0FBQztZQUNqQyxLQUFLLENBQUMsSUFBSSxDQUFDLEVBQUUsQ0FBQyxDQUFDO1lBQ2YsS0FBSyxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsT0FBTyxDQUFDLENBQUM7WUFDNUIsS0FBSyxDQUFDLElBQUksQ0FBQyxFQUFFLENBQUMsQ0FBQztRQUNqQixDQUFDO1FBRUQsT0FBTyxLQUFLLENBQUMsSUFBSSxDQUFDLElBQUksQ0FBQyxDQUFDO0lBQzFCLENBQUM7Q0FDRjtBQW5SRCxrREFtUkM7QUFFRDs7R0FFRztBQUNILFNBQWdCLG9CQUFvQjtJQUNsQyxPQUFPLElBQUksbUJBQW1CLEVBQUUsQ0FBQztBQUNuQyxDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLyoqXG4gKiBDb250ZXh0IFF1ZXJ5IEJ1aWxkZXIgZm9yIExMTSBDb250ZXh0IEFzc2VtYmx5XG4gKiBcbiAqIFRva2VuLWF3YXJlIGNvbnRleHQgYXNzZW1ibHkgd2l0aCBwcmlvcml0eS1iYXNlZCB0cnVuY2F0aW9uLlxuICovXG5cbmV4cG9ydCBlbnVtIENvbnRleHRPdXRwdXRGb3JtYXQge1xuICBUT09OID0gJ3Rvb24nLFxuICBKU09OID0gJ2pzb24nLFxuICBNQVJLRE9XTiA9ICdtYXJrZG93bicsXG59XG5cbmV4cG9ydCBlbnVtIFRydW5jYXRpb25TdHJhdGVneSB7XG4gIFRBSUxfRFJPUCA9ICd0YWlsX2Ryb3AnLCAgICAgICAvLyBEcm9wIGZyb20gZW5kXG4gIEhFQURfRFJPUCA9ICdoZWFkX2Ryb3AnLCAgICAgICAvLyBEcm9wIGZyb20gYmVnaW5uaW5nXG4gIFBST1BPUlRJT05BTCA9ICdwcm9wb3J0aW9uYWwnLCAvLyBQcm9wb3J0aW9uYWwgYWNyb3NzIHNlY3Rpb25zXG59XG5cbmludGVyZmFjZSBTZWN0aW9uIHtcbiAgbmFtZTogc3RyaW5nO1xuICBwcmlvcml0eTogbnVtYmVyO1xuICBjb250ZW50OiBzdHJpbmc7XG4gIHRva2VuQ291bnQ6IG51bWJlcjtcbn1cblxuZXhwb3J0IGludGVyZmFjZSBDb250ZXh0UmVzdWx0IHtcbiAgdGV4dDogc3RyaW5nO1xuICB0b2tlbkNvdW50OiBudW1iZXI7XG4gIHNlY3Rpb25zOiBBcnJheTx7IG5hbWU6IHN0cmluZzsgdG9rZW5Db3VudDogbnVtYmVyOyB0cnVuY2F0ZWQ6IGJvb2xlYW4gfT47XG59XG5cbi8qKlxuICogQ29udGV4dCBRdWVyeSBCdWlsZGVyIGZvciBhc3NlbWJsaW5nIExMTSBjb250ZXh0XG4gKi9cbmV4cG9ydCBjbGFzcyBDb250ZXh0UXVlcnlCdWlsZGVyIHtcbiAgcHJpdmF0ZSBzZXNzaW9uSWQ/OiBzdHJpbmc7XG4gIHByaXZhdGUgdG9rZW5CdWRnZXQ6IG51bWJlciA9IDQwOTY7XG4gIHByaXZhdGUgZm9ybWF0OiBDb250ZXh0T3V0cHV0Rm9ybWF0ID0gQ29udGV4dE91dHB1dEZvcm1hdC5UT09OO1xuICBwcml2YXRlIHRydW5jYXRpb246IFRydW5jYXRpb25TdHJhdGVneSA9IFRydW5jYXRpb25TdHJhdGVneS5UQUlMX0RST1A7XG4gIHByaXZhdGUgc2VjdGlvbnM6IFNlY3Rpb25bXSA9IFtdO1xuICBwcml2YXRlIGN1cnJlbnRTZWN0aW9uPzogU2VjdGlvbjtcblxuICAvKipcbiAgICogU2V0IHNlc3Npb24gSUQgZm9yIGNvbnRleHRcbiAgICovXG4gIGZvclNlc3Npb24oc2Vzc2lvbklkOiBzdHJpbmcpOiB0aGlzIHtcbiAgICB0aGlzLnNlc3Npb25JZCA9IHNlc3Npb25JZDtcbiAgICByZXR1cm4gdGhpcztcbiAgfVxuXG4gIC8qKlxuICAgKiBTZXQgdG9rZW4gYnVkZ2V0XG4gICAqL1xuICB3aXRoQnVkZ2V0KHRva2VuczogbnVtYmVyKTogdGhpcyB7XG4gICAgdGhpcy50b2tlbkJ1ZGdldCA9IHRva2VucztcbiAgICByZXR1cm4gdGhpcztcbiAgfVxuXG4gIC8qKlxuICAgKiBTZXQgb3V0cHV0IGZvcm1hdFxuICAgKi9cbiAgc2V0Rm9ybWF0KGZvcm1hdDogQ29udGV4dE91dHB1dEZvcm1hdCk6IHRoaXMge1xuICAgIHRoaXMuZm9ybWF0ID0gZm9ybWF0O1xuICAgIHJldHVybiB0aGlzO1xuICB9XG5cbiAgLyoqXG4gICAqIFNldCB0cnVuY2F0aW9uIHN0cmF0ZWd5XG4gICAqL1xuICBzZXRUcnVuY2F0aW9uKHN0cmF0ZWd5OiBUcnVuY2F0aW9uU3RyYXRlZ3kpOiB0aGlzIHtcbiAgICB0aGlzLnRydW5jYXRpb24gPSBzdHJhdGVneTtcbiAgICByZXR1cm4gdGhpcztcbiAgfVxuXG4gIC8qKlxuICAgKiBBZGQgbGl0ZXJhbCB0ZXh0IHNlY3Rpb25cbiAgICovXG4gIGxpdGVyYWwobmFtZTogc3RyaW5nLCBwcmlvcml0eTogbnVtYmVyLCB0ZXh0OiBzdHJpbmcpOiB0aGlzIHtcbiAgICBjb25zdCB0b2tlbkNvdW50ID0gdGhpcy5lc3RpbWF0ZVRva2Vucyh0ZXh0KTtcbiAgICB0aGlzLnNlY3Rpb25zLnB1c2goe1xuICAgICAgbmFtZSxcbiAgICAgIHByaW9yaXR5LFxuICAgICAgY29udGVudDogdGV4dCxcbiAgICAgIHRva2VuQ291bnQsXG4gICAgfSk7XG4gICAgcmV0dXJuIHRoaXM7XG4gIH1cblxuICAvKipcbiAgICogU3RhcnQgYSBuZXcgc2VjdGlvblxuICAgKi9cbiAgc2VjdGlvbihuYW1lOiBzdHJpbmcsIHByaW9yaXR5OiBudW1iZXIpOiB0aGlzIHtcbiAgICB0aGlzLmN1cnJlbnRTZWN0aW9uID0ge1xuICAgICAgbmFtZSxcbiAgICAgIHByaW9yaXR5LFxuICAgICAgY29udGVudDogJycsXG4gICAgICB0b2tlbkNvdW50OiAwLFxuICAgIH07XG4gICAgcmV0dXJuIHRoaXM7XG4gIH1cblxuICAvKipcbiAgICogQWRkIGNvbnRlbnQgdG8gY3VycmVudCBzZWN0aW9uXG4gICAqL1xuICBnZXQocGF0aDogc3RyaW5nKTogdGhpcyB7XG4gICAgaWYgKCF0aGlzLmN1cnJlbnRTZWN0aW9uKSB7XG4gICAgICB0aHJvdyBuZXcgRXJyb3IoJ05vIGFjdGl2ZSBzZWN0aW9uLiBDYWxsIHNlY3Rpb24oKSBmaXJzdC4nKTtcbiAgICB9XG4gICAgdGhpcy5jdXJyZW50U2VjdGlvbi5jb250ZW50ICs9IGBHRVQgJHtwYXRofVxcbmA7XG4gICAgcmV0dXJuIHRoaXM7XG4gIH1cblxuICAvKipcbiAgICogQWRkIGxhc3QgTiByZWNvcmRzIHF1ZXJ5XG4gICAqL1xuICBsYXN0KG46IG51bWJlciwgdGFibGU6IHN0cmluZyk6IHRoaXMge1xuICAgIGlmICghdGhpcy5jdXJyZW50U2VjdGlvbikge1xuICAgICAgdGhyb3cgbmV3IEVycm9yKCdObyBhY3RpdmUgc2VjdGlvbi4gQ2FsbCBzZWN0aW9uKCkgZmlyc3QuJyk7XG4gICAgfVxuICAgIHRoaXMuY3VycmVudFNlY3Rpb24uY29udGVudCArPSBgTEFTVCAke259IEZST00gJHt0YWJsZX1cXG5gO1xuICAgIHJldHVybiB0aGlzO1xuICB9XG5cbiAgLyoqXG4gICAqIEFkZCB3aGVyZSBlcXVhbHMgY29uZGl0aW9uXG4gICAqL1xuICB3aGVyZUVxKGZpZWxkOiBzdHJpbmcsIHZhbHVlOiBhbnkpOiB0aGlzIHtcbiAgICBpZiAoIXRoaXMuY3VycmVudFNlY3Rpb24pIHtcbiAgICAgIHRocm93IG5ldyBFcnJvcignTm8gYWN0aXZlIHNlY3Rpb24uIENhbGwgc2VjdGlvbigpIGZpcnN0LicpO1xuICAgIH1cbiAgICB0aGlzLmN1cnJlbnRTZWN0aW9uLmNvbnRlbnQgKz0gYFdIRVJFICR7ZmllbGR9ID0gJHtKU09OLnN0cmluZ2lmeSh2YWx1ZSl9XFxuYDtcbiAgICByZXR1cm4gdGhpcztcbiAgfVxuXG4gIC8qKlxuICAgKiBBZGQgdmVjdG9yIHNlYXJjaFxuICAgKi9cbiAgc2VhcmNoKGNvbGxlY3Rpb246IHN0cmluZywgZW1iZWRkaW5nOiBzdHJpbmcsIGs6IG51bWJlcik6IHRoaXMge1xuICAgIGlmICghdGhpcy5jdXJyZW50U2VjdGlvbikge1xuICAgICAgdGhyb3cgbmV3IEVycm9yKCdObyBhY3RpdmUgc2VjdGlvbi4gQ2FsbCBzZWN0aW9uKCkgZmlyc3QuJyk7XG4gICAgfVxuICAgIHRoaXMuY3VycmVudFNlY3Rpb24uY29udGVudCArPSBgU0VBUkNIICR7Y29sbGVjdGlvbn0gV0lUSCAke2VtYmVkZGluZ30gTElNSVQgJHtrfVxcbmA7XG4gICAgcmV0dXJuIHRoaXM7XG4gIH1cblxuICAvKipcbiAgICogQWRkIFNRTCBxdWVyeVxuICAgKi9cbiAgc3FsKHF1ZXJ5OiBzdHJpbmcpOiB0aGlzIHtcbiAgICBpZiAoIXRoaXMuY3VycmVudFNlY3Rpb24pIHtcbiAgICAgIHRocm93IG5ldyBFcnJvcignTm8gYWN0aXZlIHNlY3Rpb24uIENhbGwgc2VjdGlvbigpIGZpcnN0LicpO1xuICAgIH1cbiAgICB0aGlzLmN1cnJlbnRTZWN0aW9uLmNvbnRlbnQgKz0gYFNRTDogJHtxdWVyeX1cXG5gO1xuICAgIHJldHVybiB0aGlzO1xuICB9XG5cbiAgLyoqXG4gICAqIEZpbmlzaCBjdXJyZW50IHNlY3Rpb25cbiAgICovXG4gIGRvbmUoKTogdGhpcyB7XG4gICAgaWYgKHRoaXMuY3VycmVudFNlY3Rpb24pIHtcbiAgICAgIHRoaXMuY3VycmVudFNlY3Rpb24udG9rZW5Db3VudCA9IHRoaXMuZXN0aW1hdGVUb2tlbnModGhpcy5jdXJyZW50U2VjdGlvbi5jb250ZW50KTtcbiAgICAgIHRoaXMuc2VjdGlvbnMucHVzaCh0aGlzLmN1cnJlbnRTZWN0aW9uKTtcbiAgICAgIHRoaXMuY3VycmVudFNlY3Rpb24gPSB1bmRlZmluZWQ7XG4gICAgfVxuICAgIHJldHVybiB0aGlzO1xuICB9XG5cbiAgLyoqXG4gICAqIEV4ZWN1dGUgYW5kIGJ1aWxkIGNvbnRleHRcbiAgICovXG4gIGV4ZWN1dGUoKTogQ29udGV4dFJlc3VsdCB7XG4gICAgLy8gRmluaXNoIGFueSBwZW5kaW5nIHNlY3Rpb25cbiAgICBpZiAodGhpcy5jdXJyZW50U2VjdGlvbikge1xuICAgICAgdGhpcy5kb25lKCk7XG4gICAgfVxuXG4gICAgLy8gU29ydCBzZWN0aW9ucyBieSBwcmlvcml0eSAobG93ZXIgPSBoaWdoZXIgcHJpb3JpdHkpXG4gICAgY29uc3Qgc29ydGVkU2VjdGlvbnMgPSBbLi4udGhpcy5zZWN0aW9uc10uc29ydCgoYSwgYikgPT4gYS5wcmlvcml0eSAtIGIucHJpb3JpdHkpO1xuXG4gICAgLy8gQ2FsY3VsYXRlIHRvdGFsIHRva2Vuc1xuICAgIGxldCB0b3RhbFRva2VucyA9IHNvcnRlZFNlY3Rpb25zLnJlZHVjZSgoc3VtLCBzKSA9PiBzdW0gKyBzLnRva2VuQ291bnQsIDApO1xuXG4gICAgLy8gVHJ1bmNhdGUgaWYgbmVlZGVkXG4gICAgY29uc3QgdHJ1bmNhdGVkU2VjdGlvbnM6IEFycmF5PHsgbmFtZTogc3RyaW5nOyB0b2tlbkNvdW50OiBudW1iZXI7IHRydW5jYXRlZDogYm9vbGVhbiB9PiA9IFtdO1xuICAgIGNvbnN0IGluY2x1ZGVkU2VjdGlvbnM6IFNlY3Rpb25bXSA9IFtdO1xuXG4gICAgaWYgKHRvdGFsVG9rZW5zIDw9IHRoaXMudG9rZW5CdWRnZXQpIHtcbiAgICAgIC8vIE5vIHRydW5jYXRpb24gbmVlZGVkXG4gICAgICBmb3IgKGNvbnN0IHNlY3Rpb24gb2Ygc29ydGVkU2VjdGlvbnMpIHtcbiAgICAgICAgaW5jbHVkZWRTZWN0aW9ucy5wdXNoKHNlY3Rpb24pO1xuICAgICAgICB0cnVuY2F0ZWRTZWN0aW9ucy5wdXNoKHtcbiAgICAgICAgICBuYW1lOiBzZWN0aW9uLm5hbWUsXG4gICAgICAgICAgdG9rZW5Db3VudDogc2VjdGlvbi50b2tlbkNvdW50LFxuICAgICAgICAgIHRydW5jYXRlZDogZmFsc2UsXG4gICAgICAgIH0pO1xuICAgICAgfVxuICAgIH0gZWxzZSB7XG4gICAgICAvLyBBcHBseSB0cnVuY2F0aW9uIHN0cmF0ZWd5XG4gICAgICBsZXQgcmVtYWluaW5nQnVkZ2V0ID0gdGhpcy50b2tlbkJ1ZGdldDtcblxuICAgICAgaWYgKHRoaXMudHJ1bmNhdGlvbiA9PT0gVHJ1bmNhdGlvblN0cmF0ZWd5LlRBSUxfRFJPUCkge1xuICAgICAgICAvLyBJbmNsdWRlIHNlY3Rpb25zIGluIHByaW9yaXR5IG9yZGVyIHVudGlsIGJ1ZGdldCBleGhhdXN0ZWRcbiAgICAgICAgZm9yIChjb25zdCBzZWN0aW9uIG9mIHNvcnRlZFNlY3Rpb25zKSB7XG4gICAgICAgICAgaWYgKHNlY3Rpb24udG9rZW5Db3VudCA8PSByZW1haW5pbmdCdWRnZXQpIHtcbiAgICAgICAgICAgIGluY2x1ZGVkU2VjdGlvbnMucHVzaChzZWN0aW9uKTtcbiAgICAgICAgICAgIHJlbWFpbmluZ0J1ZGdldCAtPSBzZWN0aW9uLnRva2VuQ291bnQ7XG4gICAgICAgICAgICB0cnVuY2F0ZWRTZWN0aW9ucy5wdXNoKHtcbiAgICAgICAgICAgICAgbmFtZTogc2VjdGlvbi5uYW1lLFxuICAgICAgICAgICAgICB0b2tlbkNvdW50OiBzZWN0aW9uLnRva2VuQ291bnQsXG4gICAgICAgICAgICAgIHRydW5jYXRlZDogZmFsc2UsXG4gICAgICAgICAgICB9KTtcbiAgICAgICAgICB9IGVsc2Uge1xuICAgICAgICAgICAgdHJ1bmNhdGVkU2VjdGlvbnMucHVzaCh7XG4gICAgICAgICAgICAgIG5hbWU6IHNlY3Rpb24ubmFtZSxcbiAgICAgICAgICAgICAgdG9rZW5Db3VudDogMCxcbiAgICAgICAgICAgICAgdHJ1bmNhdGVkOiB0cnVlLFxuICAgICAgICAgICAgfSk7XG4gICAgICAgICAgfVxuICAgICAgICB9XG4gICAgICB9IGVsc2UgaWYgKHRoaXMudHJ1bmNhdGlvbiA9PT0gVHJ1bmNhdGlvblN0cmF0ZWd5LlBST1BPUlRJT05BTCkge1xuICAgICAgICAvLyBQcm9wb3J0aW9uYWxseSByZWR1Y2UgYWxsIHNlY3Rpb25zXG4gICAgICAgIGNvbnN0IHJhdGlvID0gdGhpcy50b2tlbkJ1ZGdldCAvIHRvdGFsVG9rZW5zO1xuICAgICAgICBmb3IgKGNvbnN0IHNlY3Rpb24gb2Ygc29ydGVkU2VjdGlvbnMpIHtcbiAgICAgICAgICBjb25zdCBhbGxvY2F0ZWRUb2tlbnMgPSBNYXRoLmZsb29yKHNlY3Rpb24udG9rZW5Db3VudCAqIHJhdGlvKTtcbiAgICAgICAgICBjb25zdCB0cnVuY2F0ZWRDb250ZW50ID0gdGhpcy50cnVuY2F0ZVRleHQoc2VjdGlvbi5jb250ZW50LCBhbGxvY2F0ZWRUb2tlbnMpO1xuICAgICAgICAgIGluY2x1ZGVkU2VjdGlvbnMucHVzaCh7XG4gICAgICAgICAgICAuLi5zZWN0aW9uLFxuICAgICAgICAgICAgY29udGVudDogdHJ1bmNhdGVkQ29udGVudCxcbiAgICAgICAgICAgIHRva2VuQ291bnQ6IGFsbG9jYXRlZFRva2VucyxcbiAgICAgICAgICB9KTtcbiAgICAgICAgICB0cnVuY2F0ZWRTZWN0aW9ucy5wdXNoKHtcbiAgICAgICAgICAgIG5hbWU6IHNlY3Rpb24ubmFtZSxcbiAgICAgICAgICAgIHRva2VuQ291bnQ6IGFsbG9jYXRlZFRva2VucyxcbiAgICAgICAgICAgIHRydW5jYXRlZDogYWxsb2NhdGVkVG9rZW5zIDwgc2VjdGlvbi50b2tlbkNvdW50LFxuICAgICAgICAgIH0pO1xuICAgICAgICB9XG4gICAgICB9XG4gICAgfVxuXG4gICAgLy8gQnVpbGQgZmluYWwgY29udGV4dCBiYXNlZCBvbiBmb3JtYXRcbiAgICBsZXQgdGV4dCA9ICcnO1xuICAgIGxldCBhY3R1YWxUb2tlbnMgPSAwO1xuXG4gICAgaWYgKHRoaXMuZm9ybWF0ID09PSBDb250ZXh0T3V0cHV0Rm9ybWF0LlRPT04pIHtcbiAgICAgIHRleHQgPSB0aGlzLmJ1aWxkVG9vbkZvcm1hdChpbmNsdWRlZFNlY3Rpb25zKTtcbiAgICB9IGVsc2UgaWYgKHRoaXMuZm9ybWF0ID09PSBDb250ZXh0T3V0cHV0Rm9ybWF0LkpTT04pIHtcbiAgICAgIHRleHQgPSB0aGlzLmJ1aWxkSnNvbkZvcm1hdChpbmNsdWRlZFNlY3Rpb25zKTtcbiAgICB9IGVsc2Uge1xuICAgICAgdGV4dCA9IHRoaXMuYnVpbGRNYXJrZG93bkZvcm1hdChpbmNsdWRlZFNlY3Rpb25zKTtcbiAgICB9XG5cbiAgICBhY3R1YWxUb2tlbnMgPSB0aGlzLmVzdGltYXRlVG9rZW5zKHRleHQpO1xuXG4gICAgcmV0dXJuIHtcbiAgICAgIHRleHQsXG4gICAgICB0b2tlbkNvdW50OiBhY3R1YWxUb2tlbnMsXG4gICAgICBzZWN0aW9uczogdHJ1bmNhdGVkU2VjdGlvbnMsXG4gICAgfTtcbiAgfVxuXG4gIC8vIEhlbHBlciBtZXRob2RzXG4gIHByaXZhdGUgZXN0aW1hdGVUb2tlbnModGV4dDogc3RyaW5nKTogbnVtYmVyIHtcbiAgICAvLyBSb3VnaCBlc3RpbWF0ZTogfjQgY2hhcmFjdGVycyBwZXIgdG9rZW4gZm9yIEVuZ2xpc2hcbiAgICByZXR1cm4gTWF0aC5jZWlsKHRleHQubGVuZ3RoIC8gNCk7XG4gIH1cblxuICBwcml2YXRlIHRydW5jYXRlVGV4dCh0ZXh0OiBzdHJpbmcsIG1heFRva2VuczogbnVtYmVyKTogc3RyaW5nIHtcbiAgICBjb25zdCBtYXhDaGFycyA9IG1heFRva2VucyAqIDQ7XG4gICAgaWYgKHRleHQubGVuZ3RoIDw9IG1heENoYXJzKSB7XG4gICAgICByZXR1cm4gdGV4dDtcbiAgICB9XG4gICAgcmV0dXJuIHRleHQuc3Vic3RyaW5nKDAsIG1heENoYXJzKSArICcuLi4nO1xuICB9XG5cbiAgcHJpdmF0ZSBidWlsZFRvb25Gb3JtYXQoc2VjdGlvbnM6IFNlY3Rpb25bXSk6IHN0cmluZyB7XG4gICAgY29uc3QgbGluZXM6IHN0cmluZ1tdID0gW107XG4gICAgXG4gICAgZm9yIChjb25zdCBzZWN0aW9uIG9mIHNlY3Rpb25zKSB7XG4gICAgICBsaW5lcy5wdXNoKGBbJHtzZWN0aW9uLm5hbWV9XWApO1xuICAgICAgbGluZXMucHVzaChzZWN0aW9uLmNvbnRlbnQpO1xuICAgICAgbGluZXMucHVzaCgnJyk7XG4gICAgfVxuICAgIFxuICAgIHJldHVybiBsaW5lcy5qb2luKCdcXG4nKTtcbiAgfVxuXG4gIHByaXZhdGUgYnVpbGRKc29uRm9ybWF0KHNlY3Rpb25zOiBTZWN0aW9uW10pOiBzdHJpbmcge1xuICAgIGNvbnN0IG9iajogUmVjb3JkPHN0cmluZywgc3RyaW5nPiA9IHt9O1xuICAgIFxuICAgIGZvciAoY29uc3Qgc2VjdGlvbiBvZiBzZWN0aW9ucykge1xuICAgICAgb2JqW3NlY3Rpb24ubmFtZV0gPSBzZWN0aW9uLmNvbnRlbnQ7XG4gICAgfVxuICAgIFxuICAgIHJldHVybiBKU09OLnN0cmluZ2lmeShvYmosIG51bGwsIDIpO1xuICB9XG5cbiAgcHJpdmF0ZSBidWlsZE1hcmtkb3duRm9ybWF0KHNlY3Rpb25zOiBTZWN0aW9uW10pOiBzdHJpbmcge1xuICAgIGNvbnN0IGxpbmVzOiBzdHJpbmdbXSA9IFtdO1xuICAgIFxuICAgIGZvciAoY29uc3Qgc2VjdGlvbiBvZiBzZWN0aW9ucykge1xuICAgICAgbGluZXMucHVzaChgIyMgJHtzZWN0aW9uLm5hbWV9YCk7XG4gICAgICBsaW5lcy5wdXNoKCcnKTtcbiAgICAgIGxpbmVzLnB1c2goc2VjdGlvbi5jb250ZW50KTtcbiAgICAgIGxpbmVzLnB1c2goJycpO1xuICAgIH1cbiAgICBcbiAgICByZXR1cm4gbGluZXMuam9pbignXFxuJyk7XG4gIH1cbn1cblxuLyoqXG4gKiBDcmVhdGUgYSBjb250ZXh0IHF1ZXJ5IGJ1aWxkZXJcbiAqL1xuZXhwb3J0IGZ1bmN0aW9uIGNyZWF0ZUNvbnRleHRCdWlsZGVyKCk6IENvbnRleHRRdWVyeUJ1aWxkZXIge1xuICByZXR1cm4gbmV3IENvbnRleHRRdWVyeUJ1aWxkZXIoKTtcbn1cbiJdfQ==