wolverine-ai 3.9.5 → 3.9.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "wolverine-ai",
-  "version": "3.9.5",
+  "version": "3.9.7",
   "description": "Self-healing Node.js server framework powered by AI. Catches crashes, diagnoses errors, generates fixes, verifies, and restarts — automatically.",
   "main": "src/index.js",
   "bin": {

package/src/core/ai-client.js

@@ -38,6 +38,11 @@ function _trackEmbedding(model, usage, latencyMs, success) {
   _tracker.record(model, "embedding", input, 0, null, latencyMs, success, 0, 0);
 }
 
+function _trackOp(model, category, inputTokens, outputTokens, tool, latencyMs, success) {
+  if (!_tracker) return;
+  _tracker.record(model, category, inputTokens || 0, outputTokens || 0, tool || null, latencyMs || 0, success, 0, 0);
+}
+
 // ── Client Management ──
 
 function getClient(provider) {
@@ -614,7 +619,7 @@ ${backupSourceCode ? `## Last Known Working Version\n\`\`\`javascript\n${backupS
 "changes" is for code edits (optional, use for actual code fixes).
 Include both if needed, or just one.`;
 
-  const result = await aiCall({ model, systemPrompt, userPrompt, maxTokens: 2048, category: "coding" });
+  const result = await aiCall({ model, systemPrompt, userPrompt, maxTokens: 2048, category: "reasoning" });
   const content = (result.content || "").trim();
 
   // Strip thinking tags (Gemma), markdown fences, and any prefix text
@@ -639,4 +644,4 @@ Include both if needed, or just one.`;
   }
 }
 
-module.exports = { requestRepair, getClient, tokenParam, aiCall, aiCallWithHistory, isResponsesModel, isAnthropicModel, setTokenTracker, getTrackerSnapshot, detectProvider, _trackEmbedding };
+module.exports = { requestRepair, getClient, tokenParam, aiCall, aiCallWithHistory, isResponsesModel, isAnthropicModel, setTokenTracker, getTrackerSnapshot, detectProvider, _trackEmbedding, _trackOp };

package/src/core/wolverine.js

@@ -1,6 +1,6 @@
 const chalk = require("chalk");
 const { parseError } = require("./error-parser");
-const { requestRepair, getClient, aiCall } = require("./ai-client");
+const { requestRepair, getClient, aiCall, _trackOp } = require("./ai-client");
 const { getModel } = require("./models");
 const { applyPatch } = require("./patcher");
 const { verifyFix } = require("./verifier");
@@ -341,6 +341,10 @@ async function _healImpl({ stderr, cwd, sandbox, notifier, rateLimiter, backupMa
             const patchResults = applyPatch(repair.changes, cwd, sandbox);
             if (!patchResults.every(r => r.success)) throw new Error("Patch failed");
 
+            // Track code generation as "coding" — the AI produced code changes
+            const codeTokens = repair.changes.reduce((s, c) => s + ((c.new || "").length / 4), 0);
+            _trackOp(getModel("coding"), "coding", 0, Math.round(codeTokens), "patch_apply", 0, true);
+
             for (const r of patchResults) console.log(chalk.green(`  ✅ Patched: ${r.file}`));
           }
 
@@ -348,6 +352,11 @@ async function _healImpl({ stderr, cwd, sandbox, notifier, rateLimiter, backupMa
           if (verification.verified) {
             backupManager.markVerified(bid);
             rateLimiter.clearSignature(errorSignature);
+            // Track tool operations: file read + patch + verify + any commands
+            // These are the same operations an agent would do with read_file/write_file/bash_exec
+            const toolOps = 1 + (repair.changes?.length || 0) + (repair.commands?.length || 0) + 1; // read + patches + commands + verify
+            const toolTokens = (sourceCode?.length || 0) / 4; // approximate tokens for file read
+            _trackOp(getModel("tool"), "tool", Math.round(toolTokens), 0, "fast_path_ops", Date.now() - healStartTime, true);
             return { healed: true, explanation: repair.explanation, backupId: bid, mode: "fast" };
           }