zen: display BYOK cost

This commit is contained in:
Frank
2026-02-23 18:18:45 -05:00
parent 34495a70d5
commit 284251ad66
19 changed files with 36 additions and 5 deletions

View File

@@ -177,7 +177,16 @@ export function UsageSection() {
<td data-slot="usage-cost">
<Show
when={usage.enrichment?.plan === "sub"}
fallback={<>${((usage.cost ?? 0) / 100000000).toFixed(4)}</>}
fallback={
<Show
when={usage.enrichment?.plan === "byok"}
fallback={<>${((usage.cost ?? 0) / 100000000).toFixed(4)}</>}
>
{i18n.t("workspace.usage.byok", {
amount: ((usage.cost ?? 0) / 100000000).toFixed(4),
})}
</Show>
}
>
{i18n.t("workspace.usage.subscription", {
amount: ((usage.cost ?? 0) / 100000000).toFixed(4),

View File

@@ -740,7 +740,11 @@ export async function handler(
cost,
keyID: authInfo.apiKeyId,
sessionID: sessionId.substring(0, 30),
enrichment: billingSource === "subscription" ? { plan: "sub" } : undefined,
enrichment: (() => {
if (billingSource === "subscription") return { plan: "sub" }
if (billingSource === "byok") return { plan: "byok" }
return undefined
})(),
}),
db
.update(KeyTable)
@@ -788,9 +792,10 @@ export async function handler(
db
.update(BillingTable)
.set({
balance: authInfo.isFree
? sql`${BillingTable.balance} - ${0}`
: sql`${BillingTable.balance} - ${cost}`,
balance:
billingSource === "free" || billingSource === "byok"
? sql`${BillingTable.balance} - ${0}`
: sql`${BillingTable.balance} - ${cost}`,
monthlyUsage: sql`
CASE
WHEN MONTH(${BillingTable.timeMonthlyUsageUpdated}) = MONTH(now()) AND YEAR(${BillingTable.timeMonthlyUsageUpdated}) = YEAR(now()) THEN ${BillingTable.monthlyUsage} + ${cost}