@trendyvideo/billing 1.0.0

package/dist/usage-checker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-checker.js","sourceRoot":"","sources":["../src/usage-checker.ts"],"names":[],"mappings":";;AA4FA,gCAkHC;AAcD,4CA0EC;AAWD,4CA8CC;AA9VD,uCAAkD;AAClD,+CAAiE;AAEjE,MAAM,aAAa,GAAkB;IACnC,KAAK,EAAE,GAAG,EAAE,GAAE,CAAC;IACf,IAAI,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC;IAChE,IAAI,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,GAAG,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC;IACjE,KAAK,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,GAAG,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC;CACpE,CAAC;AAYF;;;GAGG;AACH,KAAK,UAAU,eAAe,CAC5B,QAAwB,EACxB,KAAa,EACb,SAA2B,EAC3B,WAAmB,EACnB,OAAgC,EAChC,MAAqB;IAErB,IAAI,CAAC;QACH,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC;YAC7D,eAAe,EAAE,KAAK;YACtB,UAAU,EAAE,SAAS;YACrB,WAAW;YACX,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,OAAO;SACtB,CAAC,CAAC;QAEH,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,CAAC,IAAI,CAAC,6BAA6B,EAAE;gBACzC,KAAK;gBACL,SAAS;gBACT,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,yBAAyB;QACzB,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE;YACpD,KAAK;YACL,SAAS;YACT,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,+EAA+E;AAC/E,yBAAyB;AACzB,+EAA+E;AAE/E,MAAM,WAAW,GAA8B;IAC7C,WAAW,EAAE,CAAC;IACd,eAAe,EAAE,CAAC;IAClB,gBAAgB,EAAE,CAAC;IACnB,iBAAiB,EAAE,CAAC;IACpB,aAAa,EAAE,KAAK;IACpB,cAAc,EAAE,CAAC;CAClB,CAAC;AAEF,MAAM,YAAY,GAA6B;IAC7C,EAAE,EAAE,SAAS;IACb,IAAI,EAAE,SAAS;CAChB,CAAC;AAEF,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E;;;;;;;;;;GAUG;AACI,KAAK,UAAU,UAAU,CAC9B,QAAwB,EACxB,KAAa,EACb,SAAwB,aAAa;IAErC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,uBAAuB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAEvF,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3E,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,iBAAiB,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;IAC/F,CAAC;IAED,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;IACjD,IAAI,CAAC,GAAG,EAAE,CAAC;QACT,OAAO;YACL,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,iBAAiB;YACzB,KAAK,EAAE,WAAW;YAClB,IAAI,EAAE,EAAE,GAAG,YAAY,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE;SACpD,CAAC;IACJ,CAAC;IAED,MAAM,MAAM,GAAW,GAAG,CAAC,OAAO,IAAI,SAAS,CAAC;IAChD,MAAM,QAAQ,GAAW,GAAG,CAAC,SAAS,IAAI,SAAS,CAAC;IAEpD,mDAAmD;IACnD,MAAM,gBAAgB,GAAW,GAAG,CAAC,iBAAiB,IAAI,CAAC,CAAC;IAC5D,MAAM,WAAW,GAAG,IAAA,+BAAqB,EAAC,gBAAgB,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,IAAA,2BAAa,EAAC,IAAA,+BAAiB,EAAC,QAAQ,CAAC,CAAC,CAAC;IAC9D,wDAAwD;IACxD,MAAM,eAAe,GAAW,GAAG,CAAC,gBAAgB,IAAI,CAAC,CAAC;IAC1D,MAAM,gBAAgB,GAAG,eAAe,GAAG,UAAU,CAAC,cAAc,CAAC;IACrE,MAAM,eAAe,GAAG,UAAU,CAAC,eAAe,GAAG,gBAAgB,CAAC;IACtE,MAAM,gBAAgB,GAAG,eAAe,GAAG,WAAW,CAAC;IACvD,MAAM,iBAAiB,GAAG,eAAe,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,GAAG,eAAe,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1F,MAAM,aAAa,GAAG,WAAW,GAAG,eAAe,CAAC;IACpD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,eAAe,CAAC,CAAC;IAElE,+DAA+D;IAC/D,MAAM,WAAW,GAAW,GAAG,CAAC,iBAAiB,IAAI,CAAC,CAAC;IACvD,MAAM,YAAY,GAAW,GAAG,CAAC,kBAAkB,IAAI,CAAC,CAAC;IACzD,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,CAAC;IAE/C,MAAM,KAAK,GAA8B;QACvC,WAAW;QACX,eAAe;QACf,gBAAgB;QAChB,iBAAiB;QACjB,aAAa;QACb,cAAc;QACd,4BAA4B;QAC5B,WAAW;QACX,WAAW;QACX,YAAY;KACb,CAAC;IAEF,MAAM,IAAI,GAA6B;QACrC,EAAE,EAAE,MAAM;QACV,IAAI,EAAE,QAAQ;KACf,CAAC;IAEF,sDAAsD;IACtD,IAAI,MAAM,KAAK,MAAM,IAAI,aAAa,EAAE,CAAC;QACvC,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,gBAAgB,EAAE,iCAAiC,EAAE;YACpF,MAAM;YACN,WAAW;YACX,eAAe;YACf,iBAAiB;SAClB,EAAE,MAAM,CAAC,CAAC;QACX,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACxE,CAAC;IAED,wDAAwD;IACxD,MAAM,iBAAiB,GAAG,UAAU,CAAC,iBAAiB,CAAC;IACvD,IAAI,MAAM,KAAK,MAAM,IAAI,cAAc,IAAI,iBAAiB,IAAI,iBAAiB,GAAG,CAAC,EAAE,CAAC;QACtF,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,qBAAqB,EAAE,oCAAoC,EAAE;YAC5F,MAAM;YACN,WAAW;YACX,eAAe;YACf,cAAc;YACd,iBAAiB;SAClB,EAAE,MAAM,CAAC,CAAC;QACX,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,qBAAqB,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACxE,CAAC;IAED,+BAA+B;IAC/B,IAAI,OAAgD,CAAC;IACrD,IAAI,aAAa,EAAE,CAAC;QAClB,OAAO,GAAG,YAAY,CAAC;QACvB,sFAAsF;QACtF,IAAI,iBAAiB,IAAI,GAAG,EAAE,CAAC;YAC7B,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,eAAe,EAAE,qCAAqC,EAAE;gBACvF,MAAM;gBACN,WAAW;gBACX,eAAe;gBACf,cAAc;gBACd,OAAO,EAAE,YAAY;aACtB,EAAE,MAAM,CAAC,CAAC;QACb,CAAC;IACH,CAAC;SAAM,IAAI,iBAAiB,GAAG,EAAE,EAAE,CAAC;QAClC,OAAO,GAAG,mBAAmB,CAAC;QAC9B,2EAA2E;QAC3E,IAAI,iBAAiB,IAAI,EAAE,EAAE,CAAC;YAC5B,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,eAAe,EAAE,2CAA2C,EAAE;gBAC7F,MAAM;gBACN,WAAW;gBACX,eAAe;gBACf,iBAAiB;gBACjB,OAAO,EAAE,mBAAmB;aAC7B,EAAE,MAAM,CAAC,CAAC;QACb,CAAC;IACH,CAAC;IAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AACjD,CAAC;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E,MAAM,qBAAqB,GAAG,GAAG,CAAC;AAClC,MAAM,mBAAmB,GAAG,CAAC,CAAC;AAE9B;;;;GAIG;AACI,KAAK,UAAU,gBAAgB,CACpC,QAAwB,EACxB,KAAa,EACb,SAAwB,aAAa;IAErC,MAAM,UAAU,GAAsB;QACpC,OAAO,EAAE,KAAK;QACd,WAAW,EAAE,CAAC;QACd,UAAU,EAAE,CAAC;QACb,OAAO,EAAE,kDAAkD;KAC5D,CAAC;IAEF,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,mBAAmB,EAAE,OAAO,EAAE,EAAE,CAAC;QAChE,IAAI,CAAC;YACH,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,oBAAoB,EAAE;gBAC/D,iBAAiB,EAAE,KAAK;aACzB,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;oBAC3C,KAAK;oBACL,KAAK,EAAE,KAAK,CAAC,OAAO;oBACpB,OAAO;iBACR,CAAC,CAAC;gBAEH,IAAI,OAAO,GAAG,mBAAmB,EAAE,CAAC;oBAClC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC,CAAC;oBAC/D,SAAS;gBACX,CAAC;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC;YAED,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YACjD,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,WAAW,EAAE,CAAC;oBACd,UAAU,EAAE,CAAC;oBACb,OAAO,EAAE,mEAAmE;iBAC7E,CAAC;YACJ,CAAC;YAED,MAAM,WAAW,GAAW,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC;YAClD,MAAM,UAAU,GAAW,GAAG,CAAC,WAAW,IAAI,UAAU,CAAC;YAEzD,IAAI,WAAW,IAAI,UAAU,EAAE,CAAC;gBAC9B,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,sBAAsB,EAAE,8BAA8B,EAAE;oBACvF,WAAW;oBACX,UAAU;iBACX,EAAE,MAAM,CAAC,CAAC;gBACX,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,WAAW;oBACX,UAAU;oBACV,OAAO,EAAE,yBAAyB,WAAW,IAAI,UAAU,4CAA4C;iBACxG,CAAC;YACJ,CAAC;YAED,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;QACpD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,CAAC,KAAK,CAAC,wCAAwC,EAAE;gBACrD,KAAK;gBACL,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;gBACvD,OAAO;aACR,CAAC,CAAC;YAEH,IAAI,OAAO,GAAG,mBAAmB,EAAE,CAAC;gBAClC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC,CAAC;gBAC/D,SAAS;YACX,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;GAIG;AACI,KAAK,UAAU,gBAAgB,CACpC,QAAwB,EACxB,KAAa,EACb,KAAK,GAAG,CAAC,EACT,SAAwB,aAAa;IAErC,IAAI,CAAC;QACH,6CAA6C;QAC7C,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,QAAQ;aAClD,IAAI,CAAC,4BAA4B,CAAC;aAClC,MAAM,CAAC,0CAA0C,CAAC;aAClD,EAAE,CAAC,iBAAiB,EAAE,KAAK,CAAC;aAC5B,KAAK,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;aACzC,KAAK,CAAC,CAAC,CAAC;aACR,MAAM,EAAE,CAAC;QAEZ,IAAI,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC;YACrB,MAAM,CAAC,IAAI,CAAC,kDAAkD,EAAE;gBAC9D,KAAK;gBACL,KAAK,EAAE,QAAQ,EAAE,OAAO;aACzB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,sEAAsE;QACtE,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/B,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,wBAAwB,EAAE;gBAC7D,iBAAiB,EAAE,KAAK;gBACxB,cAAc,EAAE,GAAG,CAAC,oBAAoB;gBACxC,YAAY,EAAE,GAAG,CAAC,kBAAkB;aACrC,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,CAAC;gBACV,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE;oBAChD,KAAK;oBACL,KAAK,EAAE,KAAK,CAAC,OAAO;oBACpB,SAAS,EAAE,CAAC,GAAG,CAAC;iBACjB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE;YAC/C,KAAK;YACL,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;AACH,CAAC"}

