@supermodeltools/mcp-server 0.9.4 → 0.9.6

package/dist/constants.js

@@ -4,7 +4,7 @@
  * Single source of truth for configuration values
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MAX_SOURCE_LINES = exports.MAX_SYMBOL_RELATED = exports.MAX_SYMBOL_CALLEES = exports.MAX_SYMBOL_CALLERS = exports.MAX_OVERVIEW_HUB_FUNCTIONS = exports.MAX_OVERVIEW_DOMAINS = exports.DEFAULT_CACHE_TTL_MS = exports.DEFAULT_MAX_NODES = exports.DEFAULT_MAX_GRAPHS = exports.MAX_ZIP_SIZE_BYTES = exports.ZIP_CLEANUP_AGE_MS = exports.CONNECTION_TIMEOUT_MS = exports.DEFAULT_API_TIMEOUT_MS = void 0;
+exports.MAX_BRIEF_CALLEES = exports.MAX_BRIEF_CALLERS = exports.MAX_BATCH_SYMBOLS = exports.MAX_SOURCE_LINES = exports.MAX_SYMBOL_RELATED = exports.MAX_SYMBOL_CALLEES = exports.MAX_SYMBOL_CALLERS = exports.MAX_OVERVIEW_HUB_FUNCTIONS = exports.MAX_OVERVIEW_DOMAINS = exports.DEFAULT_CACHE_TTL_MS = exports.DEFAULT_MAX_NODES = exports.DEFAULT_MAX_GRAPHS = exports.MAX_ZIP_SIZE_BYTES = exports.ZIP_CLEANUP_AGE_MS = exports.CONNECTION_TIMEOUT_MS = exports.DEFAULT_API_TIMEOUT_MS = void 0;
 // HTTP timeout configuration
 exports.DEFAULT_API_TIMEOUT_MS = 900_000; // 15 minutes (complex repos can take 10+ min)
 exports.CONNECTION_TIMEOUT_MS = 30_000; // 30 seconds to establish connection
@@ -22,4 +22,8 @@ exports.MAX_OVERVIEW_HUB_FUNCTIONS = 10; // Top N hub functions to show
 exports.MAX_SYMBOL_CALLERS = 10; // Top N callers to show
 exports.MAX_SYMBOL_CALLEES = 10; // Top N callees to show
 exports.MAX_SYMBOL_RELATED = 8; // Related symbols in same file
-exports.MAX_SOURCE_LINES = 50; // Max lines of source code to include inline
+exports.MAX_SOURCE_LINES = 80; // Max lines of source code to include inline
+// Batch symbol_context limits
+exports.MAX_BATCH_SYMBOLS = 10; // Max symbols per batch call
+exports.MAX_BRIEF_CALLERS = 5; // Callers shown in brief mode
+exports.MAX_BRIEF_CALLEES = 5; // Callees shown in brief mode

package/dist/server.js

