vibezcheck 0.5.11 → 0.5.13

package/dist/billing/index.d.mts

@@ -1,8 +1,8 @@
-import { A as AgentSession } from '../session-D9fdP255.mjs';
-export { a as AgentSessionOptions, T as ToolTrackingOptions, c as createAgentSession } from '../session-D9fdP255.mjs';
-import { C as CustomerParam } from '../types-DHlsaLvI.mjs';
-import '../with-billing-DhHGWCAp.mjs';
-import '../client-ttJwYM7x.mjs';
+import { A as AgentSession } from '../session-CueUIV77.mjs';
+export { a as AgentSessionOptions, T as ToolTrackingOptions, c as createAgentSession } from '../session-CueUIV77.mjs';
+import { C as CustomerParam } from '../types-CqXTmCNv.mjs';
+import '../with-billing-Df6izlJ4.mjs';
+import '../client-xP86Qg_b.mjs';
 
 interface ToolOptions {
     name?: string;
@@ -33,7 +33,9 @@ declare function instrumentToolKit<T extends Record<string, any>>(tools: T, opti
     costPerActionUSD?: number;
     session?: AgentSession;
     costs?: Record<string, number>;
-}): {
+} | Record<string, {
+    costUSD: number;
+} | number | any>): {
     [K in keyof T]: T[K] & {
         costUSD: number;
     };

package/dist/billing/index.d.ts

@@ -1,8 +1,8 @@
-import { A as AgentSession } from '../session-CUnv_muf.js';
-export { a as AgentSessionOptions, T as ToolTrackingOptions, c as createAgentSession } from '../session-CUnv_muf.js';
-import { C as CustomerParam } from '../types-DHlsaLvI.js';
-import '../with-billing-CVGxl3ot.js';
-import '../client-eXSwD_kI.js';
+import { A as AgentSession } from '../session-YF67Yij2.js';
+export { a as AgentSessionOptions, T as ToolTrackingOptions, c as createAgentSession } from '../session-YF67Yij2.js';
+import { C as CustomerParam } from '../types-CqXTmCNv.js';
+import '../with-billing-DgQ380Zl.js';
+import '../client-0CuOBECi.js';
 
 interface ToolOptions {
     name?: string;
@@ -33,7 +33,9 @@ declare function instrumentToolKit<T extends Record<string, any>>(tools: T, opti
     costPerActionUSD?: number;
     session?: AgentSession;
     costs?: Record<string, number>;
-}): {
+} | Record<string, {
+    costUSD: number;
+} | number | any>): {
     [K in keyof T]: T[K] & {
         costUSD: number;
     };

package/dist/billing/index.js

@@ -600,6 +600,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 },
@@ -614,6 +618,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 },
@@ -636,6 +642,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 },
@@ -717,6 +724,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",
@@ -726,6 +736,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",
@@ -809,7 +820,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;
@@ -1094,7 +1127,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);
   }
   /**
@@ -1286,7 +1319,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
     });
@@ -1325,6 +1358,7 @@ function withBilling(model, optionsOrCustomer, extraOptions = {}) {
     const mergedMetadata = {
       ...customerMetadata,
       ...options.metadata,
+      ...options.threadId ? { threadId: options.threadId } : {},
       ...extraMeta
     };
     const event = {
@@ -1536,6 +1570,266 @@ function withBilling(model, optionsOrCustomer, extraOptions = {}) {
   return new Proxy(model, handler);
 }
 
+// src/ai-sdk/to-response.ts
+function toResponse(result, options) {
+  let baseResponse;
+  if (typeof result?.toUIMessageStreamResponse === "function") {
+    baseResponse = result.toUIMessageStreamResponse(options?.init);
+  } else if (typeof result?.toDataStreamResponse === "function") {
+    baseResponse = result.toDataStreamResponse(options?.init);
+  } else if (typeof result?.toTextStreamResponse === "function") {
+    baseResponse = result.toTextStreamResponse(options?.init);
+  } else if (result instanceof Response) {
+    baseResponse = result;
+  } else if (result && typeof result === "object" && result.body && result.headers) {
+    baseResponse = result;
+  } else {
+    baseResponse = new Response(result, options?.init);
+  }
+  if (!baseResponse.body) {
+    return baseResponse;
+  }
+  const contentType = baseResponse.headers.get("content-type") || "";
+  const isSSE = contentType.includes("text/event-stream") || baseResponse.headers.has("x-vercel-ai-ui-message-stream");
+  const isDataStream = baseResponse.headers.has("x-vercel-ai-data-stream");
+  const encoder = new TextEncoder();
+  const decoder = new TextDecoder();
+  let injected = false;
+  const resolveTelemetry = async () => {
+    if (options?.telemetry) return options.telemetry;
+    if (result?.providerMetadata) {
+      try {
+        const meta = await Promise.race([
+          result.providerMetadata,
+          new Promise((_, reject) => setTimeout(() => reject(new Error("timeout")), 800))
+        ]);
+        if (meta?.vibezcheck) {
+          return meta.vibezcheck;
+        }
+      } catch {
+      }
+    }
+    if (result?.usage) {
+      try {
+        const u = await Promise.race([
+          result.usage,
+          new Promise((_, reject) => setTimeout(() => reject(new Error("timeout")), 800))
+        ]);
+        if (u) {
+          const inputTokens = u.promptTokens ?? u.inputTokens ?? 0;
+          const outputTokens = u.completionTokens ?? u.outputTokens ?? 0;
+          const reasoningTokens = u.reasoningTokens ?? u.completionTokensDetails?.reasoningTokens;
+          return {
+            type: "vibezcheck",
+            usage: {
+              inputTokens,
+              outputTokens,
+              totalTokens: inputTokens + outputTokens,
+              reasoningTokens
+            }
+          };
+        }
+      } catch {
+      }
+    }
+    return null;
+  };
+  const transformStream = new TransformStream({
+    async transform(chunk, controller) {
+      const text = decoder.decode(chunk, { stream: true });
+      if (isSSE && text.includes("[DONE]") && !injected) {
+        injected = true;
+        const telemetry = await resolveTelemetry();
+        if (telemetry) {
+          const ssePayload = `data: ${JSON.stringify({ type: "data-vibezcheck", data: telemetry })}
+
+data: ${JSON.stringify({ type: "message-metadata", messageMetadata: { vibezcheck: telemetry } })}
+
+`;
+          const doneIndex = text.indexOf("data: [DONE]");
+          if (doneIndex !== -1) {
+            const before = text.slice(0, doneIndex);
+            const after = text.slice(doneIndex);
+            if (before) controller.enqueue(encoder.encode(before));
+            controller.enqueue(encoder.encode(ssePayload));
+            controller.enqueue(encoder.encode(after));
+            return;
+          } else {
+            controller.enqueue(encoder.encode(ssePayload));
+          }
+        }
+      } else if (isDataStream && (text.includes("d:{") || text.includes("d: {")) && !injected) {
+        injected = true;
+        const telemetry = await resolveTelemetry();
+        if (telemetry) {
+          const annotationLine = `2:[${JSON.stringify(telemetry)}]
+`;
+          controller.enqueue(encoder.encode(annotationLine));
+        }
+      }
+      controller.enqueue(chunk);
+    },
+    async flush(controller) {
+      if (!injected) {
+        injected = true;
+        const telemetry = await resolveTelemetry();
+        if (telemetry) {
+          if (isSSE) {
+            const ssePayload = `data: ${JSON.stringify({ type: "data-vibezcheck", data: telemetry })}
+
+data: ${JSON.stringify({ type: "message-metadata", messageMetadata: { vibezcheck: telemetry } })}
+
+`;
+            controller.enqueue(encoder.encode(ssePayload));
+          } else {
+            const annotationLine = `2:[${JSON.stringify(telemetry)}]
+`;
+            controller.enqueue(encoder.encode(annotationLine));
+          }
+        }
+      }
+    }
+  });
+  const mergedHeaders = new Headers(baseResponse.headers);
+  if (options?.headers) {
+    const extraHeaders = new Headers(options.headers);
+    extraHeaders.forEach((value, key) => {
+      mergedHeaders.set(key, value);
+    });
+  }
+  const status = options?.init?.status ?? baseResponse.status;
+  const statusText = options?.init?.statusText ?? baseResponse.statusText;
+  return new Response(baseResponse.body.pipeThrough(transformStream), {
+    status,
+    statusText,
+    headers: mergedHeaders
+  });
+}
+
+// src/ai-sdk/declarative.ts
+function createVibezModel(modelOrId, options = {}) {
+  if (typeof modelOrId === "object" && modelOrId !== null) {
+    return withBilling(modelOrId, options);
+  }
+  if (typeof modelOrId === "string") {
+    const rawId = modelOrId;
+    let providerName = "openai";
+    let cleanModelId = rawId;
+    if (rawId.includes("/")) {
+      const parts = rawId.split("/");
+      providerName = parts[0].toLowerCase();
+      cleanModelId = parts.slice(1).join("/");
+    } else if (rawId.startsWith("claude-")) {
+      providerName = "anthropic";
+    } else if (rawId.startsWith("gemini-")) {
+      providerName = "google";
+    } else if (rawId.startsWith("deepseek-")) {
+      providerName = "deepseek";
+    } else if (rawId.startsWith("grok-")) {
+      providerName = "xai";
+    } else if (rawId.startsWith("mistral-") || rawId.startsWith("codestral-") || rawId.startsWith("ministral-") || rawId.startsWith("magistral-") || rawId.startsWith("voxtral-") || rawId.startsWith("devstral-") || rawId.startsWith("open-mistral") || rawId.startsWith("open-mixtral")) {
+      providerName = "mistral";
+    } else if (rawId.startsWith("llama-") || rawId.startsWith("mixtral-") || rawId.startsWith("gemma") || rawId.startsWith("qwen-")) {
+      providerName = "groq";
+    } else if (rawId.startsWith("command-")) {
+      providerName = "cohere";
+    } else if (rawId.startsWith("mercury-") || rawId.startsWith("ling-")) {
+      providerName = "gateway";
+    } else if (rawId.startsWith("gpt-image-") || rawId.startsWith("gpt-6-")) {
+      providerName = "openai";
+    }
+    const apiKey = options.apiKey || process.env.AI_GATEWAY_API_KEY || (providerName === "anthropic" ? process.env.ANTHROPIC_API_KEY : void 0) || (providerName === "google" ? process.env.GOOGLE_GENERATIVE_AI_API_KEY : void 0) || (providerName === "mistral" ? process.env.MISTRAL_API_KEY : void 0) || (providerName === "groq" ? process.env.GROQ_API_KEY : void 0) || (providerName === "deepseek" ? process.env.DEEPSEEK_API_KEY : void 0) || (providerName === "xai" ? process.env.XAI_API_KEY : void 0) || (providerName === "cohere" ? process.env.COHERE_API_KEY : void 0) || process.env.OPENAI_API_KEY;
+    const baseURL = options.baseURL || process.env.AI_GATEWAY_BASE_URL || (providerName === "openai" ? "https://api.openai.com/v1" : void 0);
+    let baseModelInstance = null;
+    try {
+      if (providerName === "openai" || providerName === "gateway" || !providerName) {
+        const mod = require("@ai-sdk/openai");
+        const factory = mod.createOpenAI || mod.openai || mod.default?.createOpenAI || mod.default?.openai;
+        if (typeof factory === "function") {
+          if (mod.createOpenAI) {
+            const provider = factory({ apiKey, baseURL });
+            baseModelInstance = provider(cleanModelId);
+          } else {
+            baseModelInstance = factory(cleanModelId);
+          }
+        }
+      } else if (providerName === "anthropic") {
+        const mod = require("@ai-sdk/anthropic");
+        const factory = mod.createAnthropic || mod.anthropic || mod.default?.createAnthropic || mod.default?.anthropic;
+        if (typeof factory === "function") {
+          if (mod.createAnthropic) {
+            const provider = factory({ apiKey, baseURL });
+            baseModelInstance = provider(cleanModelId);
+          } else {
+            baseModelInstance = factory(cleanModelId);
+          }
+        }
+      } else if (providerName === "google") {
+        const mod = require("@ai-sdk/google");
+        const factory = mod.createGoogleGenerativeAI || mod.google || mod.default?.createGoogleGenerativeAI || mod.default?.google;
+        if (typeof factory === "function") {
+          if (mod.createGoogleGenerativeAI) {
+            const provider = factory({ apiKey, baseURL });
+            baseModelInstance = provider(cleanModelId);
+          } else {
+            baseModelInstance = factory(cleanModelId);
+          }
+        }
+      } else if (providerName === "mistral") {
+        const mod = require("@ai-sdk/mistral");
+        const factory = mod.createMistral || mod.mistral || mod.default?.createMistral || mod.default?.mistral;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createMistral ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "groq") {
+        const mod = require("@ai-sdk/groq");
+        const factory = mod.createGroq || mod.groq || mod.default?.createGroq || mod.default?.groq;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createGroq ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "deepseek") {
+        const mod = require("@ai-sdk/deepseek");
+        const factory = mod.createDeepSeek || mod.deepseek || mod.default?.createDeepSeek || mod.default?.deepseek;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createDeepSeek ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "xai") {
+        const mod = require("@ai-sdk/xai");
+        const factory = mod.createXai || mod.xai || mod.default?.createXai || mod.default?.xai;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createXai ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      } else if (providerName === "cohere") {
+        const mod = require("@ai-sdk/cohere");
+        const factory = mod.createCohere || mod.cohere || mod.default?.createCohere || mod.default?.cohere;
+        if (typeof factory === "function") {
+          baseModelInstance = mod.createCohere ? factory({ apiKey, baseURL })(cleanModelId) : factory(cleanModelId);
+        }
+      }
+    } catch {
+      baseModelInstance = {
+        specificationVersion: "v2",
+        provider: providerName,
+        modelId: cleanModelId
+      };
+    }
+    if (!baseModelInstance) {
+      baseModelInstance = {
+        specificationVersion: "v2",
+        provider: providerName,
+        modelId: cleanModelId
+      };
+    }
+    return withBilling(baseModelInstance, options);
+  }
+  return modelOrId;
+}
+var createVibezSession = createAgentSession;
+var vibezcheck = Object.assign(createVibezModel, {
+  session: createVibezSession,
+  toResponse
+});
+
 // src/billing/session.ts
 var AgentSession = class {
   sessionId;
@@ -1556,7 +1850,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,
@@ -1648,7 +1942,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;
@@ -1739,7 +2039,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,