@esneiderbravo/speclaw 2.0.11 → 2.0.13

package/dist/shared/output-budget.js

@@ -3,26 +3,43 @@ export const OUTPUT_BUDGET = {
     brief: 1500,
     full: 4500,
 };
+/** Brief-mode suffix: the caller can still widen the response with `mode:"full"`. */
 const TRUNCATION_SUFFIX = `\n… [truncated — use mode:"full" or narrow includes]`;
+/** Suffix under the full ceiling or an explicit cap: widening the mode would not help. */
+const FINAL_TRUNCATION_SUFFIX = `\n… [truncated — narrow includes or the query]`;
 /** Stable offline token estimate for budgeting (not a real tokenizer). */
 export function estimateTokens(text) {
     return Math.ceil(text.length / 4);
 }
 /**
- * Trim `text` to fit `mode` budget; returns the possibly shortened text and
- * whether truncation occurred.
+ * Token ceiling of a {@link TextBudget}.
+ *
+ * @param budget - Output mode or explicit cap.
+ */
+export function budgetTokens(budget) {
+    return typeof budget === "string" ? OUTPUT_BUDGET[budget] : Math.max(1, budget.maxTokens);
+}
+/**
+ * Trim `text` to fit `budget`; returns the possibly shortened text and whether
+ * truncation occurred. Under `brief` the suffix suggests `mode:"full"`; under
+ * `full` or an explicit cap it never does, since that would not widen anything.
+ *
+ * @param text - Text to budget.
+ * @param budget - Output mode (`brief` default) or `{ maxTokens }`.
  */
-export function applyTextBudget(text, mode = "brief") {
-    const budget = OUTPUT_BUDGET[mode];
+// Covers: req~find-response-budget~1
+export function applyTextBudget(text, budget = "brief") {
+    const cap = budgetTokens(budget);
     const tokens = estimateTokens(text);
-    if (tokens <= budget)
+    if (tokens <= cap)
         return { text, truncated: false, omittedChars: 0 };
-    const suffixTokens = estimateTokens(TRUNCATION_SUFFIX);
-    const bodyBudget = Math.max(1, budget - suffixTokens);
+    const suffix = budget === "brief" ? TRUNCATION_SUFFIX : FINAL_TRUNCATION_SUFFIX;
+    const suffixTokens = estimateTokens(suffix);
+    const bodyBudget = Math.max(1, cap - suffixTokens);
     const maxChars = bodyBudget * 4;
     const trimmed = text.slice(0, maxChars);
     return {
-        text: trimmed + TRUNCATION_SUFFIX,
+        text: trimmed + suffix,
         truncated: true,
         omittedChars: text.length - maxChars,
     };
@@ -37,8 +54,11 @@ export function applyTextBudget(text, mode = "brief") {
  * @param truncated - Collector for explicit truncation records.
  */
 export function budgetExploreShape(value, mode, truncated) {
-    const maxCallers = mode === "full" ? 40 : 12;
-    const maxSourceLines = mode === "full" ? 120 : 40;
+    // Full mode sizes source to the full-mode token ceiling (~4500 tokens), so a
+    // typical long function arrives whole; its hints cannot point at mode:"full".
+    const full = mode === "full";
+    const maxCallers = full ? 40 : 12;
+    const maxSourceLines = full ? 400 : 40;
     const symbol = value.symbol;
     if (symbol && typeof symbol.source === "string") {
         const lines = symbol.source.split("\n");
@@ -48,7 +68,9 @@ export function budgetExploreShape(value, mode, truncated) {
             truncated.push({
                 field: "symbol.source",
                 omitted,
-                hint: 'use mode:"full" or omit source from include',
+                hint: full
+                    ? "read the remaining lines from the file by startLine/endLine"
+                    : 'use mode:"full" or omit source from include',
             });
         }
     }
@@ -63,7 +85,9 @@ export function budgetExploreShape(value, mode, truncated) {
             truncated.push({
                 field,
                 omitted,
-                hint: 'use mode:"full" or narrow with include',
+                hint: full
+                    ? "use compass_impact for the full list"
+                    : 'use mode:"full" or narrow with include',
             });
         }
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@esneiderbravo/speclaw",
-  "version": "2.0.11",
+  "version": "2.0.13",
   "publishConfig": {
     "access": "public"
   },