@cursor/bdk 0.2.2 → 0.2.4

package/src/internal/session-cost.ts

@@ -1,208 +0,0 @@
-/**
- * Per-session cost metrics folded from a durable session event stream.
- *
- * Powers `agent-sdk cost [sessionId]`: each `turn.completed` event carries
- * the turn's token usage and the estimated dollar cost recorded by the
- * engine via `POST /v1/models/estimate-cost`, so a session's spend is a
- * pure fold over its `events.ndjson`. `GET /v1/session/:id/cost` is the
- * live billed total.
- */
-
-import type { SessionEvent, TurnCost, TurnUsage } from "../types.js";
-import { formatCostUsd, roundCostUsd } from "./trajectory.js";
-
-/** One agent loop pass (turn) with its usage and estimated cost. */
-export interface SessionCostTurn {
-  turnId: string;
-  failed: boolean;
-  usage?: TurnUsage;
-  cost?: TurnCost;
-}
-
-export interface SessionCostTotals {
-  turns: number;
-  turnFailures: number;
-  inputTokens: number;
-  outputTokens: number;
-  cacheReadTokens: number;
-  cacheWriteTokens: number;
-  /** Sum of per-turn estimated costs (USD). */
-  costUsd: number;
-  /** Turns that reported usage but had no known model rates. */
-  unpricedTurns: number;
-}
-
-export interface SessionCostReport {
-  sessionId: string;
-  turns: SessionCostTurn[];
-  totals: SessionCostTotals;
-}
-
-export function emptySessionCostTotals(): SessionCostTotals {
-  return {
-    turns: 0,
-    turnFailures: 0,
-    inputTokens: 0,
-    outputTokens: 0,
-    cacheReadTokens: 0,
-    cacheWriteTokens: 0,
-    costUsd: 0,
-    unpricedTurns: 0,
-  };
-}
-
-export function addSessionCostTotals(target: SessionCostTotals, delta: SessionCostTotals): void {
-  target.turns += delta.turns;
-  target.turnFailures += delta.turnFailures;
-  target.inputTokens += delta.inputTokens;
-  target.outputTokens += delta.outputTokens;
-  target.cacheReadTokens += delta.cacheReadTokens;
-  target.cacheWriteTokens += delta.cacheWriteTokens;
-  target.costUsd = roundCostUsd(target.costUsd + delta.costUsd);
-  target.unpricedTurns += delta.unpricedTurns;
-}
-
-/** Fold a session event stream into per-turn and total cost metrics. */
-export function buildSessionCostReport(events: readonly SessionEvent[]): SessionCostReport {
-  let sessionId = "";
-  const turns: SessionCostTurn[] = [];
-  const totals = emptySessionCostTotals();
-
-  for (const event of events) {
-    if (sessionId === "" && typeof event.sessionId === "string") {
-      sessionId = event.sessionId;
-    }
-    if (event.type !== "turn.completed" && event.type !== "turn.failed") {
-      continue;
-    }
-    const turn: SessionCostTurn = {
-      turnId: event.turnId ?? `turn_${turns.length + 1}`,
-      failed: event.type === "turn.failed",
-    };
-    totals.turns += 1;
-    if (event.type === "turn.failed") {
-      totals.turnFailures += 1;
-      turns.push(turn);
-      continue;
-    }
-    const usage = event.data.usage;
-    if (usage !== undefined) {
-      turn.usage = usage;
-      totals.inputTokens += usage.inputTokens;
-      totals.outputTokens += usage.outputTokens;
-      totals.cacheReadTokens += usage.cacheReadTokens;
-      totals.cacheWriteTokens += usage.cacheWriteTokens;
-    }
-    const cost = event.data.cost;
-    if (cost !== undefined) {
-      turn.cost = cost;
-      totals.costUsd = roundCostUsd(totals.costUsd + cost.totalUsd);
-    } else if (usage !== undefined) {
-      totals.unpricedTurns += 1;
-    }
-    turns.push(turn);
-  }
-
-  return { sessionId, turns, totals };
-}
-
-function formatCount(value: number): string {
-  return String(value);
-}
-
-/** Per-turn table for `agent-sdk cost <sessionId>`. */
-export function formatSessionCostText(report: SessionCostReport): string {
-  const lines: string[] = [];
-  lines.push(
-    `session ${report.sessionId || "(unknown)"}  cost: ${formatCostUsd(report.totals.costUsd)}`,
-  );
-  if (report.turns.length === 0) {
-    lines.push("No turns recorded.");
-    return `${lines.join("\n")}\n`;
-  }
-  const header = ["TURN", "IN", "OUT", "CACHE R", "CACHE W", "COST", "MODEL", "STATUS"];
-  const rows = report.turns.map(turn => [
-    turn.turnId,
-    turn.usage === undefined ? "—" : formatCount(turn.usage.inputTokens),
-    turn.usage === undefined ? "—" : formatCount(turn.usage.outputTokens),
-    turn.usage === undefined ? "—" : formatCount(turn.usage.cacheReadTokens),
-    turn.usage === undefined ? "—" : formatCount(turn.usage.cacheWriteTokens),
-    turn.cost === undefined ? "—" : formatCostUsd(turn.cost.totalUsd),
-    turn.cost?.model ?? "—",
-    turn.failed ? "failed" : "ok",
-  ]);
-  rows.push([
-    "total",
-    formatCount(report.totals.inputTokens),
-    formatCount(report.totals.outputTokens),
-    formatCount(report.totals.cacheReadTokens),
-    formatCount(report.totals.cacheWriteTokens),
-    formatCostUsd(report.totals.costUsd),
-    "",
-    report.totals.turnFailures > 0 ? `${report.totals.turnFailures} failed` : "",
-  ]);
-  lines.push(renderTable(header, rows));
-  appendUnpricedNote(lines, report.totals.unpricedTurns);
-  return `${lines.join("\n")}\n`;
-}
-
-export interface SessionCostRow {
-  sessionId: string;
-  channelId: string;
-  title?: string;
-  updatedAt?: string;
-  totals: SessionCostTotals;
-}
-
-/** All-sessions table for `agent-sdk cost` (no session id). */
-export function formatSessionsCostText(rows: SessionCostRow[]): string {
-  if (rows.length === 0) {
-    return "No sessions.\n";
-  }
-  const header = ["SESSION", "CHANNEL", "TURNS", "IN", "OUT", "COST", "UPDATED"];
-  const totals = emptySessionCostTotals();
-  const body = rows.map(row => {
-    addSessionCostTotals(totals, row.totals);
-    return [
-      row.sessionId,
-      row.channelId,
-      formatCount(row.totals.turns),
-      formatCount(row.totals.inputTokens),
-      formatCount(row.totals.outputTokens),
-      formatCostUsd(row.totals.costUsd),
-      row.updatedAt ?? "",
-    ];
-  });
-  body.push([
-    "total",
-    "",
-    formatCount(totals.turns),
-    formatCount(totals.inputTokens),
-    formatCount(totals.outputTokens),
-    formatCostUsd(totals.costUsd),
-    "",
-  ]);
-  const lines = [renderTable(header, body)];
-  appendUnpricedNote(lines, totals.unpricedTurns);
-  return `${lines.join("\n")}\n`;
-}
-
-function appendUnpricedNote(lines: string[], unpricedTurns: number): void {
-  if (unpricedTurns > 0) {
-    lines.push(
-      `note: ${unpricedTurns} turn(s) reported usage with no priced cost; cost totals exclude them.`,
-    );
-  }
-}
-
-function renderTable(header: string[], rows: string[][]): string {
-  const widths = header.map((h, i) =>
-    Math.max(h.length, ...rows.map(row => (row[i] ?? "").length)),
-  );
-  const renderRow = (row: string[]): string =>
-    row
-      .map((cell, i) => cell.padEnd(widths[i] ?? 0))
-      .join("  ")
-      .trimEnd();
-  return [renderRow(header), ...rows.map(renderRow)].join("\n");
-}