vibezcheck 0.3.0 → 0.4.0

package/dist/index.js

@@ -48,6 +48,7 @@ __export(src_exports, {
   createMeter: () => createMeter,
   createVibezCheck: () => createVibezCheck,
   createVibezModel: () => createVibezModel,
+  createVibezSession: () => createVibezSession,
   detectAndExtractUsage: () => detectAndExtractUsage,
   extractAnthropicResponseUsage: () => extractAnthropicResponseUsage,
   extractAuthToken: () => extractAuthToken,
@@ -63,7 +64,7 @@ __export(src_exports, {
   vibes: () => vibes,
   vibescheck: () => vibescheck,
   vibez: () => vibez,
-  vibezcheck: () => vibezcheck,
+  vibezcheck: () => vibezcheck2,
   withBilling: () => withBilling,
   wrapAnthropicStream: () => wrapAnthropicStream,
   wrapGeminiStream: () => wrapGeminiStream,
@@ -620,14 +621,26 @@ function registerModelPricing(modelName, rates) {
 
 // src/pricing/calculator.ts
 function calculateCost(params) {
-  const rates = getModelPricing(params.model);
+  let rates;
+  if (params.customRate) {
+    const outRate = params.customRate.out ?? params.customRate.output ?? 0;
+    rates = {
+      inputPer1M: params.customRate.in,
+      outputPer1M: outRate,
+      reasoningPer1M: params.customRate.reasoning ?? outRate,
+      cachedInputPer1M: params.customRate.cached ?? params.customRate.in * 0.15,
+      currency: "USD"
+    };
+  } else {
+    rates = getModelPricing(params.model);
+  }
   const inputTokens = params.inputTokens ?? 0;
   const outputTokens = params.outputTokens ?? 0;
   const reasoningTokens = params.reasoningTokens ?? 0;
   const cachedTokens = params.cachedTokens ?? 0;
   const regularInputTokens = Math.max(0, inputTokens - cachedTokens);
   const regularInputCost = regularInputTokens / 1e6 * rates.inputPer1M;
-  const cachedRate = rates.cachedInputPer1M ?? rates.inputPer1M * 0.5;
+  const cachedRate = rates.cachedInputPer1M ?? rates.inputPer1M * 0.15;
   const cachedInputCost = cachedTokens / 1e6 * cachedRate;
   const inputCostUSD = regularInputCost + cachedInputCost;
   const outputCostUSD = outputTokens / 1e6 * rates.outputPer1M;
@@ -635,20 +648,27 @@ function calculateCost(params) {
   const reasoningCostUSD = reasoningTokens / 1e6 * reasoningRate;
   const standardCacheCost = cachedTokens / 1e6 * rates.inputPer1M;
   const cachedDiscountUSD = Math.max(0, standardCacheCost - cachedInputCost);
-  const totalUSD = inputCostUSD + outputCostUSD;
+  let totalUSD = inputCostUSD + outputCostUSD;
   const markup = params.markupMultiplier ?? 1;
-  const retailUSD = markup !== 1 ? totalUSD * markup : void 0;
+  let retailUSD = void 0;
+  if (markup !== 1 || params.minimumChargeUSD !== void 0) {
+    retailUSD = totalUSD * markup;
+    if (params.minimumChargeUSD !== void 0 && retailUSD < params.minimumChargeUSD) {
+      retailUSD = params.minimumChargeUSD;
+    }
+  }
   return {
     inputCostUSD: Number(inputCostUSD.toFixed(8)),
     outputCostUSD: Number(outputCostUSD.toFixed(8)),
     reasoningCostUSD: reasoningTokens > 0 ? Number(reasoningCostUSD.toFixed(8)) : void 0,
     cachedDiscountUSD: cachedTokens > 0 ? Number(cachedDiscountUSD.toFixed(8)) : void 0,
     totalUSD: Number(totalUSD.toFixed(8)),
-    retailUSD: retailUSD ? Number(retailUSD.toFixed(8)) : void 0,
+    retailUSD: retailUSD !== void 0 ? Number(retailUSD.toFixed(8)) : void 0,
     currency: rates.currency || "USD"
   };
 }
-function calculateUsageCost(model, usage, markupMultiplier) {
+function calculateUsageCost(model, usage, optionsOrMarkup) {
+  const options = typeof optionsOrMarkup === "number" ? { markupMultiplier: optionsOrMarkup } : optionsOrMarkup || {};
   return calculateCost({
     model,
     inputTokens: usage.inputTokens,
@@ -656,7 +676,9 @@ function calculateUsageCost(model, usage, markupMultiplier) {
     reasoningTokens: usage.reasoningTokens,
     cachedTokens: usage.cachedTokens,
     cacheWriteTokens: usage.cacheWriteTokens,
-    markupMultiplier
+    markupMultiplier: options.markupMultiplier,
+    minimumChargeUSD: options.minimumChargeUSD,
+    customRate: options.customRate
   });
 }
 
@@ -917,7 +939,22 @@ function withBilling(model, options = {}) {
   const customerId = typeof options.customer === "string" ? options.customer : options.customer?.id || options.customerId;
   const modelId = model.modelId || "unknown-model";
   const provider = model.provider?.replace(/^@ai-sdk\//, "") || "ai-sdk";
-  const handleUsage = (rawUsage) => {
+  const maxCostPerCall = options.maxCostPerCallUSD !== void 0 ? options.maxCostPerCallUSD : 0.5;
+  const captureOnAbort = options.captureOnAbort !== false;
+  const scheduleFlush = () => {
+    try {
+      const flushPromise = meter.flush();
+      if (typeof globalThis !== "undefined") {
+        const g = globalThis;
+        if (typeof g.after === "function") {
+          g.after(() => flushPromise);
+          return;
+        }
+      }
+    } catch {
+    }
+  };
+  const handleUsage = (rawUsage, extraMeta) => {
     if (!rawUsage) return;
     const inputTokens = rawUsage.promptTokens ?? rawUsage.inputTokens ?? 0;
     const outputTokens = rawUsage.completionTokens ?? rawUsage.outputTokens ?? 0;
@@ -931,15 +968,19 @@ function withBilling(model, options = {}) {
       visibleOutputTokens: Math.max(0, outputTokens - reasoningTokens),
       cachedTokens: cachedTokens > 0 ? cachedTokens : void 0
     };
-    const cost = calculateUsageCost(modelId, usage);
-    if (options.maxCostPerCallUSD && cost.totalUSD > options.maxCostPerCallUSD) {
+    const cost = calculateUsageCost(modelId, usage, {
+      markupMultiplier: options.pricing?.margin,
+      minimumChargeUSD: options.pricing?.minimumChargeUSD,
+      customRate: options.rate
+    });
+    if (maxCostPerCall > 0 && maxCostPerCall !== Infinity && cost.totalUSD > maxCostPerCall) {
       const budgetEvent = {
         reason: "cost_per_call_exceeded",
-        limit: options.maxCostPerCallUSD,
+        limit: maxCostPerCall,
         current: cost.totalUSD,
         model: modelId,
         customerId,
-        message: `[vibezcheck] Circuit Breaker tripped: Call cost ($${cost.totalUSD.toFixed(4)}) exceeded maxCostPerCallUSD threshold ($${options.maxCostPerCallUSD.toFixed(4)}).`
+        message: `[vibezcheck] Circuit Breaker tripped: Call cost ($${cost.totalUSD.toFixed(4)}) exceeded budget ceiling ($${maxCostPerCall.toFixed(4)}).`
       };
       if (options.onBudgetExceeded) {
         options.onBudgetExceeded(budgetEvent);
@@ -971,7 +1012,11 @@ function withBilling(model, options = {}) {
       usage,
       cost,
       customerId,
-      metadata: options.metadata
+      metadata: {
+        ...options.metadata,
+        ...extraMeta,
+        billingMode: options.billing?.mode || "postpaid"
+      }
     };
     meter.recordUsage({
       model: modelId,
@@ -981,11 +1026,12 @@ function withBilling(model, options = {}) {
       reasoningTokens: usage.reasoningTokens,
       cachedTokens: usage.cachedTokens,
       customerId,
-      metadata: options.metadata
+      metadata: event.metadata
     });
     if (options.onUsage) {
       options.onUsage(event);
     }
+    scheduleFlush();
   };
   const handler = {
     get(target, prop, receiver) {
@@ -997,12 +1043,59 @@ function withBilling(model, options = {}) {
             return result;
           }
           const originalStream = result.stream;
+          let streamCompleted = false;
+          let accumulatedChars = 0;
+          let estimatedInputTokens = 0;
+          try {
+            if (args[0]?.prompt) {
+              const str = typeof args[0].prompt === "string" ? args[0].prompt : JSON.stringify(args[0].prompt);
+              estimatedInputTokens = Math.ceil(str.length / 4);
+            }
+          } catch {
+          }
+          const abortSignal = args[0]?.abortSignal;
+          if (abortSignal && captureOnAbort) {
+            abortSignal.addEventListener(
+              "abort",
+              () => {
+                if (!streamCompleted && accumulatedChars > 0) {
+                  streamCompleted = true;
+                  const estimatedOutputTokens = Math.ceil(accumulatedChars / 3.8);
+                  handleUsage(
+                    {
+                      inputTokens: estimatedInputTokens,
+                      outputTokens: estimatedOutputTokens
+                    },
+                    { aborted: true, partial: true }
+                  );
+                }
+              },
+              { once: true }
+            );
+          }
           const transformStream = new TransformStream({
             transform(chunk, controller) {
               controller.enqueue(chunk);
+              if (chunk.type === "text-delta" && chunk.textDelta) {
+                accumulatedChars += chunk.textDelta.length;
+              }
               if (chunk.type === "finish" && chunk.usage) {
+                streamCompleted = true;
                 handleUsage(chunk.usage);
               }
+            },
+            flush() {
+              if (!streamCompleted && captureOnAbort && accumulatedChars > 0) {
+                streamCompleted = true;
+                const estimatedOutputTokens = Math.ceil(accumulatedChars / 3.8);
+                handleUsage(
+                  {
+                    inputTokens: estimatedInputTokens,
+                    outputTokens: estimatedOutputTokens
+                  },
+                  { aborted: true, partial: true }
+                );
+              }
             }
           });
           return {
@@ -1076,6 +1169,64 @@ function createVibezModel(modelOrId, options = {}) {
   }
   return modelOrId;
 }
+function createVibezSession(sessionOptions = {}) {
+  const meter = createMeter({
+    apiKey: sessionOptions.stripeApiKey
+  });
+  const customerId = typeof sessionOptions.customer === "string" ? sessionOptions.customer : sessionOptions.customer?.id;
+  return {
+    meter,
+    model: (modelOrId, callOptions = {}) => {
+      return createVibezModel(modelOrId, {
+        customer: sessionOptions.customer,
+        pricing: sessionOptions.pricing,
+        billing: sessionOptions.billing,
+        meter,
+        metadata: {
+          ...sessionOptions.metadata,
+          ...callOptions.metadata
+        },
+        ...callOptions
+      });
+    },
+    trackTool: async (name, { costUSD, metadata }) => {
+      const event = {
+        timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+        model: `tool:${name}`,
+        provider: "tool",
+        usage: {
+          inputTokens: 0,
+          outputTokens: 0,
+          totalTokens: 0
+        },
+        cost: {
+          inputCostUSD: 0,
+          outputCostUSD: costUSD,
+          totalUSD: costUSD,
+          currency: "USD"
+        },
+        customerId,
+        metadata: {
+          ...sessionOptions.metadata,
+          ...metadata,
+          tool: name
+        }
+      };
+      meter.recordUsage({
+        model: `tool:${name}`,
+        provider: "tool",
+        inputTokens: 0,
+        outputTokens: 0,
+        customerId,
+        metadata: event.metadata
+      });
+      await meter.flush();
+    }
+  };
+}
+var vibezcheck = Object.assign(createVibezModel, {
+  session: createVibezSession
+});
 
 // src/customers/manager.ts
 var import_stripe2 = __toESM(require("stripe"));
@@ -1439,20 +1590,21 @@ var VibezCheckClient = class {
 function createVibezCheck(config = {}) {
   return new VibezCheckClient(config);
 }
-function vibezcheck(firstArg, secondArg) {
+function vibezcheck2(firstArg, secondArg) {
   if (typeof firstArg === "string" || firstArg && (firstArg.specificationVersion || firstArg.doGenerate || firstArg.doStream)) {
     return createVibezModel(firstArg, secondArg);
   }
   return new VibezCheckClient(firstArg || {});
 }
-vibezcheck.calculateCost = calculateCost;
-vibezcheck.getModelPricing = getModelPricing;
-vibezcheck.registerModelPricing = registerModelPricing;
-vibezcheck.create = createVibezCheck;
-vibezcheck.withBilling = withBilling;
-vibezcheck.createMeter = createMeter;
+vibezcheck2.calculateCost = calculateCost;
+vibezcheck2.getModelPricing = getModelPricing;
+vibezcheck2.registerModelPricing = registerModelPricing;
+vibezcheck2.create = createVibezCheck;
+vibezcheck2.withBilling = withBilling;
+vibezcheck2.createMeter = createMeter;
+vibezcheck2.session = createVibezSession;
 var vibez = new VibezCheckClient();
-var vibescheck = vibezcheck;
+var vibescheck = vibezcheck2;
 var vibes = vibez;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
@@ -1474,6 +1626,7 @@ var vibes = vibez;
   createMeter,
   createVibezCheck,
   createVibezModel,
+  createVibezSession,
   detectAndExtractUsage,
   extractAnthropicResponseUsage,
   extractAuthToken,