@nxuss/lemma 0.1.1 → 0.2.0

package/dist/esm/core/SubconsciousEngine.js

@@ -0,0 +1,278 @@
+import { ChromaClient } from 'chromadb';
+import axios from 'axios';
+import { config, getChromaURL } from '../config';
+import { createLogger } from '../utils/logger';
+const logger = createLogger('SubconsciousEngine');
+/**
+ * Subconscious Semantic Engine
+ * Provides semantic caching using ChromaDB and Ollama embeddings
+ * Implements the "short-circuit" logic for instant task resolution
+ */
+export class SubconsciousEngine {
+    constructor() {
+        this.collection = null;
+        this.isInitialized = false;
+        this.cacheHits = 0;
+        this.cacheMisses = 0;
+        const chromaURL = getChromaURL();
+        // Parse host and port from URL (e.g. "http://localhost:8000")
+        let host = 'localhost';
+        let port = 8000;
+        let ssl = false;
+        try {
+            const parsed = new URL(chromaURL);
+            host = parsed.hostname;
+            port = parseInt(parsed.port || (parsed.protocol === 'https:' ? '443' : '8000'), 10);
+            ssl = parsed.protocol === 'https:';
+        }
+        catch {
+            // fallback to defaults
+        }
+        this.client = new ChromaClient({ host, port, ssl });
+    }
+    /**
+     * Initialize the engine and create/get collection
+     */
+    async initialize() {
+        try {
+            logger.info('Initializing Subconscious Engine...');
+            // Create or get collection
+            this.collection = await this.client.getOrCreateCollection({
+                name: config.CHROMA_COLLECTION,
+                metadata: {
+                    description: 'Semantic cache for agent task responses',
+                    'hnsw:space': 'cosine',
+                },
+            });
+            this.isInitialized = true;
+            logger.info('Subconscious Engine initialized successfully', {
+                collection: config.CHROMA_COLLECTION,
+            });
+        }
+        catch (error) {
+            logger.error('Failed to initialize Subconscious Engine', error);
+            throw error;
+        }
+    }
+    /**
+     * Generate embedding using Ollama
+     */
+    async generateEmbedding(text) {
+        try {
+            const startTime = Date.now();
+            const response = await axios.post(`${config.OLLAMA_HOST}/api/embeddings`, {
+                model: config.OLLAMA_MODEL,
+                prompt: text,
+            }, {
+                timeout: 30000, // 30 second timeout
+            });
+            const embedding = response.data.embedding;
+            const duration = Date.now() - startTime;
+            logger.debug('Generated embedding', {
+                textLength: text.length,
+                embeddingDim: embedding.length,
+                duration,
+            });
+            return embedding;
+        }
+        catch (error) {
+            logger.error('Failed to generate embedding', error);
+            throw new Error('Embedding generation failed');
+        }
+    }
+    /**
+     * Search for semantically similar cached responses
+     * Returns the best match if similarity exceeds threshold
+     */
+    async search(query) {
+        if (!this.isInitialized || !this.collection) {
+            throw new Error('SubconsciousEngine not initialized');
+        }
+        if (!config.ENABLE_CACHING) {
+            this.cacheMisses++;
+            return null;
+        }
+        try {
+            const startTime = Date.now();
+            // Generate embedding for query
+            const queryEmbedding = await this.generateEmbedding(query);
+            // Search in ChromaDB
+            const results = await this.collection.query({
+                queryEmbeddings: [queryEmbedding],
+                nResults: 1,
+            });
+            const duration = Date.now() - startTime;
+            // Check if we have results
+            if (!results.ids[0] || results.ids[0].length === 0) {
+                this.cacheMisses++;
+                logger.debug('Cache miss - no results', { query, duration });
+                return null;
+            }
+            // Get similarity score (ChromaDB returns distances, convert to similarity)
+            const distance = results.distances?.[0]?.[0] ?? 1;
+            const similarity = 1 - distance; // Cosine distance to similarity
+            logger.debug('Semantic search result', {
+                similarity,
+                threshold: config.SEMANTIC_THRESHOLD,
+                duration,
+            });
+            // Check if similarity meets threshold
+            if (similarity < config.SEMANTIC_THRESHOLD) {
+                this.cacheMisses++;
+                logger.info('Cache miss - below threshold', {
+                    similarity,
+                    threshold: config.SEMANTIC_THRESHOLD,
+                    duration,
+                });
+                return null;
+            }
+            // Cache hit!
+            this.cacheHits++;
+            const id = results.ids[0][0];
+            const metadata = results.metadatas?.[0]?.[0];
+            const response = metadata?.response ? JSON.parse(metadata.response) : null;
+            logger.info('Cache HIT! 🎯', {
+                similarity,
+                duration,
+                savedTime: metadata?.executionTime || 0,
+            });
+            logger.metric('cache-hit-latency', duration);
+            return {
+                id,
+                similarity,
+                response,
+                metadata: {
+                    agentId: metadata?.agentId || 'unknown',
+                    timestamp: metadata?.timestamp || Date.now(),
+                    executionTime: metadata?.executionTime || 0,
+                    tokensUsed: metadata?.tokensUsed,
+                },
+            };
+        }
+        catch (error) {
+            logger.error('Search failed', error);
+            this.cacheMisses++;
+            return null;
+        }
+    }
+    /**
+     * Store a new response in the semantic cache
+     */
+    async store(entry) {
+        if (!this.isInitialized || !this.collection) {
+            throw new Error('SubconsciousEngine not initialized');
+        }
+        if (!config.ENABLE_CACHING) {
+            return;
+        }
+        try {
+            const startTime = Date.now();
+            // Generate embedding for the query
+            const embedding = await this.generateEmbedding(entry.query);
+            // Store in ChromaDB
+            await this.collection.add({
+                ids: [entry.id],
+                embeddings: [embedding],
+                metadatas: [{
+                        agentId: entry.metadata.agentId,
+                        timestamp: entry.metadata.timestamp,
+                        executionTime: entry.metadata.executionTime || 0,
+                        tokensUsed: entry.metadata.tokensUsed || 0,
+                        response: JSON.stringify(entry.response),
+                    }],
+                documents: [entry.query],
+            });
+            const duration = Date.now() - startTime;
+            logger.debug('Stored in cache', {
+                id: entry.id,
+                duration,
+            });
+            // Check cache size and prune if necessary
+            await this.pruneIfNeeded();
+        }
+        catch (error) {
+            logger.error('Failed to store in cache', error);
+            // Don't throw - caching failures shouldn't break the system
+        }
+    }
+    /**
+     * Prune old entries if cache exceeds max size
+     */
+    async pruneIfNeeded() {
+        if (!this.collection)
+            return;
+        try {
+            const count = await this.collection.count();
+            if (count > config.MAX_CACHE_SIZE) {
+                const toDelete = count - config.MAX_CACHE_SIZE;
+                logger.info('Pruning cache', {
+                    currentSize: count,
+                    maxSize: config.MAX_CACHE_SIZE,
+                    toDelete,
+                });
+                // Get oldest entries (this is a simplified approach)
+                // In production, you'd want more sophisticated LRU logic
+                const results = await this.collection.get({
+                    limit: toDelete,
+                });
+                if (results.ids.length > 0) {
+                    await this.collection.delete({
+                        ids: results.ids,
+                    });
+                    logger.info('Cache pruned', { deleted: results.ids.length });
+                }
+            }
+        }
+        catch (error) {
+            logger.error('Failed to prune cache', error);
+        }
+    }
+    /**
+     * Get cache statistics
+     */
+    getStats() {
+        const total = this.cacheHits + this.cacheMisses;
+        const hitRate = total > 0 ? (this.cacheHits / total) * 100 : 0;
+        return {
+            cacheHits: this.cacheHits,
+            cacheMisses: this.cacheMisses,
+            totalQueries: total,
+            hitRate: hitRate.toFixed(2) + '%',
+        };
+    }
+    /**
+     * Clear all cache entries
+     */
+    async clear() {
+        if (!this.collection)
+            return;
+        try {
+            await this.client.deleteCollection({
+                name: config.CHROMA_COLLECTION,
+            });
+            this.collection = await this.client.createCollection({
+                name: config.CHROMA_COLLECTION,
+                metadata: {
+                    description: 'Semantic cache for agent task responses',
+                    'hnsw:space': 'cosine',
+                },
+            });
+            this.cacheHits = 0;
+            this.cacheMisses = 0;
+            logger.info('Cache cleared');
+        }
+        catch (error) {
+            logger.error('Failed to clear cache', error);
+            throw error;
+        }
+    }
+    /**
+     * Shutdown the engine
+     */
+    async shutdown() {
+        logger.info('Shutting down Subconscious Engine');
+        this.isInitialized = false;
+        this.collection = null;
+    }
+}
+//# sourceMappingURL=SubconsciousEngine.js.map

