@costwise/sdk 0.0.1 → 0.0.3
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +176 -0
- package/dist/cjs/cli.d.ts +3 -0
- package/dist/cjs/cli.d.ts.map +1 -0
- package/dist/cjs/cli.js +156 -0
- package/dist/cjs/cli.js.map +1 -0
- package/dist/cjs/index.d.ts +5 -0
- package/dist/cjs/index.d.ts.map +1 -0
- package/dist/cjs/index.js +12 -0
- package/dist/cjs/index.js.map +1 -0
- package/dist/cjs/package.json +1 -0
- package/dist/cjs/src/anthropic.d.ts +10 -0
- package/dist/cjs/src/anthropic.d.ts.map +1 -0
- package/dist/cjs/src/anthropic.js +55 -0
- package/dist/cjs/src/anthropic.js.map +1 -0
- package/dist/cjs/src/openai.d.ts +9 -0
- package/dist/cjs/src/openai.d.ts.map +1 -0
- package/dist/cjs/src/openai.js +51 -0
- package/dist/cjs/src/openai.js.map +1 -0
- package/dist/cjs/src/tracker.d.ts +23 -0
- package/dist/cjs/src/tracker.d.ts.map +1 -0
- package/dist/cjs/src/tracker.js +212 -0
- package/dist/cjs/src/tracker.js.map +1 -0
- package/dist/cjs/src/types.d.ts +56 -0
- package/dist/cjs/src/types.d.ts.map +1 -0
- package/dist/cjs/src/types.js +3 -0
- package/dist/cjs/src/types.js.map +1 -0
- package/dist/esm/cli.d.ts +3 -0
- package/dist/esm/cli.d.ts.map +1 -0
- package/dist/esm/cli.js +154 -0
- package/dist/esm/cli.js.map +1 -0
- package/dist/esm/index.d.ts +5 -0
- package/dist/esm/index.d.ts.map +1 -0
- package/dist/esm/index.js +4 -0
- package/dist/esm/index.js.map +1 -0
- package/dist/esm/src/anthropic.d.ts +10 -0
- package/dist/esm/src/anthropic.d.ts.map +1 -0
- package/dist/esm/src/anthropic.js +48 -0
- package/dist/esm/src/anthropic.js.map +1 -0
- package/dist/esm/src/openai.d.ts +9 -0
- package/dist/esm/src/openai.d.ts.map +1 -0
- package/dist/esm/src/openai.js +44 -0
- package/dist/esm/src/openai.js.map +1 -0
- package/dist/esm/src/tracker.d.ts +23 -0
- package/dist/esm/src/tracker.d.ts.map +1 -0
- package/dist/esm/src/tracker.js +207 -0
- package/dist/esm/src/tracker.js.map +1 -0
- package/dist/esm/src/types.d.ts +56 -0
- package/dist/esm/src/types.d.ts.map +1 -0
- package/dist/esm/src/types.js +2 -0
- package/dist/esm/src/types.js.map +1 -0
- package/package.json +36 -2
- package/index.js +0 -3
|
@@ -0,0 +1,212 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.PRICING = exports.MODEL_CAPACITIES = exports.CostwiseTracker = void 0;
|
|
4
|
+
const promises_1 = require("node:fs/promises");
|
|
5
|
+
const OPENAI_PRICING = {
|
|
6
|
+
"gpt-5.4": { input: 2.5, output: 15 },
|
|
7
|
+
"gpt-5.4-mini": { input: 0.75, output: 4.5 },
|
|
8
|
+
"gpt-5.4-nano": { input: 0.2, output: 1.25 },
|
|
9
|
+
"gpt-5.4-pro": { input: 30, output: 180 },
|
|
10
|
+
"gpt-5.2": { input: 1.75, output: 14 },
|
|
11
|
+
"gpt-5.2-pro": { input: 21, output: 168 },
|
|
12
|
+
"o1": { input: 15, output: 60 },
|
|
13
|
+
"o1-pro": { input: 150, output: 600 },
|
|
14
|
+
"o3-mini": { input: 1.1, output: 4.4 },
|
|
15
|
+
"gpt-4o": { input: 2.5, output: 10 },
|
|
16
|
+
"gpt-4o-mini": { input: 0.15, output: 0.6 }
|
|
17
|
+
};
|
|
18
|
+
const ANTHROPIC_PRICING = {
|
|
19
|
+
"claude-opus-4.6": { input: 5, output: 25 },
|
|
20
|
+
"claude-opus-4.5": { input: 5, output: 25 },
|
|
21
|
+
"claude-opus-4.1": { input: 15, output: 75 },
|
|
22
|
+
"claude-opus-4": { input: 15, output: 75 },
|
|
23
|
+
"claude-sonnet-4.6": { input: 3, output: 15 },
|
|
24
|
+
"claude-sonnet-4.5": { input: 3, output: 15 },
|
|
25
|
+
"claude-sonnet-4": { input: 3, output: 15 },
|
|
26
|
+
"claude-haiku-4.5": { input: 1, output: 5 },
|
|
27
|
+
"claude-haiku-3.5": { input: 0.8, output: 4 },
|
|
28
|
+
"claude-haiku-3": { input: 0.25, output: 1.25 }
|
|
29
|
+
};
|
|
30
|
+
const PRICING = {
|
|
31
|
+
...OPENAI_PRICING,
|
|
32
|
+
...ANTHROPIC_PRICING
|
|
33
|
+
};
|
|
34
|
+
exports.PRICING = PRICING;
|
|
35
|
+
const MODEL_CAPACITIES = {
|
|
36
|
+
"gpt-5.4": { contextWindowTokens: 270_000, capacityClass: "frontier" },
|
|
37
|
+
"gpt-5.4-mini": { contextWindowTokens: 270_000, capacityClass: "high" },
|
|
38
|
+
"gpt-5.4-nano": { contextWindowTokens: 270_000, capacityClass: "nano" },
|
|
39
|
+
"gpt-5.4-pro": { contextWindowTokens: 270_000, capacityClass: "frontier" },
|
|
40
|
+
"gpt-5.2": { contextWindowTokens: 270_000, capacityClass: "high" },
|
|
41
|
+
"gpt-5.2-pro": { contextWindowTokens: 270_000, capacityClass: "frontier" },
|
|
42
|
+
o1: { capacityClass: "frontier" },
|
|
43
|
+
"o1-pro": { capacityClass: "frontier" },
|
|
44
|
+
"o3-mini": { capacityClass: "mini" },
|
|
45
|
+
"gpt-4o": { capacityClass: "high" },
|
|
46
|
+
"gpt-4o-mini": { capacityClass: "mini" },
|
|
47
|
+
"claude-opus-4.6": { contextWindowTokens: 1_000_000, capacityClass: "frontier" },
|
|
48
|
+
"claude-opus-4.5": { contextWindowTokens: 200_000, capacityClass: "frontier" },
|
|
49
|
+
"claude-opus-4.1": { contextWindowTokens: 200_000, capacityClass: "frontier" },
|
|
50
|
+
"claude-opus-4": { contextWindowTokens: 200_000, capacityClass: "frontier" },
|
|
51
|
+
"claude-sonnet-4.6": { contextWindowTokens: 1_000_000, capacityClass: "high" },
|
|
52
|
+
"claude-sonnet-4.5": { contextWindowTokens: 1_000_000, capacityClass: "high" },
|
|
53
|
+
"claude-sonnet-4": { contextWindowTokens: 1_000_000, capacityClass: "high" },
|
|
54
|
+
"claude-haiku-4.5": { contextWindowTokens: 200_000, capacityClass: "mini" },
|
|
55
|
+
"claude-haiku-3.5": { contextWindowTokens: 200_000, capacityClass: "mini" },
|
|
56
|
+
"claude-haiku-3": { contextWindowTokens: 200_000, capacityClass: "mini" }
|
|
57
|
+
};
|
|
58
|
+
exports.MODEL_CAPACITIES = MODEL_CAPACITIES;
|
|
59
|
+
const INGEST_URL = "https://api.costwise.dev/ingest";
|
|
60
|
+
class CostwiseTracker {
|
|
61
|
+
apiKey;
|
|
62
|
+
tag;
|
|
63
|
+
localUsagePath;
|
|
64
|
+
constructor(options = {}) {
|
|
65
|
+
this.apiKey = options.costwiseKey ?? options.costswiseKey ?? options.apiKey ?? process.env.COSTWISE_API_KEY;
|
|
66
|
+
this.tag = options.tag;
|
|
67
|
+
this.localUsagePath = options.localUsagePath ?? process.env.COSTWISE_LOCAL_USAGE_PATH;
|
|
68
|
+
}
|
|
69
|
+
trackUsage(snapshot) {
|
|
70
|
+
const payload = this.createPayload(snapshot);
|
|
71
|
+
if (!payload) {
|
|
72
|
+
return;
|
|
73
|
+
}
|
|
74
|
+
if (this.apiKey) {
|
|
75
|
+
void fetch(INGEST_URL, {
|
|
76
|
+
method: "POST",
|
|
77
|
+
headers: {
|
|
78
|
+
"content-type": "application/json",
|
|
79
|
+
authorization: `Bearer ${this.apiKey}`
|
|
80
|
+
},
|
|
81
|
+
body: JSON.stringify(payload)
|
|
82
|
+
}).catch(() => { });
|
|
83
|
+
}
|
|
84
|
+
if (this.localUsagePath) {
|
|
85
|
+
const localEvent = {
|
|
86
|
+
model: payload.model,
|
|
87
|
+
input_tokens: payload.input_tokens,
|
|
88
|
+
output_tokens: payload.output_tokens,
|
|
89
|
+
estimated_cost: payload.estimated_cost,
|
|
90
|
+
tag: payload.tag,
|
|
91
|
+
timestamp: payload.timestamp
|
|
92
|
+
};
|
|
93
|
+
void (0, promises_1.appendFile)(this.localUsagePath, `${JSON.stringify(localEvent)}\n`, "utf8").catch(() => { });
|
|
94
|
+
}
|
|
95
|
+
}
|
|
96
|
+
compareCosts(snapshot) {
|
|
97
|
+
const currentCost = this.estimateCost(snapshot.model, snapshot.inputTokens, snapshot.outputTokens);
|
|
98
|
+
if (currentCost === null) {
|
|
99
|
+
return [];
|
|
100
|
+
}
|
|
101
|
+
return Object.entries(PRICING)
|
|
102
|
+
.map(([model, rates]) => {
|
|
103
|
+
const estimatedCost = this.calculateCost(snapshot.inputTokens, snapshot.outputTokens, rates);
|
|
104
|
+
return {
|
|
105
|
+
model,
|
|
106
|
+
inputRatePerMTok: rates.input,
|
|
107
|
+
outputRatePerMTok: rates.output,
|
|
108
|
+
estimatedCost,
|
|
109
|
+
deltaFromCurrentCost: estimatedCost - currentCost,
|
|
110
|
+
capacity: MODEL_CAPACITIES[model] ?? { capacityClass: "standard" }
|
|
111
|
+
};
|
|
112
|
+
})
|
|
113
|
+
.sort((a, b) => a.estimatedCost - b.estimatedCost);
|
|
114
|
+
}
|
|
115
|
+
compareCostsByCapacity(snapshot) {
|
|
116
|
+
const all = this.compareCosts(snapshot);
|
|
117
|
+
if (all.length === 0) {
|
|
118
|
+
return { byClass: {}, cheapestPerClass: {} };
|
|
119
|
+
}
|
|
120
|
+
const byClass = {};
|
|
121
|
+
for (const c of all) {
|
|
122
|
+
const cls = c.capacity.capacityClass;
|
|
123
|
+
const list = byClass[cls] ?? [];
|
|
124
|
+
list.push(c);
|
|
125
|
+
byClass[cls] = list;
|
|
126
|
+
}
|
|
127
|
+
for (const cls of Object.keys(byClass)) {
|
|
128
|
+
const list = byClass[cls];
|
|
129
|
+
if (list) {
|
|
130
|
+
list.sort((a, b) => a.estimatedCost - b.estimatedCost);
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
const cheapestPerClass = {};
|
|
134
|
+
for (const cls of Object.keys(byClass)) {
|
|
135
|
+
const list = byClass[cls];
|
|
136
|
+
if (list?.[0]) {
|
|
137
|
+
cheapestPerClass[cls] = list[0];
|
|
138
|
+
}
|
|
139
|
+
}
|
|
140
|
+
return { byClass, cheapestPerClass };
|
|
141
|
+
}
|
|
142
|
+
summarizeComparison(snapshot) {
|
|
143
|
+
const comparisons = this.compareCosts(snapshot);
|
|
144
|
+
if (comparisons.length === 0) {
|
|
145
|
+
return `No pricing entry found for model ${snapshot.model}.`;
|
|
146
|
+
}
|
|
147
|
+
const cheapest = comparisons[0];
|
|
148
|
+
const priciest = comparisons[comparisons.length - 1];
|
|
149
|
+
return `For ${snapshot.inputTokens} input + ${snapshot.outputTokens} output tokens, cheapest is ${cheapest.model} ($${cheapest.estimatedCost.toFixed(6)}), priciest is ${priciest.model} ($${priciest.estimatedCost.toFixed(6)}).`;
|
|
150
|
+
}
|
|
151
|
+
formatComparisonForCli(snapshot) {
|
|
152
|
+
const comparisons = this.compareCosts(snapshot);
|
|
153
|
+
if (comparisons.length === 0) {
|
|
154
|
+
const text = `Costwise: no pricing entry found for model ${snapshot.model}.`;
|
|
155
|
+
return { text, lines: [text] };
|
|
156
|
+
}
|
|
157
|
+
const byCapacity = this.compareCostsByCapacity(snapshot);
|
|
158
|
+
const cheapest = comparisons[0];
|
|
159
|
+
const current = comparisons.find((c) => c.model === snapshot.model) ?? null;
|
|
160
|
+
const delta = current ? current.estimatedCost - cheapest.estimatedCost : 0;
|
|
161
|
+
const lines = [
|
|
162
|
+
`Costwise comparison for ${snapshot.model}`,
|
|
163
|
+
`Usage: ${snapshot.inputTokens} input / ${snapshot.outputTokens} output tokens`,
|
|
164
|
+
`Current model cost: $${(current?.estimatedCost ?? 0).toFixed(6)}`,
|
|
165
|
+
`Cheapest overall: ${cheapest.model} ($${cheapest.estimatedCost.toFixed(6)})`,
|
|
166
|
+
`Potential savings: $${Math.max(0, delta).toFixed(6)}`
|
|
167
|
+
];
|
|
168
|
+
const classes = ["nano", "mini", "standard", "high", "frontier"];
|
|
169
|
+
for (const cls of classes) {
|
|
170
|
+
const best = byCapacity.cheapestPerClass[cls];
|
|
171
|
+
if (best) {
|
|
172
|
+
lines.push(`Best ${cls}: ${best.model} ($${best.estimatedCost.toFixed(6)})`);
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
return { text: lines.join("\n"), lines };
|
|
176
|
+
}
|
|
177
|
+
createPayload(snapshot) {
|
|
178
|
+
const cost = this.estimateCost(snapshot.model, snapshot.inputTokens, snapshot.outputTokens);
|
|
179
|
+
if (cost === null) {
|
|
180
|
+
return null;
|
|
181
|
+
}
|
|
182
|
+
const input_tokens = Math.max(0, snapshot.inputTokens || 0);
|
|
183
|
+
const output_tokens = Math.max(0, snapshot.outputTokens || 0);
|
|
184
|
+
const tag = snapshot.tag ?? this.tag;
|
|
185
|
+
const timestamp = new Date().toISOString();
|
|
186
|
+
return {
|
|
187
|
+
model: snapshot.model,
|
|
188
|
+
input_tokens,
|
|
189
|
+
output_tokens,
|
|
190
|
+
cost,
|
|
191
|
+
estimated_cost: cost,
|
|
192
|
+
tag,
|
|
193
|
+
endpoint_tag: tag,
|
|
194
|
+
ts: timestamp,
|
|
195
|
+
timestamp
|
|
196
|
+
};
|
|
197
|
+
}
|
|
198
|
+
estimateCost(model, inputTokens, outputTokens) {
|
|
199
|
+
const rates = PRICING[model];
|
|
200
|
+
if (!rates) {
|
|
201
|
+
return null;
|
|
202
|
+
}
|
|
203
|
+
return this.calculateCost(inputTokens, outputTokens, rates);
|
|
204
|
+
}
|
|
205
|
+
calculateCost(inputTokens, outputTokens, rates) {
|
|
206
|
+
const input = Math.max(0, inputTokens || 0);
|
|
207
|
+
const output = Math.max(0, outputTokens || 0);
|
|
208
|
+
return (input / 1_000_000) * rates.input + (output / 1_000_000) * rates.output;
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
exports.CostwiseTracker = CostwiseTracker;
|
|
212
|
+
//# sourceMappingURL=tracker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracker.js","sourceRoot":"","sources":["../../../src/tracker.ts"],"names":[],"mappings":";;;AAWA,+CAA8C;AAO9C,MAAM,cAAc,GAA+B;IACjD,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE;IACrC,cAAc,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE;IAC5C,cAAc,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE;IAC5C,aAAa,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE;IACzC,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE;IACtC,aAAa,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE;IACzC,IAAI,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IAC/B,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IACrC,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE;IACtC,QAAQ,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE;IACpC,aAAa,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE;CAC5C,CAAC;AAEF,MAAM,iBAAiB,GAA+B;IACpD,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC3C,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC3C,iBAAiB,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IAC5C,eAAe,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;IAC1C,mBAAmB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC7C,mBAAmB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC7C,iBAAiB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;IAC3C,kBAAkB,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;IAC3C,kBAAkB,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,CAAC,EAAE;IAC7C,gBAAgB,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;CAChD,CAAC;AAEF,MAAM,OAAO,GAA+B;IAC1C,GAAG,cAAc;IACjB,GAAG,iBAAiB;CACrB,CAAC;AA4MyB,0BAAO;AA1MlC,MAAM,gBAAgB,GAAkC;IACtD,SAAS,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE;IACtE,cAAc,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE;IACvE,cAAc,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE;IACvE,aAAa,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE;IAC1E,SAAS,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE;IAClE,aAAa,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE;IAC1E,EAAE,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE;IACjC,QAAQ,EAAE,EAAE,aAAa,EAAE,UAAU,EAAE;IACvC,SAAS,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;IACpC,QAAQ,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;IACnC,aAAa,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE;IACxC,iBAAiB,EAAE,EAAE,mBAAmB,EAAE,SAAS,EAAE,aAAa,EAAE,UAAU,EAAE;IAChF,iBAAiB,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE;IAC9E,iBAAiB,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE;IAC9E,eAAe,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE;IAC5E,mBAAmB,EAAE,EAAE,mBAAmB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE;IAC9E,mBAAmB,EAAE,EAAE,mBAAmB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE;IAC9E,iBAAiB,EAAE,EAAE,mBAAmB,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE;IAC5E,kBAAkB,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE;IAC3E,kBAAkB,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE;IAC3E,gBAAgB,EAAE,EAAE,mBAAmB,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE;CAC1E,CAAC;AAoLO,4CAAgB;AAlLzB,MAAM,UAAU,GAAG,iCAAiC,CAAC;AAErD,MAAa,eAAe;IACT,MAAM,CAAU;IAChB,GAAG,CAAU;IACb,cAAc,CAAU;IAEzC,YAAY,UAA2B,EAAE;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;QAC5G,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC;IACxF,CAAC;IAED,UAAU,CAAC,QAAuB;QAChC,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,KAAK,KAAK,CAAC,UAAU,EAAE;gBACrB,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;iBACvC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;aAC9B,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACrB,CAAC;QAED,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,UAAU,GAAoB;gBAClC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,YAAY,EAAE,OAAO,CAAC,YAAY;gBAClC,aAAa,EAAE,OAAO,CAAC,aAAa;gBACpC,cAAc,EAAE,OAAO,CAAC,cAAc;gBACtC,GAAG,EAAE,OAAO,CAAC,GAAG;gBAChB,SAAS,EAAE,OAAO,CAAC,SAAS;aAC7B,CAAC;YACF,KAAK,IAAA,qBAAU,EAAC,IAAI,CAAC,cAAc,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAClG,CAAC;IACH,CAAC;IAED,YAAY,CAAC,QAAuB;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,WAAW,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC;QACnG,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;YACzB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,OAAO,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC;aAC3B,GAAG,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE;YACtB,MAAM,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,WAAW,EAAE,QAAQ,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;YAC7F,OAAO;gBACL,KAAK;gBACL,gBAAgB,EAAE,KAAK,CAAC,KAAK;gBAC7B,iBAAiB,EAAE,KAAK,CAAC,MAAM;gBAC/B,aAAa;gBACb,oBAAoB,EAAE,aAAa,GAAG,WAAW;gBACjD,QAAQ,EAAE,gBAAgB,CAAC,KAAK,CAAC,IAAI,EAAE,aAAa,EAAE,UAAU,EAAE;aACnE,CAAC;QACJ,CAAC,CAAC;aACD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC;IACvD,CAAC;IAED,sBAAsB,CAAC,QAAuB;QAC5C,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,CAAC;QAC/C,CAAC;QAED,MAAM,OAAO,GAA2D,EAAE,CAAC;QAC3E,KAAK,MAAM,CAAC,IAAI,GAAG,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC;YACrC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;QACtB,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAyB,EAAE,CAAC;YAC/D,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;YAC1B,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QAED,MAAM,gBAAgB,GAAyD,EAAE,CAAC;QAClF,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAyB,EAAE,CAAC;YAC/D,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;YAC1B,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACd,gBAAgB,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;QAED,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;IACvC,CAAC;IAED,mBAAmB,CAAC,QAAuB;QACzC,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO,oCAAoC,QAAQ,CAAC,KAAK,GAAG,CAAC;QAC/D,CAAC;QAED,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,WAAW,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACrD,OAAO,OAAO,QAAQ,CAAC,WAAW,YAAY,QAAQ,CAAC,YAAY,+BAA+B,QAAQ,CAAC,KAAK,MAAM,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB,QAAQ,CAAC,KAAK,MAAM,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IACrO,CAAC;IAED,sBAAsB,CAAC,QAAuB;QAC5C,MAAM,WAAW,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,8CAA8C,QAAQ,CAAC,KAAK,GAAG,CAAC;YAC7E,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACjC,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,OAAO,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC;QAC5E,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QAE3E,MAAM,KAAK,GAAa;YACtB,2BAA2B,QAAQ,CAAC,KAAK,EAAE;YAC3C,UAAU,QAAQ,CAAC,WAAW,YAAY,QAAQ,CAAC,YAAY,gBAAgB;YAC/E,wBAAwB,CAAC,OAAO,EAAE,aAAa,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAClE,qBAAqB,QAAQ,CAAC,KAAK,MAAM,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;YAC7E,uBAAuB,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;SACvD,CAAC;QAEF,MAAM,OAAO,GAAyB,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACvF,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,UAAU,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;YAC9C,IAAI,IAAI,EAAE,CAAC;gBACT,KAAK,CAAC,IAAI,CAAC,QAAQ,GAAG,KAAK,IAAI,CAAC,KAAK,MAAM,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/E,CAAC;QACH,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC;IAC3C,CAAC;IAEO,aAAa,CAAC,QAAuB;QAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,WAAW,EAAE,QAAQ,CAAC,YAAY,CAAC,CAAC;QAC5F,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;QAC5D,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC;QAE9D,MAAM,GAAG,GAAG,QAAQ,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;QACrC,MAAM,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAE3C,OAAO;YACL,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,YAAY;YACZ,aAAa;YACb,IAAI;YACJ,cAAc,EAAE,IAAI;YACpB,GAAG;YACH,YAAY,EAAE,GAAG;YACjB,EAAE,EAAE,SAAS;YACb,SAAS;SACV,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,KAAa,EAAE,WAAmB,EAAE,YAAoB;QAC3E,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,YAAY,EAAE,KAAK,CAAC,CAAC;IAC9D,CAAC;IAEO,aAAa,CAAC,WAAmB,EAAE,YAAoB,EAAE,KAAiB;QAChF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,IAAI,CAAC,CAAC,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC,CAAC;QAC9C,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,KAAK,GAAG,CAAC,MAAM,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,MAAM,CAAC;IACjF,CAAC;CACF;AA9KD,0CA8KC"}
|
|
@@ -0,0 +1,56 @@
|
|
|
1
|
+
export type CostwiseOptions = {
|
|
2
|
+
apiKey?: string;
|
|
3
|
+
costwiseKey?: string;
|
|
4
|
+
costswiseKey?: string;
|
|
5
|
+
tag?: string;
|
|
6
|
+
localUsagePath?: string;
|
|
7
|
+
};
|
|
8
|
+
export type IngestPayload = {
|
|
9
|
+
model: string;
|
|
10
|
+
input_tokens: number;
|
|
11
|
+
output_tokens: number;
|
|
12
|
+
cost: number;
|
|
13
|
+
estimated_cost: number;
|
|
14
|
+
tag?: string;
|
|
15
|
+
endpoint_tag?: string;
|
|
16
|
+
ts: string;
|
|
17
|
+
timestamp: string;
|
|
18
|
+
};
|
|
19
|
+
export type UsageSnapshot = {
|
|
20
|
+
model: string;
|
|
21
|
+
inputTokens: number;
|
|
22
|
+
outputTokens: number;
|
|
23
|
+
tag?: string;
|
|
24
|
+
};
|
|
25
|
+
export type ModelCapacity = {
|
|
26
|
+
contextWindowTokens?: number;
|
|
27
|
+
outputLimitTokens?: number;
|
|
28
|
+
capacityClass: "nano" | "mini" | "standard" | "high" | "frontier";
|
|
29
|
+
};
|
|
30
|
+
export type ModelComparison = {
|
|
31
|
+
model: string;
|
|
32
|
+
inputRatePerMTok: number;
|
|
33
|
+
outputRatePerMTok: number;
|
|
34
|
+
estimatedCost: number;
|
|
35
|
+
deltaFromCurrentCost: number;
|
|
36
|
+
capacity: ModelCapacity;
|
|
37
|
+
};
|
|
38
|
+
export type ModelCapacityClass = ModelCapacity["capacityClass"];
|
|
39
|
+
/** Grouped comparisons: each class lists models cheapest-first; cheapestPerClass picks one model per class. */
|
|
40
|
+
export type ModelComparisonByCapacity = {
|
|
41
|
+
byClass: Partial<Record<ModelCapacityClass, ModelComparison[]>>;
|
|
42
|
+
cheapestPerClass: Partial<Record<ModelCapacityClass, ModelComparison>>;
|
|
43
|
+
};
|
|
44
|
+
export type CliComparisonFormat = {
|
|
45
|
+
text: string;
|
|
46
|
+
lines: string[];
|
|
47
|
+
};
|
|
48
|
+
export type LocalUsageEvent = {
|
|
49
|
+
model: string;
|
|
50
|
+
input_tokens: number;
|
|
51
|
+
output_tokens: number;
|
|
52
|
+
estimated_cost: number;
|
|
53
|
+
tag?: string;
|
|
54
|
+
timestamp: string;
|
|
55
|
+
};
|
|
56
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,eAAe,GAAG;IAC5B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,IAAI,EAAE,MAAM,CAAC;IACb,cAAc,EAAE,MAAM,CAAC;IACvB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,GAAG,CAAC,EAAE,MAAM,CAAC;CACd,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,aAAa,EAAE,MAAM,GAAG,MAAM,GAAG,UAAU,GAAG,MAAM,GAAG,UAAU,CAAC;CACnE,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,gBAAgB,EAAE,MAAM,CAAC;IACzB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,QAAQ,EAAE,aAAa,CAAC;CACzB,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG,aAAa,CAAC,eAAe,CAAC,CAAC;AAEhE,+GAA+G;AAC/G,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,kBAAkB,EAAE,eAAe,EAAE,CAAC,CAAC,CAAC;IAChE,gBAAgB,EAAE,OAAO,CAAC,MAAM,CAAC,kBAAkB,EAAE,eAAe,CAAC,CAAC,CAAC;CACxE,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,EAAE,MAAM,CAAC;IACrB,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,CAAC;IACvB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACnB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cli.d.ts","sourceRoot":"","sources":["../../cli.ts"],"names":[],"mappings":""}
|
package/dist/esm/cli.js
ADDED
|
@@ -0,0 +1,154 @@
|
|
|
1
|
+
#!/usr/bin/env node
|
|
2
|
+
import { readFile } from "node:fs/promises";
|
|
3
|
+
import { CostwiseTracker, PRICING } from "./src/tracker.js";
|
|
4
|
+
function help(exitCode) {
|
|
5
|
+
process.stdout.write([
|
|
6
|
+
"Costwise CLI (actual usage only)",
|
|
7
|
+
"",
|
|
8
|
+
"Usage:",
|
|
9
|
+
" costwise usage [--file <path>] [--json]",
|
|
10
|
+
" costwise compare-actual [--file <path>] [--json]",
|
|
11
|
+
"",
|
|
12
|
+
"Notes:",
|
|
13
|
+
" - Reads usage events from JSONL recorded by SDK (`COSTWISE_LOCAL_USAGE_PATH`).",
|
|
14
|
+
" - No manual token input flags are supported."
|
|
15
|
+
].join("\n") + "\n");
|
|
16
|
+
process.exit(exitCode);
|
|
17
|
+
}
|
|
18
|
+
function parse(argv) {
|
|
19
|
+
if (argv.length === 0 || argv.includes("--help") || argv.includes("-h")) {
|
|
20
|
+
help(0);
|
|
21
|
+
}
|
|
22
|
+
const options = { json: false };
|
|
23
|
+
const first = argv[0];
|
|
24
|
+
if (first === "usage" || first === "compare-actual") {
|
|
25
|
+
options.command = first;
|
|
26
|
+
}
|
|
27
|
+
else {
|
|
28
|
+
help(1);
|
|
29
|
+
}
|
|
30
|
+
for (let i = 1; i < argv.length; i += 1) {
|
|
31
|
+
const arg = argv[i];
|
|
32
|
+
const next = argv[i + 1];
|
|
33
|
+
if (arg === "--file" && next) {
|
|
34
|
+
options.filePath = next;
|
|
35
|
+
i += 1;
|
|
36
|
+
continue;
|
|
37
|
+
}
|
|
38
|
+
if (arg === "--json") {
|
|
39
|
+
options.json = true;
|
|
40
|
+
continue;
|
|
41
|
+
}
|
|
42
|
+
help(1);
|
|
43
|
+
}
|
|
44
|
+
return options;
|
|
45
|
+
}
|
|
46
|
+
async function loadEvents(path) {
|
|
47
|
+
const content = await readFile(path, "utf8");
|
|
48
|
+
const lines = content.split("\n").map((line) => line.trim()).filter(Boolean);
|
|
49
|
+
return lines.map((line) => JSON.parse(line));
|
|
50
|
+
}
|
|
51
|
+
function runUsage(events, asJson) {
|
|
52
|
+
const totals = events.reduce((acc, event) => {
|
|
53
|
+
acc.input += event.input_tokens;
|
|
54
|
+
acc.output += event.output_tokens;
|
|
55
|
+
acc.cost += event.estimated_cost;
|
|
56
|
+
acc.byModel[event.model] = (acc.byModel[event.model] ?? 0) + event.estimated_cost;
|
|
57
|
+
return acc;
|
|
58
|
+
}, { input: 0, output: 0, cost: 0, byModel: {} });
|
|
59
|
+
const topModels = Object.entries(totals.byModel)
|
|
60
|
+
.map(([model, cost]) => ({ model, cost }))
|
|
61
|
+
.sort((a, b) => b.cost - a.cost)
|
|
62
|
+
.slice(0, 5);
|
|
63
|
+
if (asJson) {
|
|
64
|
+
process.stdout.write(`${JSON.stringify({
|
|
65
|
+
events: events.length,
|
|
66
|
+
input_tokens: totals.input,
|
|
67
|
+
output_tokens: totals.output,
|
|
68
|
+
estimated_cost: totals.cost,
|
|
69
|
+
top_models: topModels
|
|
70
|
+
}, null, 2)}\n`);
|
|
71
|
+
return;
|
|
72
|
+
}
|
|
73
|
+
const lines = [
|
|
74
|
+
`Events: ${events.length}`,
|
|
75
|
+
`Input tokens: ${totals.input}`,
|
|
76
|
+
`Output tokens: ${totals.output}`,
|
|
77
|
+
`Estimated cost: $${totals.cost.toFixed(6)}`,
|
|
78
|
+
"Top models by spend:"
|
|
79
|
+
];
|
|
80
|
+
for (const model of topModels) {
|
|
81
|
+
lines.push(` - ${model.model}: $${model.cost.toFixed(6)}`);
|
|
82
|
+
}
|
|
83
|
+
process.stdout.write(`${lines.join("\n")}\n`);
|
|
84
|
+
}
|
|
85
|
+
function runCompareActual(events, asJson) {
|
|
86
|
+
const totalInput = events.reduce((sum, event) => sum + event.input_tokens, 0);
|
|
87
|
+
const totalOutput = events.reduce((sum, event) => sum + event.output_tokens, 0);
|
|
88
|
+
const actualCost = events.reduce((sum, event) => sum + event.estimated_cost, 0);
|
|
89
|
+
const alternatives = Object.entries(PRICING)
|
|
90
|
+
.map(([model, rates]) => {
|
|
91
|
+
const estimated_cost = (totalInput / 1_000_000) * rates.input + (totalOutput / 1_000_000) * rates.output;
|
|
92
|
+
return {
|
|
93
|
+
model,
|
|
94
|
+
estimated_cost,
|
|
95
|
+
delta_vs_actual: estimated_cost - actualCost
|
|
96
|
+
};
|
|
97
|
+
})
|
|
98
|
+
.sort((a, b) => a.estimated_cost - b.estimated_cost);
|
|
99
|
+
const tracker = new CostwiseTracker();
|
|
100
|
+
const formatted = tracker.formatComparisonForCli({
|
|
101
|
+
model: alternatives[0]?.model ?? "unknown",
|
|
102
|
+
inputTokens: totalInput,
|
|
103
|
+
outputTokens: totalOutput
|
|
104
|
+
});
|
|
105
|
+
if (asJson) {
|
|
106
|
+
process.stdout.write(`${JSON.stringify({
|
|
107
|
+
actual: {
|
|
108
|
+
events: events.length,
|
|
109
|
+
input_tokens: totalInput,
|
|
110
|
+
output_tokens: totalOutput,
|
|
111
|
+
estimated_cost: actualCost
|
|
112
|
+
},
|
|
113
|
+
alternatives: alternatives.slice(0, 10)
|
|
114
|
+
}, null, 2)}\n`);
|
|
115
|
+
return;
|
|
116
|
+
}
|
|
117
|
+
const lines = [
|
|
118
|
+
`Actual usage events: ${events.length}`,
|
|
119
|
+
`Actual estimated cost: $${actualCost.toFixed(6)}`,
|
|
120
|
+
"Top alternative models (same total tokens):",
|
|
121
|
+
...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)})`),
|
|
122
|
+
"",
|
|
123
|
+
"Cheapest-model summary:",
|
|
124
|
+
formatted.text
|
|
125
|
+
];
|
|
126
|
+
process.stdout.write(`${lines.join("\n")}\n`);
|
|
127
|
+
}
|
|
128
|
+
async function main() {
|
|
129
|
+
const options = parse(process.argv.slice(2));
|
|
130
|
+
const usagePath = options.filePath ?? process.env.COSTWISE_LOCAL_USAGE_PATH;
|
|
131
|
+
if (!usagePath) {
|
|
132
|
+
process.stderr.write("Missing usage file. Set COSTWISE_LOCAL_USAGE_PATH or pass --file.\n");
|
|
133
|
+
process.exit(1);
|
|
134
|
+
}
|
|
135
|
+
let events = [];
|
|
136
|
+
try {
|
|
137
|
+
events = await loadEvents(usagePath);
|
|
138
|
+
}
|
|
139
|
+
catch {
|
|
140
|
+
process.stderr.write(`Unable to read usage file: ${usagePath}\n`);
|
|
141
|
+
process.exit(1);
|
|
142
|
+
}
|
|
143
|
+
if (events.length === 0) {
|
|
144
|
+
process.stdout.write("No usage events found.\n");
|
|
145
|
+
return;
|
|
146
|
+
}
|
|
147
|
+
if (options.command === "usage") {
|
|
148
|
+
runUsage(events, options.json);
|
|
149
|
+
return;
|
|
150
|
+
}
|
|
151
|
+
runCompareActual(events, options.json);
|
|
152
|
+
}
|
|
153
|
+
void main();
|
|
154
|
+
//# sourceMappingURL=cli.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cli.js","sourceRoot":"","sources":["../../cli.ts"],"names":[],"mappings":";AAEA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;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,QAAQ,CAAC,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,OAAO,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,eAAe,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"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { CostwiseAnthropic } from "./src/anthropic.js";
|
|
2
|
+
export { CostwiseOpenAI } from "./src/openai.js";
|
|
3
|
+
export { CostwiseTracker, MODEL_CAPACITIES, PRICING } from "./src/tracker.js";
|
|
4
|
+
export type { CostwiseOptions, IngestPayload, CliComparisonFormat, LocalUsageEvent, ModelCapacity, ModelCapacityClass, ModelComparison, ModelComparisonByCapacity } from "./src/types.js";
|
|
5
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","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"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import Anthropic from "@anthropic-ai/sdk";
|
|
2
|
+
import { CostwiseOptions } from "./types.js";
|
|
3
|
+
export declare class CostwiseAnthropic extends Anthropic {
|
|
4
|
+
private readonly tracker;
|
|
5
|
+
private readonly tag?;
|
|
6
|
+
constructor(options?: ConstructorParameters<typeof Anthropic>[0] & CostwiseOptions);
|
|
7
|
+
private wrapMessagesCreate;
|
|
8
|
+
private hasFinalMessage;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=anthropic.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
import Anthropic from "@anthropic-ai/sdk";
|
|
2
|
+
import { CostwiseTracker } from "./tracker.js";
|
|
3
|
+
export class CostwiseAnthropic extends Anthropic {
|
|
4
|
+
tracker;
|
|
5
|
+
tag;
|
|
6
|
+
constructor(options = {}) {
|
|
7
|
+
super(options);
|
|
8
|
+
this.tracker = new CostwiseTracker(options);
|
|
9
|
+
this.tag = options.tag;
|
|
10
|
+
this.wrapMessagesCreate();
|
|
11
|
+
}
|
|
12
|
+
wrapMessagesCreate() {
|
|
13
|
+
const originalCreate = this.messages.create.bind(this.messages);
|
|
14
|
+
this.messages.create = (async (...args) => {
|
|
15
|
+
const response = await originalCreate(...args);
|
|
16
|
+
if (this.hasFinalMessage(response)) {
|
|
17
|
+
void response
|
|
18
|
+
.finalMessage()
|
|
19
|
+
.then((message) => {
|
|
20
|
+
this.tracker.trackUsage({
|
|
21
|
+
model: message.model,
|
|
22
|
+
inputTokens: message.usage?.input_tokens ?? 0,
|
|
23
|
+
outputTokens: message.usage?.output_tokens ?? 0,
|
|
24
|
+
tag: this.tag
|
|
25
|
+
});
|
|
26
|
+
})
|
|
27
|
+
.catch(() => { });
|
|
28
|
+
}
|
|
29
|
+
else {
|
|
30
|
+
const message = response;
|
|
31
|
+
this.tracker.trackUsage({
|
|
32
|
+
model: message.model,
|
|
33
|
+
inputTokens: message.usage?.input_tokens ?? 0,
|
|
34
|
+
outputTokens: message.usage?.output_tokens ?? 0,
|
|
35
|
+
tag: this.tag
|
|
36
|
+
});
|
|
37
|
+
}
|
|
38
|
+
return response;
|
|
39
|
+
});
|
|
40
|
+
}
|
|
41
|
+
hasFinalMessage(value) {
|
|
42
|
+
return (typeof value === "object" &&
|
|
43
|
+
value !== null &&
|
|
44
|
+
"finalMessage" in value &&
|
|
45
|
+
typeof value.finalMessage === "function");
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
//# sourceMappingURL=anthropic.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../../src/anthropic.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAE1C,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAyB/C,MAAM,OAAO,iBAAkB,SAAQ,SAAS;IAC7B,OAAO,CAAkB;IACzB,GAAG,CAAU;IAE9B,YAAY,UAAwE,EAAE;QACpF,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,IAAI,eAAe,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"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import OpenAI from "openai";
|
|
2
|
+
import { CostwiseOptions } from "./types.js";
|
|
3
|
+
export declare class CostwiseOpenAI extends OpenAI {
|
|
4
|
+
private readonly tracker;
|
|
5
|
+
constructor(options?: ConstructorParameters<typeof OpenAI>[0] & CostwiseOptions);
|
|
6
|
+
private wrapChatCompletionsCreate;
|
|
7
|
+
private hasFinalChatCompletion;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=openai.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
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"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import OpenAI from "openai";
|
|
2
|
+
import { CostwiseTracker } from "./tracker.js";
|
|
3
|
+
export class CostwiseOpenAI extends OpenAI {
|
|
4
|
+
tracker;
|
|
5
|
+
constructor(options = {}) {
|
|
6
|
+
super(options);
|
|
7
|
+
this.tracker = new CostwiseTracker(options);
|
|
8
|
+
this.wrapChatCompletionsCreate();
|
|
9
|
+
}
|
|
10
|
+
wrapChatCompletionsCreate() {
|
|
11
|
+
const originalCreate = this.chat.completions.create.bind(this.chat.completions);
|
|
12
|
+
this.chat.completions.create = (async (...args) => {
|
|
13
|
+
const response = await originalCreate(...args);
|
|
14
|
+
if (this.hasFinalChatCompletion(response)) {
|
|
15
|
+
void response
|
|
16
|
+
.finalChatCompletion()
|
|
17
|
+
.then((completion) => {
|
|
18
|
+
this.tracker.trackUsage({
|
|
19
|
+
model: completion.model,
|
|
20
|
+
inputTokens: completion.usage?.prompt_tokens ?? 0,
|
|
21
|
+
outputTokens: completion.usage?.completion_tokens ?? 0
|
|
22
|
+
});
|
|
23
|
+
})
|
|
24
|
+
.catch(() => { });
|
|
25
|
+
}
|
|
26
|
+
else {
|
|
27
|
+
const completion = response;
|
|
28
|
+
this.tracker.trackUsage({
|
|
29
|
+
model: completion.model,
|
|
30
|
+
inputTokens: completion.usage?.prompt_tokens ?? 0,
|
|
31
|
+
outputTokens: completion.usage?.completion_tokens ?? 0
|
|
32
|
+
});
|
|
33
|
+
}
|
|
34
|
+
return response;
|
|
35
|
+
});
|
|
36
|
+
}
|
|
37
|
+
hasFinalChatCompletion(value) {
|
|
38
|
+
return (typeof value === "object" &&
|
|
39
|
+
value !== null &&
|
|
40
|
+
"finalChatCompletion" in value &&
|
|
41
|
+
typeof value.finalChatCompletion === "function");
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=openai.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../../src/openai.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAgB/C,MAAM,OAAO,cAAe,SAAQ,MAAM;IACvB,OAAO,CAAkB;IAE1C,YAAY,UAAqE,EAAE;QACjF,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,IAAI,eAAe,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"}
|