code-graph-context 3.0.5 → 3.0.6

package/dist/core/embeddings/embedding-sidecar.js

@@ -128,7 +128,11 @@ export class EmbeddingSidecar {
         });
         this.process.on('exit', (code, signal) => {
             if (!this.stopping) {
-                console.error(`[embedding-sidecar] Process exited unexpectedly (code=${code}, signal=${signal})`);
+                console.error(`[embedding-sidecar] Process exited unexpectedly (pid=${childPid}, code=${code}, signal=${signal}, ` +
+                    `inflight=${this.inflight}, queued=${this.waitQueue.length})`);
+            }
+            else {
+                console.error(`[embedding-sidecar] Process stopped (pid=${childPid})`);
             }
             this.cleanup();
         });
@@ -322,6 +326,13 @@ export class EmbeddingSidecar {
             process.removeListener('exit', this._exitHandler);
             this._exitHandler = null;
         }
+        // Drain the wait queue — reject all queued callers so they don't hang forever
+        while (this.waitQueue.length > 0) {
+            const waiter = this.waitQueue.shift();
+            if (waiter)
+                waiter(); // unblock; they'll fail on the dead connection and can retry
+        }
+        this.inflight = 0;
         this.process = null;
         this.readyPromise = null;
     }

package/dist/core/embeddings/local-embeddings.service.js

@@ -43,18 +43,38 @@ export class LocalEmbeddingsService {
         for (let i = 0; i < texts.length; i += httpLimit) {
             const batch = texts.slice(i, i + httpLimit);
             const batchNum = Math.floor(i / httpLimit) + 1;
-            try {
-                const results = await sidecar.embed(batch, gpuBatchSize);
+            let results;
+            let lastError;
+            // Retry once on transient failures (sidecar crash, connection refused)
+            for (let attempt = 0; attempt < 2; attempt++) {
+                try {
+                    results = await sidecar.embed(batch, gpuBatchSize);
+                    break;
+                }
+                catch (error) {
+                    lastError = error instanceof Error ? error : new Error(String(error));
+                    const isTransient = lastError.message.includes('ECONNREFUSED') ||
+                        lastError.message.includes('ECONNRESET') ||
+                        lastError.message.includes('timed out') ||
+                        lastError.message.includes('fetch failed');
+                    if (isTransient && attempt === 0) {
+                        console.error(`[embedding] Transient failure on batch ${batchNum}/${httpBatches} (${lastError.message}), restarting sidecar and retrying...`);
+                        await sidecar.stop();
+                        await sidecar.start();
+                        continue;
+                    }
+                    const textLengths = batch.map((t) => t.length);
+                    console.error(`[embedding] FAILED HTTP batch ${batchNum}/${httpBatches} (${batch.length} texts, gpuBatchSize=${gpuBatchSize}, ` +
+                        `textLengths=[min=${Math.min(...textLengths)}, max=${Math.max(...textLengths)}, total=${textLengths.reduce((a, b) => a + b, 0)}]): ${lastError.message}`);
+                    throw lastError;
+                }
+            }
+            if (results) {
                 allResults.push(...results);
                 if (httpBatches > 1) {
                     console.error(`[embedding] HTTP batch ${batchNum}/${httpBatches}: ${batch.length} texts embedded`);
                 }
             }
-            catch (error) {
-                const msg = error instanceof Error ? error.message : String(error);
-                console.error(`[embedding] FAILED HTTP batch ${batchNum}/${httpBatches} (${batch.length} texts, gpuBatchSize=${gpuBatchSize}): ${msg}`);
-                throw error;
-            }
         }
         return allResults;
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-context",
-  "version": "3.0.5",
+  "version": "3.0.6",
   "description": "MCP server that builds code graphs to provide rich context to LLMs",
   "type": "module",
   "homepage": "https://github.com/andrew-hernandez-paragon/code-graph-context#readme",