@@ -40,7 +40,7 @@ exports.Server = void 0;
 /**
  * MCP Server implementation for the Supermodel codebase analysis tools.
  * Redesigned for maximum SWE-bench performance:
- *  - 2 tools (overview, symbol_context) instead of 10
+ *  - 1 tool (symbol_context) with batch support; overview injected into instructions
  *  - Pre-computed graph support for sub-second response times
  *  - On-demand API fallback when no cache exists
  * @module server
@@ -48,7 +48,7 @@ exports.Server = void 0;
 const mcp_js_1 = require("@modelcontextprotocol/sdk/server/mcp.js");
 const stdio_js_1 = require("@modelcontextprotocol/sdk/server/stdio.js");
 const sdk_1 = require("@supermodeltools/sdk");
-const overview_1 = __importDefault(require("./tools/overview"));
+const overview_1 = require("./tools/overview");
 const symbol_context_1 = __importDefault(require("./tools/symbol-context"));
 const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
 const zip_repository_1 = require("./utils/zip-repository");
@@ -89,23 +89,28 @@ class Server {
             capabilities: { tools: {}, logging: {} },
             instructions: `# Supermodel: Codebase Intelligence
 
-Two tools for instant codebase understanding. Pre-computed graphs enable sub-second responses.
+One read-only tool for instant codebase understanding. Pre-computed graphs enable sub-second responses.
+
+The codebase overview is included below in these instructions — you already have the architecture map.
 
 ## Recommended workflow
-1. \`overview\` first to learn architecture and key symbols (1 call)
-2. \`symbol_context\` on 1-2 symbols from the issue to see source, callers, callees (1-2 calls)
-3. Stop calling MCP tools. Use the results to make your fix.
+1. Identify symbols from the issue/overview and call \`symbol_context\` to explore them.
+   Batch via \`symbols\` array or issue multiple calls in parallel (read-only, safe).
+2. Stop calling MCP tools. Start editing by turn 3. Max 3 MCP calls total.
 
-## Anti-patterns
-- >3 MCP calls total = diminishing returns. Aim for 1 overview + 1-2 symbol lookups.
-- Chasing callers-of-callers burns iterations without helping.
+## Rules
+- Do NOT use TodoWrite. Act directly.
+- Use the Task tool to delegate subtasks (e.g. running tests, exploring tangential code).
+- >2 MCP turns = diminishing returns. Explore everything you need in one turn.
 
 ## After fixing
-Run the project's existing test suite (e.g. pytest). Do NOT write standalone test scripts.
+Run the full related test suite to catch regressions. Do NOT write standalone test scripts.
 
 ## Tool reference
-- \`overview\`: Architecture map, domains, hub functions, file counts. Zero-arg, sub-second.
-- \`symbol_context\`: Source, callers, callees, domain for any function/class/method. Supports "Class.method" and partial matching.`,
+- \`symbol_context\`: Source, callers, callees, domain for any function/class/method.
+  Supports "Class.method", partial matching, and batch lookups via \`symbols\` array.
+  Use \`brief: true\` for compact output when looking up 3+ symbols.
+  Read-only — safe to call in parallel.`,
         });
         const config = new sdk_1.Configuration({
             basePath: process.env.SUPERMODEL_BASE_URL || 'https://api.supermodeltools.com',
@@ -126,7 +131,6 @@ Run the project's existing test suite (e.g. pytest). Do NOT write standalone tes
     }
     setupHandlers() {
         const allTools = [
-            overview_1.default,
             symbol_context_1.default,
         ];
         // Create a map for quick handler lookup
@@ -151,6 +155,37 @@ Run the project's existing test suite (e.g. pytest). Do NOT write standalone tes
             throw new Error(`Unknown tool: ${name}`);
         });
     }
+    getTestHint(primaryLanguage) {
+        switch (primaryLanguage.toLowerCase()) {
+            case 'python': return '\n\n**Test with:** `python -m pytest <test_file> -x`';
+            case 'javascript':
+            case 'typescript': return '\n\n**Test with:** `npm test`';
+            case 'go': return '\n\n**Test with:** `go test ./...`';
+            case 'rust': return '\n\n**Test with:** `cargo test`';
+            case 'java': return '\n\n**Test with:** `mvn test` or `gradle test`';
+            case 'ruby': return '\n\n**Test with:** `bundle exec rake test`';
+            default: return '';
+        }
+    }
+    injectOverviewInstructions(repoMap) {
+        if (repoMap.size === 0)
+            return;
+        // Only inject if there's exactly 1 unique graph (SWE-bench always has exactly 1 repo)
+        const uniqueGraphs = new Set([...repoMap.values()]);
+        if (uniqueGraphs.size !== 1)
+            return;
+        const graph = [...uniqueGraphs][0];
+        try {
+            const overview = (0, overview_1.renderOverview)(graph);
+            const testHint = this.getTestHint(graph.summary.primaryLanguage);
+            const current = this.server.server._instructions;
+            this.server.server._instructions = (current || '') + '\n\n' + overview + testHint;
+            logger.debug('Injected overview into server instructions');
+        }
+        catch (err) {
+            logger.warn('Failed to render overview for instructions:', err.message || err);
+        }
+    }
     async start() {
         // Clean up any stale ZIP files from previous sessions
         await (0, zip_repository_1.cleanupOldZips)(constants_1.ZIP_CLEANUP_AGE_MS);
@@ -162,6 +197,7 @@ Run the project's existing test suite (e.g. pytest). Do NOT write standalone tes
                 const repoMap = await (0, graph_cache_1.loadCacheFromDisk)(cacheDir, graph_cache_1.graphCache);
                 (0, graph_cache_1.setRepoMap)(repoMap);
                 logger.debug(`Loaded ${repoMap.size} repo mappings`);
+                this.injectOverviewInstructions(repoMap);
             }
             catch (err) {
                 logger.warn('Failed to load cache directory:', err.message || err);

package/dist/tools/overview.js

@@ -11,6 +11,7 @@
  */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.handler = exports.tool = void 0;
+exports.renderOverview = renderOverview;
 const types_1 = require("../types");
 const graph_cache_1 = require("../cache/graph-cache");
 const api_helpers_1 = require("../utils/api-helpers");
@@ -28,6 +29,9 @@ exports.tool = {
         },
         required: [],
     },
