agent-trace 0.2.3 → 0.2.5

package/agent-trace.cjs

@@ -25708,6 +25708,12 @@ function toNonNegativeInteger2(value) {
   }
   return Math.floor(value);
 }
+function toNonNegativeDecimal(value) {
+  if (!Number.isFinite(value) || value <= 0) {
+    return 0;
+  }
+  return Number(value.toFixed(6));
+}
 function toSessionStatus(trace) {
   return trace.endedAt !== void 0 ? "completed" : "active";
 }
@@ -25731,7 +25737,7 @@ function toPostgresCommitRow(trace, commit) {
     message: toNullableString(commit.message),
     lines_added: toNonNegativeInteger2(commit.linesAdded),
     lines_removed: toNonNegativeInteger2(commit.linesRemoved),
-    chain_cost_usd: 0,
+    chain_cost_usd: toNonNegativeDecimal(trace.metrics.totalCostUsd),
     committed_at: toNullableString(commit.committedAt)
   };
 }
@@ -26013,6 +26019,109 @@ function toTimelineEventFromClickHouseRow(row) {
   };
 }
 
+// packages/schema/src/pricing.ts
+var MODEL_PRICING = [
+  // ── Opus family ──────────────────────────────────────────
+  // Opus 4.5 / 4.6 — $5 in, $25 out
+  ["claude-opus-4-6", {
+    inputPerToken: 5 / 1e6,
+    outputPerToken: 25 / 1e6,
+    cacheReadPerToken: 0.5 / 1e6,
+    cacheWritePerToken: 6.25 / 1e6
+  }],
+  ["claude-opus-4-5", {
+    inputPerToken: 5 / 1e6,
+    outputPerToken: 25 / 1e6,
+    cacheReadPerToken: 0.5 / 1e6,
+    cacheWritePerToken: 6.25 / 1e6
+  }],
+  // Opus 4.1 / 4.0 — $15 in, $75 out
+  ["claude-opus-4-1", {
+    inputPerToken: 15 / 1e6,
+    outputPerToken: 75 / 1e6,
+    cacheReadPerToken: 1.5 / 1e6,
+    cacheWritePerToken: 18.75 / 1e6
+  }],
+  ["claude-opus-4", {
+    inputPerToken: 15 / 1e6,
+    outputPerToken: 75 / 1e6,
+    cacheReadPerToken: 1.5 / 1e6,
+    cacheWritePerToken: 18.75 / 1e6
+  }],
+  // ── Sonnet family ────────────────────────────────────────
+  // Sonnet 4.x — $3 in, $15 out
+  ["claude-sonnet-4", {
+    inputPerToken: 3 / 1e6,
+    outputPerToken: 15 / 1e6,
+    cacheReadPerToken: 0.3 / 1e6,
+    cacheWritePerToken: 3.75 / 1e6
+  }],
+  // ── Haiku family ─────────────────────────────────────────
+  // Haiku 4.5 — $1 in, $5 out
+  ["claude-haiku-4", {
+    inputPerToken: 1 / 1e6,
+    outputPerToken: 5 / 1e6,
+    cacheReadPerToken: 0.1 / 1e6,
+    cacheWritePerToken: 1.25 / 1e6
+  }],
+  // ── Legacy 3.x models ───────────────────────────────────
+  ["claude-3-5-sonnet", {
+    inputPerToken: 3 / 1e6,
+    outputPerToken: 15 / 1e6,
+    cacheReadPerToken: 0.3 / 1e6,
+    cacheWritePerToken: 3.75 / 1e6
+  }],
+  ["claude-3-5-haiku", {
+    inputPerToken: 0.8 / 1e6,
+    outputPerToken: 4 / 1e6,
+    cacheReadPerToken: 0.08 / 1e6,
+    cacheWritePerToken: 1 / 1e6
+  }],
+  ["claude-3-opus", {
+    inputPerToken: 15 / 1e6,
+    outputPerToken: 75 / 1e6,
+    cacheReadPerToken: 1.5 / 1e6,
+    cacheWritePerToken: 18.75 / 1e6
+  }],
+  ["claude-3-sonnet", {
+    inputPerToken: 3 / 1e6,
+    outputPerToken: 15 / 1e6,
+    cacheReadPerToken: 0.3 / 1e6,
+    cacheWritePerToken: 3.75 / 1e6
+  }],
+  ["claude-3-haiku", {
+    inputPerToken: 0.25 / 1e6,
+    outputPerToken: 1.25 / 1e6,
+    cacheReadPerToken: 0.03 / 1e6,
+    cacheWritePerToken: 0.3 / 1e6
+  }]
+];
+var SORTED_PRICING = [...MODEL_PRICING].sort((a, b) => b[0].length - a[0].length);
+function lookupModelPricing(model) {
+  const normalized = model.toLowerCase();
+  for (const [pattern, pricing] of SORTED_PRICING) {
+    if (normalized.startsWith(pattern)) {
+      return pricing;
+    }
+  }
+  return void 0;
+}
+function calculateCostUsd(input) {
+  if (input.model === void 0) {
+    return 0;
+  }
+  const pricing = lookupModelPricing(input.model);
+  if (pricing === void 0) {
+    return 0;
+  }
+  const baseInput = Math.max(0, input.inputTokens);
+  const output = Math.max(0, input.outputTokens);
+  const cacheRead = Math.max(0, input.cacheReadTokens ?? 0);
+  const cacheWrite = Math.max(0, input.cacheWriteTokens ?? 0);
+  const cost = baseInput * pricing.inputPerToken + cacheRead * pricing.cacheReadPerToken + cacheWrite * pricing.cacheWritePerToken + output * pricing.outputPerToken;
+  return Number(cost.toFixed(6));
+}
+
 // packages/runtime/src/runtime.ts
 var import_node_http2 = __toESM(require("node:http"));
 
