codesesh 0.9.0 → 0.9.1

package/dist/{chunk-LXHUIQZI.js → chunk-5UOLRSFH.js}

@@ -704,7 +704,7 @@ function withEstimatedSessionCost(stats, model) {
 function estimateTokenCost(model, tokens) {
   return estimateCostForTokens(model, tokens)?.cost ?? null;
 }
-var HEAD_INDEX_VERSION = "claudecode-head-v1";
+var HEAD_INDEX_VERSION = "claudecode-head-v2";
 function parseTimestampMs(data) {
   const raw = String(data["timestamp"] ?? "").trim();
   if (!raw) return 0;
@@ -714,6 +714,25 @@ function parseTimestampMs(data) {
     return 0;
   }
 }
+function numericUsage(value) {
+  return typeof value === "number" && Number.isFinite(value) ? value : 0;
+}
+function extractClaudeUsage(data, msg) {
+  const usage = msg["usage"];
+  if (!usage || typeof usage !== "object") return null;
+  const u = usage;
+  const requestId = typeof data["requestId"] === "string" ? data["requestId"].trim() : "";
+  const uuid = typeof data["uuid"] === "string" ? data["uuid"].trim() : "";
+  const key = requestId || uuid;
+  if (!key) return null;
+  return {
+    key,
+    input: numericUsage(u["input_tokens"]),
+    output: numericUsage(u["output_tokens"]),
+    cacheRead: numericUsage(u["cache_read_input_tokens"]),
+    cacheCreate: numericUsage(u["cache_creation_input_tokens"])
+  };
+}
 var ClaudeCodeAgent = class extends BaseAgent {
   name = "claudecode";
   displayName = "Claude Code";
@@ -823,6 +842,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
     const pendingToolCalls = /* @__PURE__ */ new Map();
     const ignoredToolCallIds = /* @__PURE__ */ new Set();
     const assistantUuidToToolCalls = /* @__PURE__ */ new Map();
+    const countedUsageKeys = /* @__PURE__ */ new Set();
     const assistantState = {
       currentIndex: null,
       latestTextIndex: null
@@ -840,6 +860,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
           pendingToolCalls,
           ignoredToolCallIds,
           assistantUuidToToolCalls,
+          countedUsageKeys,
           assistantState
         );
       } catch {
@@ -1057,6 +1078,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
     let totalCacheCreateTokens = 0;
     let totalCost = 0;
     const modelUsageMap = {};
+    const countedUsageKeys = /* @__PURE__ */ new Set();
     for (const line of lines) {
       try {
         const data = JSON.parse(line);
@@ -1077,12 +1099,13 @@ var ClaudeCodeAgent = class extends BaseAgent {
             if (typeof m === "string" && m.trim()) model = m.trim();
           }
           if (role === "assistant") {
-            const usage = msg["usage"];
-            if (usage && typeof usage === "object") {
-              const inputTokens = usage["input_tokens"] ?? 0;
-              const cacheRead = usage["cache_read_input_tokens"] ?? 0;
-              const cacheCreate = usage["cache_creation_input_tokens"] ?? 0;
-              const outputTokens = usage["output_tokens"] ?? 0;
+            const usage = extractClaudeUsage(data, msg);
+            if (usage && !countedUsageKeys.has(usage.key)) {
+              countedUsageKeys.add(usage.key);
+              const inputTokens = usage.input;
+              const cacheRead = usage.cacheRead;
+              const cacheCreate = usage.cacheCreate;
+              const outputTokens = usage.output;
               totalInputTokens += inputTokens + cacheRead + cacheCreate;
               totalOutputTokens += outputTokens;
               totalCacheReadTokens += cacheRead;
@@ -1156,7 +1179,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
     return null;
   }
   // --- Record conversion ---
-  convertRecord(data, messages, pendingToolCalls, ignoredToolCallIds, assistantUuidToToolCalls, assistantState) {
+  convertRecord(data, messages, pendingToolCalls, ignoredToolCallIds, assistantUuidToToolCalls, countedUsageKeys, assistantState) {
     if (data["isMeta"] === true) return;
     const msgType = String(data["type"] ?? "");
     if (isInternalEventType(msgType)) return;
@@ -1167,6 +1190,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
         pendingToolCalls,
         ignoredToolCallIds,
         assistantUuidToToolCalls,
+        countedUsageKeys,
         assistantState
       );
     } else if (msgType === "user") {
@@ -1182,7 +1206,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
       this.convertToolResultRecord(data, messages, assistantState);
     }
   }
-  convertAssistantRecord(data, messages, pendingToolCalls, ignoredToolCallIds, assistantUuidToToolCalls, assistantState) {
+  convertAssistantRecord(data, messages, pendingToolCalls, ignoredToolCallIds, assistantUuidToToolCalls, countedUsageKeys, assistantState) {
     const msg = data["message"] ?? {};
     const timestampMs = parseTimestampMs(data);
     const rawContent = msg["content"] ?? [];
@@ -1200,7 +1224,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
           if (text) {
             currentAssistantIndex = this.appendAssistantReasoning(
               messages,
-              { messageId: uuid, msg, timestampMs, text },
+              { messageId: uuid, data, msg, timestampMs, text, countedUsageKeys },
               currentAssistantIndex
             );
           }
@@ -1211,7 +1235,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
           if (text) {
             currentAssistantIndex = this.appendAssistantText(
               messages,
-              { messageId: uuid, msg, timestampMs, text },
+              { messageId: uuid, data, msg, timestampMs, text, countedUsageKeys },
               currentAssistantIndex
             );
             latestAssistantTextIndex = currentAssistantIndex;
@@ -1228,10 +1252,12 @@ var ClaudeCodeAgent = class extends BaseAgent {
         const toolPart = this.buildToolPart(part, timestampMs);
         const [msgIndex, partIndex] = this.attachToolCallToLatestAssistant(messages, {
           messageId: uuid,
+          data,
           msg,
           timestampMs,
           toolPart,
-          latestTextIndex: latestAssistantTextIndex
+          latestTextIndex: latestAssistantTextIndex,
+          countedUsageKeys
         });
         currentAssistantIndex = msgIndex;
         if (toolCallId) {
@@ -1353,21 +1379,19 @@ var ClaudeCodeAgent = class extends BaseAgent {
       time_created: timestampMs
     };
   }
-  applyAssistantMetadata(message, msg) {
+  applyAssistantMetadata(message, data, msg, countedUsageKeys) {
     const model = msg["model"];
     if (model && typeof model === "string" && !message.model) {
       message.model = model;
     }
-    const usage = msg["usage"];
-    if (usage && typeof usage === "object" && !message.tokens) {
-      const u = usage;
-      const cacheRead = u["cache_read_input_tokens"] ?? 0;
-      const cacheCreate = u["cache_creation_input_tokens"] ?? 0;
+    const usage = extractClaudeUsage(data, msg);
+    if (usage && !message.tokens && !countedUsageKeys.has(usage.key)) {
+      countedUsageKeys.add(usage.key);
       message.tokens = {
-        input: (u["input_tokens"] ?? 0) + cacheCreate + cacheRead,
-        output: u["output_tokens"] ?? 0,
-        cache_read: cacheRead,
-        cache_create: cacheCreate
+        input: usage.input + usage.cacheCreate + usage.cacheRead,
+        output: usage.output,
+        cache_read: usage.cacheRead,
+        cache_create: usage.cacheCreate
       };
       const cost = estimateTokenCost(message.model, message.tokens);
       if (cost !== null) {
@@ -1385,7 +1409,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
       const hasTool = message2.parts.some((p) => p.type === "tool");
       if (!hasText && !hasTool) {
         this.appendPartIfNew(message2, part);
-        this.applyAssistantMetadata(message2, opts.msg);
+        this.applyAssistantMetadata(message2, opts.data, opts.msg, opts.countedUsageKeys);
         return currentIndex;
       }
     }
@@ -1396,7 +1420,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
       parts: [part],
       agent: "claude"
     });
-    this.applyAssistantMetadata(message, opts.msg);
+    this.applyAssistantMetadata(message, opts.data, opts.msg, opts.countedUsageKeys);
     messages.push(message);
     return messages.length - 1;
   }
@@ -1407,7 +1431,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
       const hasTool = message2.parts.some((p) => p.type === "tool");
       if (!hasTool) {
         this.appendPartIfNew(message2, part);
-        this.applyAssistantMetadata(message2, opts.msg);
+        this.applyAssistantMetadata(message2, opts.data, opts.msg, opts.countedUsageKeys);
         return currentIndex;
       }
     }
@@ -1418,7 +1442,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
       parts: [part],
       agent: "claude"
     });
-    this.applyAssistantMetadata(message, opts.msg);
+    this.applyAssistantMetadata(message, opts.data, opts.msg, opts.countedUsageKeys);
     messages.push(message);
     return messages.length - 1;
   }
@@ -1426,7 +1450,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
     if (opts.latestTextIndex !== null) {
       const message2 = messages[opts.latestTextIndex];
       message2.parts.push(opts.toolPart);
-      this.applyAssistantMetadata(message2, opts.msg);
+      this.applyAssistantMetadata(message2, opts.data, opts.msg, opts.countedUsageKeys);
       return [opts.latestTextIndex, message2.parts.length - 1];
     }
     const message = this.buildMessage({
@@ -1437,7 +1461,7 @@ var ClaudeCodeAgent = class extends BaseAgent {
       agent: "claude",
       mode: "tool"
     });
-    this.applyAssistantMetadata(message, opts.msg);
+    this.applyAssistantMetadata(message, opts.data, opts.msg, opts.countedUsageKeys);
     messages.push(message);
     return [messages.length - 1, 0];
   }
@@ -9148,4 +9172,4 @@ export {
   importBookmarks,
   deleteBookmark
 };
-//# sourceMappingURL=chunk-LXHUIQZI.js.map
+//# sourceMappingURL=chunk-5UOLRSFH.js.map