code-graph-context 3.0.8 → 3.0.9

package/dist/mcp/constants.js

@@ -199,7 +199,10 @@ Actions: send (post or broadcast), read (retrieve), acknowledge (mark read). Cat
 export const DEFAULTS = {
     traversalDepth: 3,
     skipOffset: 0,
-    batchSize: 500,
+    /** Batch size for node imports (larger payloads due to embeddings) */
+    nodeBatchSize: 500,
+    /** Batch size for edge imports (lightweight — just IDs and type) */
+    edgeBatchSize: 2000,
     maxResultsDisplayed: 30,
     codeSnippetLength: 500, // Reduced from 1000 to control output size
     chainSnippetLength: 700,

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

@@ -18,6 +18,7 @@ export class GraphGeneratorHandler {
     neo4jService;
     embeddingsService;
     static EMBEDDED_LABEL = 'Embedded';
+    static GRAPH_NODE_LABEL = 'GraphNode';
     projectId = null;
     constructor(neo4jService, embeddingsService) {
         this.neo4jService = neo4jService;
@@ -29,10 +30,10 @@ export class GraphGeneratorHandler {
     setProjectId(projectId) {
         this.projectId = projectId;
     }
-    async generateGraph(graphJsonPath, batchSize = DEFAULTS.batchSize, clearExisting = true) {
+    async generateGraph(graphJsonPath, clearExisting = true) {
         console.error(`Generating graph from JSON file: ${graphJsonPath}`);
         const graphData = await this.loadGraphData(graphJsonPath);
-        return this.generateGraphFromData(graphData.nodes, graphData.edges, batchSize, clearExisting, graphData.metadata);
+        return this.generateGraphFromData(graphData.nodes, graphData.edges, clearExisting, graphData.metadata);
     }
     /**
      * Import nodes and edges directly from in-memory data.
@@ -41,11 +42,12 @@ export class GraphGeneratorHandler {
      * @param skipIndexes - When true, skips index creation (caller manages indexes).
      *   Use this for chunked imports where indexes are created once before/after all chunks.
      */
-    async generateGraphFromData(nodes, edges, batchSize = DEFAULTS.batchSize, clearExisting = true, metadata = {}, skipIndexes = false) {
+    async generateGraphFromData(nodes, edges, clearExisting = true, metadata = {}, skipIndexes = false) {
         await debugLog('Starting graph generation', {
             nodeCount: nodes.length,
             edgeCount: edges.length,
-            batchSize,
+            nodeBatchSize: DEFAULTS.nodeBatchSize,
+            edgeBatchSize: DEFAULTS.edgeBatchSize,
             clearExisting,
             skipIndexes,
             projectId: this.projectId,
@@ -58,8 +60,8 @@ export class GraphGeneratorHandler {
             if (!skipIndexes) {
                 await this.createProjectIndexes();
             }
-            await this.importNodes(nodes, batchSize);
-            await this.importEdges(edges, batchSize);
+            await this.importNodes(nodes);
+            await this.importEdges(edges);
             if (!skipIndexes) {
                 await this.createVectorIndexes();
             }
@@ -77,7 +79,7 @@ export class GraphGeneratorHandler {
             throw error;
         }
     }
-    async ingestConfigFiles(projectPath, projectId, globs, excludeGlobs, maxFileSizeBytes, batchSize = DEFAULTS.batchSize) {
+    async ingestConfigFiles(projectPath, projectId, globs, excludeGlobs, maxFileSizeBytes) {
         const files = await glob(globs, {
             cwd: projectPath,
             ignore: [...EXCLUDE_PATTERNS_GLOB, ...excludeGlobs],
@@ -129,7 +131,7 @@ export class GraphGeneratorHandler {
             });
         }
         console.error(`[config-ingest] Importing ${nodes.length} config file nodes`);
-        await this.importNodes(nodes, batchSize);
+        await this.importNodes(nodes);
         return { nodesCreated: nodes.length };
     }
     /**
@@ -161,14 +163,16 @@ export class GraphGeneratorHandler {
         await this.neo4jService.run(QUERIES.CREATE_PROJECT_INDEX_SOURCEFILE);
         await this.neo4jService.run(QUERIES.CREATE_PROJECT_ID_INDEX_EMBEDDED);
         await this.neo4jService.run(QUERIES.CREATE_PROJECT_ID_INDEX_SOURCEFILE);
+        await this.neo4jService.run(QUERIES.CREATE_PROJECT_ID_INDEX_GRAPHNODE);
         await this.neo4jService.run(QUERIES.CREATE_NORMALIZED_HASH_INDEX);
         await this.neo4jService.run(QUERIES.CREATE_SESSION_BOOKMARK_INDEX);
         await this.neo4jService.run(QUERIES.CREATE_SESSION_NOTE_INDEX);
         await this.neo4jService.run(QUERIES.CREATE_SESSION_NOTE_CATEGORY_INDEX);
         await debugLog('Project indexes created');
     }
-    async importNodes(nodes, batchSize) {
-        console.error(`Importing ${nodes.length} nodes with embeddings...`);
+    async importNodes(nodes) {
+        const batchSize = DEFAULTS.nodeBatchSize;
+        console.error(`Importing ${nodes.length} nodes with embeddings (batch size: ${batchSize})...`);
         // Pipelined: write batch N to Neo4j while embedding batch N+1.
         // This overlaps GPU work with Neo4j I/O.
         let pendingWrite = null;
@@ -218,7 +222,7 @@ export class GraphGeneratorHandler {
                 // Node doesn't need embedding - prepare it immediately
                 nodeResults[index] = {
                     ...node,
-                    labels: node.labels,
+                    labels: [...node.labels, GraphGeneratorHandler.GRAPH_NODE_LABEL],
                     properties: {
                         ...this.flattenProperties(node.properties),
                         embedding: null,
@@ -242,7 +246,7 @@ export class GraphGeneratorHandler {
                         embeddedCount++;
                     nodeResults[item.index] = {
                         ...item.node,
-                        labels: embedding ? [...item.node.labels, GraphGeneratorHandler.EMBEDDED_LABEL] : item.node.labels,
+                        labels: embedding ? [...item.node.labels, GraphGeneratorHandler.EMBEDDED_LABEL, GraphGeneratorHandler.GRAPH_NODE_LABEL] : [...item.node.labels, GraphGeneratorHandler.GRAPH_NODE_LABEL],
                         properties: {
                             ...this.flattenProperties(item.node.properties),
                             embedding,
@@ -265,8 +269,9 @@ export class GraphGeneratorHandler {
         }
         return nodeResults;
     }
-    async importEdges(edges, batchSize) {
-        console.error(`Importing ${edges.length} edges using APOC...`);
+    async importEdges(edges) {
+        const batchSize = DEFAULTS.edgeBatchSize;
+        console.error(`Importing ${edges.length} edges using APOC (batch size: ${batchSize})...`);
         for (let i = 0; i < edges.length; i += batchSize) {
             const batch = edges.slice(i, i + batchSize).map((edge) => ({
                 ...edge,

package/dist/mcp/handlers/incremental-parse.handler.js

@@ -10,7 +10,7 @@ import { ParserFactory } from '../../core/parsers/parser-factory.js';
 import { detectChangedFiles } from '../../core/utils/file-change-detection.js';
 import { resolveProjectId, getProjectName, UPSERT_PROJECT_QUERY } from '../../core/utils/project-id.js';
 import { Neo4jService, QUERIES } from '../../storage/neo4j/neo4j.service.js';
-import { DEFAULTS, FILE_PATHS, LOG_CONFIG } from '../constants.js';
+import { FILE_PATHS, LOG_CONFIG } from '../constants.js';
 import { debugLog } from '../utils.js';
 import { deleteSourceFileSubgraphs, loadExistingNodesForEdgeDetection, getCrossFileEdges, } from './cross-file-edge.helpers.js';
 import { GraphGeneratorHandler } from './graph-generator.handler.js';
@@ -109,7 +109,7 @@ export const performIncrementalParse = async (projectPath, projectId, tsconfigPa
             await debugLog('Incremental parse: starting graph import', {});
             graphHandler.setProjectId(resolvedId);
             try {
-                const result = await graphHandler.generateGraph(outputPath, DEFAULTS.batchSize, false);
+                const result = await graphHandler.generateGraph(outputPath, false);
                 nodesImported = result.nodesImported;
                 edgesImported = result.edgesImported;
                 await debugLog('Incremental parse: graph import completed', { nodesImported, edgesImported });

package/dist/mcp/handlers/parallel-import.handler.js

@@ -117,6 +117,6 @@ export class ParallelImportHandler {
     async importToNeo4j(nodes, edges) {
         if (nodes.length === 0 && edges.length === 0)
             return;
-        await this.graphGeneratorHandler.generateGraphFromData(nodes, edges, 100, false, {}, true);
+        await this.graphGeneratorHandler.generateGraphFromData(nodes, edges, false, {}, true);
     }
 }

package/dist/mcp/handlers/streaming-import.handler.js

@@ -3,7 +3,6 @@
  * Orchestrates chunked parsing and import for large codebases
  */
 import { ProgressReporter } from '../../core/utils/progress-reporter.js';
-import { DEFAULTS } from '../constants.js';
 import { debugLog } from '../utils.js';
 export class StreamingImportHandler {
     graphGeneratorHandler;
@@ -120,11 +119,11 @@ export class StreamingImportHandler {
         return result;
     }
     async importChunkToNeo4j(nodes, edges) {
-        await this.graphGeneratorHandler.generateGraphFromData(nodes, edges, DEFAULTS.batchSize, false, {}, true);
+        await this.graphGeneratorHandler.generateGraphFromData(nodes, edges, false, {}, true);
     }
     async importEdgesToNeo4j(edges) {
         if (edges.length === 0)
             return;
-        await this.graphGeneratorHandler.generateGraphFromData([], edges, DEFAULTS.batchSize, false, {}, true);
+        await this.graphGeneratorHandler.generateGraphFromData([], edges, false, {}, true);
     }
 }

package/dist/mcp/tools/parse-typescript-project.tool.js

@@ -16,7 +16,7 @@ import { ParserFactory } from '../../core/parsers/parser-factory.js';
 import { detectChangedFiles } from '../../core/utils/file-change-detection.js';
 import { resolveProjectId, getProjectName, UPSERT_PROJECT_QUERY, UPDATE_PROJECT_STATUS_QUERY, } from '../../core/utils/project-id.js';
 import { Neo4jService, QUERIES } from '../../storage/neo4j/neo4j.service.js';
-import { TOOL_NAMES, TOOL_METADATA, DEFAULTS, FILE_PATHS, LOG_CONFIG, PARSING, CONFIG_FILE_PATTERNS, } from '../constants.js';
+import { TOOL_NAMES, TOOL_METADATA, FILE_PATHS, LOG_CONFIG, PARSING, CONFIG_FILE_PATTERNS, } from '../constants.js';
 import { deleteSourceFileSubgraphs, loadExistingNodesForEdgeDetection, getCrossFileEdges, } from '../handlers/cross-file-edge.helpers.js';
 import { GraphGeneratorHandler } from '../handlers/graph-generator.handler.js';
 import { StreamingImportHandler } from '../handlers/streaming-import.handler.js';
@@ -315,7 +315,7 @@ export const createParseTypescriptProjectTool = (server) => {
             try {
                 // Set projectId for project-scoped operations (clear, indexes)
                 graphGeneratorHandler.setProjectId(finalProjectId);
-                const result = await graphGeneratorHandler.generateGraph(outputPath, DEFAULTS.batchSize, clearExisting);
+                const result = await graphGeneratorHandler.generateGraph(outputPath, clearExisting);
                 // Recreate cross-file edges after incremental parse
                 if (!clearExisting && savedCrossFileEdges.length > 0) {
                     await debugLog('Recreating cross-file edges', { edgesToRecreate: savedCrossFileEdges.length });

package/dist/storage/neo4j/neo4j.service.js

@@ -87,6 +87,8 @@ export const QUERIES = {
     // Create composite indexes on projectId + id for efficient lookups
     CREATE_PROJECT_ID_INDEX_EMBEDDED: 'CREATE INDEX project_id_embedded_idx IF NOT EXISTS FOR (n:Embedded) ON (n.projectId, n.id)',
     CREATE_PROJECT_ID_INDEX_SOURCEFILE: 'CREATE INDEX project_id_sourcefile_idx IF NOT EXISTS FOR (n:SourceFile) ON (n.projectId, n.id)',
+    // Create composite index on GraphNode for efficient edge resolution lookups
+    CREATE_PROJECT_ID_INDEX_GRAPHNODE: 'CREATE INDEX project_id_graphnode_idx IF NOT EXISTS FOR (n:GraphNode) ON (n.projectId, n.id)',
     // Create index on normalizedHash for efficient structural duplicate detection
     CREATE_NORMALIZED_HASH_INDEX: 'CREATE INDEX normalized_hash_idx IF NOT EXISTS FOR (n:Embedded) ON (n.normalizedHash)',
     CREATE_NODE: `
@@ -96,8 +98,8 @@ export const QUERIES = {
   `,
     CREATE_RELATIONSHIP: `
     UNWIND $edges AS edgeData
-    MATCH (start) WHERE start.id = edgeData.startNodeId AND start.projectId = $projectId
-    MATCH (end) WHERE end.id = edgeData.endNodeId AND end.projectId = $projectId
+    MATCH (start:GraphNode) WHERE start.id = edgeData.startNodeId AND start.projectId = $projectId
+    MATCH (end:GraphNode) WHERE end.id = edgeData.endNodeId AND end.projectId = $projectId
     WITH start, end, edgeData
     CALL apoc.create.relationship(start, edgeData.type, edgeData.properties, end) YIELD rel
     RETURN count(*) as created

package/package.json

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