aboutsummaryrefslogtreecommitdiff
path: root/modules/core/module-billing.js
blob: 87f8dc37870ebac586152a59249d36e3355f4596 (plain)
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
// Billing / pricing surface for newOS4 (2026-10-28)
//
// Why: the terminal could show a model's name but never what it costs. Prices and the anonymous
// message allowance are public data, so they need no login — the account BALANCE does, and newOS4
// has no session layer, so that is reported honestly as "not signed in" rather than guessed.
//
// Commands: cost [on|off], usage
class BillingModule {
    constructor() {
        this.commands = {};
        this.terminal = null;
        this._prices = null;
        this._pricesAt = 0;
        this._installed = false;
    }

    getHelpInfo() {
        return {
            'cost': 'cost [on|off] — AI prices, today\'s free-message allowance and this session\'s spend',
            'usage': 'usage — same as cost',
        };
    }

    registerCommands() {
        this.commands['cost'] = (args) => this.run(args);
        this.commands['usage'] = (args) => this.run(args);
    }

    init(terminal) {
        this.terminal = terminal;
        this._hookUsage();
    }

    // --- data -----------------------------------------------------------------
    async _loadPrices() {
        if (this._prices && Date.now() - this._pricesAt < 300000) return this._prices;
        try {
            const r = await fetch('/api/billing/prices');
            if (!r.ok) return null;
            this._prices = await r.json();
            this._pricesAt = Date.now();
            return this._prices;
        } catch (e) {
            return null;
        }
    }

    async _loadDemo() {
        try {
            const r = await fetch('/api/billing/demo-usage');
            if (!r.ok) return null;
            return await r.json();
        } catch (e) {
            return null;
        }
    }

    async _loadBalance() {
        // Only answers when a session exists; newOS4 currently has none, so this is expected to 401.
        try {
            const r = await fetch('/api/billing/subscription', { credentials: 'include' });
            if (!r.ok) return null;
            const d = await r.json();
            return d && d.balance_usd != null ? d : null;
        } catch (e) {
            return null;
        }
    }

    // --- session spend, measured where the tokens actually are ----------------
    _hookUsage() {
        const om = this.terminal && this.terminal.ollamaManager;
        if (!om || om.__billingHooked) return;
        const orig = om._logUsage ? om._logUsage.bind(om) : null;
        if (!orig) return;
        om._billingSession = { msgs: 0, prompt: 0, completion: 0, cost: 0, byModel: {} };
        om._logUsage = (provider, model, latency, promptTokens, completionTokens) => {
            try { this._accumulate(model, promptTokens, completionTokens); } catch (e) { /* never break a reply */ }
            return orig(provider, model, latency, promptTokens, completionTokens);
        };
        om.__billingHooked = true;
    }

    _rate(model) {
        const cat = (this._prices && this._prices.models) || [];
        const bare = String(model || '').replace(/:cloud$|-cloud$/, '');
        return cat.find(m => m.model === model) || cat.find(m => m.model === bare) || null;
    }

    _accumulate(rawModel, promptTokens, completionTokens) {
        const om = this.terminal.ollamaManager;
        const s = om._billingSession;
        if (!s) return;
        const model = rawModel || 'unknown';
        const pt = Number(promptTokens) || 0;
        const ct = Number(completionTokens) || 0;
        if (!pt && !ct) return;                       // metadata calls carry no tokens
        const row = this._rate(model);
        const r = row ? row.retail_per_million : null;
        const cost = r ? (pt * r.input + ct * r.output) / 1e6 : 0;
        s.msgs += 1; s.prompt += pt; s.completion += ct; s.cost += cost;
        const m = s.byModel[model] || (s.byModel[model] = { msgs: 0, tokens: 0, cost: 0 });
        m.msgs += 1; m.tokens += pt + ct; m.cost += cost;

        if (this._costLineOn() && ct > 0 && cost > 0) {
            this.terminal.print(
                `💵 $${cost.toFixed(4)} · ${model} · ${(pt / 1000).toFixed(1)}k in / ${(ct / 1000).toFixed(1)}k out (retail est.)`,
                'info');
        }
    }

