fix(knowledge): key the usage gate cache by billing period source

Author: waleedlatif1

apps/sim/lib/billing/core/usage-gate-cache.test.ts

@@ -85,7 +85,7 @@ describe('checkIngestionUsageLimits', () => {
     expect(mockCheck).toHaveBeenCalledTimes(2)
   })
 
-  it('separates answers by actor, period, payer and plan', async () => {
+  it('separates answers by actor, period, period source, payer and plan', async () => {
     await checkIngestionUsageLimits(ATTRIBUTION)
     await checkIngestionUsageLimits({ ...ATTRIBUTION, actorUserId: 'member-2' })
     await checkIngestionUsageLimits({
@@ -97,12 +97,16 @@ describe('checkIngestionUsageLimits', () => {
       billedAccountUserId: 'owner-2',
       billingEntity: { type: 'user', id: 'owner-2' },
     })
+    await checkIngestionUsageLimits({
+      ...ATTRIBUTION,
+      billingPeriod: { ...ATTRIBUTION.billingPeriod, source: 'reporting' },
+    })
     await checkIngestionUsageLimits({ ...ATTRIBUTION, payerSubscription: SUBSCRIPTION })
     await checkIngestionUsageLimits({
       ...ATTRIBUTION,
       payerSubscription: { ...SUBSCRIPTION, plan: 'enterprise' },
     })
-    expect(mockCheck).toHaveBeenCalledTimes(6)
+    expect(mockCheck).toHaveBeenCalledTimes(7)
   })
 
   it('does not cache a failed read', async () => {

apps/sim/lib/billing/core/usage-gate-cache.ts

@@ -28,10 +28,10 @@ const gateCache = new LRUCache<string, AttributedUsageLimitsResult>({
 })
 
 /**
- * The gate depends on who pays, for which period, under which plan, and which
- * member acts: the payer pool, its limit and the per-member cap are all part of
- * the answer. The workspace is not, so every workspace of one payer shares an
- * entry.
+ * The gate depends on who pays, for which period (and how that period was
+ * derived), under which plan, and which member acts: the payer pool, its limit
+ * and the per-member cap are all part of the answer. The workspace is not, so
+ * every workspace of one payer shares an entry.
  */
 function gateKey(attribution: BillingAttributionSnapshot): string {
   const subscription = attribution.payerSubscription
@@ -40,6 +40,7 @@ function gateKey(attribution: BillingAttributionSnapshot): string {
     attribution.billingEntity.id,
     attribution.billingPeriod.start,
     attribution.billingPeriod.end,
+    attribution.billingPeriod.source ?? '',
     attribution.billedAccountUserId,
     attribution.actorUserId,
     subscription?.id ?? '',