@costwise/sdk 0.0.1 → 0.0.3

package/README.md

@@ -0,0 +1,176 @@
+# @costwise/sdk
+
+Track Anthropic and OpenAI API usage costs with a drop-in SDK wrapper.
+
+## Install
+
+```bash
+npm install @costwise/sdk
+```
+
+## Quick start
+
+Replace:
+
+```ts
+import OpenAI from "openai";
+```
+
+with:
+
+```ts
+import { CostwiseOpenAI } from "@costwise/sdk";
+```
+
+Then initialize as usual:
+
+```ts
+import { CostwiseOpenAI, CostwiseAnthropic } from "@costwise/sdk";
+
+const openai = new CostwiseOpenAI({
+  apiKey: process.env.OPENAI_API_KEY,
+  costwiseKey: process.env.COSTWISE_API_KEY,
+  tag: "my-app"
+});
+
+const anthropic = new CostwiseAnthropic({
+  apiKey: process.env.ANTHROPIC_API_KEY,
+  costwiseKey: process.env.COSTWISE_API_KEY,
+  tag: "my-app"
+});
+```
+
+After each response, Costwise sends a non-blocking ingest event to:
+
+`https://api.costwise.dev/ingest`
+
+Tracking is fire-and-forget and never awaited, so request latency is not impacted. If ingest is unreachable, it fails silently.
+
+## Environment variables
+
+- `COSTWISE_API_KEY` (optional if `costwiseKey` is provided in constructor)
+- Constructor key aliases supported: `costwiseKey` (canonical) and `costswiseKey` (compatibility alias)
+- `COSTWISE_LOCAL_USAGE_PATH` (optional JSONL file path for local usage history used by CLI)
+
+## Ingest payload
+
+The SDK emits:
+
+- `model`
+- `input_tokens`
+- `output_tokens`
+- `cost` and `estimated_cost`
+- `tag` and `endpoint_tag` (optional)
+- `ts` and `timestamp`
+
+## Compare model costs and capacity
+
+Use tracker utilities to estimate what other models would have cost for the same token usage:
+
+```ts
+import { CostwiseTracker } from "@costwise/sdk";
+
+const tracker = new CostwiseTracker({
+  costwiseKey: process.env.COSTWISE_API_KEY
+});
+
+const comparisons = tracker.compareCosts({
+  model: "gpt-4o",
+  inputTokens: 12000,
+  outputTokens: 2500
+});
+
+const summary = tracker.summarizeComparison({
+  model: "gpt-4o",
+  inputTokens: 12000,
+  outputTokens: 2500
+});
+```
+
+`compareCosts()` returns each model's estimated cost and capacity metadata (context window and capacity class), sorted cheapest-first.
+
+Group by capacity tier and read the cheapest model per tier:
+
+```ts
+const { byClass, cheapestPerClass } = tracker.compareCostsByCapacity({
+  model: "gpt-4o",
+  inputTokens: 12000,
+  outputTokens: 2500
+});
+
+// e.g. cheapestPerClass.mini?.model, cheapestPerClass.high?.estimatedCost
+```
+
+`byClass` maps `nano` | `mini` | `standard` | `high` | `frontier` to full lists (cheapest-first within each class). `cheapestPerClass` is the single best-priced model in each class that has at least one priced model.
+
+Format a ready-to-print CLI summary (without logging inside the SDK):
+
+```ts
+const cli = tracker.formatComparisonForCli({
+  model: "gpt-4o",
+  inputTokens: 12000,
+  outputTokens: 2500
+});
+
+// App decides if/where to print:
+// console.log(cli.text)
+```
+
+## Smoke test
+
+Run a local end-to-end smoke test (build + mocked ingest + comparison output):
+
+```bash
+npm run smoke
+```
+
+## CLI for actual usage
+
+The CLI reads **real usage events** from a local JSONL file written by the SDK at runtime.
+
+1) Set a usage file path in your app environment:
+
+```bash
+export COSTWISE_LOCAL_USAGE_PATH=.costwise-usage.jsonl
+```
+
+2) Run your app so events are recorded.
+
+3) Query usage and compare alternatives:
+
+```bash
+costwise usage
+costwise compare-actual
+```
+
+Optional:
+
+```bash
+costwise usage --file .costwise-usage.jsonl --json
+costwise compare-actual --file .costwise-usage.jsonl --json
+```
+
+## Automated tests
+
+Run the Node test suite (build + unit tests):
+
+```bash
+npm test
+```
+
+Current test coverage includes:
+
+- Tracker cost math correctness for known models
+- Unknown model handling (no ingest + empty comparison)
+- Fire-and-forget tracking behavior (`fetch` not awaited)
+- Silent ingest failure handling (`.catch(() => {})`)
+- `compareCostsByCapacity()` output shape
+- `formatComparisonForCli()` output shape
+- OpenAI wrapper behavior:
+  - non-stream response returned untouched
+  - stream response returned untouched
+  - stream tracking via `finalChatCompletion()`
+- Anthropic wrapper behavior:
+  - non-stream response returned untouched
+  - stream response returned untouched
+  - stream tracking via `finalMessage()`