@@ -26891,6 +27000,10 @@ function buildNormalizedPayload(record, message, eventType) {
   if (cacheReadTokens !== void 0) {
     payload["cache_read_tokens"] = cacheReadTokens;
   }
+  const cacheWriteTokens = readNumber(usage, ["cache_creation_input_tokens", "cacheCreationInputTokens"]);
+  if (cacheWriteTokens !== void 0) {
+    payload["cache_write_tokens"] = cacheWriteTokens;
+  }
   const promptText = readPromptText(message);
   if (promptText !== void 0) {
     payload["prompt_text"] = promptText;
@@ -28045,11 +28158,33 @@ function buildNormalizedDetails(payload) {
   addCamelAlias(normalized, "file_path", "filePath");
   return normalized;
 }
+function computeEventCost(payload) {
+  const explicit = readNumber3(payload, ["cost_usd", "costUsd"]);
+  if (explicit !== void 0 && explicit > 0) {
+    return explicit;
+  }
+  const model = readString4(payload, ["model"]);
+  const inputTokens = readNumber3(payload, ["input_tokens", "inputTokens"]);
+  const outputTokens = readNumber3(payload, ["output_tokens", "outputTokens"]);
+  if (model === void 0 || inputTokens === void 0 || outputTokens === void 0) {
+    return explicit;
+  }
+  const cacheReadTokens = readNumber3(payload, ["cache_read_tokens", "cacheReadTokens", "cache_read_input_tokens"]);
+  const cacheWriteTokens = readNumber3(payload, ["cache_write_tokens", "cacheWriteTokens", "cache_creation_input_tokens"]);
+  const calculated = calculateCostUsd({
+    model,
+    inputTokens,
+    outputTokens,
+    cacheReadTokens: cacheReadTokens ?? 0,
+    cacheWriteTokens: cacheWriteTokens ?? 0
+  });
+  return calculated > 0 ? calculated : explicit;
+}
 function toTimelineEvent(envelope) {
   const payload = asRecord5(envelope.payload);
   const inputTokens = readNumber3(payload, ["input_tokens", "inputTokens"]);
   const outputTokens = readNumber3(payload, ["output_tokens", "outputTokens"]);
-  const costUsd = readNumber3(payload, ["cost_usd", "costUsd"]);
+  const costUsd = computeEventCost(payload);
   const details = buildNormalizedDetails(envelope.payload);
   return {
     id: envelope.eventId,
@@ -28131,7 +28266,7 @@ function toUpdatedTrace(existing, envelope) {
   const mergedTimeline = [...existing.timeline, timelineEvent];
   const endedAt = shouldMarkEnded(envelope.eventType) ? envelope.eventTimestamp : existing.endedAt;
   const latestTime = endedAt ?? envelope.eventTimestamp;
-  const cost = readNumber3(payload, ["cost_usd", "costUsd"]) ?? 0;
+  const cost = computeEventCost(payload) ?? 0;
   const inputTokens = readNumber3(payload, ["input_tokens", "inputTokens"]) ?? 0;
   const outputTokens = readNumber3(payload, ["output_tokens", "outputTokens"]) ?? 0;
   const linesAdded = readNumber3(payload, ["lines_added", "linesAdded"]) ?? 0;
@@ -28429,11 +28564,26 @@ function hydrateFromSqlite(runtime, sqlite, limit, eventLimit) {
       const extra = commits.filter((c) => !pgShas.has(c.sha));
       commits = [...mapped, ...extra];
     }
-    return {
+    let hydratedTrace = {
       ...trace,
       timeline,
       git: { ...trace.git, commits }
     };
+    if (hydratedTrace.metrics.totalCostUsd === 0 && (hydratedTrace.metrics.totalInputTokens > 0 || hydratedTrace.metrics.totalOutputTokens > 0) && hydratedTrace.metrics.modelsUsed.length > 0) {
+      const model = String(hydratedTrace.metrics.modelsUsed[0]);
+      const recalculated = calculateCostUsd({
+        model,
+        inputTokens: hydratedTrace.metrics.totalInputTokens,
+        outputTokens: hydratedTrace.metrics.totalOutputTokens
+      });
+      if (recalculated > 0) {
+        hydratedTrace = {
+          ...hydratedTrace,
+          metrics: { ...hydratedTrace.metrics, totalCostUsd: recalculated }
+        };
+      }
+    }
+    return hydratedTrace;
   });
   for (const trace of traces) {
     runtime.sessionRepository.upsert(trace);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-trace",
-  "version": "0.2.3",
+  "version": "0.2.5",
   "description": "Self-hosted observability for AI coding agents. One command, zero config.",
   "license": "Apache-2.0",
   "bin": {