tokentracker-cli 0.56.0 → 0.57.1

package/src/lib/rollout.js

@@ -2520,6 +2520,36 @@ function readOpencodeDbMessages(dbPath, sqliteOptions = {}) {
   return out;
 }
 
+// mimocode mirrors the user's Claude Code + claude-mem history into its own
+// `message` table — via an explicit `claude_import` AND a live observer /
+// session sync — so the overwhelming majority of rows are anthropic-endpoint
+// turns (providerID="anthropic") the Claude parser ALREADY counts as
+// source=claude. On the dev's box that's ~3.9B mirrored tokens vs ~22M genuine
+// mimo tokens. Counting the mirror under "mimo" double-counts and mislabels
+// Claude usage (user saw claude-* rows under the MIMO provider).
+//
+// The discriminator is providerID, NOT the model id. mimo's own runtime tags
+// turns providerID="mimo" (its auto router) or "xiaomi". providerID="anthropic"
+// means the turn went through a Claude-compatible endpoint — plain Claude Code,
+// OR a mimo-named model the user picked IN Claude Code (e.g. model=mimo-v2.5-pro
+// run inside Claude Code, logged in ~/.claude, counted as source=claude). Keying
+// off the model id would wrongly re-count that mimo-v2.5-pro. claude_import is
+// irrelevant — it never covers the observer/session mirror; the provider rule
+// subsumes it.
+function isMimoNativeMessage(data) {
+  if (!data) return false;
+  const provider = String(data.providerID || "").toLowerCase();
+  return provider === "mimo" || provider === "xiaomi";
+}
+
+// Read only genuine mimo assistant messages (mimo's own models), dropping the
+// mirrored Claude/claude-mem rows. See isMimoNativeMessage for why.
+function readMimoDbMessages(dbPath, sqliteOptions = {}) {
+  if (!dbPath || !fssync.existsSync(dbPath)) return [];
+  const all = readOpencodeDbMessages(dbPath, sqliteOptions);
+  return all.filter((m) => isMimoNativeMessage(m.data));
+}
+
 async function parseOpencodeDbIncremental({
   dbMessages,
   cursors,
@@ -9033,6 +9063,7 @@ module.exports = {
   listGeminiSessionFiles,
   listOpencodeMessageFiles,
   readOpencodeDbMessages,
+  readMimoDbMessages,
   resolveKiroDbPath,
   resolveKiroJsonlPath,
   resolveHermesPath,