code-graph-context 2.12.6 → 2.12.8

package/README.md

@@ -190,7 +190,7 @@ If you prefer to edit the config files directly:
 | `NEO4J_USER` | No | `neo4j` | Neo4j username |
 | `NEO4J_PASSWORD` | No | `PASSWORD` | Neo4j password |
 | `EMBEDDING_MODEL` | No | `codesage/codesage-base-v2` | Local embedding model (see [Embedding Configuration](#embedding-configuration)) |
-| `EMBEDDING_BATCH_SIZE` | No | `16` | Texts per embedding batch (lower = less memory, higher = faster) |
+| `EMBEDDING_BATCH_SIZE` | No | `8` | Texts per embedding batch (lower = less memory, higher = faster) |
 | `EMBEDDING_SIDECAR_PORT` | No | `8787` | Port for local embedding server |
 | `EMBEDDING_DEVICE` | No | auto (`mps`/`cpu`) | Device for embeddings. Auto-detects MPS on Apple Silicon |
 | `EMBEDDING_HALF_PRECISION` | No | `false` | Set `true` for float16 (uses ~0.5x memory) |

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

@@ -175,7 +175,9 @@ export class EmbeddingSidecar {
             }
             return false;
         }
-        catch {
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(`[embedding-sidecar] Health check failed: ${msg}`);
             return false;
         }
     }
@@ -186,6 +188,7 @@ export class EmbeddingSidecar {
         await this.start();
         const controller = new AbortController();
         const timeout = setTimeout(() => controller.abort(), this.config.requestTimeoutMs);
+        const startTime = Date.now();
         try {
             const res = await fetch(`${this.baseUrl}/embed`, {
                 method: 'POST',
@@ -202,21 +205,24 @@ export class EmbeddingSidecar {
                     console.error('[embedding-sidecar] OOM detected, restarting sidecar to reclaim GPU memory');
                     await this.stop();
                 }
+                console.error(`[embedding-sidecar] Embed failed after ${Date.now() - startTime}ms: status=${res.status}, texts=${texts.length}, oom=${isOOM}, detail=${detail}`);
                 throw new Error(`Sidecar embed failed (${res.status}): ${detail}`);
             }
             const data = (await res.json());
             if (data.dimensions)
                 this._dimensions = data.dimensions;
+            console.error(`[embedding-sidecar] Embedded ${texts.length} texts in ${Date.now() - startTime}ms (dims=${data.dimensions})`);
             this.resetIdleTimer();
             return data.embeddings;
         }
         catch (err) {
             if (err instanceof Error && err.name === 'AbortError') {
-                // Timeout likely means the sidecar is stuck — kill it
-                console.error('[embedding-sidecar] Request timed out, restarting sidecar');
+                console.error(`[embedding-sidecar] Request timed out after ${Date.now() - startTime}ms (limit=${this.config.requestTimeoutMs}ms, texts=${texts.length}), restarting sidecar`);
                 await this.stop();
                 throw new Error(`Embedding request timed out after ${this.config.requestTimeoutMs}ms`);
             }
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(`[embedding-sidecar] Embed error after ${Date.now() - startTime}ms: ${msg} (url=${this.baseUrl}, running=${this.isRunning}, texts=${texts.length})`);
             throw err;
         }
         finally {

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

@@ -6,7 +6,7 @@
 import { debugLog } from '../../mcp/utils.js';
 import { getEmbeddingSidecar } from './embedding-sidecar.js';
 const BATCH_CONFIG = {
-    maxBatchSize: parseInt(process.env.EMBEDDING_BATCH_SIZE ?? '', 10) || 16,
+    maxBatchSize: parseInt(process.env.EMBEDDING_BATCH_SIZE ?? '', 10) || 8,
 };
 export class LocalEmbeddingsService {
     async embedText(text) {
@@ -28,15 +28,23 @@ export class LocalEmbeddingsService {
         const totalBatches = Math.ceil(texts.length / safeBatchSize);
         for (let i = 0; i < texts.length; i += safeBatchSize) {
             const batch = texts.slice(i, i + safeBatchSize);
-            const batchIndex = Math.floor(i / batchSize) + 1;
+            const batchIndex = Math.floor(i / safeBatchSize) + 1;
+            console.error(`[embedding] Batch ${batchIndex}/${totalBatches} (${batch.length} texts)`);
             await debugLog('Embedding batch progress', {
                 provider: 'local',
                 batchIndex,
                 totalBatches,
                 batchSize: batch.length,
             });
-            const batchResults = await sidecar.embed(batch);
-            results.push(...batchResults);
+            try {
+                const batchResults = await sidecar.embed(batch);
+                results.push(...batchResults);
+            }
+            catch (error) {
+                const msg = error instanceof Error ? error.message : String(error);
+                console.error(`[embedding] Batch ${batchIndex}/${totalBatches} FAILED (${batch.length} texts, batchSize=${safeBatchSize}): ${msg}`);
+                throw error;
+            }
         }
         return results;
     }

package/dist/mcp/handlers/graph-generator.handler.js

@@ -132,11 +132,17 @@ export class GraphGeneratorHandler {
         // Batch embed all texts that need it
         if (nodesNeedingEmbedding.length > 0) {
             const texts = nodesNeedingEmbedding.map((n) => n.text);
+            const effectiveBatchSize = parseInt(process.env.EMBEDDING_BATCH_SIZE ?? '', 10) || EMBEDDING_BATCH_CONFIG.maxBatchSize;
+            const totalBatches = Math.ceil(texts.length / effectiveBatchSize);
+            console.error(`[embedding] Starting ${texts.length} texts in ~${totalBatches} batches (effective_batch_size=${effectiveBatchSize}, config_max=${EMBEDDING_BATCH_CONFIG.maxBatchSize})`);
             try {
                 const embeddings = await this.embeddingsService.embedTextsInBatches(texts, EMBEDDING_BATCH_CONFIG.maxBatchSize);
                 // Map embeddings back to their nodes
+                let embeddedCount = 0;
                 nodesNeedingEmbedding.forEach((item, i) => {
                     const embedding = embeddings[i];
+                    if (embedding)
+                        embeddedCount++;
                     nodeResults[item.index] = {
                         ...item.node,
                         labels: embedding ? [...item.node.labels, GraphGeneratorHandler.EMBEDDED_LABEL] : item.node.labels,
@@ -146,15 +152,17 @@ export class GraphGeneratorHandler {
                         },
                     };
                 });
+                console.error(`[embedding] Done: ${embeddedCount}/${texts.length} nodes embedded`);
                 await debugLog('Batch embedding completed', {
                     totalNodes: nodes.length,
-                    nodesEmbedded: nodesNeedingEmbedding.length,
-                    batchesUsed: Math.ceil(texts.length / EMBEDDING_BATCH_CONFIG.maxBatchSize),
+                    nodesEmbedded: embeddedCount,
+                    batchesUsed: totalBatches,
                 });
             }
             catch (error) {
-                // DON'T silently continue - propagate the error so user knows what's wrong
-                await debugLog('Embedding failed', { error: error instanceof Error ? error.message : String(error) });
+                const msg = error instanceof Error ? error.message : String(error);
+                console.error(`[embedding] FAILED: ${msg}`);
+                await debugLog('Embedding failed', { error: msg });
                 throw error;
             }
         }

package/package.json

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