mirror/puter
1
0
Fork 0
mirror of https://github.com/HeyPuter/puter.git synced 2026-01-16 10:01:26 -06:00
Code Issues Packages Projects Releases Wiki Activity

fix: lessen alert triggering for metering (#1837)

Browse Source
This commit is contained in:
Daniel Salazar
2025-10-27 12:41:48 -07:00
committed by GitHub
parent 2c7343115a
commit fd8a88b8f4
1 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
src/backend/src/services/MeteringService/MeteringService.ts
View File

@@ -79,9 +79,9 @@ export class MeteringService {
return this.#superUserService.sudo(async () => {
const totalCost = (costOverride ?? (COST_MAPS[usageType as keyof typeof COST_MAPS] || 0) * usageAmount) || 0; // TODO DS: apply our policy discounts here eventually
if ( totalCost === 0 && costOverride === undefined ) {
if ( COST_MAPS[usageType as keyof typeof COST_MAPS] !== 0 && totalCost === 0 && costOverride === undefined ) {
// could be something is off, there are some models that cost nothing from openrouter, but then our overrides should not be undefined, so will flag
this.#alarmService.create('metering-service-warning', 'potential abuse vector', {
this.#alarmService.create('metering-service-0-cost-warning', 'potential abuse vector', {
actor,
usageType,
usageAmount,
@@ -380,12 +380,11 @@ export class MeteringService {
});
}
// eslint-disable-next-line
async #updateAddonCredit(actor: Actor, tokenAmount: number) {
if ( !actor.type?.user?.uuid ) {
throw new Error('Actor must be a user to update extra credits');
async updateAddonCredit(userId:string, tokenAmount: number) {
if ( !userId ) {
throw new Error('User needed to update extra credits');
}
const key = `${POLICY_PREFIX}:actor:${actor.type.user?.uuid}:addons`;
const key = `${POLICY_PREFIX}:actor:${userId}:addons`;
return this.#superUserService.sudo(async () => {
await this.#kvStore.incr({
key,
@@ -395,13 +394,4 @@ export class MeteringService {
});
});
}
handlePolicyPurchase(_actor: Actor, _policyType: keyof typeof SUB_POLICIES) {
// TODO DS: this should leverage extensions to call billing implementations
}
handleTokenPurchase(_actor: Actor, _tokenAmount: number) {
// TODO DS: this should leverage extensions to call billing implementations
}
}