code-graph-context 2.12.6 → 2.12.7

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

@@ -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: ${msg}`);
+                throw error;
+            }
         }
         return results;
     }

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

@@ -132,11 +132,16 @@ 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})`);
             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 +151,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.7",
   "description": "MCP server that builds code graphs to provide rich context to LLMs",
   "type": "module",
   "homepage": "https://github.com/drewdrewH/code-graph-context#readme",