vibezcheck 0.5.11 → 0.5.13

package/dist/index.mjs

@@ -808,6 +808,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 },
@@ -822,6 +826,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 },
@@ -844,6 +850,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 },
@@ -925,6 +932,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",
@@ -934,6 +944,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",
@@ -1022,7 +1033,29 @@ function registerModelPricing(modelName, rates) {
 }
 
 // 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;
@@ -1307,7 +1340,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);
   }
   /**
@@ -1509,7 +1542,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
     });
@@ -1548,6 +1581,7 @@ function withBilling(model, optionsOrCustomer, extraOptions = {}) {
     const mergedMetadata = {
       ...customerMetadata,
       ...options.metadata,
+      ...options.threadId ? { threadId: options.threadId } : {},
       ...extraMeta
     };
     const event = {
@@ -2160,7 +2194,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,
@@ -2252,7 +2286,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;
@@ -2484,7 +2524,13 @@ function instrumentToolKit(tools, options = {}) {
   const instrumented = {};
   const cost = options.costPerActionUSD ?? 5e-3;
   for (const [toolName, toolDef] of Object.entries(tools)) {
-    const actionCost = options.costs?.[toolName] ?? cost;
+    let actionCost = cost;
+    if (options && "costs" in options && options.costs?.[toolName] !== void 0) {
+      actionCost = options.costs[toolName];
+    } else if (options && toolName in options) {
+      const val = options[toolName];
+      actionCost = typeof val === "number" ? val : val?.costUSD ?? cost;
+    }
     instrumented[toolName] = wrapTool({
       name: toolName,
       costUSD: actionCost,