package/dist/usage-tracker.d.ts

@@ -0,0 +1,25 @@
+import { SupabaseClient, UsageEvent, BillingLogger } from './types';
+/**
+ * Record token usage after an API response.
+ *
+ * Uses the `record_usage_atomic` Postgres RPC to insert the usage_event AND
+ * upsert the usage_records aggregate in a single transaction. This prevents
+ * token drift where events exist without corresponding aggregated totals.
+ *
+ * Falls back to the legacy two-step approach if the RPC is unavailable.
+ *
+ * Never throws — all errors are caught and logged internally.
+ */
+export declare function recordUsage(supabase: SupabaseClient, event: UsageEvent, logger?: BillingLogger): Promise<void>;
+export interface VertexUsageMetadata {
+    promptTokenCount?: number;
+    candidatesTokenCount?: number;
+    totalTokenCount?: number;
+    cachedContentTokenCount?: number;
+}
+/**
+ * Extract token counts from a Vertex AI generateContent() response.
+ * Works with both typed SDK responses and raw JSON.
+ */
+export declare function extractVertexUsageMetadata(response: unknown): VertexUsageMetadata;
+//# sourceMappingURL=usage-tracker.d.ts.map

package/dist/usage-tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracker.d.ts","sourceRoot":"","sources":["../src/usage-tracker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAUpE;;;;;;;;;;GAUG;AACH,wBAAsB,WAAW,CAC/B,QAAQ,EAAE,cAAc,EACxB,KAAK,EAAE,UAAU,EACjB,MAAM,GAAE,aAA6B,GACpC,OAAO,CAAC,IAAI,CAAC,CAuKf;AAMD,MAAM,WAAW,mBAAmB;IAClC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,uBAAuB,CAAC,EAAE,MAAM,CAAC;CAClC;AAED;;;GAGG;AACH,wBAAgB,0BAA0B,CAAC,QAAQ,EAAE,OAAO,GAAG,mBAAmB,CASjF"}