package/dist/cjs/cli.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=cli.d.ts.map

package/dist/cjs/cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../../cli.ts"],"names":[],"mappings":""}

package/dist/cjs/cli.js

@@ -0,0 +1,156 @@
+#!/usr/bin/env node
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const promises_1 = require("node:fs/promises");
+const tracker_js_1 = require("./src/tracker.js");
+function help(exitCode) {
+    process.stdout.write([
+        "Costwise CLI (actual usage only)",
+        "",
+        "Usage:",
+        "  costwise usage [--file <path>] [--json]",
+        "  costwise compare-actual [--file <path>] [--json]",
+        "",
+        "Notes:",
+        "  - Reads usage events from JSONL recorded by SDK (`COSTWISE_LOCAL_USAGE_PATH`).",
+        "  - No manual token input flags are supported."
+    ].join("\n") + "\n");
+    process.exit(exitCode);
+}
+function parse(argv) {
+    if (argv.length === 0 || argv.includes("--help") || argv.includes("-h")) {
+        help(0);
+    }
+    const options = { json: false };
+    const first = argv[0];
+    if (first === "usage" || first === "compare-actual") {
+        options.command = first;
+    }
+    else {
+        help(1);
+    }
+    for (let i = 1; i < argv.length; i += 1) {
+        const arg = argv[i];
+        const next = argv[i + 1];
+        if (arg === "--file" && next) {
+            options.filePath = next;
+            i += 1;
+            continue;
+        }
+        if (arg === "--json") {
+            options.json = true;
+            continue;
+        }
+        help(1);
+    }
+    return options;
+}
+async function loadEvents(path) {
+    const content = await (0, promises_1.readFile)(path, "utf8");
+    const lines = content.split("\n").map((line) => line.trim()).filter(Boolean);
+    return lines.map((line) => JSON.parse(line));
+}
+function runUsage(events, asJson) {
+    const totals = events.reduce((acc, event) => {
+        acc.input += event.input_tokens;
+        acc.output += event.output_tokens;
+        acc.cost += event.estimated_cost;
+        acc.byModel[event.model] = (acc.byModel[event.model] ?? 0) + event.estimated_cost;
+        return acc;
+    }, { input: 0, output: 0, cost: 0, byModel: {} });
+    const topModels = Object.entries(totals.byModel)
+        .map(([model, cost]) => ({ model, cost }))
+        .sort((a, b) => b.cost - a.cost)
+        .slice(0, 5);
+    if (asJson) {
+        process.stdout.write(`${JSON.stringify({
+            events: events.length,
+            input_tokens: totals.input,
+            output_tokens: totals.output,
+            estimated_cost: totals.cost,
+            top_models: topModels
+        }, null, 2)}\n`);
+        return;
+    }
+    const lines = [
+        `Events: ${events.length}`,
+        `Input tokens: ${totals.input}`,
+        `Output tokens: ${totals.output}`,
+        `Estimated cost: $${totals.cost.toFixed(6)}`,
+        "Top models by spend:"
+    ];
+    for (const model of topModels) {
+        lines.push(`  - ${model.model}: $${model.cost.toFixed(6)}`);
+    }
+    process.stdout.write(`${lines.join("\n")}\n`);
+}
+function runCompareActual(events, asJson) {
+    const totalInput = events.reduce((sum, event) => sum + event.input_tokens, 0);
+    const totalOutput = events.reduce((sum, event) => sum + event.output_tokens, 0);
+    const actualCost = events.reduce((sum, event) => sum + event.estimated_cost, 0);
+    const alternatives = Object.entries(tracker_js_1.PRICING)
+        .map(([model, rates]) => {
+        const estimated_cost = (totalInput / 1_000_000) * rates.input + (totalOutput / 1_000_000) * rates.output;
+        return {
+            model,
+            estimated_cost,
+            delta_vs_actual: estimated_cost - actualCost
+        };
+    })
+        .sort((a, b) => a.estimated_cost - b.estimated_cost);
+    const tracker = new tracker_js_1.CostwiseTracker();
+    const formatted = tracker.formatComparisonForCli({
+        model: alternatives[0]?.model ?? "unknown",
+        inputTokens: totalInput,
+        outputTokens: totalOutput
+    });
+    if (asJson) {
+        process.stdout.write(`${JSON.stringify({
+            actual: {
+                events: events.length,
+                input_tokens: totalInput,
+                output_tokens: totalOutput,
+                estimated_cost: actualCost
+            },
+            alternatives: alternatives.slice(0, 10)
+        }, null, 2)}\n`);
+        return;
+    }
+    const lines = [
+        `Actual usage events: ${events.length}`,
+        `Actual estimated cost: $${actualCost.toFixed(6)}`,
+        "Top alternative models (same total tokens):",
+        ...alternatives.slice(0, 5).map((alt) => `  - ${alt.model}: $${alt.estimated_cost.toFixed(6)} (delta ${alt.delta_vs_actual >= 0 ? "+" : ""}${alt.delta_vs_actual.toFixed(6)})`),
+        "",
+        "Cheapest-model summary:",
+        formatted.text
+    ];
+    process.stdout.write(`${lines.join("\n")}\n`);
+}
+async function main() {
+    const options = parse(process.argv.slice(2));
+    const usagePath = options.filePath ?? process.env.COSTWISE_LOCAL_USAGE_PATH;
+    if (!usagePath) {
+        process.stderr.write("Missing usage file. Set COSTWISE_LOCAL_USAGE_PATH or pass --file.\n");
+        process.exit(1);
+    }
+    let events = [];
+    try {
+        events = await loadEvents(usagePath);
+    }
+    catch {
+        process.stderr.write(`Unable to read usage file: ${usagePath}\n`);
+        process.exit(1);
+    }
+    if (events.length === 0) {
+        process.stdout.write("No usage events found.\n");
+        return;
+    }
+    if (options.command === "usage") {
+        runUsage(events, options.json);
+        return;
+    }
+    runCompareActual(events, options.json);
+}
+void main();
+//# sourceMappingURL=cli.js.map