package/dist/esm/core/SubconsciousEngine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SubconsciousEngine.js","sourceRoot":"","sources":["../../../src/core/SubconsciousEngine.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAc,MAAM,UAAU,CAAC;AACpD,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AAEjD,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,MAAM,GAAG,YAAY,CAAC,oBAAoB,CAAC,CAAC;AAElD;;;;GAIG;AACH,MAAM,OAAO,kBAAkB;IAO7B;QALQ,eAAU,GAAsB,IAAI,CAAC;QACrC,kBAAa,GAAY,KAAK,CAAC;QAC/B,cAAS,GAAW,CAAC,CAAC;QACtB,gBAAW,GAAW,CAAC,CAAC;QAG9B,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;QACjC,8DAA8D;QAC9D,IAAI,IAAI,GAAG,WAAW,CAAC;QACvB,IAAI,IAAI,GAAG,IAAI,CAAC;QAChB,IAAI,GAAG,GAAG,KAAK,CAAC;QAChB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC;YAClC,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC;YACvB,IAAI,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;YACpF,GAAG,GAAG,MAAM,CAAC,QAAQ,KAAK,QAAQ,CAAC;QACrC,CAAC;QAAC,MAAM,CAAC;YACP,uBAAuB;QACzB,CAAC;QACD,IAAI,CAAC,MAAM,GAAG,IAAI,YAAY,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,EAAS,CAAC,CAAC;IAC7D,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,UAAU;QACrB,IAAI,CAAC;YACH,MAAM,CAAC,IAAI,CAAC,qCAAqC,CAAC,CAAC;YAEnD,2BAA2B;YAC3B,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC;gBACxD,IAAI,EAAE,MAAM,CAAC,iBAAiB;gBAC9B,QAAQ,EAAE;oBACR,WAAW,EAAE,yCAAyC;oBACtD,YAAY,EAAE,QAAQ;iBACvB;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,8CAA8C,EAAE;gBAC1D,UAAU,EAAE,MAAM,CAAC,iBAAiB;aACrC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;YAChE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAC1C,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,GAAG,MAAM,CAAC,WAAW,iBAAiB,EACtC;gBACE,KAAK,EAAE,MAAM,CAAC,YAAY;gBAC1B,MAAM,EAAE,IAAI;aACb,EACD;gBACE,OAAO,EAAE,KAAK,EAAE,oBAAoB;aACrC,CACF,CAAC;YAEF,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC;YAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE;gBAClC,UAAU,EAAE,IAAI,CAAC,MAAM;gBACvB,YAAY,EAAE,SAAS,CAAC,MAAM;gBAC9B,QAAQ;aACT,CAAC,CAAC;YAEH,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACpD,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,MAAM,CAAC,KAAa;QAC/B,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC3B,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,+BAA+B;YAC/B,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAE3D,qBAAqB;YACrB,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC1C,eAAe,EAAE,CAAC,cAAc,CAAC;gBACjC,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,2BAA2B;YAC3B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACnD,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;gBAC7D,OAAO,IAAI,CAAC;YACd,CAAC;YAED,2EAA2E;YAC3E,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAClD,MAAM,UAAU,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC,gCAAgC;YAEjE,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE;gBACrC,UAAU;gBACV,SAAS,EAAE,MAAM,CAAC,kBAAkB;gBACpC,QAAQ;aACT,CAAC,CAAC;YAEH,sCAAsC;YACtC,IAAI,UAAU,GAAG,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC3C,IAAI,CAAC,WAAW,EAAE,CAAC;gBACnB,MAAM,CAAC,IAAI,CAAC,8BAA8B,EAAE;oBAC1C,UAAU;oBACV,SAAS,EAAE,MAAM,CAAC,kBAAkB;oBACpC,QAAQ;iBACT,CAAC,CAAC;gBACH,OAAO,IAAI,CAAC;YACd,CAAC;YAED,aAAa;YACb,IAAI,CAAC,SAAS,EAAE,CAAC;YAEjB,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAQ,CAAC;YACpD,MAAM,QAAQ,GAAG,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAE3E,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE;gBAC3B,UAAU;gBACV,QAAQ;gBACR,SAAS,EAAE,QAAQ,EAAE,aAAa,IAAI,CAAC;aACxC,CAAC,CAAC;YAEH,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,QAAQ,CAAC,CAAC;YAE7C,OAAO;gBACL,EAAE;gBACF,UAAU;gBACV,QAAQ;gBACR,QAAQ,EAAE;oBACR,OAAO,EAAE,QAAQ,EAAE,OAAO,IAAI,SAAS;oBACvC,SAAS,EAAE,QAAQ,EAAE,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE;oBAC5C,aAAa,EAAE,QAAQ,EAAE,aAAa,IAAI,CAAC;oBAC3C,UAAU,EAAE,QAAQ,EAAE,UAAU;iBACjC;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;YACrC,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,KAAK,CAAC,KAAiB;QAClC,IAAI,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;YAC5C,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACxD,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,mCAAmC;YACnC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAE5D,oBAAoB;YACpB,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBACxB,GAAG,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;gBACf,UAAU,EAAE,CAAC,SAAS,CAAC;gBACvB,SAAS,EAAE,CAAC;wBACV,OAAO,EAAE,KAAK,CAAC,QAAQ,CAAC,OAAO;wBAC/B,SAAS,EAAE,KAAK,CAAC,QAAQ,CAAC,SAAS;wBACnC,aAAa,EAAE,KAAK,CAAC,QAAQ,CAAC,aAAa,IAAI,CAAC;wBAChD,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,UAAU,IAAI,CAAC;wBAC1C,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;qBACzC,CAAC;gBACF,SAAS,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC;aACzB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAExC,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE;gBAC9B,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,QAAQ;aACT,CAAC,CAAC;YAEH,0CAA0C;YAC1C,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAK,CAAC,CAAC;YAChD,4DAA4D;QAC9D,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO;QAE7B,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YAE5C,IAAI,KAAK,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;gBAClC,MAAM,QAAQ,GAAG,KAAK,GAAG,MAAM,CAAC,cAAc,CAAC;gBAC/C,MAAM,CAAC,IAAI,CAAC,eAAe,EAAE;oBAC3B,WAAW,EAAE,KAAK;oBAClB,OAAO,EAAE,MAAM,CAAC,cAAc;oBAC9B,QAAQ;iBACT,CAAC,CAAC;gBAEH,qDAAqD;gBACrD,yDAAyD;gBACzD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;oBACxC,KAAK,EAAE,QAAQ;iBAChB,CAAC,CAAC;gBAEH,IAAI,OAAO,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3B,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;wBAC3B,GAAG,EAAE,OAAO,CAAC,GAAG;qBACjB,CAAC,CAAC;oBAEH,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;IAED;;OAEG;IACI,QAAQ;QACb,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,WAAW,CAAC;QAChD,MAAM,OAAO,GAAG,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAE/D,OAAO;YACL,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,YAAY,EAAE,KAAK;YACnB,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG;SAClC,CAAC;IACJ,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,KAAK;QAChB,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO;QAE7B,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;gBACjC,IAAI,EAAE,MAAM,CAAC,iBAAiB;aAC/B,CAAC,CAAC;YAEH,IAAI,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;gBACnD,IAAI,EAAE,MAAM,CAAC,iBAAiB;gBAC9B,QAAQ,EAAE;oBACR,WAAW,EAAE,yCAAyC;oBACtD,YAAY,EAAE,QAAQ;iBACvB;aACF,CAAC,CAAC;YAEH,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;YACnB,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC;YAErB,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;YAC7C,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,QAAQ;QACnB,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;QACjD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC;QAC3B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;IACzB,CAAC;CACF"}

