@soulcraft/brainy 0.39.0 → 0.40.0

package/dist/unified.js

@@ -4939,7 +4939,8 @@ class HNSWIndex {
         const noun = {
             id,
             vector,
-            connections: new Map()
+            connections: new Map(),
+            level: nounLevel
         };
         // Initialize empty connection sets for each level
         for (let level = 0; level <= nounLevel; level++) {
@@ -5264,6 +5265,29 @@ class HNSWIndex {
     getConfig() {
         return { ...this.config };
     }
+    /**
+     * Get index health metrics
+     */
+    getIndexHealth() {
+        let totalConnections = 0;
+        const layerCounts = new Array(this.maxLevel + 1).fill(0);
+        // Count connections and layer distribution
+        this.nouns.forEach(noun => {
+            // Count connections at each layer
+            for (let level = 0; level <= noun.level; level++) {
+                totalConnections += noun.connections.get(level)?.size || 0;
+                layerCounts[level]++;
+            }
+        });
+        const totalNodes = this.nouns.size;
+        const averageConnections = totalNodes > 0 ? totalConnections / totalNodes : 0;
+        return {
+            averageConnections,
+            layerDistribution: layerCounts,
+            maxLayer: this.maxLevel,
+            totalNodes
+        };
+    }
     /**
      * Search within a specific layer
      * Returns a map of noun IDs to distances, sorted by distance
@@ -5721,7 +5745,8 @@ class HNSWIndexOptimized extends HNSWIndex {
             const noun = {
                 id,
                 vector,
-                connections: new Map()
+                connections: new Map(),
+                level: 0
             };
             // Store the noun
             this.storage.saveNoun(noun).catch((error) => {
@@ -6869,7 +6894,8 @@ class MemoryStorage extends BaseStorage {
         const nounCopy = {
             id: noun.id,
             vector: [...noun.vector],
-            connections: new Map()
+            connections: new Map(),
+            level: noun.level || 0
         };
         // Copy connections
         for (const [level, connections] of noun.connections.entries()) {
@@ -6892,7 +6918,8 @@ class MemoryStorage extends BaseStorage {
         const nounCopy = {
             id: noun.id,
             vector: [...noun.vector],
-            connections: new Map()
+            connections: new Map(),
+            level: noun.level || 0
         };
         // Copy connections
         for (const [level, connections] of noun.connections.entries()) {
@@ -6911,7 +6938,8 @@ class MemoryStorage extends BaseStorage {
             const nounCopy = {
                 id: noun.id,
                 vector: [...noun.vector],
-                connections: new Map()
+                connections: new Map(),
+                level: noun.level || 0
             };
             // Copy connections
             for (const [level, connections] of noun.connections.entries()) {
@@ -6986,7 +7014,8 @@ class MemoryStorage extends BaseStorage {
             const nounCopy = {
                 id: noun.id,
                 vector: [...noun.vector],
-                connections: new Map()
+                connections: new Map(),
+                level: noun.level || 0
             };
             // Copy connections
             for (const [level, connections] of noun.connections.entries()) {
@@ -7294,6 +7323,9 @@ class MemoryStorage extends BaseStorage {
         this.nounMetadata.clear();
         this.verbMetadata.clear();
         this.statistics = null;
+        // Clear the statistics cache
+        this.statisticsCache = null;
+        this.statisticsModified = false;
     }
     /**
      * Get information about storage usage and capacity
@@ -7527,7 +7559,8 @@ class OPFSStorage extends BaseStorage {
             return {
                 id: data.id,
                 vector: data.vector,
-                connections
+                connections,
+                level: data.level || 0
             };
         }
         catch (error) {
@@ -7558,7 +7591,8 @@ class OPFSStorage extends BaseStorage {
                         allNouns.push({
                             id: data.id,
                             vector: data.vector,
-                            connections
+                            connections,
+                            level: data.level || 0
                         });
                     }
                     catch (error) {
@@ -7609,7 +7643,8 @@ class OPFSStorage extends BaseStorage {
                             nodes.push({
                                 id: data.id,
                                 vector: data.vector,
-                                connections
+                                connections,
+                                level: data.level || 0
                             });
                         }
                     }
@@ -7979,6 +8014,9 @@ class OPFSStorage extends BaseStorage {
             await removeDirectoryContents(this.verbMetadataDir);
             // Remove all files in the index directory
             await removeDirectoryContents(this.indexDir);
+            // Clear the statistics cache
+            this.statisticsCache = null;
+            this.statisticsModified = false;
         }
         catch (error) {
             console.error('Error clearing storage:', error);
@@ -10306,7 +10344,8 @@ class S3CompatibleStorage extends BaseStorage {
                 const node = {
                     id: parsedNode.id,
                     vector: parsedNode.vector,
-                    connections
+                    connections,
+                    level: parsedNode.level || 0
                 };
                 console.log(`Successfully retrieved node ${id}:`, node);
                 return node;
@@ -11204,6 +11243,9 @@ class S3CompatibleStorage extends BaseStorage {
             await deleteObjectsWithPrefix(this.metadataPrefix);
             // Delete all objects in the index directory
             await deleteObjectsWithPrefix(this.indexPrefix);
+            // Clear the statistics cache
+            this.statisticsCache = null;
+            this.statisticsModified = false;
         }
         catch (error) {
             console.error('Failed to clear storage:', error);
@@ -12080,7 +12122,8 @@ class FileSystemStorage extends BaseStorage {
             return {
                 id: parsedNode.id,
                 vector: parsedNode.vector,
-                connections
+                connections,
+                level: parsedNode.level || 0
             };
         }
         catch (error) {
@@ -12111,7 +12154,8 @@ class FileSystemStorage extends BaseStorage {
                     allNodes.push({
                         id: parsedNode.id,
                         vector: parsedNode.vector,
-                        connections
+                        connections,
+                        level: parsedNode.level || 0
                     });
                 }
             }
@@ -12150,7 +12194,8 @@ class FileSystemStorage extends BaseStorage {
                         nouns.push({
                             id: parsedNode.id,
                             vector: parsedNode.vector,
-                            connections
+                            connections,
+                            level: parsedNode.level || 0
                         });
                     }
                 }
@@ -12415,6 +12460,9 @@ class FileSystemStorage extends BaseStorage {
         await removeDirectoryContents(this.verbMetadataDir);
         // Remove all files in the index directory
         await removeDirectoryContents(this.indexDir);
+        // Clear the statistics cache
+        this.statisticsCache = null;
+        this.statisticsModified = false;
     }
     /**
      * Get information about storage usage and capacity
@@ -17110,6 +17158,230 @@ class CacheAutoConfigurator {
     }
 }
 
+/**
+ * Lightweight statistics collector for Brainy
+ * Designed to have minimal performance impact even with millions of entries
+ */
+class StatisticsCollector {
+    constructor() {
+        // Content type tracking (lightweight counters)
+        this.contentTypes = new Map();
+        // Data freshness tracking (only track timestamps, not full data)
+        this.oldestTimestamp = Date.now();
+        this.newestTimestamp = Date.now();
+        this.updateTimestamps = [];
+        // Search performance tracking (rolling window)
+        this.searchMetrics = {
+            totalSearches: 0,
+            totalSearchTimeMs: 0,
+            searchTimestamps: [],
+            topSearchTerms: new Map()
+        };
+        // Verb type tracking
+        this.verbTypes = new Map();
+        // Storage size estimates (updated periodically, not on every operation)
+        this.storageSizeCache = {
+            lastUpdated: 0,
+            sizes: {
+                nouns: 0,
+                verbs: 0,
+                metadata: 0,
+                index: 0
+            }
+        };
+        this.MAX_TIMESTAMPS = 1000; // Keep last 1000 timestamps
+        this.MAX_SEARCH_TERMS = 100; // Track top 100 search terms
+        this.SIZE_UPDATE_INTERVAL = 60000; // Update sizes every minute
+    }
+    /**
+     * Track content type (very lightweight)
+     */
+    trackContentType(type) {
+        this.contentTypes.set(type, (this.contentTypes.get(type) || 0) + 1);
+    }
+    /**
+     * Track data update timestamp (lightweight)
+     */
+    trackUpdate(timestamp) {
+        const ts = timestamp || Date.now();
+        // Update oldest/newest
+        if (ts < this.oldestTimestamp)
+            this.oldestTimestamp = ts;
+        if (ts > this.newestTimestamp)
+            this.newestTimestamp = ts;
+        // Add to rolling window
+        this.updateTimestamps.push({ timestamp: ts, count: 1 });
+        // Keep window size limited
+        if (this.updateTimestamps.length > this.MAX_TIMESTAMPS) {
+            this.updateTimestamps.shift();
+        }
+    }
+    /**
+     * Track search performance (lightweight)
+     */
+    trackSearch(searchTerm, durationMs) {
+        this.searchMetrics.totalSearches++;
+        this.searchMetrics.totalSearchTimeMs += durationMs;
+        // Add to rolling window
+        this.searchMetrics.searchTimestamps.push({
+            timestamp: Date.now(),
+            count: 1
+        });
+        // Keep window size limited
+        if (this.searchMetrics.searchTimestamps.length > this.MAX_TIMESTAMPS) {
+            this.searchMetrics.searchTimestamps.shift();
+        }
+        // Track search term (limit to top N)
+        const termCount = (this.searchMetrics.topSearchTerms.get(searchTerm) || 0) + 1;
+        this.searchMetrics.topSearchTerms.set(searchTerm, termCount);
+        // Prune if too many terms
+        if (this.searchMetrics.topSearchTerms.size > this.MAX_SEARCH_TERMS * 2) {
+            this.pruneSearchTerms();
+        }
+    }
+    /**
+     * Track verb type (lightweight)
+     */
+    trackVerbType(type) {
+        this.verbTypes.set(type, (this.verbTypes.get(type) || 0) + 1);
+    }
+    /**
+     * Update storage size estimates (called periodically, not on every operation)
+     */
+    updateStorageSizes(sizes) {
+        this.storageSizeCache = {
+            lastUpdated: Date.now(),
+            sizes
+        };
+    }
+    /**
+     * Get comprehensive statistics
+     */
+    getStatistics() {
+        const now = Date.now();
+        const hourAgo = now - 3600000;
+        const dayAgo = now - 86400000;
+        const weekAgo = now - 604800000;
+        const monthAgo = now - 2592000000;
+        // Calculate data freshness
+        const updatesLastHour = this.updateTimestamps.filter(t => t.timestamp > hourAgo).length;
+        const updatesLastDay = this.updateTimestamps.filter(t => t.timestamp > dayAgo).length;
+        // Calculate age distribution
+        const ageDistribution = {
+            last24h: 0,
+            last7d: 0,
+            last30d: 0,
+            older: 0
+        };
+        // Estimate based on update patterns (not scanning all data)
+        const totalUpdates = this.updateTimestamps.length;
+        if (totalUpdates > 0) {
+            const recentUpdates = this.updateTimestamps.filter(t => t.timestamp > dayAgo).length;
+            const weekUpdates = this.updateTimestamps.filter(t => t.timestamp > weekAgo).length;
+            const monthUpdates = this.updateTimestamps.filter(t => t.timestamp > monthAgo).length;
+            ageDistribution.last24h = Math.round((recentUpdates / totalUpdates) * 100);
+            ageDistribution.last7d = Math.round(((weekUpdates - recentUpdates) / totalUpdates) * 100);
+            ageDistribution.last30d = Math.round(((monthUpdates - weekUpdates) / totalUpdates) * 100);
+            ageDistribution.older = 100 - ageDistribution.last24h - ageDistribution.last7d - ageDistribution.last30d;
+        }
+        // Calculate search metrics
+        const searchesLastHour = this.searchMetrics.searchTimestamps.filter(t => t.timestamp > hourAgo).length;
+        const searchesLastDay = this.searchMetrics.searchTimestamps.filter(t => t.timestamp > dayAgo).length;
+        const avgSearchTime = this.searchMetrics.totalSearches > 0
+            ? this.searchMetrics.totalSearchTimeMs / this.searchMetrics.totalSearches
+            : 0;
+        // Get top search terms
+        const topSearchTerms = Array.from(this.searchMetrics.topSearchTerms.entries())
+            .sort((a, b) => b[1] - a[1])
+            .slice(0, 10)
+            .map(([term]) => term);
+        // Calculate storage metrics
+        const totalSize = Object.values(this.storageSizeCache.sizes).reduce((a, b) => a + b, 0);
+        return {
+            contentTypes: Object.fromEntries(this.contentTypes),
+            dataFreshness: {
+                oldestEntry: new Date(this.oldestTimestamp).toISOString(),
+                newestEntry: new Date(this.newestTimestamp).toISOString(),
+                updatesLastHour,
+                updatesLastDay,
+                ageDistribution
+            },
+            storageMetrics: {
+                totalSizeBytes: totalSize,
+                nounsSizeBytes: this.storageSizeCache.sizes.nouns,
+                verbsSizeBytes: this.storageSizeCache.sizes.verbs,
+                metadataSizeBytes: this.storageSizeCache.sizes.metadata,
+                indexSizeBytes: this.storageSizeCache.sizes.index
+            },
+            searchMetrics: {
+                totalSearches: this.searchMetrics.totalSearches,
+                averageSearchTimeMs: avgSearchTime,
+                searchesLastHour,
+                searchesLastDay,
+                topSearchTerms
+            },
+            verbStatistics: {
+                totalVerbs: Array.from(this.verbTypes.values()).reduce((a, b) => a + b, 0),
+                verbTypes: Object.fromEntries(this.verbTypes),
+                averageConnectionsPerVerb: 2 // Verbs connect 2 nouns
+            }
+        };
+    }
+    /**
+     * Merge statistics from storage (for distributed systems)
+     */
+    mergeFromStorage(stored) {
+        // Merge content types
+        if (stored.contentTypes) {
+            for (const [type, count] of Object.entries(stored.contentTypes)) {
+                this.contentTypes.set(type, count);
+            }
+        }
+        // Merge verb types
+        if (stored.verbStatistics?.verbTypes) {
+            for (const [type, count] of Object.entries(stored.verbStatistics.verbTypes)) {
+                this.verbTypes.set(type, count);
+            }
+        }
+        // Merge search metrics
+        if (stored.searchMetrics) {
+            this.searchMetrics.totalSearches = stored.searchMetrics.totalSearches || 0;
+            this.searchMetrics.totalSearchTimeMs = (stored.searchMetrics.averageSearchTimeMs || 0) * this.searchMetrics.totalSearches;
+        }
+        // Merge data freshness
+        if (stored.dataFreshness) {
+            this.oldestTimestamp = new Date(stored.dataFreshness.oldestEntry).getTime();
+            this.newestTimestamp = new Date(stored.dataFreshness.newestEntry).getTime();
+        }
+    }
+    /**
+     * Reset statistics (for testing)
+     */
+    reset() {
+        this.contentTypes.clear();
+        this.verbTypes.clear();
+        this.updateTimestamps = [];
+        this.searchMetrics = {
+            totalSearches: 0,
+            totalSearchTimeMs: 0,
+            searchTimestamps: [],
+            topSearchTerms: new Map()
+        };
+        this.oldestTimestamp = Date.now();
+        this.newestTimestamp = Date.now();
+    }
+    pruneSearchTerms() {
+        // Keep only top N search terms
+        const sorted = Array.from(this.searchMetrics.topSearchTerms.entries())
+            .sort((a, b) => b[1] - a[1])
+            .slice(0, this.MAX_SEARCH_TERMS);
+        this.searchMetrics.topSearchTerms.clear();
+        for (const [term, count] of sorted) {
+            this.searchMetrics.topSearchTerms.set(term, count);
+        }
+    }
+}
+
 /**
  * BrainyData
  * Main class that provides the vector database functionality
@@ -17170,6 +17442,8 @@ class BrainyData {
         this.operationalMode = null;
         this.domainDetector = null;
         this.healthMonitor = null;
+        // Statistics collector
+        this.statisticsCollector = new StatisticsCollector();
         // Set dimensions to fixed value of 512 (Universal Sentence Encoder dimension)
         this._dimensions = 512;
         // Set distance function
@@ -17433,7 +17707,8 @@ class BrainyData {
             }
             // Adapt cache configuration based on performance (every few updates)
             // Only adapt every 5th update to avoid over-optimization
-            const updateCount = Math.floor((Date.now() - (this.lastUpdateTime || 0)) / this.realtimeUpdateConfig.interval);
+            const updateCount = Math.floor((Date.now() - (this.lastUpdateTime || 0)) /
+                this.realtimeUpdateConfig.interval);
             if (updateCount % 5 === 0) {
                 this.adaptCacheConfiguration();
             }
@@ -17751,6 +18026,16 @@ class BrainyData {
                     // Continue initialization even if remote connection fails
                 }
             }
+            // Initialize statistics collector with existing data
+            try {
+                const existingStats = await this.storage.getStatistics();
+                if (existingStats) {
+                    this.statisticsCollector.mergeFromStorage(existingStats);
+                }
+            }
+            catch (e) {
+                // Ignore errors loading existing statistics
+            }
             this.isInitialized = true;
             this.isInitializing = false;
             // Start real-time updates if enabled
@@ -17962,13 +18247,15 @@ class BrainyData {
                 try {
                     if (this.writeOnly) {
                         // In write-only mode, check storage directly
-                        existingNoun = await this.storage.getNoun(options.id) ?? undefined;
+                        existingNoun =
+                            (await this.storage.getNoun(options.id)) ?? undefined;
                     }
                     else {
                         // In normal mode, check index first, then storage
                         existingNoun = this.index.getNouns().get(options.id);
                         if (!existingNoun) {
-                            existingNoun = await this.storage.getNoun(options.id) ?? undefined;
+                            existingNoun =
+                                (await this.storage.getNoun(options.id)) ?? undefined;
                         }
                     }
                     if (existingNoun) {
@@ -18003,6 +18290,7 @@ class BrainyData {
                     id,
                     vector,
                     connections: new Map(),
+                    level: 0, // Default level for new nodes
                     metadata: undefined // Will be set separately
                 };
             }
@@ -18076,17 +18364,24 @@ class BrainyData {
                                 // Domain already specified, keep it
                                 const domainInfo = this.domainDetector.detectDomain(metadataToSave);
                                 if (domainInfo.domainMetadata) {
-                                    metadataToSave.domainMetadata = domainInfo.domainMetadata;
+                                    ;
+                                    metadataToSave.domainMetadata =
+                                        domainInfo.domainMetadata;
                                 }
                             }
                             else {
                                 // Try to detect domain from the data
-                                const dataToAnalyze = Array.isArray(vectorOrData) ? metadata : vectorOrData;
+                                const dataToAnalyze = Array.isArray(vectorOrData)
+                                    ? metadata
+                                    : vectorOrData;
                                 const domainInfo = this.domainDetector.detectDomain(dataToAnalyze);
                                 if (domainInfo.domain) {
+                                    ;
                                     metadataToSave.domain = domainInfo.domain;
                                     if (domainInfo.domainMetadata) {
-                                        metadataToSave.domainMetadata = domainInfo.domainMetadata;
+                                        ;
+                                        metadataToSave.domainMetadata =
+                                            domainInfo.domainMetadata;
                                     }
                                 }
                             }
@@ -18101,10 +18396,19 @@ class BrainyData {
                     // Track metadata statistics
                     const metadataService = this.getServiceName(options);
                     await this.storage.incrementStatistic('metadata', metadataService);
+                    // Track content type if it's a GraphNoun
+                    if (metadataToSave &&
+                        typeof metadataToSave === 'object' &&
+                        'noun' in metadataToSave) {
+                        this.statisticsCollector.trackContentType(metadataToSave.noun);
+                    }
+                    // Track update timestamp
+                    this.statisticsCollector.trackUpdate();
                 }
             }
-            // Update HNSW index size (excluding verbs)
-            await this.storage.updateHnswIndexSize(await this.getNounCount());
+            // Update HNSW index size with actual index size
+            const indexSize = this.index.size();
+            await this.storage.updateHnswIndexSize(indexSize);
             // Update health metrics if in distributed mode
             if (this.healthMonitor) {
                 const vectorCount = await this.getNounCount();
@@ -18578,7 +18882,7 @@ class BrainyData {
         let startIndex = 0;
         // If cursor provided, find starting position
         if (options.cursor) {
-            startIndex = allResults.findIndex(r => r.id === options.cursor.lastId &&
+            startIndex = allResults.findIndex((r) => r.id === options.cursor.lastId &&
                 Math.abs(r.score - options.cursor.lastScore) < 0.0001);
             if (startIndex >= 0) {
                 startIndex += 1; // Start after the cursor position
@@ -18595,7 +18899,8 @@ class BrainyData {
         }
         // Create cursor for next page
         let nextCursor;
-        const hasMoreResults = (startIndex + results.length) < allResults.length || allResults.length >= searchK;
+        const hasMoreResults = startIndex + results.length < allResults.length ||
+            allResults.length >= searchK;
         if (results.length > 0 && hasMoreResults) {
             const lastResult = results[results.length - 1];
             nextCursor = {
@@ -18679,7 +18984,7 @@ class BrainyData {
             });
         }
         // Filter out placeholder nouns from search results
-        searchResults = searchResults.filter(result => {
+        searchResults = searchResults.filter((result) => {
             if (result.metadata && typeof result.metadata === 'object') {
                 const metadata = result.metadata;
                 // Exclude placeholder nouns from search results
@@ -18787,7 +19092,7 @@ class BrainyData {
             // In write-only mode, query storage directly since index is not loaded
             if (this.writeOnly) {
                 try {
-                    noun = await this.storage.getNoun(id) ?? undefined;
+                    noun = (await this.storage.getNoun(id)) ?? undefined;
                 }
                 catch (storageError) {
                     // If storage lookup fails, return null (noun doesn't exist)
@@ -18800,7 +19105,7 @@ class BrainyData {
                 // If not found in index, fallback to storage (for race conditions)
                 if (!noun && this.storage) {
                     try {
-                        noun = await this.storage.getNoun(id) ?? undefined;
+                        noun = (await this.storage.getNoun(id)) ?? undefined;
                     }
                     catch (storageError) {
                         // Storage lookup failed, noun doesn't exist
@@ -19192,6 +19497,7 @@ class BrainyData {
                     id: sourceId,
                     vector: sourcePlaceholderVector,
                     connections: new Map(),
+                    level: 0,
                     metadata: sourceMetadata
                 };
                 // Create placeholder target noun
@@ -19212,6 +19518,7 @@ class BrainyData {
                     id: targetId,
                     vector: targetPlaceholderVector,
                     connections: new Map(),
+                    level: 0,
                     metadata: targetMetadata
                 };
                 // Save placeholder nouns to storage (but skip indexing for speed)
@@ -19448,8 +19755,11 @@ class BrainyData {
             // Track verb statistics
             const serviceForStats = this.getServiceName(options);
             await this.storage.incrementStatistic('verb', serviceForStats);
-            // Update HNSW index size (excluding verbs)
-            await this.storage.updateHnswIndexSize(await this.getNounCount());
+            // Track verb type
+            this.statisticsCollector.trackVerbType(verbMetadata.verb);
+            // Update HNSW index size with actual index size
+            const indexSize = this.index.size();
+            await this.storage.updateHnswIndexSize(indexSize);
             // Invalidate search cache since verb data has changed
             this.searchCache.invalidateOnDataChange('add');
             return id;
@@ -19674,6 +19984,8 @@ class BrainyData {
             await this.index.clear();
             // Clear storage
             await this.storage.clear();
+            // Reset statistics collector
+            this.statisticsCollector = new StatisticsCollector();
             // Clear search cache since all data has been removed
             this.searchCache.invalidateOnDataChange('delete');
         }
@@ -19727,7 +20039,8 @@ class BrainyData {
             // Apply new cache configuration
             this.searchCache.updateConfig(newConfig.cacheConfig);
             // Apply new real-time update configuration if needed
-            if (newConfig.realtimeConfig.enabled !== this.realtimeUpdateConfig.enabled ||
+            if (newConfig.realtimeConfig.enabled !==
+                this.realtimeUpdateConfig.enabled ||
                 newConfig.realtimeConfig.interval !== this.realtimeUpdateConfig.interval) {
                 const wasEnabled = this.realtimeUpdateConfig.enabled;
                 this.realtimeUpdateConfig = {
@@ -19817,6 +20130,41 @@ class BrainyData {
         // Call the flushStatisticsToStorage method on the storage adapter
         await this.storage.flushStatisticsToStorage();
     }
+    /**
+     * Update storage sizes if needed (called periodically for performance)
+     */
+    async updateStorageSizesIfNeeded() {
+        // Only update every minute to avoid performance impact
+        const now = Date.now();
+        const lastUpdate = this.lastStorageSizeUpdate || 0;
+        if (now - lastUpdate < 60000) {
+            return; // Skip if updated recently
+        }
+        this.lastStorageSizeUpdate = now;
+        try {
+            // Estimate sizes based on counts and average sizes
+            const stats = await this.storage.getStatistics();
+            if (stats) {
+                const avgNounSize = 2048; // ~2KB per noun (vector + metadata)
+                const avgVerbSize = 512; // ~0.5KB per verb
+                const avgMetadataSize = 256; // ~0.25KB per metadata entry
+                const avgIndexEntrySize = 128; // ~128 bytes per index entry
+                // Calculate total counts
+                const totalNouns = Object.values(stats.nounCount).reduce((a, b) => a + b, 0);
+                const totalVerbs = Object.values(stats.verbCount).reduce((a, b) => a + b, 0);
+                const totalMetadata = Object.values(stats.metadataCount).reduce((a, b) => a + b, 0);
+                this.statisticsCollector.updateStorageSizes({
+                    nouns: totalNouns * avgNounSize,
+                    verbs: totalVerbs * avgVerbSize,
+                    metadata: totalMetadata * avgMetadataSize,
+                    index: stats.hnswIndexSize * avgIndexEntrySize
+                });
+            }
+        }
+        catch (error) {
+            // Ignore errors in size calculation
+        }
+    }
     /**
      * Get statistics about the current state of the database
      * @param options Additional options for retrieving statistics
@@ -19891,6 +20239,40 @@ class BrainyData {
                     relate: result.verbCount,
                     total: result.nounCount + result.verbCount + result.metadataCount
                 };
+                // Add extended statistics if requested
+                if (true) {
+                    // Always include for now
+                    // Add index health metrics
+                    try {
+                        const indexHealth = this.index.getIndexHealth();
+                        result.indexHealth = indexHealth;
+                    }
+                    catch (e) {
+                        // Index health not available
+                    }
+                    // Add cache metrics
+                    try {
+                        const cacheStats = this.searchCache.getStats();
+                        result.cacheMetrics = cacheStats;
+                    }
+                    catch (e) {
+                        // Cache stats not available
+                    }
+                    // Add memory usage
+                    if (typeof process !== 'undefined' && process.memoryUsage) {
+                        ;
+                        result.memoryUsage = process.memoryUsage().heapUsed;
+                    }
+                    // Add last updated timestamp
+                    ;
+                    result.lastUpdated =
+                        stats.lastUpdated || new Date().toISOString();
+                    // Add enhanced statistics from collector
+                    const collectorStats = this.statisticsCollector.getStatistics();
+                    Object.assign(result, collectorStats);
+                    // Update storage sizes if needed (only periodically for performance)
+                    await this.updateStorageSizesIfNeeded();
+                }
                 return result;
             }
             // If statistics are not available, return zeros instead of calculating on-demand
@@ -20250,15 +20632,20 @@ class BrainyData {
         await this.ensureInitialized();
         // Check if database is in write-only mode
         this.checkWriteOnly();
+        const searchStartTime = Date.now();
         try {
             // Embed the query text
             const queryVector = await this.embed(query);
             // Search using the embedded vector
-            return await this.search(queryVector, k, {
+            const results = await this.search(queryVector, k, {
                 nounTypes: options.nounTypes,
                 includeVerbs: options.includeVerbs,
                 searchMode: options.searchMode
             });
+            // Track search performance
+            const duration = Date.now() - searchStartTime;
+            this.statisticsCollector.trackSearch(query, duration);
+            return results;
         }
         catch (error) {
             console.error('Failed to search with text query:', error);