@goodandready/dsh-context-lens 0.1.7 → 0.1.8

package/README.md

@@ -65,12 +65,15 @@ graph LR
 ## ✨ Key Capabilities & Modules
 
 ### 1. 🧬 Multi-Language AST Code Skeletonizer (`lib/ast/skeletonizer.js`)
-* Automatically extracts structural interfaces, function signatures, classes, types, and exports across **TypeScript, JavaScript, Python, and Go**;
+* Automatically extracts structural interfaces, function signatures, classes, types, and exports across **TypeScript, JavaScript, Python, Go, Rust, and Java**;
+* Supports `pub async fn`, `async fn`, `pub(crate)` and `pub(super)` in Rust;
+* Retains JSDoc, docstrings, and structural comments preceding definitions;
 * Drops internal function implementations, loops, and repetitive boilerplate while preserving indentation and export declarations;
 * Allows the agent to understand entire multi-package repository architectures without loading tens of thousands of implementation tokens.
 
 ### 2. 🗜️ Fast Heuristic Log Condenser (`lib/compression/log-compressor.js`)
-* High-performance $O(n)$ heuristic line filter for test runners and build tools (Jest, Vitest, Pytest, Go test, NPM, Webpack, Cargo);
+* High-performance $O(n)$ heuristic line filter for test runners and build tools (Jest, Vitest, Pytest, Go test, NPM, Webpack, Cargo, Maven/Gradle);
+* Strips terminal ANSI escape color sequences before evaluating regex patterns;
 * Automatically filters out passing test noise (`PASS`, `✓`, `ok`) and build notices;
 * Retains critical error lines, stack traces, assertion failures (`Expected ... Received ...`), and failure context windows;
 * 3 Aggressiveness Modes: `raw`, `balanced`, and `aggressive`.
@@ -81,7 +84,8 @@ graph LR
 
 ### 4. 📊 Token Savings Tracking & Dashboard (`lib/tokens/tracker.js` & `lib/client.js`)
 * Measures exact token counts before and after compression;
-* Calculates cumulative session token savings and displays live efficiency percentage badges in the DSH interface.
+* Calculates cumulative session token savings and displays live efficiency percentage badges in the DSH interface;
+* Enforces Token Budget Guard limits with alert warnings when exceeding 90% budget.
 
 ---
 
@@ -90,9 +94,9 @@ graph LR
 | Tool Name | Parameters | Description |
 |---|---|---|
 | `context_lens_focus` | `paths: string[]`, `maxDepth?: number` | Designates active focus files/folders; collapses surrounding workspace into AST skeletons |
-| `context_lens_compress_log` | `log: string`, `mode?: "raw"\|"balanced"\|"aggressive"` | Condenses terminal/test outputs, keeping only stack traces and failure windows |
-| `context_lens_compress_code` | `code: string`, `language?: string`, `maxDepth?: number` | Generates a clean structural AST skeleton from raw source code |
-| `context_lens_stats` | *(none)* | Returns real-time cumulative token savings, original token count, and efficiency % |
+| `context_lens_compress_log` | `text: string` *(or `log`)*, `mode?: "raw"\|"balanced"\|"aggressive"`, `maxLines?: number`, `auto?: boolean` | Condenses terminal/test outputs, keeping only stack traces and failure windows |
+| `context_lens_compress_code` | `code: string`, `language?: string`, `maxDepth?: number`, `filePath?: string` | Generates a clean structural AST skeleton from raw source code |
+| `context_lens_stats` | *(none)* | Returns real-time cumulative token savings, history, and budget status |
 
 ---
 
@@ -111,13 +115,27 @@ dsh plugin --profile web add @goodandready/dsh-context-lens
 
 ```yaml
 dsh-context-lens:
-  compressionMode: balanced      # 'raw', 'balanced', or 'aggressive'
-  astSkeletonMaxDepth: 3        # Maximum depth level for AST signature traversal
-  tokenSavingsTracking: true    # Track and display live token savings
+  compressionMode: balanced        # 'raw', 'balanced', or 'aggressive'
+  astSkeletonMaxDepth: 3          # Maximum depth level for AST signature traversal (1..10)
+  tokenSavingsTracking: true      # Track and display live token savings
+  autoCompressThreshold: 4000     # Auto-compression character threshold (0 to disable)
+  budgetLimit: 100000             # Session token budget limit
+  autoCollapse: true              # Auto-collapse UI when budget is nearly exhausted
 ```
 
 ---
 
