vibezcheck 0.4.0 → 0.4.2

package/dist/ai-sdk/index.js

@@ -650,6 +650,58 @@ function calculateUsageCost(model, usage, optionsOrMarkup) {
   });
 }
 
+// src/customers/helpers.ts
+function normalizeCustomer(customer, extraCustomerId) {
+  if (!customer && !extraCustomerId) {
+    return {
+      customerId: void 0,
+      customerEmail: void 0,
+      customerObj: void 0,
+      customerMetadata: {}
+    };
+  }
+  if (typeof customer === "string") {
+    const isEmail = customer.includes("@");
+    const customerObj2 = {
+      id: customer,
+      email: isEmail ? customer : void 0
+    };
+    return {
+      customerId: customer,
+      customerEmail: isEmail ? customer : void 0,
+      customerObj: customerObj2,
+      customerMetadata: {}
+    };
+  }
+  const customerObj = customer || (extraCustomerId ? { id: extraCustomerId } : void 0);
+  const customerId = customerObj?.id || customerObj?.userId || extraCustomerId;
+  const customerEmail = customerObj?.email;
+  const customerMetadata = {};
+  if (customerObj) {
+    const orgId = customerObj.orgId || customerObj.organizationId;
+    if (orgId) customerMetadata.org_id = String(orgId);
+    const teamId = customerObj.teamId || customerObj.workspaceId;
+    if (teamId) customerMetadata.team_id = String(teamId);
+    if (customerObj.plan) customerMetadata.plan = String(customerObj.plan);
+    if (customerObj.tier) customerMetadata.tier = String(customerObj.tier);
+    if (customerObj.role) customerMetadata.role = String(customerObj.role);
+    if (customerObj.orgName) customerMetadata.org_name = String(customerObj.orgName);
+    if (customerObj.metadata) {
+      for (const [key, val] of Object.entries(customerObj.metadata)) {
+        if (val !== null && val !== void 0) {
+          customerMetadata[key] = val;
+        }
+      }
+    }
+  }
+  return {
+    customerId,
+    customerEmail,
+    customerObj,
+    customerMetadata
+  };
+}
+
 // src/meter/stream.ts
 function wrapOpenAIStream(stream, options, onComplete) {
   let detectedModel = options.model || "gpt-4o";
@@ -670,15 +722,20 @@ function wrapOpenAIStream(stream, options, onComplete) {
       } finally {
         if (finalUsage) {
           const cost = calculateUsageCost(detectedModel, finalUsage);
-          const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+          const normalized = normalizeCustomer(options.customer, options.customerId);
           const event = {
             timestamp: (/* @__PURE__ */ new Date()).toISOString(),
             model: detectedModel,
             provider: "openai",
             usage: finalUsage,
             cost,
-            customerId,
-            metadata: options.metadata
+            customerId: normalized.customerId,
+            customerEmail: normalized.customerEmail,
+            customer: normalized.customerObj,
+            metadata: {
+              ...normalized.customerMetadata,
+              ...options.metadata
+            }
           };
           onComplete(event);
           if (options.onUsage) {
@@ -703,15 +760,20 @@ function wrapAnthropicStream(stream, options, onComplete) {
         const extracted = accumulator.getUsage();
         const model = options.model || extracted.model;
         const cost = calculateUsageCost(model, extracted.usage);
-        const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+        const normalized = normalizeCustomer(options.customer, options.customerId);
         const usageEvent = {
           timestamp: (/* @__PURE__ */ new Date()).toISOString(),
           model,
           provider: "anthropic",
           usage: extracted.usage,
           cost,
-          customerId,
-          metadata: options.metadata
+          customerId: normalized.customerId,
+          customerEmail: normalized.customerEmail,
+          customer: normalized.customerObj,
+          metadata: {
+            ...normalized.customerMetadata,
+            ...options.metadata
+          }
         };
         onComplete(usageEvent);
         if (options.onUsage) {
@@ -740,15 +802,20 @@ function wrapGeminiStream(result, options, onComplete) {
         const extracted = extractGeminiResponseUsage(lastChunkWithUsage, model);
         if (extracted) {
           const cost = calculateUsageCost(extracted.model, extracted.usage);
-          const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+          const normalized = normalizeCustomer(options.customer, options.customerId);
           const event = {
             timestamp: (/* @__PURE__ */ new Date()).toISOString(),
             model: extracted.model,
             provider: "google",
             usage: extracted.usage,
             cost,
-            customerId,
-            metadata: options.metadata
+            customerId: normalized.customerId,
+            customerEmail: normalized.customerEmail,
+            customer: normalized.customerObj,
+            metadata: {
+              ...normalized.customerMetadata,
+              ...options.metadata
+            }
           };
           onComplete(event);
           if (options.onUsage) {
@@ -791,7 +858,7 @@ var VibezMeter = class {
       this.stripeClient = new import_stripe.default(key, {
         appInfo: {
           name: "vibezcheck",
-          version: "0.1.0",
+          version: "0.4.2",
           url: "https://vibezcheck.xyz"
         }
       });
@@ -811,15 +878,21 @@ var VibezMeter = class {
     }
     const model = options.model || extracted.model;
     const cost = calculateUsageCost(model, extracted.usage, this.markupMultiplier);
-    const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+    const normalized = normalizeCustomer(options.customer, options.customerId);
+    const mergedMeta = {
+      ...normalized.customerMetadata,
+      ...options.metadata
+    };
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model,
       provider: extracted.provider,
       usage: extracted.usage,
       cost,
-      customerId,
-      metadata: options.metadata
+      customerId: normalized.customerId,
+      customerEmail: normalized.customerEmail,
+      customer: normalized.customerObj,
+      metadata: mergedMeta
     };
     this.batcher.enqueue(event);
     return event;
@@ -849,15 +922,21 @@ var VibezMeter = class {
       cachedTokens
     };
     const cost = calculateUsageCost(options.model, usage, this.markupMultiplier);
-    const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+    const normalized = normalizeCustomer(options.customer, options.customerId);
+    const mergedMeta = {
+      ...normalized.customerMetadata,
+      ...options.metadata
+    };
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model: options.model,
       provider: options.provider || "custom",
       usage,
       cost,
-      customerId,
-      metadata: options.metadata
+      customerId: normalized.customerId,
+      customerEmail: normalized.customerEmail,
+      customer: normalized.customerObj,
+      metadata: mergedMeta
     };
     this.batcher.enqueue(event);
     return event;
@@ -894,7 +973,11 @@ function withBilling(model, options = {}) {
     apiKey: options.stripeApiKey,
     eventName: options.eventName
   });
-  const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
+  const normalized = normalizeCustomer(options.customer, options.customerId);
+  const customerId = normalized.customerId;
+  const customerEmail = normalized.customerEmail;
+  const customerObj = normalized.customerObj;
+  const customerMetadata = normalized.customerMetadata;
   const modelId = model.modelId || "unknown-model";
   const provider = model.provider?.replace(/^@ai-sdk\//, "") || "ai-sdk";
   const maxCostPerCall = options.maxCostPerCallUSD !== void 0 ? options.maxCostPerCallUSD : 0.5;
@@ -963,6 +1046,12 @@ function withBilling(model, options = {}) {
         throw new VibezCircuitBreakerError(budgetEvent);
       }
     }
+    const mergedMetadata = {
+      ...customerMetadata,
+      ...options.metadata,
+      ...extraMeta,
+      billingMode: options.billing?.mode || "postpaid"
+    };
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model: modelId,
@@ -970,11 +1059,9 @@ function withBilling(model, options = {}) {
       usage,
       cost,
       customerId,
-      metadata: {
-        ...options.metadata,
-        ...extraMeta,
-        billingMode: options.billing?.mode || "postpaid"
-      }
+      customerEmail,
+      customer: customerObj,
+      metadata: mergedMetadata
     };
     meter.recordUsage({
       model: modelId,
@@ -984,11 +1071,84 @@ function withBilling(model, options = {}) {
       reasoningTokens: usage.reasoningTokens,
       cachedTokens: usage.cachedTokens,
       customerId,
+      customer: customerObj,
       metadata: event.metadata
     });
     if (options.onUsage) {
       options.onUsage(event);
     }
+    if (options.database) {
+      const dbTarget = options.database;
+      const executeDbWrite = async () => {
+        try {
+          const supabaseClient = typeof dbTarget === "object" && dbTarget !== null && typeof dbTarget.from === "function" ? dbTarget : typeof dbTarget === "object" && dbTarget !== null && typeof dbTarget.client?.from === "function" ? dbTarget.client : null;
+          if (supabaseClient) {
+            const tableName = dbTarget?.table || "vibez_usage";
+            await supabaseClient.from(tableName).insert({
+              customer_id: customerId,
+              user_id: customerObj?.userId || customerId,
+              model: modelId,
+              input_tokens: usage.inputTokens,
+              output_tokens: usage.outputTokens,
+              total_tokens: usage.totalTokens,
+              cached_tokens: usage.cachedTokens,
+              reasoning_tokens: usage.reasoningTokens,
+              cost_usd: cost.totalUSD,
+              created_at: event.timestamp,
+              metadata: event.metadata
+            });
+            return;
+          }
+          if (typeof dbTarget === "object" && dbTarget !== null && typeof dbTarget.save === "function") {
+            await dbTarget.save(event);
+            return;
+          }
+          if (typeof dbTarget === "function") {
+            await dbTarget(event);
+            return;
+          }
+        } catch (err) {
+          console.warn(`[vibezcheck] Database record failed safely:`, err?.message || err);
+        }
+      };
+      if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
+        globalThis.after(() => executeDbWrite());
+      } else {
+        executeDbWrite().catch(() => {
+        });
+      }
+    }
+    if (options.billing?.charge && typeof options.billing.charge === "function") {
+      const chargeFn = options.billing.charge;
+      const executeCharge = async () => {
+        try {
+          await chargeFn(cost.totalUSD, event);
+        } catch (err) {
+          console.warn(`[vibezcheck] Custom charge handler failed safely:`, err?.message || err);
+        }
+      };
+      if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
+        globalThis.after(() => executeCharge());
+      } else {
+        executeCharge().catch(() => {
+        });
+      }
+    } else if (options.billing?.provider && typeof options.billing.provider === "object" && typeof options.billing.provider.charge === "function") {
+      const providerCharge = options.billing.provider.charge;
+      const executeCharge = async () => {
+        try {
+          await providerCharge(cost.totalUSD, event);
+        } catch (err) {
+          console.warn(`[vibezcheck] Payment provider charge failed safely:`, err?.message || err);
+        }
+      };
+      if (typeof globalThis !== "undefined" && typeof globalThis.after === "function") {
+        globalThis.after(() => executeCharge());
+      } else {
+        executeCharge().catch(() => {
+        });
+      }
+    }
     scheduleFlush();
   };
   const handler = {
@@ -1036,6 +1196,8 @@ function withBilling(model, options = {}) {
               controller.enqueue(chunk);
               if (chunk.type === "text-delta" && chunk.textDelta) {
                 accumulatedChars += chunk.textDelta.length;
+              } else if (chunk.type === "reasoning" && (chunk.textDelta || chunk.reasoning)) {
+                accumulatedChars += (chunk.textDelta || chunk.reasoning || "").length;
               }
               if (chunk.type === "finish" && chunk.usage) {
                 streamCompleted = true;