package/dist/cjs/cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../cli.ts"],"names":[],"mappings":";;;AAEA,+CAA4C;AAC5C,iDAA4D;AAS5D,SAAS,IAAI,CAAC,QAAgB;IAC5B,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB;QACE,kCAAkC;QAClC,EAAE;QACF,QAAQ;QACR,2CAA2C;QAC3C,oDAAoD;QACpD,EAAE;QACF,QAAQ;QACR,kFAAkF;QAClF,gDAAgD;KACjD,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CACpB,CAAC;IACF,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AACzB,CAAC;AAED,SAAS,KAAK,CAAC,IAAc;IAC3B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACxE,IAAI,CAAC,CAAC,CAAC,CAAC;IACV,CAAC;IAED,MAAM,OAAO,GAAe,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACtB,IAAI,KAAK,KAAK,OAAO,IAAI,KAAK,KAAK,gBAAgB,EAAE,CAAC;QACpD,OAAO,CAAC,OAAO,GAAG,KAAK,CAAC;IAC1B,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,CAAC,CAAC,CAAC;IACV,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QACzB,IAAI,GAAG,KAAK,QAAQ,IAAI,IAAI,EAAE,CAAC;YAC7B,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC;YACxB,CAAC,IAAI,CAAC,CAAC;YACP,SAAS;QACX,CAAC;QACD,IAAI,GAAG,KAAK,QAAQ,EAAE,CAAC;YACrB,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;YACpB,SAAS;QACX,CAAC;QACD,IAAI,CAAC,CAAC,CAAC,CAAC;IACV,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,KAAK,UAAU,UAAU,CAAC,IAAY;IACpC,MAAM,OAAO,GAAG,MAAM,IAAA,mBAAQ,EAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC7E,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoB,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,QAAQ,CAAC,MAAyB,EAAE,MAAe;IAC1D,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAC1B,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QACb,GAAG,CAAC,KAAK,IAAI,KAAK,CAAC,YAAY,CAAC;QAChC,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC;QAClC,GAAG,CAAC,IAAI,IAAI,KAAK,CAAC,cAAc,CAAC;QACjC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,GAAG,KAAK,CAAC,cAAc,CAAC;QAClF,OAAO,GAAG,CAAC;IACb,CAAC,EACD,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAA4B,EAAE,CACxE,CAAC;IAEF,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC;SAC7C,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;SACzC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC;SAC/B,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAEf,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,GAAG,IAAI,CAAC,SAAS,CACf;YACE,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,YAAY,EAAE,MAAM,CAAC,KAAK;YAC1B,aAAa,EAAE,MAAM,CAAC,MAAM;YAC5B,cAAc,EAAE,MAAM,CAAC,IAAI;YAC3B,UAAU,EAAE,SAAS;SACtB,EACD,IAAI,EACJ,CAAC,CACF,IAAI,CACN,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,KAAK,GAAG;QACZ,WAAW,MAAM,CAAC,MAAM,EAAE;QAC1B,iBAAiB,MAAM,CAAC,KAAK,EAAE;QAC/B,kBAAkB,MAAM,CAAC,MAAM,EAAE;QACjC,oBAAoB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;QAC5C,sBAAsB;KACvB,CAAC;IACF,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,OAAO,KAAK,CAAC,KAAK,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IAC9D,CAAC;IACD,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAyB,EAAE,MAAe;IAClE,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;IAC9E,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;IAChF,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IAEhF,MAAM,YAAY,GAAG,MAAM,CAAC,OAAO,CAAC,oBAAO,CAAC;SACzC,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE;QACtB,MAAM,cAAc,GAAG,CAAC,UAAU,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;QACzG,OAAO;YACL,KAAK;YACL,cAAc;YACd,eAAe,EAAE,cAAc,GAAG,UAAU;SAC7C,CAAC;IACJ,CAAC,CAAC;SACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC,CAAC;IAEvD,MAAM,OAAO,GAAG,IAAI,4BAAe,EAAE,CAAC;IACtC,MAAM,SAAS,GAAG,OAAO,CAAC,sBAAsB,CAAC;QAC/C,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,SAAS;QAC1C,WAAW,EAAE,UAAU;QACvB,YAAY,EAAE,WAAW;KAC1B,CAAC,CAAC;IAEH,IAAI,MAAM,EAAE,CAAC;QACX,OAAO,CAAC,MAAM,CAAC,KAAK,CAClB,GAAG,IAAI,CAAC,SAAS,CACf;YACE,MAAM,EAAE;gBACN,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,YAAY,EAAE,UAAU;gBACxB,aAAa,EAAE,WAAW;gBAC1B,cAAc,EAAE,UAAU;aAC3B;YACD,YAAY,EAAE,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;SACxC,EACD,IAAI,EACJ,CAAC,CACF,IAAI,CACN,CAAC;QACF,OAAO;IACT,CAAC;IAED,MAAM,KAAK,GAAG;QACZ,wBAAwB,MAAM,CAAC,MAAM,EAAE;QACvC,2BAA2B,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;QAClD,6CAA6C;QAC7C,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAC7B,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,CAAC,KAAK,MAAM,GAAG,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,GAAG,CAAC,eAAe,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAC/I;QACD,EAAE;QACF,yBAAyB;QACzB,SAAS,CAAC,IAAI;KACf,CAAC;IACF,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChD,CAAC;AAED,KAAK,UAAU,IAAI;IACjB,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7C,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC;IAC5E,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,qEAAqE,CAAC,CAAC;QAC5F,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,MAAM,GAAsB,EAAE,CAAC;IACnC,IAAI,CAAC;QACH,MAAM,GAAG,MAAM,UAAU,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,SAAS,IAAI,CAAC,CAAC;QAClE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;QACjD,OAAO;IACT,CAAC;IAED,IAAI,OAAO,CAAC,OAAO,KAAK,OAAO,EAAE,CAAC;QAChC,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QAC/B,OAAO;IACT,CAAC;IACD,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;AACzC,CAAC;AAED,KAAK,IAAI,EAAE,CAAC"}

