viberag 0.4.2 → 0.4.4

package/dist/cli/components/StatusBar.js

@@ -77,7 +77,7 @@ function deriveIndexingDisplay(indexing) {
         percent: indexing.percent,
         stage: indexing.stage,
         chunkInfo,
-        throttleInfo: indexing.throttleMessage,
+        throttleInfo: indexing.throttleMessage !== null ? 'Requests retrying...' : null,
         color,
     };
 }

package/dist/daemon/services/indexing.d.ts

@@ -74,6 +74,7 @@ export declare class IndexingService extends TypedEmitter<IndexingServiceEvents>
     private logger;
     private debugLogger;
     private readonly externalStorage;
+    private suppressEvents;
     constructor(projectRoot: string, options?: IndexingServiceOptions | Logger);
     /**
      * Run the indexing pipeline.

package/dist/daemon/services/indexing.js

@@ -107,6 +107,12 @@ export class IndexingService extends TypedEmitter {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "suppressEvents", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
         this.projectRoot = projectRoot;
         this.debugLogger = createServiceLogger(projectRoot, 'indexer');
         // Handle both old (logger) and new (options) signatures for backward compatibility
@@ -157,6 +163,7 @@ export class IndexingService extends TypedEmitter {
         const { force = false } = options;
         const failedFilesThisRun = new Set();
         const failedBatches = [];
+        this.suppressEvents = false;
         // Emit start event
         this.emit('start');
         this.emit('slots-reset');
@@ -265,6 +272,9 @@ export class IndexingService extends TypedEmitter {
                     let chunksProcessed = 0;
                     // Emit progress helper
                     const emitProgress = () => {
+                        if (this.suppressEvents) {
+                            return;
+                        }
                         this.emit('progress', {
                             current: chunksProcessed,
                             total: totalChunks,
@@ -275,9 +285,10 @@ export class IndexingService extends TypedEmitter {
                     // Wire throttle callback for rate limit feedback
                     if ('onThrottle' in embeddings) {
                         embeddings.onThrottle = message => {
-                            if (message) {
-                                this.emit('throttle', { message });
+                            if (this.suppressEvents) {
+                                return;
                             }
+                            this.emit('throttle', { message });
                         };
                     }
                     // Wire slot progress callbacks for API providers
@@ -319,6 +330,9 @@ export class IndexingService extends TypedEmitter {
                                 // Wire up per-chunk progress callback
                                 if ('onBatchProgress' in embeddings) {
                                     embeddings.onBatchProgress = (processed, _total) => {
+                                        if (this.suppressEvents) {
+                                            return;
+                                        }
                                         const delta = processed - lastReportedWithinBatch;
                                         if (delta > 0) {
                                             chunksProcessed += delta;
@@ -426,6 +440,7 @@ export class IndexingService extends TypedEmitter {
             return stats;
         }
         catch (error) {
+            this.suppressEvents = true;
             this.log('error', 'Indexing failed', error);
             this.emit('error', {
                 error: error instanceof Error ? error : new Error(String(error)),
@@ -440,21 +455,33 @@ export class IndexingService extends TypedEmitter {
         const provider = embeddings;
         if ('onSlotProcessing' in provider) {
             provider.onSlotProcessing = (slot, batchInfo) => {
+                if (this.suppressEvents) {
+                    return;
+                }
                 this.emit('slot-processing', { slot, batchInfo });
             };
         }
         if ('onSlotRateLimited' in provider) {
             provider.onSlotRateLimited = (slot, _batchInfo, retryInfo) => {
+                if (this.suppressEvents) {
+                    return;
+                }
                 this.emit('slot-rate-limited', { slot, retryInfo });
             };
         }
         if ('onSlotIdle' in provider) {
             provider.onSlotIdle = slot => {
+                if (this.suppressEvents) {
+                    return;
+                }
                 this.emit('slot-idle', { slot });
             };
         }
         if ('onSlotFailure' in provider) {
             provider.onSlotFailure = data => {
+                if (this.suppressEvents) {
+                    return;
+                }
                 const failure = {
                     batchInfo: data.batchInfo,
                     files: data.files,
@@ -474,6 +501,9 @@ export class IndexingService extends TypedEmitter {
         }
         if ('onResetSlots' in provider) {
             provider.onResetSlots = () => {
+                if (this.suppressEvents) {
+                    return;
+                }
                 this.emit('slots-reset');
             };
         }
@@ -614,6 +644,7 @@ export class IndexingService extends TypedEmitter {
      * Close all resources.
      */
     close() {
+        this.suppressEvents = true;
         // Only close storage if we created it (not external)
         if (!this.externalStorage) {
             this.storage?.close();

package/dist/daemon/services/storage/index.js

@@ -296,7 +296,12 @@ export class Storage {
         this.ensureConnected();
         if (entries.length === 0)
             return;
-        const rows = entries.map(embeddingToRow);
+        // Deduplicate by content hash to avoid merge-insert ambiguity.
+        const uniqueEntries = new Map();
+        for (const entry of entries) {
+            uniqueEntries.set(entry.contentHash, entry);
+        }
+        const rows = Array.from(uniqueEntries.values()).map(embeddingToRow);
         // Use merge insert for upsert behavior
         await this.getCacheTable()
             .mergeInsert('content_hash')

package/dist/daemon/services/types.d.ts

@@ -45,9 +45,9 @@ export interface IndexingEvents {
     'chunk-progress': [data: {
         chunksProcessed: number;
     }];
-    /** Rate limiting message from embedding provider */
+    /** Rate limiting message from embedding provider (null clears) */
     throttle: [data: {
-        message: string;
+        message: string | null;
     }];
     /** Indexing completed successfully */
     complete: [data: {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "viberag",
-	"version": "0.4.2",
+	"version": "0.4.4",
 	"description": "Local code RAG for AI coding assistants - semantic search via MCP server",
 	"license": "AGPL-3.0",
 	"keywords": [