Repository navigation
Expand file tree
/
Copy pathai-pricing.test.ts
More file actions
69 lines (65 loc) · 2.02 KB
/
Copy pathai-pricing.test.ts
File metadata and controls
69 lines (65 loc) · 2.02 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import { describe, expect, it } from "vitest";
import {
AI_PRICING_VERSION,
mergeAiUsageCharge,
priceRealtimeUsage,
priceSearchUsage,
priceTranscriptionUsage,
} from "./ai-pricing";
describe("AI pricing snapshots", () => {
it("prices current Realtime and transcription models in integer nano-USD", () => {
expect(priceRealtimeUsage({
model: "gpt-realtime",
textInputTokens: 1_000_000,
cachedTextInputTokens: 1_000_000,
audioInputTokens: 1_000_000,
cachedAudioInputTokens: 1_000_000,
textOutputTokens: 1_000_000,
})).toEqual({
estimatedNanoUsd: 52_800_000_000,
unpricedEvents: 0,
pricingVersions: [AI_PRICING_VERSION],
});
expect(priceTranscriptionUsage({
model: "gpt-4o-mini-transcribe",
audioInputTokens: 1_000_000,
textOutputTokens: 1_000_000,
}).estimatedNanoUsd).toBe(6_250_000_000);
});
it("prices gpt-5.5 tokens and only reported web-search actions", () => {
expect(priceSearchUsage({
model: "gpt-5.5-2026-07-01",
inputTokens: 1_000_000,
cachedInputTokens: 200_000,
outputTokens: 1_000_000,
webSearchCalls: 2,
})).toEqual({
estimatedNanoUsd: 34_120_000_000,
unpricedEvents: 0,
pricingVersions: [AI_PRICING_VERSION],
});
});
it("keeps known action fees but marks unknown-model token usage unpriced", () => {
expect(priceSearchUsage({
model: "future-model",
inputTokens: 10,
cachedInputTokens: 0,
outputTokens: 5,
webSearchCalls: 1,
})).toEqual({
estimatedNanoUsd: 10_000_000,
unpricedEvents: 1,
pricingVersions: [AI_PRICING_VERSION],
});
});
it("merges charges without duplicating pricing versions", () => {
expect(mergeAiUsageCharge(
{ estimatedNanoUsd: 10, unpricedEvents: 1, pricingVersions: ["v1"] },
{ estimatedNanoUsd: 20, unpricedEvents: 2, pricingVersions: ["v1", "v2"] },
)).toEqual({
estimatedNanoUsd: 30,
unpricedEvents: 3,
pricingVersions: ["v1", "v2"],
});
});
});