@papyruslabsai/seshat-mcp 0.3.0 → 0.3.1

package/dist/graph.js

@@ -21,7 +21,10 @@ export function buildCallGraph(entities) {
         callers.set(entity.id, new Set());
         callees.set(entity.id, new Set());
         if (entity.context?.module) {
-            entityByModule.set(entity.context.module, entity);
+            const mod = entity.context.module;
+            if (!entityByModule.has(mod))
+                entityByModule.set(mod, []);
+            entityByModule.get(mod).push(entity);
         }
     }
     // Build call edges from ε dimension
@@ -42,12 +45,27 @@ export function buildCallGraph(entities) {
             }
             // Strategy 2: Method match (target = "module.method")
             else if (target.includes('.')) {
-                const [modulePart, methodPart] = target.split('.');
-                const moduleEntity = entityByModule.get(modulePart);
-                if (moduleEntity) {
-                    calleeId = moduleEntity.id;
+                const dotIdx = target.indexOf('.');
+                const modulePart = target.substring(0, dotIdx);
+                const methodPart = target.substring(dotIdx + 1);
+                // First try "module.method" as a full entity ID
+                if (entityById.has(target)) {
+                    calleeId = target;
                 }
-                // Also try "module.method" as a full ID
+                // Search within the module's entities for one named methodPart
+                if (!calleeId) {
+                    const moduleEntities = entityByModule.get(modulePart);
+                    if (moduleEntities) {
+                        const match = moduleEntities.find(e => {
+                            const name = typeof e.struct === 'string' ? e.struct : e.struct?.name;
+                            return e.id === methodPart || name === methodPart;
+                        });
+                        if (match) {
+                            calleeId = match.id;
+                        }
+                    }
+                }
+                // Also try "module.method" concatenated as an ID
                 if (!calleeId && entityById.has(`${modulePart}.${methodPart}`)) {
                     calleeId = `${modulePart}.${methodPart}`;
                 }

package/dist/index.js

@@ -367,7 +367,7 @@ async function main() {
     }
     const server = new Server({
         name: serverLabel,
-        version: '0.3.0',
+        version: '0.3.1',
     }, {
         capabilities: {
             tools: {},

package/dist/loader.js

@@ -27,6 +27,18 @@ export class BundleLoader {
         const raw = fs.readFileSync(bundlePath, 'utf-8');
         const bundle = JSON.parse(raw);
         this.entities = bundle.entities || [];
+        // Remap bundle field names to internal _ prefixed names.
+        // The extraction pipeline outputs `sourceFile`, `sourceLanguage`, `_jstfFilename`
+        // but our JstfEntity type expects `_sourceFile`, `_sourceLanguage`.
+        for (const e of this.entities) {
+            const raw = e;
+            if (raw.sourceFile && !e._sourceFile) {
+                e._sourceFile = raw.sourceFile;
+            }
+            if (raw.sourceLanguage && !e._sourceLanguage) {
+                e._sourceLanguage = raw.sourceLanguage;
+            }
+        }
         // Load topology if available
         const topoPath = path.join(this.seshatDir, '_topology.json');
         if (fs.existsSync(topoPath)) {

package/dist/tools/index.js

@@ -74,7 +74,8 @@ export function entityLayer(e) {
         return 'model';
     if (src.includes('/schema') || explicit === 'schema')
         return 'schema';
-    if (src.includes('/test') || src.includes('.test.') || src.includes('.spec.'))
+    if (src.includes('/test/') || src.includes('/tests/') || src.includes('/__tests__/') ||
+        src.includes('.test.') || src.includes('.spec.'))
         return 'test';
     return explicit || 'other';
 }
@@ -91,7 +92,10 @@ export function entitySummary(e) {
         async: typeof e.struct !== 'string' ? e.struct?.async : undefined,
         exported: typeof e.struct !== 'string' ? e.struct?.exported : undefined,
         constraints: constraintTags.length > 0 ? constraintTags : undefined,
-        callCount: Array.isArray(e.edges?.calls) ? e.edges.calls.length : 0,
+        callExpressions: Array.isArray(e.edges?.calls) ? e.edges.calls.length : 0,
+        uniqueCallees: Array.isArray(e.edges?.calls)
+            ? new Set(e.edges.calls.map(c => c.target)).size
+            : 0,
     };
 }
 export function normalizeConstraints(constraints) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@papyruslabsai/seshat-mcp",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "Semantic MCP server — exposes a codebase's 9D JSTF-T coordinate space as queryable tools",
   "type": "module",
   "bin": {