package/dist/esm/core/WebSocketServer.d.ts

@@ -0,0 +1,89 @@
+import { EventEmitter } from 'events';
+import { AgentConnection, ServerConfig, ConnectionStats } from '../types';
+import { SecurityMiddleware, SecurityConfig } from '../security';
+/**
+ * High-performance WebSocket server for agent connections
+ * Handles connection lifecycle, heartbeat, reconnection logic, and security
+ */
+export declare class WebSocketServer extends EventEmitter {
+    private wss;
+    private connections;
+    private heartbeatInterval;
+    private config;
+    private security;
+    private stats;
+    constructor(config: ServerConfig, securityConfig?: SecurityConfig);
+    /**
+     * Start the WebSocket server
+     */
+    start(): Promise<void>;
+    /**
+     * Handle new WebSocket connection
+     */
+    private handleConnection;
+    /**
+     * Handle incoming message from agent
+     */
+    private handleMessage;
+    /**
+     * Handle pong response (heartbeat)
+     */
+    private handlePong;
+    /**
+     * Handle connection close
+     */
+    private handleDisconnection;
+    /**
+     * Handle WebSocket error
+     */
+    private handleError;
+    /**
+     * Start heartbeat mechanism (ping/pong)
+     */
+    private startHeartbeat;
+    /**
+     * Send message to specific connection
+     */
+    sendToConnection(connectionId: string, message: any): boolean;
+    /**
+     * Send message to specific agent
+     */
+    sendToAgent(agentId: string, message: any): boolean;
+    /**
+     * Broadcast message to all connected agents
+     */
+    broadcast(message: any, excludeConnectionId?: string): void;
+    /**
+     * Send error message to connection
+     */
+    private sendError;
+    /**
+     * Update agent information for a connection
+     */
+    updateAgentInfo(connectionId: string, agentId: string, capabilities: string[]): boolean;
+    /**
+     * Get connection by ID
+     */
+    getConnection(connectionId: string): AgentConnection | undefined;
+    /**
+     * Get all active connections
+     */
+    getConnections(): AgentConnection[];
+    /**
+     * Get connection statistics
+     */
+    getStats(): ConnectionStats;
+    /**
+     * Generate unique connection ID
+     */
+    private generateConnectionId;
+    /**
+     * Stop the WebSocket server
+     */
+    stop(): Promise<void>;
+    /**
+     * Get security middleware (for key management)
+     */
+    getSecurity(): SecurityMiddleware | null;
+}
+//# sourceMappingURL=WebSocketServer.d.ts.map