package/dist/cjs/index.d.ts

@@ -0,0 +1,5 @@
+export { CostwiseAnthropic } from "./src/anthropic.js";
+export { CostwiseOpenAI } from "./src/openai.js";
+export { CostwiseTracker, MODEL_CAPACITIES, PRICING } from "./src/tracker.js";
+export type { CostwiseOptions, IngestPayload, CliComparisonFormat, LocalUsageEvent, ModelCapacity, ModelCapacityClass, ModelComparison, ModelComparisonByCapacity } from "./src/types.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/cjs/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACjD,OAAO,EAAE,eAAe,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC9E,YAAY,EACV,eAAe,EACf,aAAa,EACb,mBAAmB,EACnB,eAAe,EACf,aAAa,EACb,kBAAkB,EAClB,eAAe,EACf,yBAAyB,EAC1B,MAAM,gBAAgB,CAAC"}

package/dist/cjs/index.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PRICING = exports.MODEL_CAPACITIES = exports.CostwiseTracker = exports.CostwiseOpenAI = exports.CostwiseAnthropic = void 0;
+var anthropic_js_1 = require("./src/anthropic.js");
+Object.defineProperty(exports, "CostwiseAnthropic", { enumerable: true, get: function () { return anthropic_js_1.CostwiseAnthropic; } });
+var openai_js_1 = require("./src/openai.js");
+Object.defineProperty(exports, "CostwiseOpenAI", { enumerable: true, get: function () { return openai_js_1.CostwiseOpenAI; } });
+var tracker_js_1 = require("./src/tracker.js");
+Object.defineProperty(exports, "CostwiseTracker", { enumerable: true, get: function () { return tracker_js_1.CostwiseTracker; } });
+Object.defineProperty(exports, "MODEL_CAPACITIES", { enumerable: true, get: function () { return tracker_js_1.MODEL_CAPACITIES; } });
+Object.defineProperty(exports, "PRICING", { enumerable: true, get: function () { return tracker_js_1.PRICING; } });
+//# sourceMappingURL=index.js.map

