@yhong91/vibetime 0.1.18 → 0.1.19

package/bin/vibetime.mjs

@@ -1202,7 +1202,7 @@ function countTextLines(text) {
 }
 
 // src/lib/constants.ts
-var PACKAGE_VERSION = true ? "0.1.18" : "0.1.1";
+var PACKAGE_VERSION = true ? "0.1.19" : "0.1.1";
 var DEFAULT_API_URL = "http://121.196.224.82:3001";
 var DEFAULT_BACKFILL_BATCH_SIZE = 50;
 var DEFAULT_BACKFILL_BATCH_BYTES = 800 * 1024;
@@ -3431,8 +3431,9 @@ async function parseCodebuddyTraceFile(filePath, options) {
   }
   const trace = traceFile.trace;
   const spans = Array.isArray(traceFile.spans) ? traceFile.spans : [];
+  const hasGenerationSpans = spans.some((s) => s.type === "generation");
   const hasFunctionSpans = spans.some((s) => s.type === "function");
-  const isTrivial = !trace.sessionId && !trace.modelInfo && spans.length <= 3 && !hasFunctionSpans;
+  const isTrivial = !hasGenerationSpans && !hasFunctionSpans;
   if (isTrivial) {
     return [];
   }
@@ -3608,7 +3609,7 @@ async function parseCodebuddyTraceFile(filePath, options) {
     }
     if (span.type === "generation") {
       generationCount++;
-      const usage = modelUsageFromTrace(trace, generationCount, totalGenerations);
+      const usage = extractUsageFromGenerationSpan(span) ?? modelUsageFromTrace(trace, generationCount, totalGenerations);
       push(baseEvent({
         ts,
         type: "model.usage",
@@ -3787,6 +3788,32 @@ function normalizeModelCandidate(value) {
   }
   return trimmed;
 }
+function extractUsageFromGenerationSpan(span) {
+  const parsed = parseEmbeddedJson(span.toolOutput);
+  if (!Array.isArray(parsed) || parsed.length === 0) {
+    return void 0;
+  }
+  const first = parsed[0];
+  if (!isPlainObject(first)) {
+    return void 0;
+  }
+  const usage = objectField(first, "usage");
+  if (!isPlainObject(usage)) {
+    return void 0;
+  }
+  const inputTokens = numberField(usage, "prompt_tokens");
+  const outputTokens = numberField(usage, "completion_tokens");
+  const totalTokens = numberField(usage, "total_tokens");
+  const details = objectField(usage, "prompt_tokens_details");
+  const cachedTokens = isPlainObject(details) ? numberField(details, "cached_tokens") : void 0;
+  return {
+    tokensInput: inputTokens || void 0,
+    tokensCachedInput: cachedTokens || void 0,
+    tokensOutput: outputTokens || void 0,
+    tokensTotal: totalTokens || void 0,
+    modelCalls: 1
+  };
+}
 function modelUsageFromTrace(trace, generationIndex, totalGenerations) {
   const info = trace.modelInfo;
   if (!info || !info.totalInputTokens && !info.totalOutputTokens && !trace.totalTokens) {

package/package.json

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