+    annotations: {
+        readOnlyHint: true,
+    },
 };
 const handler = async (client, args, defaultWorkdir) => {
     const rawDir = args?.directory;

package/dist/tools/symbol-context.js

@@ -48,6 +48,7 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.handler = exports.tool = void 0;
 exports.findSymbol = findSymbol;
+exports.renderBriefSymbolContext = renderBriefSymbolContext;
 exports.renderSymbolContext = renderSymbolContext;
 exports.languageFromExtension = languageFromExtension;
 const fs_1 = require("fs");
@@ -58,7 +59,7 @@ const api_helpers_1 = require("../utils/api-helpers");
 const constants_1 = require("../constants");
 exports.tool = {
     name: 'symbol_context',
-    description: `Strictly better than grep for understanding a function, class, or method. Given a symbol name, instantly returns its source code, definition location, all callers, all callees, architectural domain, and related symbols in the same file -- structural context that grep cannot reconstruct. Sub-second, zero cost. Supports partial matching ("filter" finds "QuerySet.filter", "filter_queryset", etc.) and "ClassName.method" syntax. Use this whenever you have a symbol name from a stack trace, issue, or search result and need to understand how it connects to the rest of the codebase.`,
+    description: `Strictly better than grep for understanding a function, class, or method. Given a symbol name, instantly returns its source code, definition location, all callers, all callees, architectural domain, and related symbols in the same file -- structural context that grep cannot reconstruct. Sub-second, zero cost, read-only. Supports partial matching ("filter" finds "QuerySet.filter", "filter_queryset", etc.) and "ClassName.method" syntax. You can issue multiple calls in parallel (read-only, no side effects) or batch symbols into one call via the "symbols" array.`,
     inputSchema: {
         type: 'object',
         properties: {
@@ -66,27 +67,54 @@ exports.tool = {
                 type: 'string',
                 description: 'Name of the function, class, or method to look up. Supports "ClassName.method" syntax.',
             },
+            symbols: {
+                type: 'array',
+                items: { type: 'string' },
+                description: 'Array of symbol names to look up in parallel. More efficient than multiple calls.',
+                maxItems: 10,
+            },
             directory: {
                 type: 'string',
                 description: 'Path to the repository directory. Omit if the MCP server was started with a default workdir.',
             },
+            brief: {
+                type: 'boolean',
+                description: 'Return compact output (definition + callers only, no source code). Recommended when looking up 3+ symbols.',
+            },
         },
-        required: ['symbol'],
+        required: [],
+    },
+    annotations: {
+        readOnlyHint: true,
     },
 };
 const handler = async (client, args, defaultWorkdir) => {
-    const symbol = typeof args?.symbol === 'string' ? args.symbol.trim() : '';
-    const rawDir = args?.directory;
-    const directory = (rawDir && rawDir.trim()) || defaultWorkdir || process.cwd();
-    if (!symbol) {
+    // Extract symbol list: prefer `symbols` array, fall back to single `symbol`
+    let symbolList;
+    const symbolsArg = args?.symbols;
+    const symbolArg = typeof args?.symbol === 'string' ? args.symbol.trim() : '';
+    if (Array.isArray(symbolsArg) && symbolsArg.length > 0) {
+        symbolList = symbolsArg
+            .filter((s) => typeof s === 'string')
+            .map(s => s.trim())
+            .filter(s => s.length > 0)
+            .slice(0, constants_1.MAX_BATCH_SYMBOLS);
+    }
+    else if (symbolArg) {
+        symbolList = [symbolArg];
+    }
+    else {
         return (0, types_1.asErrorResult)({
             type: 'validation_error',
-            message: 'Missing required "symbol" parameter.',
+            message: 'Missing required "symbol" or "symbols" parameter.',
             code: 'MISSING_SYMBOL',
             recoverable: false,
             suggestion: 'Provide the name of a function, class, or method to look up.',
         });
     }
+    const brief = !!args?.brief;
+    const rawDir = args?.directory;
+    const directory = (rawDir && rawDir.trim()) || defaultWorkdir || process.cwd();
     if (!directory || typeof directory !== 'string') {
         return (0, types_1.asErrorResult)({
             type: 'validation_error',
@@ -103,22 +131,33 @@ const handler = async (client, args, defaultWorkdir) => {
     catch (error) {
         return (0, types_1.asErrorResult)((0, api_helpers_1.classifyApiError)(error));
     }
-    // Find the symbol
-    const matches = findSymbol(graph, symbol);
-    if (matches.length === 0) {
-        return (0, types_1.asTextContentResult)(`No symbol matching "${symbol}" found in the code graph.\n\n` +
-            `Try:\n` +
-            `- A different spelling or casing\n` +
-            `- Just the function name without the class prefix\n` +
-            `- Use the \`overview\` tool to see available domains and key functions`);
-    }
-    // Render results for top matches (usually 1, sometimes a few)
-    const renderedParts = await Promise.all(matches.slice(0, 3).map(node => renderSymbolContext(graph, node, directory)));
-    const rendered = renderedParts.join('\n---\n\n');
-    if (matches.length > 3) {
-        return (0, types_1.asTextContentResult)(rendered + `\n\n*... and ${matches.length - 3} more matches. Use a more specific name to narrow results.*`);
+    const isBatch = symbolList.length > 1;
+    const useBrief = brief || isBatch;
+    const allRendered = [];
+    for (const sym of symbolList) {
+        const matches = findSymbol(graph, sym);
+        if (matches.length === 0) {
+            allRendered.push(`## ${sym}\n\nNo symbol matching "${sym}" found in the code graph.`);
+            continue;
+        }
+        if (useBrief) {
+            // Brief mode: compact output for each top match
+            const parts = matches.slice(0, 3).map(node => renderBriefSymbolContext(graph, node));
+            allRendered.push(parts.join('\n---\n\n'));
+            if (matches.length > 3) {
+                allRendered.push(`*... and ${matches.length - 3} more matches for "${sym}".*`);
+            }
+        }
+        else {
+            // Full mode: detailed output (single symbol, not brief)
+            const parts = await Promise.all(matches.slice(0, 3).map(node => renderSymbolContext(graph, node, directory)));
+            allRendered.push(parts.join('\n---\n\n'));
+            if (matches.length > 3) {
+                allRendered.push(`*... and ${matches.length - 3} more matches. Use a more specific name to narrow results.*`);
+            }
+        }
     }
-    return (0, types_1.asTextContentResult)(rendered);
+    return (0, types_1.asTextContentResult)(allRendered.join('\n\n---\n\n'));
 };
 exports.handler = handler;
 // ── Symbol lookup ──
@@ -215,6 +254,45 @@ function callerCount(graph, node) {
     return graph.callAdj.get(node.id)?.in.length || 0;
 }
 // ── Rendering ──
+function renderBriefSymbolContext(graph, node) {
+    const name = node.properties?.name || '(unknown)';
+    const rawFilePath = node.properties?.filePath || '';
+    const filePath = (0, graph_cache_1.normalizePath)(rawFilePath);
+    const startLine = node.properties?.startLine || 0;
+    const endLine = node.properties?.endLine || 0;
+    const kind = node.properties?.kind || node.labels?.[0]?.toLowerCase() || 'symbol';
+    const language = node.properties?.language || '';
+    const lines = [];
+    lines.push(`## ${name}`);
+    lines.push(`**Defined in:** ${filePath}${startLine ? ':' + startLine : ''}${endLine ? '-' + endLine : ''}`);
+    lines.push(`**Type:** ${kind}${language ? ' (' + language + ')' : ''}`);
+    const domain = findDomain(graph, node.id);
+    if (domain) {
+        lines.push(`**Domain:** ${domain}`);
+    }
+    // Callers (inline, max MAX_BRIEF_CALLERS)
+    const adj = graph.callAdj.get(node.id);
+    if (adj && adj.in.length > 0) {
+        const callerNames = adj.in
+            .map(id => graph.nodeById.get(id))
+            .filter((n) => !!n)
+            .map(n => `\`${n.properties?.name || '(unknown)'}\``)
+            .slice(0, constants_1.MAX_BRIEF_CALLERS);
+        const suffix = adj.in.length > constants_1.MAX_BRIEF_CALLERS ? `, ... (${adj.in.length} total)` : '';
+        lines.push(`**Called by:** ${callerNames.join(', ')}${suffix}`);
+    }
+    // Callees (inline, max MAX_BRIEF_CALLEES)
+    if (adj && adj.out.length > 0) {
+        const calleeNames = adj.out
+            .map(id => graph.nodeById.get(id))
+            .filter((n) => !!n)
+            .map(n => `\`${n.properties?.name || '(unknown)'}\``)
+            .slice(0, constants_1.MAX_BRIEF_CALLEES);
+        const suffix = adj.out.length > constants_1.MAX_BRIEF_CALLEES ? `, ... (${adj.out.length} total)` : '';
+        lines.push(`**Calls:** ${calleeNames.join(', ')}${suffix}`);
+    }
+    return lines.join('\n');
+}
 async function renderSymbolContext(graph, node, directory) {
     const name = node.properties?.name || '(unknown)';
     const rawFilePath = node.properties?.filePath || '';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@supermodeltools/mcp-server",
-  "version": "0.9.4",
+  "version": "0.9.6",
   "description": "MCP server for Supermodel API - code graph generation for AI agents",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -38,7 +38,7 @@
   ],
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.1",
-    "@supermodeltools/sdk": "0.9.4",
+    "@supermodeltools/sdk": "0.9.6",
     "archiver": "^7.0.1",
     "ignore": "^7.0.5",
     "jq-web": "^0.6.2",