package/dist/cjs/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../index.ts"],"names":[],"mappings":";;;AAAA,mDAAuD;AAA9C,iHAAA,iBAAiB,OAAA;AAC1B,6CAAiD;AAAxC,2GAAA,cAAc,OAAA;AACvB,+CAA8E;AAArE,6GAAA,eAAe,OAAA;AAAE,8GAAA,gBAAgB,OAAA;AAAE,qGAAA,OAAO,OAAA"}

package/dist/cjs/package.json

@@ -0,0 +1 @@
+{"type":"commonjs"}

package/dist/cjs/src/anthropic.d.ts

@@ -0,0 +1,10 @@
+import Anthropic from "@anthropic-ai/sdk";
+import { CostwiseOptions } from "./types.js";
+export declare class CostwiseAnthropic extends Anthropic {
+    private readonly tracker;
+    private readonly tag?;
+    constructor(options?: ConstructorParameters<typeof Anthropic>[0] & CostwiseOptions);
+    private wrapMessagesCreate;
+    private hasFinalMessage;
+}
+//# sourceMappingURL=anthropic.d.ts.map

package/dist/cjs/src/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../../src/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AA0B7C,qBAAa,iBAAkB,SAAQ,SAAS;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAkB;IAC1C,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAS;gBAElB,OAAO,GAAE,qBAAqB,CAAC,OAAO,SAAS,CAAC,CAAC,CAAC,CAAC,GAAG,eAAoB;IAOtF,OAAO,CAAC,kBAAkB;IAoC1B,OAAO,CAAC,eAAe;CAQxB"}

