trace-mcp 1.7.0 → 1.8.0

package/dist/index.js

@@ -4,10 +4,16 @@ var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __require = /* @__PURE__ */ ((x) => typeof require !== "undefined" ? require : typeof Proxy !== "undefined" ? new Proxy(x, {
+  get: (a, b) => (typeof require !== "undefined" ? require : a)[b]
+}) : x)(function(x) {
+  if (typeof require !== "undefined") return require.apply(this, arguments);
+  throw Error('Dynamic require of "' + x + '" is not supported');
+});
 var __esm = (fn, res) => function __init() {
   return fn && (res = (0, fn[__getOwnPropNames(fn)[0]])(fn = 0)), res;
 };
-var __commonJS = (cb, mod) => function __require() {
+var __commonJS = (cb, mod) => function __require2() {
   return mod || (0, cb[__getOwnPropNames(cb)[0]])((mod = { exports: {} }).exports, mod), mod.exports;
 };
 var __export = (target, all) => {
@@ -1106,7 +1112,7 @@ var require_parse = __commonJS({
       }
       return { risky: false };
     };
-    var parse = (input, options) => {
+    var parse2 = (input, options) => {
       if (typeof input !== "string") {
         throw new TypeError("Expected a string");
       }
@@ -1276,7 +1282,7 @@ var require_parse = __commonJS({
             output = token.close = `)$))${extglobStar}`;
           }
           if (token.inner.includes("*") && (rest = remaining()) && /^\.[^\\/.]+$/.test(rest)) {
-            const expression = parse(rest, { ...options, fastpaths: false }).output;
+            const expression = parse2(rest, { ...options, fastpaths: false }).output;
             output = token.close = `)${expression})${extglobStar})`;
           }
           if (token.prev.type === "bos") {
@@ -1798,7 +1804,7 @@ var require_parse = __commonJS({
       }
       return state;
     };
-    parse.fastpaths = (input, options) => {
+    parse2.fastpaths = (input, options) => {
       const opts = { ...options };
       const max = typeof opts.maxLength === "number" ? Math.min(MAX_LENGTH, opts.maxLength) : MAX_LENGTH;
       const len = input.length;
@@ -1863,7 +1869,7 @@ var require_parse = __commonJS({
       }
       return source;
     };
-    module.exports = parse;
+    module.exports = parse2;
   }
 });
 
@@ -1872,7 +1878,7 @@ var require_picomatch = __commonJS({
   "node_modules/picomatch/lib/picomatch.js"(exports, module) {
     "use strict";
     var scan = require_scan();
-    var parse = require_parse();
+    var parse2 = require_parse();
     var utils = require_utils();
     var constants = require_constants();
     var isObject = (val) => val && typeof val === "object" && !Array.isArray(val);
@@ -1960,7 +1966,7 @@ var require_picomatch = __commonJS({
     picomatch3.isMatch = (str2, patterns, options) => picomatch3(patterns, options)(str2);
     picomatch3.parse = (pattern, options) => {
       if (Array.isArray(pattern)) return pattern.map((p) => picomatch3.parse(p, options));
-      return parse(pattern, { ...options, fastpaths: false });
+      return parse2(pattern, { ...options, fastpaths: false });
     };
     picomatch3.scan = (input, options) => scan(input, options);
     picomatch3.compileRe = (state, options, returnOutput = false, returnState = false) => {
@@ -1986,10 +1992,10 @@ var require_picomatch = __commonJS({
       }
       let parsed = { negated: false, fastpaths: true };
       if (options.fastpaths !== false && (input[0] === "." || input[0] === "*")) {
-        parsed.output = parse.fastpaths(input, options);
+        parsed.output = parse2.fastpaths(input, options);
       }
       if (!parsed.output) {
-        parsed = parse(input, options);
+        parsed = parse2(input, options);
       }
       return picomatch3.compileRe(parsed, options, returnOutput, returnState);
     };
@@ -7363,9 +7369,9 @@ var DEFAULT_CONFIG_JSONC = `{
     "provider": "ollama",                       // "ollama" | "openai"
     // "base_url": "http://localhost:11434",     // custom endpoint
     // "api_key": "",                            // required for openai; or set OPENAI_API_KEY env
-    // "inference_model": "",                    // ollama: "gemma4-e4b", openai: "gpt-4o-mini"
-    // "fast_model": "",                         // ollama: "gemma4-e4b", openai: "gpt-4o-mini"
-    // "embedding_model": "",                    // ollama: "qwen3-embedding:0.6b", openai: "text-embedding-3-small"
+    "inference_model": "gemma4-e4b",             // ollama: "gemma4-e4b", openai: "gpt-4o-mini"
+    "fast_model": "gemma4-e4b",                  // ollama: "gemma4-e4b", openai: "gpt-4o-mini"
+    "embedding_model": "qwen3-embedding:0.6b",   // ollama: "qwen3-embedding:0.6b", openai: "text-embedding-3-small"
     // "embedding_dimensions": 1536,             // provider-specific
     "summarize_on_index": true,
     "summarize_batch_size": 20,
@@ -9988,10 +9994,34 @@ var DEFAULT_WEIGHTS = { primary: 0.4, dependencies: 0.3, callers: 0.2, typeConte
 function assembleStructuredContext(request) {
   const weights = request.budgetWeights ?? DEFAULT_WEIGHTS;
   const budget = request.totalBudget;
-  const primaryResult = assembleContext(request.primary, Math.floor(budget * weights.primary));
-  const depsResult = assembleContext(request.dependencies, Math.floor(budget * weights.dependencies));
-  const callersResult = assembleContext(request.callers, Math.floor(budget * weights.callers));
-  const typeResult = assembleContext(request.typeContext, Math.floor(budget * weights.typeContext));
+  const categories = ["primary", "dependencies", "callers", "typeContext"];
+  const itemCounts = {
+    primary: request.primary.length,
+    dependencies: request.dependencies.length,
+    callers: request.callers.length,
+    typeContext: request.typeContext.length
+  };
+  let surplusWeight = 0;
+  let activeWeight = 0;
+  for (const cat of categories) {
+    if (itemCounts[cat] === 0) {
+      surplusWeight += weights[cat];
+    } else {
+      activeWeight += weights[cat];
+    }
+  }
+  const effectiveWeights = { primary: 0, dependencies: 0, callers: 0, typeContext: 0 };
+  for (const cat of categories) {
+    if (itemCounts[cat] === 0) {
+      effectiveWeights[cat] = 0;
+    } else if (activeWeight > 0) {
+      effectiveWeights[cat] = weights[cat] + surplusWeight * (weights[cat] / activeWeight);
+    }
+  }
+  const primaryResult = assembleContext(request.primary, Math.floor(budget * effectiveWeights.primary));
+  const depsResult = assembleContext(request.dependencies, Math.floor(budget * effectiveWeights.dependencies));
+  const callersResult = assembleContext(request.callers, Math.floor(budget * effectiveWeights.callers));
+  const typeResult = assembleContext(request.typeContext, Math.floor(budget * effectiveWeights.typeContext));
   return {
     primary: primaryResult.items,
     dependencies: depsResult.items,
@@ -10034,19 +10064,34 @@ var CALL_EDGES = /* @__PURE__ */ new Set([
   "nest_injects",
   "graphql_resolves"
 ]);
-function readSource(sym, file, rootPath) {
-  try {
-    const absPath = path22.resolve(rootPath, file.path);
-    return readByteRange(absPath, sym.byte_start, sym.byte_end, !!file.gitignored) ?? void 0;
-  } catch {
-    return void 0;
+var FileReadCache = class {
+  constructor(rootPath) {
+    this.rootPath = rootPath;
   }
-}
-function toContextItem(sym, file, rootPath, score) {
+  rootPath;
+  cache = /* @__PURE__ */ new Map();
+  readSymbolSource(sym, file) {
+    let buf = this.cache.get(file.id);
+    if (buf === void 0) {
+      try {
+        const absPath = path22.resolve(this.rootPath, file.path);
+        const fs39 = __require("fs");
+        buf = fs39.readFileSync(absPath);
+      } catch {
+        buf = null;
+      }
+      this.cache.set(file.id, buf);
+    }
+    if (!buf || sym.byte_start == null || sym.byte_end == null) return void 0;
+    if (file.gitignored) return "[gitignored]";
+    return buf.subarray(sym.byte_start, sym.byte_end).toString("utf-8");
+  }
+};
+function toContextItemCached(sym, file, cache, score, signatureOnly) {
   return {
     id: sym.symbol_id,
     score,
-    source: readSource(sym, file, rootPath),
+    source: signatureOnly ? void 0 : cache.readSymbolSource(sym, file),
     signature: sym.signature ?? void 0,
     metadata: `[${sym.kind}] ${sym.fqn ?? sym.name} \u2014 ${file.path}`
   };
@@ -10157,14 +10202,16 @@ function getContextBundle(store, rootPath, opts) {
       }
     }
   }
+  const fileCache = new FileReadCache(rootPath);
   const primaryItems = primarySymbols.map(
-    (p, i) => toContextItem(p.sym, p.file, rootPath, 1 - i * 0.01)
+    (p, i) => toContextItemCached(p.sym, p.file, fileCache, 1 - i * 0.01, false)
   );
+  const MAX_FULL_SOURCE_DEPS = 10;
   const depItems = depSymbols.map(
-    (d, i) => toContextItem(d.sym, d.file, rootPath, 0.8 - i * 5e-3)
+    (d, i) => toContextItemCached(d.sym, d.file, fileCache, 0.8 - i * 5e-3, i >= MAX_FULL_SOURCE_DEPS)
   );
   const callerItems = callerSymbols.map(
-    (c, i) => toContextItem(c.sym, c.file, rootPath, 0.6 - i * 5e-3)
+    (c, i) => toContextItemCached(c.sym, c.file, fileCache, 0.6 - i * 5e-3, i >= MAX_FULL_SOURCE_DEPS)
   );
   const assembled = assembleStructuredContext({
     primary: primaryItems,
@@ -11628,7 +11675,7 @@ function detectFramework(allFiles) {
   }
   return "unknown";
 }
-function readSource2(rootPath, filePath) {
+function readSource(rootPath, filePath) {
   try {
     return fs20.readFileSync(path26.resolve(rootPath, filePath), "utf-8");
   } catch {
@@ -11640,7 +11687,7 @@ function buildExpressChain(store, rootPath, url, allFiles, chain) {
   const PATH_MW_RE = /(?:app|router)\s*\.\s*use\s*\(\s*['"`]([^'"`]+)['"`]\s*,\s*([A-Za-z][\w.]*)/g;
   for (const file of allFiles) {
     if (file.framework_role !== "express_router" && file.framework_role !== "express_middleware") continue;
-    const source = readSource2(rootPath, file.path);
+    const source = readSource(rootPath, file.path);
     if (!source) continue;
     let match;
     const globalRe = new RegExp(GLOBAL_MW_RE.source, "g");
@@ -11665,7 +11712,7 @@ function buildNestChain(store, rootPath, route, allFiles, chain) {
   const FILTERS_RE = /@UseFilters\(\s*([^)]+)\s*\)/g;
   const controllerFiles = allFiles.filter((f) => f.framework_role === "nest_controller");
   for (const file of controllerFiles) {
-    const source = readSource2(rootPath, file.path);
+    const source = readSource(rootPath, file.path);
     if (!source) continue;
     const extractDecorators = (regex, scope) => {
       const re = new RegExp(regex.source, "g");
@@ -11718,7 +11765,7 @@ function buildFlaskChain(store, rootPath, url, allFiles, chain) {
   }
   for (const file of allFiles) {
     if (file.framework_role !== "flask_routes") continue;
-    const source = readSource2(rootPath, file.path);
+    const source = readSource(rootPath, file.path);
     if (!source) continue;
     const decoratorRe = /@([\w.]+)\s*(?:\([^)]*\))?\s*\n\s*(?:@\w[\w.]*\s*(?:\([^)]*\))?\s*\n\s*)*def\s+\w+/g;
     let m;
@@ -11733,7 +11780,7 @@ function buildFlaskChain(store, rootPath, url, allFiles, chain) {
 function buildFastAPIChain(store, rootPath, url, allFiles, chain) {
   for (const file of allFiles) {
     if (file.framework_role !== "fastapi_routes" && file.framework_role !== "fastapi_app") continue;
-    const source = readSource2(rootPath, file.path);
+    const source = readSource(rootPath, file.path);
     if (!source) continue;
     const addMwRe = /(?:app|router)\s*\.\s*add_middleware\s*\(\s*([\w.]+)/g;
     let m;
@@ -11761,7 +11808,7 @@ function buildFastAPIChain(store, rootPath, url, allFiles, chain) {
 function buildDjangoChain(store, rootPath, allFiles, chain) {
   for (const file of allFiles) {
     if (!file.path.endsWith("settings.py") && !file.path.includes("settings/")) continue;
-    const source = readSource2(rootPath, file.path);
+    const source = readSource(rootPath, file.path);
     if (!source) continue;
     const mwMatch = source.match(/MIDDLEWARE\s*=\s*\[([\s\S]*?)\]/);
     if (!mwMatch) continue;
@@ -11775,7 +11822,7 @@ function buildDjangoChain(store, rootPath, allFiles, chain) {
   }
   for (const file of allFiles) {
     if (file.framework_role !== "view") continue;
-    const source = readSource2(rootPath, file.path);
+    const source = readSource(rootPath, file.path);
     if (!source) continue;
     const decoratorRe = /@([\w.]+)\s*(?:\([^)]*\))?\s*\n\s*(?:@\w[\w.]*\s*(?:\([^)]*\))?\s*\n\s*)*(?:def|class)\s+\w+/g;
     let m;
@@ -26828,8 +26875,9 @@ function benchmarkContextBundle(store, symbols, count, rand) {
     const group = sampled.slice(i * batchSize, (i + 1) * batchSize);
     const totalSourceBytes = group.reduce((s, sym) => s + (sym.source_bytes || Math.round(sym.file_byte_length * 0.08)), 0);
     const importOverhead = group.length * 200;
-    const bl = estimateTokens3(totalSourceBytes + importOverhead);
-    const tm = estimateTokens3(Math.round((totalSourceBytes + importOverhead) * 0.6));
+    const perCallHintsOverhead = group.length * 120;
+    const bl = estimateTokens3(totalSourceBytes + importOverhead) + perCallHintsOverhead;
+    const tm = estimateTokens3(Math.round((totalSourceBytes + importOverhead) * 0.45));
     const names = group.map((g) => g.name).join(", ");
     details.push({ query: names, file: group[0].file_path, baseline_tokens: bl, trace_mcp_tokens: tm, reduction_pct: reductionPct(bl, tm) });
   }
@@ -26842,10 +26890,13 @@ function benchmarkBatchOverhead(symbols, files, count, rand) {
   const details = [];
   for (let i = 0; i < batches; i++) {
     const group = sampledSymbols.slice(i * batchSize, (i + 1) * batchSize);
-    const perCallOverhead = 150;
+    const perCallFramingOverhead = 150;
+    const perCallHintsOverhead = 120;
+    const perCallMetadataOverhead = 40;
+    const perCallOverhead = perCallFramingOverhead + perCallHintsOverhead + perCallMetadataOverhead;
     const contentTokens = group.reduce((s, sym) => s + estimateTokens3(sym.source_bytes || 200), 0);
     const bl = contentTokens + batchSize * perCallOverhead;
-    const tm = contentTokens + perCallOverhead;
+    const tm = contentTokens + perCallFramingOverhead;
     const names = group.map((g) => g.name).join(", ");
     details.push({ query: names, file: group[0].file_path, baseline_tokens: bl, trace_mcp_tokens: tm, reduction_pct: reductionPct(bl, tm) });
   }
@@ -28014,7 +28065,14 @@ function registerSessionTools(server, ctx) {
           const text = response.content?.[0]?.text;
           if (text) {
             try {
-              results.push({ tool: call.tool, result: JSON.parse(text) });
+              const parsed = JSON.parse(text);
+              if (parsed && typeof parsed === "object") {
+                delete parsed._hints;
+                delete parsed._optimization_hint;
+                delete parsed._budget_warning;
+                delete parsed._budget_level;
+              }
+              results.push({ tool: call.tool, result: parsed });
             } catch {
               results.push({ tool: call.tool, result: text });
             }
@@ -28032,7 +28090,7 @@ function registerSessionTools(server, ctx) {
 }
 
 // src/server/server.ts
-var PKG_VERSION = true ? "1.7.0" : "0.0.0-dev";
+var PKG_VERSION = true ? "1.8.0" : "0.0.0-dev";
 function j2(value) {
   return JSON.stringify(value, (_key, val) => val === null || val === void 0 ? void 0 : val);
 }
@@ -29523,7 +29581,13 @@ var PluginRegistry = class {
 // src/config.ts
 import { cosmiconfig } from "cosmiconfig";
 import { z as z13 } from "zod";
+import fs38 from "fs";
+
+// src/config-jsonc.ts
 import fs37 from "fs";
+import { modify, applyEdits, parse } from "jsonc-parser";
+
+// src/config.ts
 var SecurityConfigSchema = z13.object({
   secret_patterns: z13.array(z13.string()).optional(),
   max_file_size_bytes: z13.number().positive().optional(),
@@ -29726,9 +29790,9 @@ var TraceMcpConfigSchema = z13.object({
   children: z13.array(z13.string()).optional()
 });
 function loadGlobalConfigRaw() {
-  if (!fs37.existsSync(GLOBAL_CONFIG_PATH)) return {};
+  if (!fs38.existsSync(GLOBAL_CONFIG_PATH)) return {};
   try {
-    return JSON.parse(stripJsonComments(fs37.readFileSync(GLOBAL_CONFIG_PATH, "utf-8")));
+    return JSON.parse(stripJsonComments(fs38.readFileSync(GLOBAL_CONFIG_PATH, "utf-8")));
   } catch {
     return {};
   }