agent-trace 0.2.4 → 0.2.5

package/agent-trace.cjs

@@ -26019,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"));
 
@@ -26897,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;
@@ -27967,92 +28074,6 @@ 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)) {
@@ -28543,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.4",
+  "version": "0.2.5",
   "description": "Self-hosted observability for AI coding agents. One command, zero config.",
   "license": "Apache-2.0",
   "bin": {