package/dist/cjs/src/anthropic.js

@@ -0,0 +1,55 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CostwiseAnthropic = void 0;
+const sdk_1 = __importDefault(require("@anthropic-ai/sdk"));
+const tracker_js_1 = require("./tracker.js");
+class CostwiseAnthropic extends sdk_1.default {
+    tracker;
+    tag;
+    constructor(options = {}) {
+        super(options);
+        this.tracker = new tracker_js_1.CostwiseTracker(options);
+        this.tag = options.tag;
+        this.wrapMessagesCreate();
+    }
+    wrapMessagesCreate() {
+        const originalCreate = this.messages.create.bind(this.messages);
+        this.messages.create = (async (...args) => {
+            const response = await originalCreate(...args);
+            if (this.hasFinalMessage(response)) {
+                void response
+                    .finalMessage()
+                    .then((message) => {
+                    this.tracker.trackUsage({
+                        model: message.model,
+                        inputTokens: message.usage?.input_tokens ?? 0,
+                        outputTokens: message.usage?.output_tokens ?? 0,
+                        tag: this.tag
+                    });
+                })
+                    .catch(() => { });
+            }
+            else {
+                const message = response;
+                this.tracker.trackUsage({
+                    model: message.model,
+                    inputTokens: message.usage?.input_tokens ?? 0,
+                    outputTokens: message.usage?.output_tokens ?? 0,
+                    tag: this.tag
+                });
+            }
+            return response;
+        });
+    }
+    hasFinalMessage(value) {
+        return (typeof value === "object" &&
+            value !== null &&
+            "finalMessage" in value &&
+            typeof value.finalMessage === "function");
+    }
+}
+exports.CostwiseAnthropic = CostwiseAnthropic;
+//# sourceMappingURL=anthropic.js.map

package/dist/cjs/src/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/anthropic.ts"],"names":[],"mappings":";;;;;;AAAA,4DAA0C;AAE1C,6CAA+C;AAyB/C,MAAa,iBAAkB,SAAQ,aAAS;IAC7B,OAAO,CAAkB;IACzB,GAAG,CAAU;IAE9B,YAAY,UAAwE,EAAE;QACpF,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,IAAI,4BAAe,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAEO,kBAAkB;QACxB,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAoB,CAAC;QAEnF,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,KAAK,EAAE,GAAG,IAAiC,EAAE,EAAE;YACrE,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;YAE/C,IAAI,IAAI,CAAC,eAAe,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACnC,KAAK,QAAQ;qBACV,YAAY,EAAE;qBACd,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;oBAChB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;wBACtB,KAAK,EAAE,OAAO,CAAC,KAAK;wBACpB,WAAW,EAAE,OAAO,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;wBAC7C,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;wBAC/C,GAAG,EAAE,IAAI,CAAC,GAAG;qBACd,CAAC,CAAC;gBACL,CAAC,CAAC;qBACD,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACrB,CAAC;iBAAM,CAAC;gBACN,MAAM,OAAO,GAAG,QAGf,CAAC;gBAEF,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;oBACtB,KAAK,EAAE,OAAO,CAAC,KAAK;oBACpB,WAAW,EAAE,OAAO,CAAC,KAAK,EAAE,YAAY,IAAI,CAAC;oBAC7C,YAAY,EAAE,OAAO,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBAC/C,GAAG,EAAE,IAAI,CAAC,GAAG;iBACd,CAAC,CAAC;YACL,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAoB,CAAC;IACxB,CAAC;IAEO,eAAe,CAAC,KAAc;QACpC,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;YACzB,KAAK,KAAK,IAAI;YACd,cAAc,IAAI,KAAK;YACvB,OAAQ,KAA6B,CAAC,YAAY,KAAK,UAAU,CAClE,CAAC;IACJ,CAAC;CACF;AAvDD,8CAuDC"}

