code-graph-context 2.4.0 → 2.4.1

package/dist/core/config/fairsquare-framework-schema.js

@@ -525,13 +525,11 @@ export const FAIRSQUARE_FRAMEWORK_SCHEMA = {
                 {
                     type: 'function',
                     pattern: (parsedNode) => {
-                        const node = parsedNode.sourceNode;
-                        if (!node || !Node.isVariableDeclaration(node))
-                            return false;
-                        const name = node.getName();
-                        const typeNode = node.getTypeNode();
+                        // Use pre-extracted properties (works after AST cleanup in streaming/chunking)
+                        const name = parsedNode.properties.name;
+                        const typeAnnotation = parsedNode.properties.typeAnnotation;
                         // Check if variable name ends with "Routes" AND has type ModuleRoute[]
-                        return !!name.endsWith('Routes') && !!typeNode?.getText().includes('ModuleRoute');
+                        return !!name?.endsWith('Routes') && !!typeAnnotation?.includes('ModuleRoute');
                     },
                     confidence: 1.0,
                     priority: 10,
@@ -696,14 +694,8 @@ export const FAIRSQUARE_FRAMEWORK_SCHEMA = {
                 if (matchingRoutes.length === 0)
                     return false;
                 // CRITICAL FIX: Verify the method belongs to the correct controller
-                // Find the parent class of this method by checking the AST node
-                const targetNode = parsedTargetNode.sourceNode;
-                if (!targetNode || !Node.isMethodDeclaration(targetNode))
-                    return false;
-                const parentClass = targetNode.getParent();
-                if (!parentClass || !Node.isClassDeclaration(parentClass))
-                    return false;
-                const parentClassName = parentClass.getName();
+                // Use pre-extracted parentClassName property (works after AST cleanup in streaming/chunking)
+                const parentClassName = parsedTargetNode.properties.parentClassName;
                 if (!parentClassName)
                     return false;
                 // Check if any matching route's controller name matches the parent class

package/dist/core/config/nestjs-framework-schema.js

@@ -434,7 +434,7 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
                 },
                 {
                     type: 'function',
-                    pattern: (node) => node.getName()?.endsWith('Service'),
+                    pattern: (parsedNode) => parsedNode.sourceNode?.getName()?.endsWith('Service'),
                     confidence: 0.7,
                     priority: 7,
                 },
@@ -516,7 +516,10 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
             detectionPatterns: [
                 {
                     type: 'function',
-                    pattern: (node) => {
+                    pattern: (parsedNode) => {
+                        const node = parsedNode.sourceNode;
+                        if (!node)
+                            return false;
                         const decorators = node.getDecorators?.() ?? [];
                         const messageDecorators = ['MessagePattern', 'EventPattern'];
                         return decorators.some((d) => messageDecorators.includes(d.getName()));
@@ -567,7 +570,10 @@ export const NESTJS_FRAMEWORK_SCHEMA = {
             detectionPatterns: [
                 {
                     type: 'function',
-                    pattern: (node) => {
+                    pattern: (parsedNode) => {
+                        const node = parsedNode.sourceNode;
+                        if (!node)
+                            return false;
                         const decorators = node.getDecorators?.() ?? [];
                         const httpDecorators = ['Get', 'Post', 'Put', 'Delete', 'Patch', 'Head', 'Options'];
                         return decorators.some((d) => httpDecorators.includes(d.getName()));

package/dist/core/config/schema.js

@@ -480,6 +480,12 @@ export const CORE_TYPESCRIPT_SCHEMA = {
                     extraction: { method: 'static', defaultValue: false }, // We'll set this manually
                     neo4j: { indexed: true, unique: false, required: true },
                 },
+                {
+                    name: 'typeAnnotation',
+                    type: 'string',
+                    extraction: { method: 'ast', source: 'getTypeNode', transform: 'getText' },
+                    neo4j: { indexed: false, unique: false, required: false },
+                },
             ],
             relationships: [],
             children: {},

package/dist/core/embeddings/natural-language-to-cypher.service.js

@@ -277,16 +277,24 @@ Provide ONLY the JSON response with no additional text, markdown formatting, or
                 frameworkHint = '\nFRAMEWORK DETECTED: React/functional codebase. Use Function nodes for components.';
             }
             return `
-ACTUAL GRAPH SCHEMA (use these exact labels):
+=== VALID NODE LABELS (use ONLY these after the colon) ===
+${nodeTypes}
 
-Node Types: ${nodeTypes}
-Relationship Types: ${relTypes}
-Semantic Types: ${semTypes}
+=== VALID RELATIONSHIP TYPES ===
+${relTypes}
+
+=== SEMANTIC TYPES (these are PROPERTY values, NOT labels) ===
+${semTypes}
+Query semantic types via property: WHERE n.semanticType = 'TypeName'
 ${frameworkHint}
-CRITICAL: Use ONLY these node labels. Do NOT invent labels like :DbService, :UserService, etc.
-For queries about specific classes/services, use: (n:Class {name: 'ClassName'})
-For inheritance: (child:Class)-[:EXTENDS]->(parent:Class {name: 'ParentName'})
-For decorator-based queries: Use semanticType property with values from the discovered semantic types above.
+
+=== CRITICAL RULES ===
+1. Use ONLY the labels listed above after the colon (:Label)
+2. Semantic types are PROPERTY values, NOT labels
+3. Class/service names are PROPERTY values, NOT labels
+4. WRONG: (n:MyService), (n:MyController) - names as labels
+5. CORRECT: (n:Service {name: 'MyService'}), (n:Controller {name: 'MyController'})
+6. CORRECT: (n:Class) WHERE n.semanticType = 'Service'
 `.trim();
         }
         catch (error) {
@@ -507,7 +515,7 @@ Remember to include WHERE n.projectId = $projectId for all node patterns.
     }
     /**
      * Load valid labels dynamically from the schema file.
-     * Returns all keys from rawSchema which represent actual Neo4j labels.
+     * Returns all keys from rawSchema AND discoveredSchema.nodeTypes which represent actual Neo4j labels.
      */
     loadValidLabelsFromSchema() {
         // Fallback to core TypeScript labels if schema not available
@@ -535,12 +543,21 @@ Remember to include WHERE n.projectId = $projectId for all node patterns.
         try {
             const content = fs.readFileSync(this.schemaPath, 'utf-8');
             const schema = JSON.parse(content);
-            if (!schema.rawSchema?.records?.[0]?._fields?.[0]) {
-                return coreLabels;
+            const allLabels = new Set(coreLabels);
+            // Extract labels from rawSchema keys
+            if (schema.rawSchema?.records?.[0]?._fields?.[0]) {
+                const schemaLabels = Object.keys(schema.rawSchema.records[0]._fields[0]);
+                schemaLabels.forEach((label) => allLabels.add(label));
+            }
+            // Also extract labels from discoveredSchema.nodeTypes (includes framework labels)
+            if (schema.discoveredSchema?.nodeTypes) {
+                for (const nodeType of schema.discoveredSchema.nodeTypes) {
+                    if (nodeType.label) {
+                        allLabels.add(nodeType.label);
+                    }
+                }
             }
-            // Extract all keys from rawSchema - these are the valid labels
-            const schemaLabels = Object.keys(schema.rawSchema.records[0]._fields[0]);
-            return new Set([...coreLabels, ...schemaLabels]);
+            return allLabels;
         }
         catch {
             return coreLabels;

package/dist/core/parsers/typescript-parser.js

@@ -542,6 +542,11 @@ export class TypeScriptParser {
                         return astNode.getName();
                     }
                     break;
+                case CoreNodeType.VARIABLE_DECLARATION:
+                    if (Node.isVariableDeclaration(astNode)) {
+                        return astNode.getName();
+                    }
+                    break;
                 default:
                     return astNode.getKindName();
             }
@@ -552,13 +557,18 @@ export class TypeScriptParser {
         return 'Unknown';
     }
     extractProperty(astNode, propDef) {
-        const { method, source, defaultValue } = propDef.extraction;
+        const { method, source, defaultValue, transform } = propDef.extraction;
         try {
             switch (method) {
                 case 'ast':
                     if (typeof source === 'string') {
                         const fn = astNode[source];
-                        return typeof fn === 'function' ? fn.call(astNode) : defaultValue;
+                        let result = typeof fn === 'function' ? fn.call(astNode) : defaultValue;
+                        // Apply transform if specified (e.g., 'getText' on a returned node)
+                        if (result && transform && typeof result[transform] === 'function') {
+                            result = result[transform]();
+                        }
+                        return result ?? defaultValue;
                     }
                     return defaultValue;
                 case 'function':
@@ -1284,11 +1294,9 @@ export class TypeScriptParser {
                         }
                     case 'function':
                         if (typeof pattern.pattern === 'function') {
-                            // Pass the AST sourceNode to pattern functions, not the ParsedNode wrapper
-                            const astNode = node.sourceNode;
-                            if (!astNode)
-                                return false;
-                            return pattern.pattern(astNode);
+                            // Pass the ParsedNode to pattern functions
+                            // Patterns should use pre-extracted properties for cross-chunk compatibility
+                            return pattern.pattern(node);
                         }
                         return false;
                     case 'classname':

package/package.json

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