+## 📝 Version History
+
+### v0.1.8
+* **Fix**: Support both `text` and `log` parameter names in `context_lens_compress_log`.
+* **Fix**: Cross-platform path resolution in unit tests on Windows (`fileURLToPath`).
+* **Fix**: Dynamic propagation of `budgetLimit` configuration into token tracker.
+* **Fix**: ANSI terminal escape sequence stripping for colored logs.
+* **Fix**: Expanded Rust syntax support (`pub async fn`, `pub(crate)`) and proper `#` comment prefix for Python.
+
+---
+
 ## 📄 License
 
 MIT © [GooDAnDReaDY](https://github.com/GooDAnDReaDY)

package/README.ru.md

@@ -65,13 +65,16 @@ graph LR
 ## ✨ Ключевые возможности
 
 ### 1. 🧬 Мультиязычный AST-скелетонизатор кода (`lib/ast/skeletonizer.js`)
-* Извлекает структурные интерфейсы, сигнатуры функций, методы классов, типы и экспорты для **TypeScript, JavaScript, Python и Go**;
+* Извлекает структурные интерфейсы, сигнатуры функций, методы классов, типы и экспорты для **TypeScript, JavaScript, Python, Go, Rust и Java**;
+* Поддерживает `pub async fn`, `async fn`, `pub(crate)` и `pub(super)` в Rust;
+* Сохраняет JSDoc, docstrings и структурные комментарии над определениями;
 * Удаляет внутренние тела функций и комментарии реализации, оставляя точную архитектуру файла;
 * Позволяет агенту обозревать всю структуру проекта без загрузки лишних десятков тысяч токенов.
 
 ### 2. 🗜️ Быстрый компрессор логов (`lib/compression/log-compressor.js`)
-* Высокопроизводительный $O(n)$ фильтр шума логов тестирования и сборки (Jest, Vitest, Pytest, Go test, NPM, Webpack, Cargo);
-* Автоматически вырезает успешные проверки (`PASS`, `✓`, `ok`) и служебные уведомления;
+* Высокопроизводительный $O(n)$ фильтр шума логов тестирования и сборки (Jest, Vitest, Pytest, Go test, NPM, Webpack, Cargo, Maven/Gradle);
+* Автоматическая очистка терминальных ANSI-эскейп последовательностей;
+* Вырезает успешные проверки (`PASS`, `✓`, `ok`) и служебные уведомления;
 * Сохраняет строки падений, стек-трейсы, расхождения в утверждениях (`Expected ... Received ...`) и контекстное окружение ошибки;
 * 3 Режима сжатия: `raw`, `balanced`, `aggressive`.
 
@@ -81,7 +84,8 @@ graph LR
 
 ### 4. 📊 Трекер экономии токенов (`lib/tokens/tracker.js` & `lib/client.js`)
 * Фиксация точного числа токенов до и после сжатия;
-* Подсчёт накопленной экономии за сессию с отображением процента эффективности в Web UI.
+* Подсчёт накопленной экономии за сессию с отображением процента эффективности в Web UI;
+* Контроль сессионного бюджета токенов (Token Budget Guard) с предупреждением при превышении 90%.
 
 ---
 
@@ -90,9 +94,9 @@ graph LR
 | Имя инструмента | Параметры | Описание |
 |---|---|---|
 | `context_lens_focus` | `paths: string[]`, `maxDepth?: number` | Задаёт пути активного фокуса; всё остальное сворачивается в AST-скелеты |
-| `context_lens_compress_log` | `log: string`, `mode?: "raw"\|"balanced"\|"aggressive"` | Сжимает вывод тестов и терминала, сохраняя стек-трейсы и ошибки |
-| `context_lens_compress_code` | `code: string`, `language?: string`, `maxDepth?: number` | Генерирует структурный AST-скелет из переданного исходного кода |
-| `context_lens_stats` | *(нет)* | Возвращает метрики сэкономленных токенов и процент эффективности |
+| `context_lens_compress_log` | `text: string` *(или `log`)*, `mode?: "raw"\|"balanced"\|"aggressive"`, `maxLines?: number`, `auto?: boolean` | Сжимает вывод тестов и терминала, сохраняя стек-трейсы и ошибки |
+| `context_lens_compress_code` | `code: string`, `language?: string`, `maxDepth?: number`, `filePath?: string` | Генерирует структурный AST-скелет из переданного исходного кода |
+| `context_lens_stats` | *(нет)* | Возвращает метрики сэкономленных токенов, историю и статус бюджета |
 
 ---
 
@@ -108,13 +112,27 @@ dsh plugin --profile web add @goodandready/dsh-context-lens
 
 ```yaml
 dsh-context-lens:
-  compressionMode: balanced      # 'raw', 'balanced' или 'aggressive'
-  astSkeletonMaxDepth: 3        # Максимальная глубина обхода AST-сигнатур
-  tokenSavingsTracking: true    # Включить трекинг сэкономленных токенов
+  compressionMode: balanced        # 'raw', 'balanced' или 'aggressive'
+  astSkeletonMaxDepth: 3          # Максимальная глубина обхода AST-сигнатур (1..10)
+  tokenSavingsTracking: true      # Включить трекинг сэкономленных токенов
+  autoCompressThreshold: 4000     # Порог авто-сжатия в символах (0 для отключения)
+  budgetLimit: 100000             # Сессионный лимит бюджета токенов
+  autoCollapse: true              # Автоматически сворачивать UI при исчерпании бюджета
 ```
 
 ---
 
+## 📝 История версий
+
+### v0.1.8
+* **Fix**: Поддержка как `text`, так и `log` в параметрах инструмента `context_lens_compress_log`.
+* **Fix**: Кроссплатформенное разрешение путей в тестах на Windows (`fileURLToPath`).
+* **Fix**: Динамическая передача и учёт `budgetLimit` в трекере токенов.
+* **Fix**: Корректная обработка цветных логов с терминальными ANSI-кодами.
+* **Fix**: Расширена поддержка Rust (`pub async fn`, `pub(crate)`) и правильные комментарии `#` для Python в AST-скелетонизаторе.
+
+---
+
 ## 📄 Лицензия
 
 MIT © [GooDAnDReaDY](https://github.com/GooDAnDReaDY)

package/lib/ast/skeletonizer.js

@@ -2,8 +2,9 @@
 const JS_FUNC_RE = /^\s*(export\s+)?(async\s+)?(function\s+(\w+)|const\s+(\w+)\s*=\s*(async\s+)?\([^)]*\)\s*=>|(\w+)\s*:\s*\([^)]*\)\s*=>|class\s+(\w+)|interface\s+(\w+)|type\s+(\w+)\s*=)/;
 const PY_RE = /^\s*(def\s+(\w+)\s*\([^)]*\)|class\s+(\w+).*?:|async def\s+(\w+)\s*\([^)]*\))/;
 const GO_RE = /^\s*(func\s+(\([^)]+\)\s+)?(\w+)\s*\([^)]*\)|type\s+(\w+)\s+(struct|interface))/;
