agent-trace 0.2.3 → 0.2.4

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)
   };
 }
@@ -27961,6 +27967,92 @@ var InMemoryRuntimePersistence = class extends WriterBackedRuntimePersistence {
   }
 };
 
+// packages/schema/src/pricing.ts
+var MODEL_PRICING = [
+  // Opus 4 / 4.6
+  ["claude-opus-4", {
+    inputPerToken: 15 / 1e6,
+    outputPerToken: 75 / 1e6,
+    cacheReadPerToken: 1.5 / 1e6,
+    cacheWritePerToken: 18.75 / 1e6
+  }],
+  // Sonnet 4 / 4.6
+  ["claude-sonnet-4", {
+    inputPerToken: 3 / 1e6,
+    outputPerToken: 15 / 1e6,
+    cacheReadPerToken: 0.3 / 1e6,
+    cacheWritePerToken: 3.75 / 1e6
+  }],
+  // Haiku 4.5
+  ["claude-haiku-4", {
+    inputPerToken: 0.8 / 1e6,
+    outputPerToken: 4 / 1e6,
+    cacheReadPerToken: 0.08 / 1e6,
+    cacheWritePerToken: 1 / 1e6
+  }],
+  // Claude 3.5 Sonnet
+  ["claude-3-5-sonnet", {
+    inputPerToken: 3 / 1e6,
+    outputPerToken: 15 / 1e6,
+    cacheReadPerToken: 0.3 / 1e6,
+    cacheWritePerToken: 3.75 / 1e6
+  }],
+  // Claude 3.5 Haiku
+  ["claude-3-5-haiku", {
+    inputPerToken: 0.8 / 1e6,
+    outputPerToken: 4 / 1e6,
+    cacheReadPerToken: 0.08 / 1e6,
+    cacheWritePerToken: 1 / 1e6
+  }],
+  // Claude 3 Opus
+  ["claude-3-opus", {
+    inputPerToken: 15 / 1e6,
+    outputPerToken: 75 / 1e6,
+    cacheReadPerToken: 1.5 / 1e6,
+    cacheWritePerToken: 18.75 / 1e6
+  }],
+  // Claude 3 Sonnet
+  ["claude-3-sonnet", {
+    inputPerToken: 3 / 1e6,
+    outputPerToken: 15 / 1e6,
+    cacheReadPerToken: 0.3 / 1e6,
+    cacheWritePerToken: 3.75 / 1e6
+  }],
+  // Claude 3 Haiku
+  ["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 totalInput = 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 regularInput = Math.max(0, totalInput - cacheRead - cacheWrite);
+  const cost = regularInput * pricing.inputPerToken + cacheRead * pricing.cacheReadPerToken + cacheWrite * pricing.cacheWritePerToken + output * pricing.outputPerToken;
+  return Number(cost.toFixed(6));
+}
+
 // packages/runtime/src/projector.ts
 function asRecord5(value) {
   if (typeof value !== "object" || value === null || Array.isArray(value)) {
@@ -28045,11 +28137,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 +28245,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;

package/package.json

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