@zuvia-software-solutions/code-mapper 2.3.12 → 2.4.1

package/models/mlx-embedder.py

@@ -228,7 +228,7 @@ def batch_mode(db_path, dims=256, max_tokens=2048):
         db.execute("ALTER TABLE embeddings ADD COLUMN textHash TEXT")
 
     # Query embeddable nodes — skip test/fixture files (BM25 covers them)
-    labels = ('Function', 'Class', 'Method', 'Interface')
+    labels = ('Function', 'Class', 'Method', 'Interface', 'Const', 'Enum', 'TypeAlias', 'Namespace', 'Module', 'Struct')
     placeholders = ','.join('?' * len(labels))
     all_rows = db.execute(
         f"SELECT id, name, label, filePath, content, startLine, endLine, nameExpanded FROM nodes WHERE label IN ({placeholders})",
@@ -272,7 +272,23 @@ def batch_mode(db_path, dims=256, max_tokens=2048):
         for row in db.execute(f"SELECT e.sourceId, c.heuristicLabel FROM edges e JOIN nodes c ON c.id = e.targetId WHERE e.sourceId IN ({ph}) AND e.type = 'MEMBER_OF' AND c.label = 'Community' LIMIT {len(chunk_ids)}", chunk_ids):
             module_map[row[0]] = row[1]
 
-    print(json.dumps({"phase": "context", "with_callers": len(caller_map), "with_module": len(module_map)}), flush=True)
+    # Batch fetch import names per file
+    import_map = {}
+    for ci in range(0, len(node_ids), CHUNK):
+        chunk_ids = node_ids[ci:ci+CHUNK]
+        ph = ','.join('?' * len(chunk_ids))
+        # Get unique file paths for these nodes
+        file_paths = [r[3] for r in rows if r[0] in set(chunk_ids)]
+        unique_files = list(set(file_paths))
+        if unique_files:
+            fph = ','.join('?' * len(unique_files))
+            for row in db.execute(
+                f"SELECT DISTINCT n.filePath, tn.name FROM nodes n JOIN edges e ON e.sourceId = n.id AND e.type = 'IMPORTS' JOIN nodes tn ON tn.id = e.targetId WHERE n.filePath IN ({fph}) LIMIT {len(unique_files)*10}",
+                unique_files
+            ):
+                import_map.setdefault(row[0], []).append(row[1])
+
+    print(json.dumps({"phase": "context", "with_callers": len(caller_map), "with_module": len(module_map), "with_imports": len(import_map)}), flush=True)
 
     # Get existing text hashes for skip detection
     existing_hashes = {}
@@ -367,13 +383,24 @@ def batch_mode(db_path, dims=256, max_tokens=2048):
         if comment:
             parts.append(comment)
 
+        # Directory context
+        dir_parts = filePath.rsplit('/', 2)
+        dir_context = '/'.join(dir_parts[:-1])[-40:] if '/' in filePath else ''
+
         # File + module location
         loc = f"File: {file_name}"
+        if dir_context:
+            loc += f" in {dir_context}"
         module = module_map.get(nid, "")
         if module:
             loc += f" | Module: {module}"
         parts.append(loc)
 
+        # Import context
+        file_imports = import_map.get(filePath, [])[:5]
+        if file_imports:
+            parts.append(f"Imports: {', '.join(file_imports)}")
+
         # Graph context
         callers = caller_map.get(nid, [])[:5]
         callees = callee_map.get(nid, [])[:5]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zuvia-software-solutions/code-mapper",
-  "version": "2.3.12",
+  "version": "2.4.1",
   "description": "Graph-powered code intelligence for AI agents. Index any codebase, query via MCP or CLI.",
   "author": "Abhigyan Patwari",
   "license": "PolyForm-Noncommercial-1.0.0",