-const IMPORT_RE = /^\s*import\s+.*from\s+['"].*['"]|^\s*import\s+['"].*['"]|^\s*export\s+.*from\s+['"]/;
-const COMMENT_RE = /^\s*(\/\/.*|\/\*.*\*\/|#.*)/;
+const RUST_RE = /^\s*(pub(\([^)]+\))?\s+)?(async\s+)?(fn|struct|enum|impl|trait|type|const|static)\s+\w+/;
+const IMPORT_RE = /^\s*import\s+.*from\s+['"].*['"]|^\s*import\s+['"].*['"]|^\s*export\s+.*from\s+['"]|^\s*use\s+[a-zA-Z0-9_:]+/;
+const COMMENT_RE = /^\s*(\/\/.*|\/\*.*\*\/|\/\*.*|\*.*|#.*)/;
 
 function indentDepth(line) {
   const m = line.match(/^(\s*)/);
@@ -39,13 +40,13 @@ export function skeletonize(text, { maxDepth = 3, language, includeImports = tru
       while (nextIdx < lines.length && !lines[nextIdx].trim()) nextIdx++;
       if (nextIdx < lines.length) {
         const nextLine = lines[nextIdx];
-        if (JS_FUNC_RE.test(nextLine) || PY_RE.test(nextLine) || GO_RE.test(nextLine) || /^\s*(class|function|interface|type|def |func )/.test(nextLine)) {
+        if (JS_FUNC_RE.test(nextLine) || PY_RE.test(nextLine) || GO_RE.test(nextLine) || RUST_RE.test(nextLine) || /^\s*(class|function|interface|type|def |func |pub |fn )/.test(nextLine)) {
           pendingComment = line.trim().slice(0, 120);
           continue;
         }
       }
       // Otherwise, keep top-level comments (up to 5)
-      if (out.length < 5 && line.trim().startsWith('//')) {
+      if (out.length < 5 && (line.trim().startsWith('//') || line.trim().startsWith('#') || line.trim().startsWith('/*'))) {
         const sig = line.trim().slice(0, 120);
         if (!seen.has(sig)) {
           seen.add(sig);
@@ -57,9 +58,9 @@ export function skeletonize(text, { maxDepth = 3, language, includeImports = tru
     }
     let sig = null;
     // try JS/TS
-    if (!language || language === 'js' || language === 'ts' || language === 'rust' || language === 'java') {
+    if (!language || language === 'js' || language === 'ts') {
       const m = line.match(/^\s*(export\s+)?(async\s+)?(function\s+\w+[^\n]*|const\s+\w+\s*=.*=>.*|class\s+\w+.*|interface\s+\w+.*|type\s+\w+\s*=.*|(?:public|private|protected)?\s*(async\s+)?\w+\s*\([^)]*\)\s*[:{]|import\s+.*|export\s+.*)/);
-      if (m) sig = line.trim().replace(/\s*\{.*$/, '').replace(/\s+$/, '') + (line.trim().endsWith('{') ? '' : '');
+      if (m) sig = line.trim().replace(/\s*\{.*$/, '').replace(/\s+$/, '');
       if (!sig && /^\s*(export\s+)?(async\s+)?function\s+\w+/.test(line)) sig = line.trim();
       if (!sig && /^\s*class\s+\w+/.test(line)) sig = line.trim();
       if (!sig && /^\s*import\s+/.test(line)) sig = line.trim().slice(0, 120);
@@ -72,14 +73,15 @@ export function skeletonize(text, { maxDepth = 3, language, includeImports = tru
       const m = line.match(GO_RE);
       if (m) sig = line.trim();
     }
-    // Rust/Java fallback
-    if (!sig && (language === 'rust' || language === 'java')) {
-      if (/^\s*(pub\s+)?(fn|struct|enum|impl|trait)\s+\w+/.test(line)) sig = line.trim();
-      if (/^\s*(public\s+)?(class|interface|enum)\s+\w+/.test(line)) sig = line.trim();
+    if (!sig && (!language || language === 'rust')) {
+      if (RUST_RE.test(line)) sig = line.trim();
+    }
+    if (!sig && (!language || language === 'java')) {
+      if (/^\s*(public|protected|private|static|\s)+\s+(class|interface|enum|record|\w+)\s+\w+/.test(line)) sig = line.trim();
     }
     // generic fallback: if no specific language, try all
     if (!sig && !language) {
-      if (JS_FUNC_RE.test(line) || PY_RE.test(line) || GO_RE.test(line)) sig = line.trim();
+      if (JS_FUNC_RE.test(line) || PY_RE.test(line) || GO_RE.test(line) || RUST_RE.test(line)) sig = line.trim();
     }
     if (sig) {
       // If we have a pending comment, prepend it
@@ -112,7 +114,8 @@ export function skeletonize(text, { maxDepth = 3, language, includeImports = tru
   const total = lines.filter(l => l.trim()).length;
   const removed = total - out.length;
   if (removed > 0 && out.length > 0) {
-    out.push(`// ... ${removed} lines collapsed (skeleton, maxDepth=${maxDepth})`);
+    const commentPrefix = (language === 'py' || language === 'python') ? '# ' : '// ';
+    out.push(`${commentPrefix}... ${removed} lines collapsed (skeleton, maxDepth=${maxDepth})`);
   }
   return out.join('\n');
 }

package/lib/client.js

@@ -76,8 +76,7 @@ window.__ModuleLoader__.load({
           ...history.slice(0, 3).map((h) => React.createElement('div', { key: h.id, style: { fontSize: 11, color: 'var(--dsw-alias-label-secondary)', padding: '2px 0' } }, `−${h.savedTokens} tk (${h.savedPercent}%) · ${(h.preview || '').slice(0, 40)}`))
         ) : null,
         React.createElement('div', { style: { marginTop: 12, display: 'flex', gap: 8 } },
-          React.createElement('button', { onClick: refresh, style: { fontSize: 12, padding: '4px 8px', borderRadius: 6, border: '1px solid var(--dsw-alias-border-l2)', background: 'var(--dsw-alias-bg-layer-3)', cursor: 'pointer' } }, 'Refresh'),
-          React.createElement('button', { onClick: () => { try { _ctx.betterSidebar.openTab({ type: 'dsh-context-lens:tab', title: 'Lens' }) } catch(e){} }, style: { fontSize: 12, padding: '4px 8px', borderRadius: 6, border: '1px solid var(--dsw-alias-border-l2)', background: 'transparent', cursor: 'pointer' } }, 'Settings')
+          React.createElement('button', { onClick: refresh, style: { fontSize: 12, padding: '4px 10px', borderRadius: 6, border: '1px solid var(--dsw-alias-border-l2)', background: 'var(--dsw-alias-bg-layer-3)', color: 'var(--dsw-alias-label-primary)', cursor: 'pointer' } }, 'Refresh')
         )
       );
     }

package/lib/compression/log-compressor.js

@@ -1,13 +1,17 @@
 // ponytail: heuristic line filter, not ML — O(n) scan, no deps
+const ANSI_RE = /\u001b\[[0-9;]*[a-zA-Z]/g;
 const KEEP_RE = /(FAIL|FAILED|Error|AssertionError|Exception|Traceback|panic|npm ERR!|ERR!|Expected|Received|missing|×|●|✕|FAIL:|--- FAIL|not ok|at\s+.*:\d+:\d+|stack|Caused by|test result:\s*FAILED|running \d+ test|test .* \.\.\. FAILED|BUILD FAILED|Tests run:|FAILURE:|cargo:.*error|error\[E\d+\]|thread '.*' panicked)/i;
 const PASS_RE = /^(PASS|\s*✓|\s*✔|\s*ok\s|…+\s*$|\s*\.\s*$|test result:\s*ok|running \d+ test.*ok)/i;
 const NOISE_RE = /^\s*(npm (notice|warn|info)|Browserslist|cached|Downloading|Done in|Compiling\s|cargo:.*Finished)/i;
 const STACK_RE = /^\s*(at\s+|File ".*", line \d+|#\d+\s+0x|goroutine \d+ \[|thread '.*' panicked|note: run with)/;
 
-function keepLine(line, mode) {
-  if (KEEP_RE.test(line) || STACK_RE.test(line)) return true;
-  if (mode === 'raw') return !(PASS_RE.test(line) || NOISE_RE.test(line)) ? true : false;
-  // balanced/aggressive rely on context window, not bare line
+function cleanAnsi(str) {
+  return (str || '').replace(ANSI_RE, '');
+}
+
+function keepLine(line) {
+  const clean = cleanAnsi(line);
+  if (KEEP_RE.test(clean) || STACK_RE.test(clean)) return true;
   return false;
 }
 
@@ -20,8 +24,13 @@ export function compressLog(text, { mode = 'balanced', maxLines = 400 } = {}) {
   const lines = text.split(/\r?\n/);
   const totalLines = lines.length;
   if (mode === 'raw') {
-    // raw: drop only obvious noise, keep rest
-    const filtered = lines.filter((l) => !NOISE_RE.test(l) || KEEP_RE.test(l));
+    // raw: drop only obvious noise and non-failure passes, keep rest
+    const filtered = lines.filter((l) => {
+      const clean = cleanAnsi(l);
+      if (KEEP_RE.test(clean) || STACK_RE.test(clean)) return true;
+      if (NOISE_RE.test(clean)) return false;
+      return true;
+    });
     const truncated = filtered.length > maxLines ? [...filtered.slice(0, maxLines - 1), `… truncated ${filtered.length - maxLines + 1} lines`] : filtered;
     const compressed = truncated.join('\n');
     const originalTokens = estimateTokens(text);
@@ -32,7 +41,7 @@ export function compressLog(text, { mode = 'balanced', maxLines = 400 } = {}) {
   const keep = new Array(lines.length).fill(false);
   const context = mode === 'aggressive' ? 1 : 2;
   for (let i = 0; i < lines.length; i++) {
-    if (keepLine(lines[i], mode)) {
+    if (keepLine(lines[i])) {
       const s = Math.max(0, i - context);
       const e = Math.min(lines.length - 1, i + context);
       for (let j = s; j <= e; j++) keep[j] = true;
@@ -50,8 +59,9 @@ export function compressLog(text, { mode = 'balanced', maxLines = 400 } = {}) {
   let out = [];
   for (let i = 0; i < lines.length; i++) if (keep[i]) {
     const l = lines[i];
+    const clean = cleanAnsi(l);
     // aggressive drops more noise even inside window
-    if (mode === 'aggressive' && (PASS_RE.test(l) || NOISE_RE.test(l)) && !KEEP_RE.test(l) && !STACK_RE.test(l)) continue;
+    if (mode === 'aggressive' && (PASS_RE.test(clean) || NOISE_RE.test(clean)) && !KEEP_RE.test(clean) && !STACK_RE.test(clean)) continue;
     out.push(l);
   }
 

package/lib/index.js

@@ -67,28 +67,29 @@ export function apply(ctx, config) {
 
     ctx.tools.register({
       name: 'context_lens_compress_log',
-      description: 'Compress a large log/test output block, keeping failures and stacktraces (Jest/Pytest/Go/nnpm)',
+      description: 'Compress a large log/test output block, keeping failures and stacktraces (Jest/Pytest/Go/npm)',
       parameters: {
         type: 'object',
         properties: {
           text: { type: 'string', description: 'Raw log text to compress' },
+          log: { type: 'string', description: 'Alias for text' },
           mode: { type: 'string', enum: ['raw', 'balanced', 'aggressive'], description: 'Compression aggressiveness' },
           maxLines: { type: 'number', description: 'Max output lines' },
           auto: { type: 'boolean', description: 'Auto-compress if large (uses threshold)' }
-        },
-        required: ['text']
+        }
       },
       output: { schema: OUTPUT_SCHEMA, render: renderOutput },
       execute: async (params) => {
         const cfg = getConfig();
+        const inputText = params.text || params.log || '';
         const mode = params.mode || cfg.compressionMode || 'balanced';
         const maxLines = params.maxLines || 400;
         // Auto-compress check (#15)
         const threshold = cfg.autoCompressThreshold ?? 4000;
-        const useAuto = params.auto !== false && shouldAutoCompress(params.text, threshold);
-        const effectiveMode = useAuto ? mode : mode;
-        const res = compressLog(params.text, { mode: effectiveMode, maxLines });
-        maybeTrack(params.text, res.compressed);
+        const useAuto = params.auto !== false && shouldAutoCompress(inputText, threshold);
+        const effectiveMode = (useAuto && !params.mode) ? 'balanced' : mode;
+        const res = compressLog(inputText, { mode: effectiveMode, maxLines });
+        maybeTrack(inputText, res.compressed);
         return { success: true, mode: effectiveMode, autoCompressed: useAuto, ...res };
       }
     });
@@ -128,9 +129,9 @@ export function apply(ctx, config) {
       parameters: { type: 'object', properties: {} },
       output: { schema: OUTPUT_SCHEMA, render: renderOutput },
       execute: async () => {
-        const stats = tracker.getStats();
-        const history = tracker.getHistory();
         const cfg = getConfig();
+        const stats = tracker.getStats(cfg.budgetLimit);
+        const history = tracker.getHistory();
         return { success: true, ...stats, history, trackingEnabled: cfg.tokenSavingsTracking !== false, autoCollapse: cfg.autoCollapse !== false, focus: focusState, autoCompressThreshold: cfg.autoCompressThreshold };
       }
     });
@@ -142,8 +143,9 @@ export function apply(ctx, config) {
       kind: 'exact',
       path: '/dsh-context-lens/status',
       handler: (req, res) => {
+        const cfg = getConfig();
         res.setHeader('content-type', 'application/json');
-        res.end(JSON.stringify({ ok: true, plugin: 'dsh-context-lens', stats: tracker.getStats(), history: tracker.getHistory(), focus: focusState }));
+        res.end(JSON.stringify({ ok: true, plugin: 'dsh-context-lens', stats: tracker.getStats(cfg.budgetLimit), history: tracker.getHistory(), focus: focusState }));
       }
     }), 'dsh-context-lens status route');
   }

package/lib/tokens/tracker.js

@@ -4,12 +4,18 @@ let totalCompressed = 0;
 let calls = 0;
 const history = []; // last 10
 const MAX_HISTORY = 10;
-const BUDGET_LIMIT = 100000; // tokens, ~100k default budget
+let defaultBudgetLimit = 100000; // tokens, ~100k default budget
 
 export function estimateTokens(text) {
   return Math.ceil((text || '').length / 4);
 }
 
+export function setBudgetLimit(limit) {
+  if (typeof limit === 'number' && limit > 0) {
+    defaultBudgetLimit = limit;
+  }
+}
+
 export function record(originalText, compressedText) {
   const o = estimateTokens(originalText);
   const c = estimateTokens(compressedText);
@@ -31,14 +37,28 @@ export function record(originalText, compressedText) {
   return { originalTokens: o, compressedTokens: c, savedTokens: saved };
 }
 
-export function getStats() {
+export function getStats(budgetLimitOverride) {
+  const limit = (typeof budgetLimitOverride === 'number' && budgetLimitOverride > 0)
+    ? budgetLimitOverride
+    : defaultBudgetLimit;
   const saved = Math.max(0, totalOriginal - totalCompressed);
   const pct = totalOriginal ? Math.round((1 - totalCompressed / totalOriginal) * 100) : 0;
   const budgetUsed = totalCompressed;
-  const budgetRemaining = Math.max(0, BUDGET_LIMIT - budgetUsed);
-  const budgetPercent = Math.min(100, Math.round((budgetUsed / BUDGET_LIMIT) * 100));
+  const budgetRemaining = Math.max(0, limit - budgetUsed);
+  const budgetPercent = Math.min(100, Math.round((budgetUsed / limit) * 100));
   const lowBudget = budgetPercent > 90;
-  return { totalOriginal, totalCompressed, savedTokens: saved, savedPercent: pct, calls, budgetUsed, budgetRemaining, budgetPercent, lowBudget, budgetLimit: BUDGET_LIMIT };
+  return {
+    totalOriginal,
+    totalCompressed,
+    savedTokens: saved,
+    savedPercent: pct,
+    calls,
+    budgetUsed,
+    budgetRemaining,
+    budgetPercent,
+    lowBudget,
+    budgetLimit: limit
+  };
 }
 
 export function getHistory() {
@@ -54,6 +74,7 @@ export function reset() {
   totalCompressed = 0;
   calls = 0;
   history.length = 0;
+  defaultBudgetLimit = 100000;
 }
 
-export default { estimateTokens, record, getStats, getHistory, clearHistory, reset };
+export default { estimateTokens, setBudgetLimit, record, getStats, getHistory, clearHistory, reset };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-context-lens",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "DSH plugin for AST context compression, test log filtering, and token budget guard",
   "type": "module",
   "main": "lib/index.js",