0sec-cli 0.15.0 → 0.17.0

package/chunks/chunk-SZJCPG2I.js

@@ -0,0 +1,110 @@
+#!/usr/bin/env node
+import { createRequire as __0secCreateRequire } from "node:module";
+const require = __0secCreateRequire(import.meta.url);
+import {
+  estimateCost,
+  modelProvider,
+  splitCost
+} from "./chunk-KKGE5RQE.js";
+
+// packages/core/dist/agent/cost-ledger.js
+var ScanCostLedger = class {
+  inputTokens = 0;
+  outputTokens = 0;
+  cachedInputTokens = 0;
+  perModel = /* @__PURE__ */ new Map();
+  /**
+   * Fold one turn's token usage into the cumulative total. `model` (the
+   * session's pricing model) keys the per-model buckets the cost_breakdown
+   * is derived from; sessions without a model pick land in "unknown".
+   */
+  add(usage, model) {
+    this.inputTokens += usage.inputTokens;
+    this.outputTokens += usage.outputTokens;
+    this.cachedInputTokens += usage.cachedInputTokens ?? 0;
+    const key = model ?? "unknown";
+    const bucket = this.perModel.get(key) ?? { inputTokens: 0, outputTokens: 0, cachedInputTokens: 0 };
+    bucket.inputTokens += usage.inputTokens;
+    bucket.outputTokens += usage.outputTokens;
+    bucket.cachedInputTokens += usage.cachedInputTokens ?? 0;
+    this.perModel.set(key, bucket);
+  }
+  /** Cumulative estimated cost (USD) across every contributing session. */
+  costUsd(model) {
+    return estimateCost({
+      inputTokens: this.inputTokens,
+      outputTokens: this.outputTokens,
+      cachedInputTokens: this.cachedInputTokens
+    }, model);
+  }
+  /** Exact cumulative spend derived from the per-model buckets. */
+  totalCostUsd() {
+    return this.costBreakdown()?.costUsd ?? 0;
+  }
+  /** Cumulative usage for the final cloud cost_update event. */
+  tokenUsage() {
+    return {
+      inputTokens: this.inputTokens,
+      outputTokens: this.outputTokens,
+      cachedInputTokens: this.cachedInputTokens
+    };
+  }
+  /**
+   * The model every contributing session used, when they all agree (and it
+   * is a real model id, not the "unknown" bucket). Undefined for multi-model
+   * runs and for runs whose sessions never named one — the caller then
+   * leaves the model field to its own resolution.
+   */
+  soleModel() {
+    if (this.perModel.size !== 1)
+      return void 0;
+    const [key] = this.perModel.keys();
+    return key === "unknown" ? void 0 : key;
+  }
+  /**
+   * Aggregate the recorded usage into the cloud's per-(provider, model)
+   * cost_breakdown shape — the same aggregation agentic-scanner's
+   * emitScanCompleted applies to its per-stage tally, sourced here from the
+   * true cross-session cumulative. Zero-usage buckets are skipped so the
+   * breakdown stays honest. Returns null when nothing was recorded at all,
+   * so callers omit cost_usd / cost_breakdown rather than emit a
+   * misleading $0.
+   */
+  costBreakdown() {
+    const acc = /* @__PURE__ */ new Map();
+    for (const [model, usage] of this.perModel) {
+      if (usage.inputTokens === 0 && usage.outputTokens === 0 && usage.cachedInputTokens === 0) {
+        continue;
+      }
+      const priced = model === "unknown" ? void 0 : model;
+      const provider = modelProvider(priced);
+      const key = `${provider}${model}`;
+      const split = splitCost(usage, priced);
+      const existing = acc.get(key);
+      if (existing) {
+        existing.cost_in += split.cost_in;
+        existing.cost_out += split.cost_out;
+        if (split.cost_cache_read !== void 0) {
+          existing.cost_cache_read = (existing.cost_cache_read ?? 0) + split.cost_cache_read;
+        }
+      } else {
+        acc.set(key, {
+          provider,
+          model,
+          cost_in: split.cost_in,
+          cost_out: split.cost_out,
+          ...split.cost_cache_read !== void 0 ? { cost_cache_read: split.cost_cache_read } : {}
+        });
+      }
+    }
+    if (acc.size === 0)
+      return null;
+    const breakdown = Array.from(acc.values());
+    const costUsd = breakdown.reduce((sum, e) => sum + e.cost_in + e.cost_out + (e.cost_cache_read ?? 0), 0);
+    return { costUsd, breakdown };
+  }
+};
+
+export {
+  ScanCostLedger
+};