@vibe-cafe/vibe-usage 0.2.2 → 0.2.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vibe-cafe/vibe-usage",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "Track your AI coding tool token usage and sync to vibecafe.ai",
   "type": "module",
   "bin": {

package/src/hooks.js

@@ -85,20 +85,25 @@ export function injectCodex() {
     mkdirSync(dirname(configPath), { recursive: true });
   }
 
-  const notifyLine = `notify = ["sh", "-c", "${SYNC_CMD}"]`;
+  const notifyLine = `notify = "sh -c \\"${SYNC_CMD}\\""`;
 
   if (content.includes('vibe-usage')) {
-    // Migrate broken [[notify]] / [notify] table format from previous versions
-    // to correct inline array format: notify = ["sh", "-c", "..."]
+    // Migrate broken [[notify]] / [notify] table format and array format from previous versions
+    // to correct string format: notify = "sh -c \"...\""
     content = content.replace(
       /^\[\[?notify\]\]?\n(?:command\s*=\s*["'][^"']*["']\n?)?/gm,
       notifyLine + '\n',
     );
-    // Also update existing inline notify = [...] to use latest command
+    // Migrate array format: notify = ["sh", "-c", "..."]
     content = content.replace(
       /^notify\s*=\s*\[.*vibe-usage.*\]$/gm,
       notifyLine,
     );
+    // Update existing string format notify = "..." to use latest command
+    content = content.replace(
+      /^notify\s*=\s*".*vibe-usage.*"$/gm,
+      notifyLine,
+    );
     writeFileSync(configPath, content, 'utf-8');
     return { injected: false, reason: 'already installed (updated)' };
   }
@@ -106,8 +111,8 @@ export function injectCodex() {
   // Check if any notify line already exists
   const hasNotify = /^notify\s*=/m.test(content);
   if (hasNotify) {
-    // Append our command to existing notify (replace it)
-    content = content.replace(/^notify\s*=\s*\[.*\]$/gm, notifyLine);
+    // Replace existing notify value
+    content = content.replace(/^notify\s*=\s*.+$/gm, notifyLine);
   } else {
     content += `\n${notifyLine}\n`;
   }

package/src/parsers/codex.js

@@ -72,6 +72,8 @@ export async function parse(lastSync) {
       } catch { break; }
     }
 
+    // Track model from turn_context events (fallback when token_count lacks model)
+    let turnContextModel = 'unknown';
     // Track previous cumulative totals per model to compute deltas when only total_token_usage is available
     const prevTotal = new Map();
     for (const line of content.split('\n')) {
@@ -83,7 +85,15 @@ export async function parse(lastSync) {
         if (obj.type !== 'event_msg') continue;
 
         const payload = obj.payload;
-        if (!payload || payload.type !== 'token_count') continue;
+        if (!payload) continue;
+
+        // Capture model from turn_context events
+        if (payload.type === 'turn_context' && payload.model) {
+          turnContextModel = payload.model;
+          continue;
+        }
+
+        if (payload.type !== 'token_count') continue;
 
         const info = payload.info;
         if (!info) continue;
@@ -113,7 +123,7 @@ export async function parse(lastSync) {
         }
         if (!usage) continue;
 
-        const model = info.model || payload.model || sessionModel;
+        const model = info.model || payload.model || turnContextModel || sessionModel;
 
         entries.push({
           source: 'codex',

package/src/sync.js

@@ -7,7 +7,7 @@ import { ingest } from './api.js';
 import { parsers } from './parsers/index.js';
 import { TOOLS } from './hooks.js';
 
-const BATCH_SIZE = 500;
+const BATCH_SIZE = 100;
 
 function formatBytes(bytes) {
   if (bytes < 1024) return `${bytes}B`;