vibezcheck 0.5.11 → 0.5.12

package/dist/ai-sdk/index.d.mts

@@ -1,8 +1,8 @@
-import { W as WithBillingOptions } from '../with-billing-DhHGWCAp.mjs';
-export { m as meteredModel, w as withBilling } from '../with-billing-DhHGWCAp.mjs';
-import { A as AgentSession, a as AgentSessionOptions, c as createAgentSession } from '../session-D9fdP255.mjs';
-import { C as CustomerParam } from '../types-DHlsaLvI.mjs';
-import '../client-ttJwYM7x.mjs';
+import { W as WithBillingOptions } from '../with-billing-Df6izlJ4.mjs';
+export { m as meteredModel, w as withBilling } from '../with-billing-Df6izlJ4.mjs';
+import { A as AgentSession, a as AgentSessionOptions, c as createAgentSession } from '../session-CueUIV77.mjs';
+import { C as CustomerParam } from '../types-CqXTmCNv.mjs';
+import '../client-xP86Qg_b.mjs';
 
 /**
  * ✦ VibezCheck Universal Telemetry Response Stream Injector

package/dist/ai-sdk/index.d.ts

@@ -1,8 +1,8 @@
-import { W as WithBillingOptions } from '../with-billing-CVGxl3ot.js';
-export { m as meteredModel, w as withBilling } from '../with-billing-CVGxl3ot.js';
-import { A as AgentSession, a as AgentSessionOptions, c as createAgentSession } from '../session-CUnv_muf.js';
-import { C as CustomerParam } from '../types-DHlsaLvI.js';
-import '../client-eXSwD_kI.js';
+import { W as WithBillingOptions } from '../with-billing-DgQ380Zl.js';
+export { m as meteredModel, w as withBilling } from '../with-billing-DgQ380Zl.js';
+import { A as AgentSession, a as AgentSessionOptions, c as createAgentSession } from '../session-YF67Yij2.js';
+import { C as CustomerParam } from '../types-CqXTmCNv.js';
+import '../client-0CuOBECi.js';
 
 /**
  * ✦ VibezCheck Universal Telemetry Response Stream Injector

package/dist/ai-sdk/index.js

@@ -800,7 +800,29 @@ function getModelPricing(modelName) {
 }
 
 // src/pricing/calculator.ts
-function calculateCost(params) {
+function calculateCost(modelOrParams, usage, optionsOrMarkup) {
+  let params;
+  if (typeof modelOrParams === "string") {
+    const opts = typeof optionsOrMarkup === "number" ? { markupMultiplier: optionsOrMarkup } : optionsOrMarkup || {};
+    params = {
+      model: modelOrParams,
+      inputTokens: usage?.inputTokens ?? usage?.promptTokens ?? 0,
+      outputTokens: usage?.outputTokens ?? usage?.completionTokens ?? 0,
+      reasoningTokens: usage?.reasoningTokens,
+      cachedTokens: usage?.cachedTokens,
+      cacheWriteTokens: usage?.cacheWriteTokens,
+      markupMultiplier: opts.markupMultiplier ?? opts.markup ?? opts.margin,
+      minimumChargeUSD: opts.minimumChargeUSD,
+      customRate: opts.customRate
+    };
+  } else {
+    params = {
+      ...modelOrParams,
+      inputTokens: modelOrParams.inputTokens ?? modelOrParams.promptTokens ?? 0,
+      outputTokens: modelOrParams.outputTokens ?? modelOrParams.completionTokens ?? 0,
+      markupMultiplier: modelOrParams.markupMultiplier ?? modelOrParams.markup ?? modelOrParams.margin
+    };
+  }
   let rates;
   if (params.customRate) {
     const outRate = params.customRate.out ?? params.customRate.output ?? 0;
@@ -1085,7 +1107,7 @@ var VibezMeter = class {
   batcher;
   markupMultiplier;
   constructor(options = {}) {
-    this.markupMultiplier = options.pricing?.margin ?? options.markupMultiplier;
+    this.markupMultiplier = options.pricing?.margin ?? options.pricing?.markup ?? options.markupMultiplier;
     this.batcher = new MeterBatcher(options);
   }
   /**
@@ -1277,7 +1299,7 @@ function withBilling(model, optionsOrCustomer, extraOptions = {}) {
       cacheWriteTokens: cacheWriteTokens > 0 ? cacheWriteTokens : void 0
     };
     const cost = calculateUsageCost(modelId, usage, {
-      markupMultiplier: options.pricing?.margin,
+      markupMultiplier: options.pricing?.margin ?? options.pricing?.markup,
       minimumChargeUSD: options.pricing?.minimumChargeUSD,
       customRate: options.rate
     });
@@ -1316,6 +1338,7 @@ function withBilling(model, optionsOrCustomer, extraOptions = {}) {
     const mergedMetadata = {
       ...customerMetadata,
       ...options.metadata,
+      ...options.threadId ? { threadId: options.threadId } : {},
       ...extraMeta
     };
     const event = {
@@ -1684,7 +1707,7 @@ var AgentSession = class {
    */
   model(modelOrId, modelOptions = {}) {
     const session = this;
-    return withBilling(modelOrId, {
+    return createVibezModel(modelOrId, {
       ...modelOptions,
       customer: this.customer,
       pricing: modelOptions.pricing || this.options.pricing,
@@ -1776,7 +1799,13 @@ var AgentSession = class {
     const defaultCost = options.costPerActionUSD ?? 5e-3;
     const instrumented = {};
     for (const [name, toolDef] of Object.entries(toolsRecord)) {
-      const costUSD = options.costs?.[name] ?? defaultCost;
+      let costUSD = defaultCost;
+      if (options && "costs" in options && options.costs?.[name] !== void 0) {
+        costUSD = options.costs[name];
+      } else if (options && name in options) {
+        const val = options[name];
+        costUSD = typeof val === "number" ? val : val?.costUSD ?? defaultCost;
+      }
       instrumented[name] = this.wrapTool(toolDef, { name, costUSD });
     }
     return instrumented;