package/dist/cjs/src/openai.d.ts

@@ -0,0 +1,9 @@
+import OpenAI from "openai";
+import { CostwiseOptions } from "./types.js";
+export declare class CostwiseOpenAI extends OpenAI {
+    private readonly tracker;
+    constructor(options?: ConstructorParameters<typeof OpenAI>[0] & CostwiseOptions);
+    private wrapChatCompletionsCreate;
+    private hasFinalChatCompletion;
+}
+//# sourceMappingURL=openai.d.ts.map

package/dist/cjs/src/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../../src/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAiB7C,qBAAa,cAAe,SAAQ,MAAM;IACxC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAkB;gBAE9B,OAAO,GAAE,qBAAqB,CAAC,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,eAAoB;IAMnF,OAAO,CAAC,yBAAyB;IA6BjC,OAAO,CAAC,sBAAsB;CAQ/B"}

package/dist/cjs/src/openai.js

@@ -0,0 +1,51 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CostwiseOpenAI = void 0;
+const openai_1 = __importDefault(require("openai"));
+const tracker_js_1 = require("./tracker.js");
+class CostwiseOpenAI extends openai_1.default {
+    tracker;
+    constructor(options = {}) {
+        super(options);
+        this.tracker = new tracker_js_1.CostwiseTracker(options);
+        this.wrapChatCompletionsCreate();
+    }
+    wrapChatCompletionsCreate() {
+        const originalCreate = this.chat.completions.create.bind(this.chat.completions);
+        this.chat.completions.create = (async (...args) => {
+            const response = await originalCreate(...args);
+            if (this.hasFinalChatCompletion(response)) {
+                void response
+                    .finalChatCompletion()
+                    .then((completion) => {
+                    this.tracker.trackUsage({
+                        model: completion.model,
+                        inputTokens: completion.usage?.prompt_tokens ?? 0,
+                        outputTokens: completion.usage?.completion_tokens ?? 0
+                    });
+                })
+                    .catch(() => { });
+            }
+            else {
+                const completion = response;
+                this.tracker.trackUsage({
+                    model: completion.model,
+                    inputTokens: completion.usage?.prompt_tokens ?? 0,
+                    outputTokens: completion.usage?.completion_tokens ?? 0
+                });
+            }
+            return response;
+        });
+    }
+    hasFinalChatCompletion(value) {
+        return (typeof value === "object" &&
+            value !== null &&
+            "finalChatCompletion" in value &&
+            typeof value.finalChatCompletion === "function");
+    }
+}
+exports.CostwiseOpenAI = CostwiseOpenAI;
+//# sourceMappingURL=openai.js.map

