vibezcheck 0.5.11 → 0.5.13

package/dist/ai-sdk/index.mjs

@@ -564,6 +564,10 @@ var MODEL_PRICING_TABLE = {
   "text-embedding-3-large": { inputPer1M: 0.13, outputPer1M: 0 },
   "text-embedding-ada-002": { inputPer1M: 0.1, outputPer1M: 0 },
   // --- Anthropic (Modern & Extended Thinking) ---
+  "claude-haiku-5.5": { inputPer1M: 0.1, outputPer1M: 0.5, cachedInputPer1M: 0.01 },
+  "claude-sonnet-5.5": { inputPer1M: 2, outputPer1M: 10, cachedInputPer1M: 0.1 },
+  "claude-opus-5.5": { inputPer1M: 4, outputPer1M: 20, cachedInputPer1M: 0.2 },
+  "claude-opus-5.5-fast": { inputPer1M: 8, outputPer1M: 40, cachedInputPer1M: 0.4 },
   "claude-3-7-sonnet": { inputPer1M: 0.59, outputPer1M: 2.93, cachedInputPer1M: 0.3 },
   "claude-sonnet-5": { inputPer1M: 2, outputPer1M: 10, cachedInputPer1M: 0.3 },
   "claude-3-5-sonnet": { inputPer1M: 3, outputPer1M: 15, cachedInputPer1M: 0.3 },
@@ -578,6 +582,8 @@ var MODEL_PRICING_TABLE = {
   "claude-2.0": { inputPer1M: 8, outputPer1M: 24 },
   "claude-instant-1.2": { inputPer1M: 1.63, outputPer1M: 5.51 },
   // --- Google Gemini (Modern & Thoughts) ---
+  "gemini-3.8-flash": { inputPer1M: 0.75, outputPer1M: 3.75, cachedInputPer1M: 0.075 },
+  "gemini-nano-banana-2.1": { inputPer1M: 1.5, outputPer1M: 7.5, cachedInputPer1M: 0.15 },
   "gemini-3.7-flash": { inputPer1M: 0.75, outputPer1M: 3.75, cachedInputPer1M: 0.18 },
   "gemini-3.1-pro": { inputPer1M: 2, outputPer1M: 12, cachedInputPer1M: 0.5 },
   "gemini-3.5-flash": { inputPer1M: 1.5, outputPer1M: 9, cachedInputPer1M: 0.38 },
@@ -600,6 +606,7 @@ var MODEL_PRICING_TABLE = {
   "grok-2-vision": { inputPer1M: 2, outputPer1M: 10 },
   "grok-beta": { inputPer1M: 5, outputPer1M: 15 },
   // --- Mistral ---
+  "mistral-large-4": { inputPer1M: 0.68, outputPer1M: 2.09, cachedInputPer1M: 0.07 },
   "mistral-large-3": { inputPer1M: 2, outputPer1M: 6 },
   "mistral-large-latest": { inputPer1M: 2, outputPer1M: 6 },
   "mistral-large-2411": { inputPer1M: 2, outputPer1M: 6 },
@@ -681,6 +688,9 @@ var MODEL_ALIASES = {
   "sonnet-3.5": "claude-3-5-sonnet",
   "haiku": "claude-3-5-haiku",
   "haiku-3.5": "claude-3-5-haiku",
+  "haiku-5.5": "claude-haiku-5.5",
+  "sonnet-5.5": "claude-sonnet-5.5",
+  "opus-5.5": "claude-opus-5.5",
   "opus": "claude-3-opus",
   "claude-2": "claude-2.0",
   "claude-instant": "claude-instant-1.2",
@@ -690,6 +700,7 @@ var MODEL_ALIASES = {
   // Gemini Shorthands
   "flash": "gemini-2.0-flash",
   "pro": "gemini-1.5-pro",
+  "gemini-banana": "gemini-nano-banana-2.1",
   // Meta / Groq Shorthands
   "llama-3.3-70b": "llama-3.3-70b-versatile",
   "llama-3.1-70b": "llama-3.1-70b-versatile",
@@ -773,7 +784,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;
@@ -1058,7 +1091,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);
   }
   /**
@@ -1250,7 +1283,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
     });
@@ -1289,6 +1322,7 @@ function withBilling(model, optionsOrCustomer, extraOptions = {}) {
     const mergedMetadata = {
       ...customerMetadata,
       ...options.metadata,
+      ...options.threadId ? { threadId: options.threadId } : {},
       ...extraMeta
     };
     const event = {
@@ -1657,7 +1691,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,
@@ -1749,7 +1783,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;