@opengeni/db 1.4.0 → 1.4.2

package/dist/{chunk-T3CYF3DH.js → chunk-TODVDGMO.js}

@@ -7,7 +7,7 @@ import {
   withAccountRls,
   withRlsContext,
   withWorkspaceRls
-} from "./chunk-27T4AP25.js";
+} from "./chunk-IT3EYY2C.js";
 import {
   connections,
   creditDebitAllocations,
@@ -345,7 +345,8 @@ async function getBillingBalance(db, accountId, policyRevision) {
                 else policy->'offers'->(g.metadata->>'creditOfferId')->'eligibleModelIds' end,
               policy->'defaultModelIds'
             ) from current_policy), to_jsonb(g.eligible_model_ids)),
-          'remainingMicros', g.amount_micros - coalesce(used.amount, 0)
+          'remainingMicros', g.amount_micros - coalesce(used.amount, 0),
+          'coversVoice', g.source_type = 'verified_signup_trial'
         ) order by g.created_at, g.id)
           from ${creditLedgerEntries} g
           left join (
@@ -371,30 +372,45 @@ async function getBillingBalance(db, accountId, policyRevision) {
     };
   });
 }
-function spendableCreditMicros(balance, modelId) {
+var VOICE_CREDIT_USAGE = Object.freeze({ kind: "voice" });
+function grantCovers(grant, usage) {
+  if (!usage) return false;
+  if (typeof usage === "string") return grant.eligibleModelIds.includes(usage);
+  return grant.coversVoice === true;
+}
+function spendableCreditMicros(balance, usage) {
   return Math.max(0, balance.generalBalanceMicros ?? balance.balanceMicros) + (balance.promotionalCredits ?? []).reduce(
-    (sum, grant) => sum + (modelId && grant.eligibleModelIds.includes(modelId) ? grant.remainingMicros : 0),
+    (sum, grant) => sum + (grantCovers(grant, usage) ? grant.remainingMicros : 0),
     0
   );
 }
-async function getSpendableCreditBalance(db, accountId, modelId, policyRevision) {
+async function getSpendableCreditBalance(db, accountId, usage, policyRevision) {
   const balance = await getBillingBalance(db, accountId, policyRevision);
-  return { ...balance, balanceMicros: spendableCreditMicros(balance, modelId) };
+  return { ...balance, balanceMicros: spendableCreditMicros(balance, usage) };
+}
+function allocateToCoveringGrants(balance, amountMicros, usage) {
+  let remaining = amountMicros;
+  const allocations = [];
+  for (const grant of balance.promotionalCredits ?? []) {
+    if (!remaining || !grantCovers(grant, usage) || grant.remainingMicros <= 0) continue;
+    const allocated = Math.min(remaining, grant.remainingMicros);
+    allocations.push({ grantEntryId: grant.grantId, amountMicros: allocated });
+    remaining -= allocated;
+  }
+  return allocations;
 }
-function planCreditDebit(balance, requestedMicros, modelId) {
+function planCreditDebit(balance, requestedMicros, usage) {
   if (!Number.isSafeInteger(requestedMicros) || requestedMicros <= 0) {
     throw new Error("credit debit requires a positive, safe integer micro amount");
   }
-  const debitedMicros = Math.min(requestedMicros, spendableCreditMicros(balance, modelId));
-  let remaining = debitedMicros;
-  const allocations = [];
-  for (const grant of balance.promotionalCredits ?? []) {
-    if (!remaining || !modelId || !grant.eligibleModelIds.includes(modelId)) continue;
-    const amountMicros = Math.min(remaining, grant.remainingMicros);
-    allocations.push({ grantEntryId: grant.grantId, amountMicros });
-    remaining -= amountMicros;
+  const debitedMicros = Math.min(requestedMicros, spendableCreditMicros(balance, usage));
+  return { debitedMicros, allocations: allocateToCoveringGrants(balance, debitedMicros, usage) };
+}
+function planPostUseCreditAllocations(balance, amountMicros, usage) {
+  if (!Number.isSafeInteger(amountMicros) || amountMicros <= 0) {
+    throw new Error("credit debit requires a positive, safe integer micro amount");
   }
-  return { debitedMicros, allocations };
+  return allocateToCoveringGrants(balance, amountMicros, usage);
 }
 
 // src/video-generation.ts
@@ -1378,9 +1394,11 @@ export {
   checkWorkspaceAllowance,
   maintainWorkspaceAllowances,
   getBillingBalance,
+  VOICE_CREDIT_USAGE,
   spendableCreditMicros,
   getSpendableCreditBalance,
   planCreditDebit,
+  planPostUseCreditAllocations,
   ACTIVE_VIDEO_GENERATION_STATUSES,
   VideoGenerationConflictError,
   VideoGenerationCapacityError,
@@ -1412,4 +1430,4 @@ export {
   mediaGenerationResultForStoredOperation,
   getVideoGenerationOperationSummary
 };
-//# sourceMappingURL=chunk-T3CYF3DH.js.map
+//# sourceMappingURL=chunk-TODVDGMO.js.map