package/dist/cjs/src/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/openai.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAE5B,6CAA+C;AAgB/C,MAAa,cAAe,SAAQ,gBAAM;IACvB,OAAO,CAAkB;IAE1C,YAAY,UAAqE,EAAE;QACjF,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,IAAI,4BAAe,CAAC,OAAO,CAAC,CAAC;QAC5C,IAAI,CAAC,yBAAyB,EAAE,CAAC;IACnC,CAAC;IAEO,yBAAyB;QAC/B,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAqB,CAAC;QAEpG,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,KAAK,EAAE,GAAG,IAAkC,EAAE,EAAE;YAC9E,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,GAAG,IAAI,CAAC,CAAC;YAE/C,IAAI,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC1C,KAAK,QAAQ;qBACV,mBAAmB,EAAE;qBACrB,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE;oBACnB,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;wBACtB,KAAK,EAAE,UAAU,CAAC,KAAK;wBACvB,WAAW,EAAE,UAAU,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;wBACjD,YAAY,EAAE,UAAU,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;qBACvD,CAAC,CAAC;gBACL,CAAC,CAAC;qBACD,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACrB,CAAC;iBAAM,CAAC;gBACN,MAAM,UAAU,GAAG,QAA4B,CAAC;gBAChD,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;oBACtB,KAAK,EAAE,UAAU,CAAC,KAAK;oBACvB,WAAW,EAAE,UAAU,CAAC,KAAK,EAAE,aAAa,IAAI,CAAC;oBACjD,YAAY,EAAE,UAAU,CAAC,KAAK,EAAE,iBAAiB,IAAI,CAAC;iBACvD,CAAC,CAAC;YACL,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAqB,CAAC;IACzB,CAAC;IAEO,sBAAsB,CAAC,KAAc;QAC3C,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;YACzB,KAAK,KAAK,IAAI;YACd,qBAAqB,IAAI,KAAK;YAC9B,OAAQ,KAA0B,CAAC,mBAAmB,KAAK,UAAU,CACtE,CAAC;IACJ,CAAC;CACF;AA9CD,wCA8CC"}

package/dist/cjs/src/tracker.d.ts

@@ -0,0 +1,23 @@
+import { CostwiseOptions, ModelCapacity, CliComparisonFormat, ModelComparison, ModelComparisonByCapacity, UsageSnapshot } from "./types.js";
+type ModelRates = {
+    input: number;
+    output: number;
+};
+declare const PRICING: Record<string, ModelRates>;
+declare const MODEL_CAPACITIES: Record<string, ModelCapacity>;
+export declare class CostwiseTracker {
+    private readonly apiKey?;
+    private readonly tag?;
+    private readonly localUsagePath?;
+    constructor(options?: CostwiseOptions);
+    trackUsage(snapshot: UsageSnapshot): void;
+    compareCosts(snapshot: UsageSnapshot): ModelComparison[];
+    compareCostsByCapacity(snapshot: UsageSnapshot): ModelComparisonByCapacity;
+    summarizeComparison(snapshot: UsageSnapshot): string;
+    formatComparisonForCli(snapshot: UsageSnapshot): CliComparisonFormat;
+    private createPayload;
+    private estimateCost;
+    private calculateCost;
+}
+export { MODEL_CAPACITIES, PRICING };
+//# sourceMappingURL=tracker.d.ts.map

package/dist/cjs/src/tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracker.d.ts","sourceRoot":"","sources":["../../../src/tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,eAAe,EAGf,aAAa,EAEb,mBAAmB,EACnB,eAAe,EACf,yBAAyB,EACzB,aAAa,EACd,MAAM,YAAY,CAAC;AAGpB,KAAK,UAAU,GAAG;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AA6BF,QAAA,MAAM,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,CAGvC,CAAC;AAEF,QAAA,MAAM,gBAAgB,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,CAsBnD,CAAC;AAIF,qBAAa,eAAe;IAC1B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAS;IAC9B,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAS;gBAE7B,OAAO,GAAE,eAAoB;IAMzC,UAAU,CAAC,QAAQ,EAAE,aAAa,GAAG,IAAI;IA8BzC,YAAY,CAAC,QAAQ,EAAE,aAAa,GAAG,eAAe,EAAE;IAqBxD,sBAAsB,CAAC,QAAQ,EAAE,aAAa,GAAG,yBAAyB;IAgC1E,mBAAmB,CAAC,QAAQ,EAAE,aAAa,GAAG,MAAM;IAWpD,sBAAsB,CAAC,QAAQ,EAAE,aAAa,GAAG,mBAAmB;IA+BpE,OAAO,CAAC,aAAa;IAyBrB,OAAO,CAAC,YAAY;IAQpB,OAAO,CAAC,aAAa;CAKtB;AAED,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,CAAC"}