@yhong91/vibetime 0.1.75 → 0.1.76

package/bin/vibetime.mjs

@@ -2047,7 +2047,7 @@ function claudeStyleFileMetrics(tool, input) {
 }
 
 // src/lib/constants.ts
-var PACKAGE_VERSION = true ? "0.1.75" : "0.1.1";
+var PACKAGE_VERSION = true ? "0.1.76" : "0.1.1";
 var GENERATED_MARKER = "Generated by vibetime.";
 var DEFAULT_API_URL = "http://121.196.224.82:3001";
 var DEFAULT_BACKFILL_BATCH_SIZE = 50;
@@ -12039,7 +12039,7 @@ async function loadQoderDbModelCalls(appDirName, sessionId, modelMap) {
         const modelInfo = parseJsonLine(stringField(row, "model_info") || "") || {};
         const modelKey = stringField(modelInfo, "model_key");
         const call = {
-          model: modelKey ? modelMap[modelKey] || modelKey : preferredModel,
+          model: mappedQoderModel(modelKey, modelMap) || preferredModel,
           inputTokens: numberField(usage, "prompt_tokens") || 0,
           outputTokens: numberField(usage, "completion_tokens") || 0,
           cachedTokens: numberField(usage, "cached_tokens") || 0
@@ -12075,7 +12075,14 @@ function resolveRootSessionId(db, sessionId) {
   }
   return current === sessionId ? void 0 : current;
 }
+function mappedQoderModel(raw, modelMap) {
+  if (!raw) {
+    return void 0;
+  }
+  return modelMap[raw] || raw;
+}
 function resolveSessionPreferredModel(db, sessionId, modelMap) {
+  const chain = [];
   let current = sessionId;
   for (let depth = 0; depth < 5 && current; depth++) {
     let row;
@@ -12087,15 +12094,36 @@ function resolveSessionPreferredModel(db, sessionId, modelMap) {
       return void 0;
     }
     if (!row) {
-      return void 0;
+      break;
     }
+    chain.push(current);
     const info = parseJsonLine(stringField(row, "preferred_model_info") || "") || {};
-    const preferred = stringField(info, "preferred_model");
+    const preferred = mappedQoderModel(stringField(info, "preferred_model"), modelMap);
     if (preferred) {
-      return modelMap[preferred] || preferred;
+      return preferred;
     }
     current = stringField(row, "parent_session_id");
   }
+  return resolveModelKeyFromAncestorMessages(db, chain, modelMap);
+}
+function resolveModelKeyFromAncestorMessages(db, sessionIds, modelMap) {
+  for (const sid of sessionIds) {
+    let rows;
+    try {
+      rows = db.prepare(
+        `select model_info from chat_message where session_id = ? and role = 'assistant' and instr(model_info, 'model_key') > 0 order by gmt_create desc limit 20`
+      ).all(sid);
+    } catch {
+      continue;
+    }
+    for (const row of rows) {
+      const info = parseJsonLine(stringField(row, "model_info") || "") || {};
+      const mapped = mappedQoderModel(stringField(info, "model_key"), modelMap);
+      if (mapped) {
+        return mapped;
+      }
+    }
+  }
   return void 0;
 }
 
@@ -12208,8 +12236,8 @@ async function loadQoderCnSegmentModelCalls(filePath, isSubagentSession, modelMa
           }
           if (type === "model.response.completed") {
             const ts = stringField(raw, "ts") || "";
-            const rawModel = stringField(data, "model") || "unknown";
-            const modelName = modelMap[rawModel] || rawModel;
+            const rawModel = stringField(data, "model");
+            const modelName = rawModel ? modelMap[rawModel] || rawModel : void 0;
             modelCalls.push({
               ts,
               isSubagent: currentTurnIsSubagent,
@@ -13094,8 +13122,8 @@ async function loadQoderSegmentModelCalls(filePath, isSubagentSession, modelMap)
           }
           if (type === "model.response.completed") {
             const ts = stringField(raw, "ts") || "";
-            const rawModel = stringField(data, "model") || "unknown";
-            const modelName = modelMap[rawModel] || rawModel;
+            const rawModel = stringField(data, "model");
+            const modelName = rawModel ? modelMap[rawModel] || rawModel : void 0;
             modelCalls.push({
               ts,
               isSubagent: currentTurnIsSubagent,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@yhong91/vibetime",
   "type": "module",
-  "version": "0.1.75",
+  "version": "0.1.76",
   "description": "vibetime CLI — install AI-agent hooks (Claude Code, Codex, OpenCode, Pi, Cursor) and report activity to vibetime.",
   "license": "MIT",
   "publishConfig": {