driftguard-mcp 0.1.4 → 0.1.5

package/dist/bin.js

@@ -51,17 +51,29 @@ function parseJSONL(filePath) {
       if (entry.type !== "user" && entry.type !== "assistant") continue;
       const content = entry.message?.content;
       let text = "";
+      let toolTokens = 0;
       if (typeof content === "string") {
         text = content.trim();
       } else if (Array.isArray(content)) {
-        text = content.filter((block) => block.type === "text" && typeof block.text === "string").map((block) => block.text.trim()).join("\n").trim();
+        const blocks = content;
+        text = blocks.filter((block) => block.type === "text" && typeof block.text === "string").map((block) => block.text.trim()).join("\n").trim();
+        for (const block of blocks) {
+          if (block.type === "tool_use" && block.input) {
+            const inputText = JSON.stringify(block.input);
+            toolTokens += Math.round(inputText.length / 4);
+          } else if (block.type === "tool_result") {
+            const resultText = typeof block.content === "string" ? block.content : Array.isArray(block.content) ? block.content.filter((b) => b.type === "text" && typeof b.text === "string").map((b) => b.text).join("\n") : "";
+            toolTokens += Math.round(resultText.length / 4);
+          }
+        }
       }
       if (!text.trim()) continue;
       messages.push({
         id: entry.uuid,
         role: entry.message.role,
         content: text,
-        timestamp: parseTimestamp(entry.timestamp)
+        timestamp: parseTimestamp(entry.timestamp),
+        ...toolTokens > 0 ? { toolTokens } : {}
       });
     } catch {
       skipped++;
@@ -234,14 +246,25 @@ var init_gemini_adapter = __esm({
           try {
             const entry = JSON.parse(line);
             if (entry.role !== "user" && entry.role !== "model") continue;
-            const text = (entry.parts ?? []).map((p) => p.text ?? "").join("\n").trim();
+            const parts = entry.parts ?? [];
+            const text = parts.map((p) => p.text ?? "").join("\n").trim();
             if (!text) continue;
+            let toolTokens = 0;
+            for (const part of parts) {
+              if (part.functionCall?.args) {
+                toolTokens += Math.round(JSON.stringify(part.functionCall.args).length / 4);
+              }
+              if (part.functionResponse?.response) {
+                toolTokens += Math.round(JSON.stringify(part.functionResponse.response).length / 4);
+              }
+            }
             const ts = entry.timestamp ? new Date(entry.timestamp).getTime() : Date.now();
             messages.push({
               id: `gemini-${index++}`,
               role: entry.role === "model" ? "assistant" : "user",
               content: text,
-              timestamp: isFinite(ts) ? ts : Date.now()
+              timestamp: isFinite(ts) ? ts : Date.now(),
+              ...toolTokens > 0 ? { toolTokens } : {}
             });
           } catch {
           }
@@ -313,15 +336,37 @@ var init_codex_adapter = __esm({
         const lines = raw.split("\n").filter((l) => l.trim());
         const messages = [];
         let index = 0;
+        let pendingToolTokens = 0;
         for (const line of lines) {
           try {
             const entry = JSON.parse(line);
+            if (entry.role === "tool") {
+              const resultText = typeof entry.content === "string" ? entry.content : "";
+              pendingToolTokens += Math.round(resultText.length / 4);
+              continue;
+            }
             if (entry.role !== "user" && entry.role !== "assistant") continue;
             let text = "";
+            let toolTokens = pendingToolTokens;
+            pendingToolTokens = 0;
             if (typeof entry.content === "string") {
               text = entry.content.trim();
             } else if (Array.isArray(entry.content)) {
-              text = entry.content.filter((b) => b.type === "text" && typeof b.text === "string").map((b) => b.text.trim()).join("\n").trim();
+              for (const block of entry.content) {
+                if (block.type === "text" && typeof block.text === "string") {
+                  text += (text ? "\n" : "") + block.text.trim();
+                } else if (block.type === "tool_use" && block.input) {
+                  toolTokens += Math.round(JSON.stringify(block.input).length / 4);
+                }
+              }
+              text = text.trim();
+            }
+            if (entry.tool_calls) {
+              for (const tc of entry.tool_calls) {
+                if (tc.function?.arguments) {
+                  toolTokens += Math.round(tc.function.arguments.length / 4);
+                }
+              }
             }
             if (!text) continue;
             let ts = Date.now();
@@ -333,7 +378,8 @@ var init_codex_adapter = __esm({
               id: `codex-${index++}`,
               role: entry.role,
               content: text,
-              timestamp: ts
+              timestamp: ts,
+              ...toolTokens > 0 ? { toolTokens } : {}
             });
           } catch {
           }
@@ -1431,7 +1477,7 @@ function calcMessageDecay(messages) {
     const words = msg.content.split(/\s+/).filter((w) => w.length > 0).length;
     const hasCode = msg.content.includes("```");
     const tokenEstimate = Math.round(words * 1.3 * (hasCode ? 1.5 : 1));
-    totalTokens += tokenEstimate;
+    totalTokens += tokenEstimate + (msg.toolTokens ?? 0);
   }
   if (totalTokens < 500) return 0;
   let score = Math.min(100, Math.round(15 * Math.log(totalTokens / 1500)));
@@ -2190,7 +2236,8 @@ var init_mcp_server = __esm({
         ...m,
         platform: "claude",
         tabId: 0,
-        chatId: "cli"
+        chatId: "cli",
+        ...m.toolTokens !== void 0 ? { toolTokens: m.toolTokens } : {}
       }));
       const analysis = calculateDrift(chatMessages, config.weights);
       const level = scoreToLevel(analysis.score);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "driftguard-mcp",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "Real-time AI conversation drift monitor — MCP server for Claude Code, Gemini CLI, Codex CLI, and Cursor",
   "main": "dist/bin.js",
   "bin": {