package/dist/usage-tracker.js

@@ -0,0 +1,181 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.recordUsage = recordUsage;
+exports.extractVertexUsageMetadata = extractVertexUsageMetadata;
+const pricing_1 = require("./pricing");
+const defaultLogger = {
+    debug: (msg, meta) => { },
+    info: (msg, meta) => console.log(`[billing] ${msg}`, meta ?? ''),
+    warn: (msg, meta) => console.warn(`[billing] ${msg}`, meta ?? ''),
+    error: (msg, meta) => console.error(`[billing] ${msg}`, meta ?? ''),
+};
+/**
+ * Record token usage after an API response.
+ *
+ * Uses the `record_usage_atomic` Postgres RPC to insert the usage_event AND
+ * upsert the usage_records aggregate in a single transaction. This prevents
+ * token drift where events exist without corresponding aggregated totals.
+ *
+ * Falls back to the legacy two-step approach if the RPC is unavailable.
+ *
+ * Never throws — all errors are caught and logged internally.
+ */
+async function recordUsage(supabase, event, logger = defaultLogger) {
+    try {
+        const totalTokens = event.inputTokens +
+            event.outputTokens +
+            (event.cacheReadTokens ?? 0) +
+            (event.cacheCreationTokens ?? 0);
+        const costMicrodollars = (0, pricing_1.calculateCostMicrodollars)(event.model, event.inputTokens, event.outputTokens, event.cacheReadTokens ?? 0, event.cacheCreationTokens ?? 0, event.costMicrodollarsOverride);
+        const creditsConsumed = (0, pricing_1.microdollarsToCredits)(costMicrodollars);
+        // Platform-paid metadata (auto-tag so analytics can filter)
+        const effectiveMetadata = event.platformPaid
+            ? { ...(event.metadata ?? {}), platform_paid: true }
+            : event.metadata ?? {};
+        // Platform-paid events: audit-only insert into usage_events,
+        // skip the usage_records aggregate so they never hit the user's balance.
+        if (event.platformPaid) {
+            const { error: auditError } = await supabase.from('usage_events').insert({
+                organization_id: event.organizationId,
+                ...(event.conversationId && { conversation_id: event.conversationId }),
+                service: event.service,
+                model: event.model,
+                input_tokens: event.inputTokens,
+                output_tokens: event.outputTokens,
+                cache_read_tokens: event.cacheReadTokens ?? 0,
+                cache_creation_tokens: event.cacheCreationTokens ?? 0,
+                total_tokens: totalTokens,
+                api_calls: event.apiCalls ?? 1,
+                cost_microdollars: costMicrodollars,
+                credits_consumed: 0,
+                metadata: effectiveMetadata,
+            });
+            if (auditError) {
+                logger.warn('Failed to insert platform-paid usage_event (audit-only)', {
+                    orgId: event.organizationId,
+                    error: auditError.message,
+                });
+            }
+            return;
+        }
+        // Attempt atomic insert: event + aggregate in one transaction
+        const { data: eventId, error: atomicError } = await supabase.rpc('record_usage_atomic', {
+            p_organization_id: event.organizationId,
+            p_service: event.service,
+            p_model: event.model,
+            p_input_tokens: event.inputTokens,
+            p_output_tokens: event.outputTokens,
+            p_cache_read_tokens: event.cacheReadTokens ?? 0,
+            p_cache_creation_tokens: event.cacheCreationTokens ?? 0,
+            p_api_calls: event.apiCalls ?? 1,
+            p_cost_microdollars: costMicrodollars,
+            p_credits_consumed: creditsConsumed,
+            p_conversation_id: event.conversationId ?? null,
+            p_metadata: effectiveMetadata,
+        });
+        if (!atomicError) {
+            logger.debug('Usage recorded (atomic)', {
+                orgId: event.organizationId,
+                eventId,
+                service: event.service,
+                model: event.model,
+                totalTokens,
+                costMicrodollars,
+                creditsConsumed,
+            });
+            return;
+        }
+        // -----------------------------------------------------------------------
+        // Fallback: legacy two-step approach (RPC not deployed yet)
+        // -----------------------------------------------------------------------
+        logger.warn('record_usage_atomic RPC failed, falling back to two-step', {
+            orgId: event.organizationId,
+            error: atomicError.message,
+        });
+        // 1. Insert usage event (audit trail)
+        const { error: eventError } = await supabase.from('usage_events').insert({
+            organization_id: event.organizationId,
+            ...(event.conversationId && { conversation_id: event.conversationId }),
+            service: event.service,
+            model: event.model,
+            input_tokens: event.inputTokens,
+            output_tokens: event.outputTokens,
+            cache_read_tokens: event.cacheReadTokens ?? 0,
+            cache_creation_tokens: event.cacheCreationTokens ?? 0,
+            total_tokens: totalTokens,
+            api_calls: event.apiCalls ?? 1,
+            cost_microdollars: costMicrodollars,
+            credits_consumed: creditsConsumed,
+            metadata: effectiveMetadata,
+        });
+        if (eventError) {
+            logger.error('Failed to insert usage_event', {
+                orgId: event.organizationId,
+                error: eventError.message,
+            });
+            // Do NOT return — still try to upsert the aggregate so drift is minimized
+        }
+        // 2. Get current subscription period
+        const { data: sub, error: subError } = await supabase
+            .from('organization_subscriptions')
+            .select('current_period_start, current_period_end')
+            .eq('organization_id', event.organizationId)
+            .order('created_at', { ascending: false })
+            .limit(1)
+            .single();
+        if (subError || !sub) {
+            logger.warn('No subscription found, skipping usage_records upsert', {
+                orgId: event.organizationId,
+                error: subError?.message,
+            });
+            return;
+        }
+        // 3. Upsert into usage_records via RPC
+        const { error: upsertError } = await supabase.rpc('upsert_usage_record', {
+            p_organization_id: event.organizationId,
+            p_period_start: sub.current_period_start,
+            p_period_end: sub.current_period_end,
+            p_input_tokens: event.inputTokens,
+            p_output_tokens: event.outputTokens,
+            p_cache_read_tokens: event.cacheReadTokens ?? 0,
+            p_cache_creation_tokens: event.cacheCreationTokens ?? 0,
+            p_api_call_count: event.apiCalls ?? 1,
+        });
+        if (upsertError) {
+            logger.error('Failed to upsert usage_records (legacy path)', {
+                orgId: event.organizationId,
+                error: upsertError.message,
+            });
+        }
+        logger.debug('Usage recorded (legacy)', {
+            orgId: event.organizationId,
+            service: event.service,
+            model: event.model,
+            totalTokens,
+            costMicrodollars,
+            creditsConsumed,
+        });
+    }
+    catch (err) {
+        // Never throw — billing must not break the calling service
+        logger.error('Unexpected error recording usage', {
+            orgId: event.organizationId,
+            error: err instanceof Error ? err.message : String(err),
+        });
+    }
+}
+/**
+ * Extract token counts from a Vertex AI generateContent() response.
+ * Works with both typed SDK responses and raw JSON.
+ */
+function extractVertexUsageMetadata(response) {
+    const r = response;
+    const usage = r?.response?.usageMetadata ?? r?.usageMetadata ?? {};
+    return {
+        promptTokenCount: usage.promptTokenCount ?? 0,
+        candidatesTokenCount: usage.candidatesTokenCount ?? 0,
+        totalTokenCount: usage.totalTokenCount ?? 0,
+        cachedContentTokenCount: usage.cachedContentTokenCount ?? 0,
+    };
+}
+//# sourceMappingURL=usage-tracker.js.map