    _costLineOn() {
        try { return localStorage.getItem('webx11_cost_line') === '1'; } catch (e) { return false; }
    }

    // --- the command ----------------------------------------------------------
    async run(args) {
        const p = this.terminal.print.bind(this.terminal);
        if (args && args[0] === 'on') {
            try { localStorage.setItem('webx11_cost_line', '1'); } catch (e) {}
            p('💵 cost line on — each reply will show its retail estimate. `cost off` to stop.', 'success');
            return;
        }
        if (args && args[0] === 'off') {
            try { localStorage.setItem('webx11_cost_line', '0'); } catch (e) {}
            p('💵 cost line off.', 'info');
            return;
        }

        const [prices, demo, bal] = await Promise.all([this._loadPrices(), this._loadDemo(), this._loadBalance()]);
        const s = (this.terminal.ollamaManager && this.terminal.ollamaManager._billingSession) || null;

        if (demo) {
            const cap = demo.exempt ? 'unlimited (operator exemption)'
                : `${demo.used} / ${demo.limit} messages today — ${demo.remaining} left, resets ${String(demo.resets_at).slice(0, 10)}`;
            p(`📊 Anonymous allowance: ${cap}`, 'info');   // distinct from the monthly money allowance below
        } else {
            p('📊 Free allowance: unavailable (billing API did not answer)', 'warning');
        }

        p(`💰 Balance: ${bal && bal.balance_usd != null ? '$' + Number(bal.balance_usd).toFixed(2) : 'not signed in — newOS4 sends no session token, so a balance cannot be shown here yet'}`, 'info');

        if (bal && bal.plan_allowance) {
            const pa = bal.plan_allowance;
            p(`🎁 Included usage (${pa.plan}): $${Number(pa.used_usd).toFixed(2)} of $${Number(pa.limit_usd).toFixed(2)} ${pa.period === 'day' ? 'today' : 'this month'} — $${Number(pa.remaining_usd).toFixed(2)} left, then your balance pays`, 'info');
        }

        if (s && s.msgs) {
            p(`🧮 This session: ${s.msgs} call(s), ${(s.prompt / 1000).toFixed(1)}k in / ${(s.completion / 1000).toFixed(1)}k out ≈ $${s.cost.toFixed(4)} retail`, 'info');
            Object.entries(s.byModel)
                .sort((a, b) => b[1].cost - a[1].cost)
                .slice(0, 5)
                .forEach(([m, v]) => p(`   · ${m}: ${v.msgs} call(s), ${(v.tokens / 1000).toFixed(1)}k tok, $${v.cost.toFixed(4)}`, ''));
        } else {
            p('🧮 This session: nothing yet', '');
        }

        if (prices) {
            const rows = prices.models
                .filter(m => m.retail_per_million.input > 0)
                .sort((a, b) => a.retail_per_million.output - b.retail_per_million.output);
            const body = rows.map(m => {
                const r = m.retail_per_million;
                return `<tr><td style="padding:2px 8px">${m.model}</td>
                        <td style="padding:2px 8px;text-align:right">$${r.input.toFixed(2)}</td>
                        <td style="padding:2px 8px;text-align:right">$${r.cached_input.toFixed(3)}</td>
                        <td style="padding:2px 8px;text-align:right">$${r.output.toFixed(2)}</td></tr>`;
            }).join('');
            this.terminal.printHTML(`<div style="margin:8px 0"><b>Retail prices per 1M tokens</b> (cost × ${prices.markup})<br>
                <table style="border-collapse:collapse;margin-top:4px;font-size:12px">
                <tr><th style="text-align:left;padding:2px 8px">model</th><th style="padding:2px 8px">input</th>
                    <th style="padding:2px 8px">cached</th><th style="padding:2px 8px">output</th></tr>
                ${body}</table>
                <div style="opacity:.7;margin-top:4px">Catalog ${String(prices.fetched_at || '').slice(0, 16)} · input billed as fresh
                (cached counts are not exposed upstream), so figures here are a ceiling.</div></div>`);
        } else {
            p('💲 Prices unavailable (billing API did not answer)', 'warning');
        }
    }
}

if (typeof window !== 'undefined') {
    window.BillingModule = BillingModule;
}