This commit is contained in:
Frank
2026-02-27 00:30:12 -05:00
parent 7528419172
commit 2a4ed49551

View File

@@ -245,16 +245,15 @@ export async function handler(
dataDumper?.flush() dataDumper?.flush()
await rateLimiter?.track() await rateLimiter?.track()
const usage = usageParser.retrieve() const usage = usageParser.retrieve()
let cost = "0"
if (usage) { if (usage) {
const usageInfo = providerInfo.normalizeUsage(usage) const usageInfo = providerInfo.normalizeUsage(usage)
const costInfo = calculateCost(modelInfo, usageInfo) const costInfo = calculateCost(modelInfo, usageInfo)
await trialLimiter?.track(usageInfo) await trialLimiter?.track(usageInfo)
await trackUsage(sessionId, billingSource, authInfo, modelInfo, providerInfo, usageInfo, costInfo) await trackUsage(sessionId, billingSource, authInfo, modelInfo, providerInfo, usageInfo, costInfo)
await reload(billingSource, authInfo, costInfo) await reload(billingSource, authInfo, costInfo)
cost = calculateOccuredCost(billingSource, costInfo) const cost = calculateOccuredCost(billingSource, costInfo)
}
c.enqueue(encoder.encode(usageParser.buidlCostChunk(cost))) c.enqueue(encoder.encode(usageParser.buidlCostChunk(cost)))
}
c.close() c.close()
return return
} }