package/dist/usage-tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage-tracker.js","sourceRoot":"","sources":["../src/usage-tracker.ts"],"names":[],"mappings":";;AAqBA,kCA2KC;AAiBD,gEASC;AAzND,uCAA6E;AAE7E,MAAM,aAAa,GAAkB;IACnC,KAAK,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,GAAE,CAAC;IACxB,IAAI,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,GAAG,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC;IAChE,IAAI,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,aAAa,GAAG,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC;IACjE,KAAK,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,GAAG,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC;CACpE,CAAC;AAEF;;;;;;;;;;GAUG;AACI,KAAK,UAAU,WAAW,CAC/B,QAAwB,EACxB,KAAiB,EACjB,SAAwB,aAAa;IAErC,IAAI,CAAC;QACH,MAAM,WAAW,GACf,KAAK,CAAC,WAAW;YACjB,KAAK,CAAC,YAAY;YAClB,CAAC,KAAK,CAAC,eAAe,IAAI,CAAC,CAAC;YAC5B,CAAC,KAAK,CAAC,mBAAmB,IAAI,CAAC,CAAC,CAAC;QAEnC,MAAM,gBAAgB,GAAG,IAAA,mCAAyB,EAChD,KAAK,CAAC,KAAK,EACX,KAAK,CAAC,WAAW,EACjB,KAAK,CAAC,YAAY,EAClB,KAAK,CAAC,eAAe,IAAI,CAAC,EAC1B,KAAK,CAAC,mBAAmB,IAAI,CAAC,EAC9B,KAAK,CAAC,wBAAwB,CAC/B,CAAC;QAEF,MAAM,eAAe,GAAG,IAAA,+BAAqB,EAAC,gBAAgB,CAAC,CAAC;QAEhE,4DAA4D;QAC5D,MAAM,iBAAiB,GAAG,KAAK,CAAC,YAAY;YAC1C,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,aAAa,EAAE,IAAI,EAAE;YACpD,CAAC,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC;QAEzB,6DAA6D;QAC7D,yEAAyE;QACzE,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC;gBACvE,eAAe,EAAE,KAAK,CAAC,cAAc;gBACrC,GAAG,CAAC,KAAK,CAAC,cAAc,IAAI,EAAE,eAAe,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC;gBACtE,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,YAAY,EAAE,KAAK,CAAC,WAAW;gBAC/B,aAAa,EAAE,KAAK,CAAC,YAAY;gBACjC,iBAAiB,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;gBAC7C,qBAAqB,EAAE,KAAK,CAAC,mBAAmB,IAAI,CAAC;gBACrD,YAAY,EAAE,WAAW;gBACzB,SAAS,EAAE,KAAK,CAAC,QAAQ,IAAI,CAAC;gBAC9B,iBAAiB,EAAE,gBAAgB;gBACnC,gBAAgB,EAAE,CAAC;gBACnB,QAAQ,EAAE,iBAAiB;aAC5B,CAAC,CAAC;YACH,IAAI,UAAU,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,yDAAyD,EAAE;oBACrE,KAAK,EAAE,KAAK,CAAC,cAAc;oBAC3B,KAAK,EAAE,UAAU,CAAC,OAAO;iBAC1B,CAAC,CAAC;YACL,CAAC;YACD,OAAO;QACT,CAAC;QAED,8DAA8D;QAC9D,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,MAAM,QAAQ,CAAC,GAAG,CAC9D,qBAAqB,EACrB;YACE,iBAAiB,EAAE,KAAK,CAAC,cAAc;YACvC,SAAS,EAAE,KAAK,CAAC,OAAO;YACxB,OAAO,EAAE,KAAK,CAAC,KAAK;YACpB,cAAc,EAAE,KAAK,CAAC,WAAW;YACjC,eAAe,EAAE,KAAK,CAAC,YAAY;YACnC,mBAAmB,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;YAC/C,uBAAuB,EAAE,KAAK,CAAC,mBAAmB,IAAI,CAAC;YACvD,WAAW,EAAE,KAAK,CAAC,QAAQ,IAAI,CAAC;YAChC,mBAAmB,EAAE,gBAAgB;YACrC,kBAAkB,EAAE,eAAe;YACnC,iBAAiB,EAAE,KAAK,CAAC,cAAc,IAAI,IAAI;YAC/C,UAAU,EAAE,iBAAiB;SAC9B,CACF,CAAC;QAEF,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE;gBACtC,KAAK,EAAE,KAAK,CAAC,cAAc;gBAC3B,OAAO;gBACP,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,KAAK,EAAE,KAAK,CAAC,KAAK;gBAClB,WAAW;gBACX,gBAAgB;gBAChB,eAAe;aAChB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,0EAA0E;QAC1E,4DAA4D;QAC5D,0EAA0E;QAC1E,MAAM,CAAC,IAAI,CAAC,0DAA0D,EAAE;YACtE,KAAK,EAAE,KAAK,CAAC,cAAc;YAC3B,KAAK,EAAE,WAAW,CAAC,OAAO;SAC3B,CAAC,CAAC;QAEH,sCAAsC;QACtC,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,MAAM,CAAC;YACvE,eAAe,EAAE,KAAK,CAAC,cAAc;YACrC,GAAG,CAAC,KAAK,CAAC,cAAc,IAAI,EAAE,eAAe,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC;YACtE,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,YAAY,EAAE,KAAK,CAAC,WAAW;YAC/B,aAAa,EAAE,KAAK,CAAC,YAAY;YACjC,iBAAiB,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;YAC7C,qBAAqB,EAAE,KAAK,CAAC,mBAAmB,IAAI,CAAC;YACrD,YAAY,EAAE,WAAW;YACzB,SAAS,EAAE,KAAK,CAAC,QAAQ,IAAI,CAAC;YAC9B,iBAAiB,EAAE,gBAAgB;YACnC,gBAAgB,EAAE,eAAe;YACjC,QAAQ,EAAE,iBAAiB;SAC5B,CAAC,CAAC;QAEH,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE;gBAC3C,KAAK,EAAE,KAAK,CAAC,cAAc;gBAC3B,KAAK,EAAE,UAAU,CAAC,OAAO;aAC1B,CAAC,CAAC;YACH,0EAA0E;QAC5E,CAAC;QAED,qCAAqC;QACrC,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,QAAQ;aAClD,IAAI,CAAC,4BAA4B,CAAC;aAClC,MAAM,CAAC,0CAA0C,CAAC;aAClD,EAAE,CAAC,iBAAiB,EAAE,KAAK,CAAC,cAAc,CAAC;aAC3C,KAAK,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;aACzC,KAAK,CAAC,CAAC,CAAC;aACR,MAAM,EAAE,CAAC;QAEZ,IAAI,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAC;YACrB,MAAM,CAAC,IAAI,CAAC,sDAAsD,EAAE;gBAClE,KAAK,EAAE,KAAK,CAAC,cAAc;gBAC3B,KAAK,EAAE,QAAQ,EAAE,OAAO;aACzB,CAAC,CAAC;YACH,OAAO;QACT,CAAC;QAED,uCAAuC;QACvC,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,qBAAqB,EAAE;YACvE,iBAAiB,EAAE,KAAK,CAAC,cAAc;YACvC,cAAc,EAAE,GAAG,CAAC,oBAAoB;YACxC,YAAY,EAAE,GAAG,CAAC,kBAAkB;YACpC,cAAc,EAAE,KAAK,CAAC,WAAW;YACjC,eAAe,EAAE,KAAK,CAAC,YAAY;YACnC,mBAAmB,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;YAC/C,uBAAuB,EAAE,KAAK,CAAC,mBAAmB,IAAI,CAAC;YACvD,gBAAgB,EAAE,KAAK,CAAC,QAAQ,IAAI,CAAC;SACtC,CAAC,CAAC;QAEH,IAAI,WAAW,EAAE,CAAC;YAChB,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE;gBAC3D,KAAK,EAAE,KAAK,CAAC,cAAc;gBAC3B,KAAK,EAAE,WAAW,CAAC,OAAO;aAC3B,CAAC,CAAC;QACL,CAAC;QAED,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE;YACtC,KAAK,EAAE,KAAK,CAAC,cAAc;YAC3B,OAAO,EAAE,KAAK,CAAC,OAAO;YACtB,KAAK,EAAE,KAAK,CAAC,KAAK;YAClB,WAAW;YACX,gBAAgB;YAChB,eAAe;SAChB,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,2DAA2D;QAC3D,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE;YAC/C,KAAK,EAAE,KAAK,CAAC,cAAc;YAC3B,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;SACxD,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAaD;;;GAGG;AACH,SAAgB,0BAA0B,CAAC,QAAiB;IAC1D,MAAM,CAAC,GAAG,QAA2C,CAAC;IACtD,MAAM,KAAK,GAAG,CAAC,EAAE,QAAQ,EAAE,aAAa,IAAI,CAAC,EAAE,aAAa,IAAI,EAAE,CAAC;IACnE,OAAO;QACL,gBAAgB,EAAE,KAAK,CAAC,gBAAgB,IAAI,CAAC;QAC7C,oBAAoB,EAAE,KAAK,CAAC,oBAAoB,IAAI,CAAC;QACrD,eAAe,EAAE,KAAK,CAAC,eAAe,IAAI,CAAC;QAC3C,uBAAuB,EAAE,KAAK,CAAC,uBAAuB,IAAI,CAAC;KAC5D,CAAC;AACJ,CAAC"}

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "@trendyvideo/billing",
+  "version": "1.0.0",
+  "description": "Unified billing, usage tracking, and plan limit enforcement",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "clean": "rm -rf dist",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "prepublishOnly": "npm run clean && npm run build"
+  },
+  "keywords": [
+    "trendy",
+    "billing",
+    "usage",
+    "pricing",
+    "plan-limits"
+  ],
+  "author": "Trendy Team",
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "peerDependencies": {
+    "@supabase/supabase-js": "^2.45.0"
+  },
+  "devDependencies": {
+    "@types/node": "^20.10.0",
+    "typescript": "^5.3.0",
+    "vitest": "^4.0.18"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}