code-graph-context 2.12.7 → 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) {
@@ -42,7 +42,7 @@ export class LocalEmbeddingsService {
             }
             catch (error) {
                 const msg = error instanceof Error ? error.message : String(error);
-                console.error(`[embedding] Batch ${batchIndex}/${totalBatches} FAILED: ${msg}`);
+                console.error(`[embedding] Batch ${batchIndex}/${totalBatches} FAILED (${batch.length} texts, batchSize=${safeBatchSize}): ${msg}`);
                 throw error;
             }
         }

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

@@ -132,8 +132,9 @@ export class GraphGeneratorHandler {
         // Batch embed all texts that need it
         if (nodesNeedingEmbedding.length > 0) {
             const texts = nodesNeedingEmbedding.map((n) => n.text);
-            const totalBatches = Math.ceil(texts.length / EMBEDDING_BATCH_CONFIG.maxBatchSize);
-            console.error(`[embedding] Starting ${texts.length} texts in ${totalBatches} batches (batch_size=${EMBEDDING_BATCH_CONFIG.maxBatchSize})`);
+            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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "code-graph-context",
-  "version": "2.12.7",
+  "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",