package/dist/esm/core/WebSocketServer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"WebSocketServer.d.ts","sourceRoot":"","sources":["../../../src/core/WebSocketServer.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AACtC,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,eAAe,EAAE,MAAM,UAAU,CAAC;AAC1E,OAAO,EAAE,kBAAkB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AAEjE;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,YAAY;IAC/C,OAAO,CAAC,GAAG,CAAyB;IACpC,OAAO,CAAC,WAAW,CAA2C;IAC9D,OAAO,CAAC,iBAAiB,CAA+B;IACxD,OAAO,CAAC,MAAM,CAAe;IAC7B,OAAO,CAAC,QAAQ,CAAmC;IACnD,OAAO,CAAC,KAAK,CAMX;gBAEU,MAAM,EAAE,YAAY,EAAE,cAAc,CAAC,EAAE,cAAc;IAqBjE;;OAEG;IACI,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IA8B7B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA+ExB;;OAEG;IACH,OAAO,CAAC,aAAa;IA6ErB;;OAEG;IACH,OAAO,CAAC,UAAU;IAQlB;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAiB3B;;OAEG;IACH,OAAO,CAAC,WAAW;IAKnB;;OAEG;IACH,OAAO,CAAC,cAAc;IAyBtB;;OAEG;IACI,gBAAgB,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,OAAO;IAgBpE;;OAEG;IACI,WAAW,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,GAAG,OAAO;IAY1D;;OAEG;IACI,SAAS,CAAC,OAAO,EAAE,GAAG,EAAE,mBAAmB,CAAC,EAAE,MAAM,GAAG,IAAI;IAelE;;OAEG;IACH,OAAO,CAAC,SAAS;IAQjB;;OAEG;IACI,eAAe,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,OAAO;IAY9F;;OAEG;IACI,aAAa,CAAC,YAAY,EAAE,MAAM,GAAG,eAAe,GAAG,SAAS;IAIvE;;OAEG;IACI,cAAc,IAAI,eAAe,EAAE;IAI1C;;OAEG;IACI,QAAQ,IAAI,eAAe;IAIlC;;OAEG;IACH,OAAO,CAAC,oBAAoB;IAI5B;;OAEG;IACU,IAAI,IAAI,OAAO,CAAC,IAAI,CAAC;IA8BlC;;OAEG;IACI,WAAW,IAAI,kBAAkB,GAAG,IAAI;CAGhD"}