@traits-dev/cli 0.3.0 → 0.4.0

package/dist/traits.js

@@ -22,6 +22,8 @@ function printCompileUsage(out = process.stderr) {
       "  --model <model>           Model target (required)",
       "  --json                    Output structured JSON",
       "  --strict                  Treat warnings as compile-blocking",
+      "  --budget                  Print estimated token count (chars/4)",
+      "  --budget-limit <tokens>   Warn to stderr if estimate exceeds limit",
       "  --explain                 Include compilation trace output",
       "  --context key=value       Activate context adaptation (repeatable)",
       "  --knowledge-base-dir      Directory containing compiler pattern files",
@@ -52,6 +54,8 @@ function parseCompileArgs(args) {
     model: null,
     strict: false,
     json: false,
+    budget: false,
+    budgetLimit: null,
     explain: false,
     verbose: false,
     noColor: false,
@@ -70,6 +74,10 @@ function parseCompileArgs(args) {
       result.json = true;
       continue;
     }
+    if (arg === "--budget") {
+      result.budget = true;
+      continue;
+    }
     if (arg === "--explain") {
       result.explain = true;
       continue;
@@ -82,7 +90,7 @@ function parseCompileArgs(args) {
       result.noColor = true;
       continue;
     }
-    if (arg === "--model" || arg === "--bundled-profiles-dir" || arg === "--context" || arg === "--knowledge-base-dir") {
+    if (arg === "--model" || arg === "--bundled-profiles-dir" || arg === "--context" || arg === "--knowledge-base-dir" || arg === "--budget-limit") {
       const value = args[index + 1];
       if (!value) return { error: `Missing value for "${arg}"` };
       if (arg === "--model") {
@@ -91,6 +99,12 @@ function parseCompileArgs(args) {
         result.bundledProfilesDir = value;
       } else if (arg === "--knowledge-base-dir") {
         result.knowledgeBaseDir = value;
+      } else if (arg === "--budget-limit") {
+        const parsedBudgetLimit = Number(value);
+        if (!Number.isFinite(parsedBudgetLimit) || parsedBudgetLimit <= 0) {
+          return { error: `Invalid value for "--budget-limit": "${value}"` };
+        }
+        result.budgetLimit = Math.round(parsedBudgetLimit);
       } else {
         const parsedContext = parseContextArg(value);
         if ("error" in parsedContext) return { error: parsedContext.error };
@@ -111,8 +125,14 @@ function parseCompileArgs(args) {
   if (!result.model) {
     return { error: 'Missing required option "--model"' };
   }
+  if (result.budgetLimit != null) {
+    result.budget = true;
+  }
   return { value: result };
 }
+function estimateBudgetTokens(text) {
+  return Math.ceil(String(text ?? "").length / 4);
+}
 function runCompile(args, io = process) {
   const parsed = parseCompileArgs(args);
   if ("error" in parsed) {
@@ -156,10 +176,32 @@ function runCompile(args, io = process) {
     if (options.json) {
       io.stdout.write(`${JSON.stringify(compiled, null, 2)}
 `);
+      if (options.budget) {
+        const budgetEstimate = estimateBudgetTokens(compiled.text);
+        io.stderr.write(`Estimated token count: ${budgetEstimate}
+`);
+        if (options.budgetLimit != null && budgetEstimate > options.budgetLimit) {
+          io.stderr.write(
+            `Warning: Estimated token count ${budgetEstimate} exceeds budget limit ${options.budgetLimit}
+`
+          );
+        }
+      }
       return 0;
     }
     io.stdout.write(`${compiled.text}
 `);
+    if (options.budget) {
+      const budgetEstimate = estimateBudgetTokens(compiled.text);
+      io.stderr.write(`Estimated token count: ${budgetEstimate}
+`);
+      if (options.budgetLimit != null && budgetEstimate > options.budgetLimit) {
+        io.stderr.write(
+          `Warning: Estimated token count ${budgetEstimate} exceeds budget limit ${options.budgetLimit}
+`
+        );
+      }
+    }
     if (options.explain && compiled.trace) {
       io.stdout.write(`
 [TRACE]

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@traits-dev/cli",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "description": "traits.dev command-line interface for voice profile init, validate, compile, eval, and import workflows.",
   "keywords": [
     "traits-dev",
@@ -41,7 +41,7 @@
     "provenance": true
   },
   "dependencies": {
-    "@traits-dev/core": "^0.3.0"
+    "@traits-dev/core": "^0.4.0"
   },
   "devDependencies": {
     "@types/node": "^25.2.3",