vibezcheck 0.2.0 → 0.3.0

package/dist/index.d.mts

@@ -1,16 +1,17 @@
-import { M as MeterOptions, S as StreamWrapOptions, C as CustomerParam, U as UsageEvent, a as UsageSummary } from './types-CSrSmsd1.mjs';
-export { I as InferenceCost, b as ModelPricingRates, R as RecordUsageOptions, T as TokenUsage } from './types-CSrSmsd1.mjs';
-import { V as VibezMeter } from './client-MJ3tl7bz.mjs';
-export { c as createMeter } from './client-MJ3tl7bz.mjs';
+export { AnthropicStreamAccumulator, ExtractedUsage, MeterBatcher, UsageDispatchedCallback, detectAndExtractUsage, extractAnthropicResponseUsage, extractGeminiResponseUsage, extractGenericResponseUsage, extractOpenAIResponseUsage, inspectOpenAIStreamChunk, trackTokens, wrapAnthropicStream, wrapGeminiStream, wrapOpenAIStream, wrapStream, wrapUniversalStream } from './meter/index.mjs';
+import { VibezCheckModelOptions, withBilling } from './ai-sdk/index.mjs';
+export { WithBillingOptions, createVibezModel, meteredModel } from './ai-sdk/index.mjs';
+import { calculateCost, getModelPricing, registerModelPricing } from './pricing/index.mjs';
+export { CalculateCostParams, MODEL_PRICING_TABLE, calculateUsageCost, normalizeModelKey } from './pricing/index.mjs';
+import { M as MeterOptions, S as StreamWrapOptions, C as CustomerParam, U as UsageEvent, a as UsageSummary } from './types-DCuzLVyc.mjs';
+export { B as BudgetExceededEvent, b as CircuitBreakerOptions, I as InferenceCost, K as KnownModel, c as ModelPricingRates, R as RecordUsageOptions, d as SupportedAnthropicModel, e as SupportedDeepSeekModel, f as SupportedGoogleModel, g as SupportedModel, h as SupportedOpenAIModel, T as TokenUsage, V as VibezCircuitBreakerError } from './types-DCuzLVyc.mjs';
+import { V as VibezMeter, c as createMeter } from './client-C67DW_tR.mjs';
 import { CustomerManager } from './customers/index.mjs';
 export { CustomerCache, CustomerManagerOptions, GetOrCreateCustomerParams, createCustomerManager } from './customers/index.mjs';
 import { ApiKeyAuth } from './auth/index.mjs';
 export { ApiKeyRecord, CreateApiKeyParams, VerifyAuthResult, createApiKeyAuth, extractAuthToken } from './auth/index.mjs';
 import { BillingHelper } from './billing/index.mjs';
 export { CreateCheckoutSessionParams, CreatePortalSessionParams, CreateTopUpSessionParams, createBillingHelper } from './billing/index.mjs';
-export { AnthropicStreamAccumulator, ExtractedUsage, MeterBatcher, UsageDispatchedCallback, detectAndExtractUsage, extractAnthropicResponseUsage, extractGeminiResponseUsage, extractGenericResponseUsage, extractOpenAIResponseUsage, inspectOpenAIStreamChunk, trackTokens, wrapAnthropicStream, wrapGeminiStream, wrapOpenAIStream, wrapStream, wrapUniversalStream } from './meter/index.mjs';
-export { CalculateCostParams, MODEL_PRICING_TABLE, calculateCost, calculateUsageCost, getModelPricing, normalizeModelKey, registerModelPricing } from './pricing/index.mjs';
-export { WithBillingOptions, meteredModel, withBilling } from './ai-sdk/index.mjs';
 import 'stripe';
 
 /**
@@ -48,17 +49,9 @@ declare class VibezCheckClient {
         customer?: CustomerParam;
     }): T;
     /**
-     * 1-Line Universal Stream Responder for API Routes (Next.js, Express, Hono)
+     * Declarative model resolver for Vercel AI SDK
      */
-    stream(params: {
-        model: string;
-        messages: Array<{
-            role: string;
-            content: string;
-        }>;
-        customer?: CustomerParam;
-        temperature?: number;
-    }): Promise<Response>;
+    model(modelOrId: any, options?: VibezCheckModelOptions): any;
     /**
      * Flush pending meter events (vital for serverless runtimes)
      */
@@ -73,20 +66,50 @@ declare class VibezCheckClient {
  */
 declare function createVibezCheck(config?: VibezCheckConfig): VibezCheckClient;
 /**
- * vibezcheck factory alias
+ * Declarative 1-line factory & model resolver for Vercel AI SDK
+ *
+ * @example
+ * ```typescript
+ * import { generateText, streamText } from 'ai';
+ * import { vibezcheck } from 'vibezcheck';
+ *
+ * // 1. Use string model identifier:
+ * const { text } = await generateText({
+ *   model: vibezcheck('openai/gpt-4o-mini', { customer: 'alex@example.com' }),
+ *   prompt: 'What is love?',
+ * });
+ *
+ * // 2. Use with streamText:
+ * const result = streamText({
+ *   model: vibezcheck('gpt-4o-mini', { customer: 'alex@example.com' }),
+ *   messages,
+ * });
+ *
+ * // 3. Wrap existing model instance:
+ * const result = streamText({
+ *   model: vibezcheck(openai('gpt-4o-mini'), { customer: 'alex@example.com' }),
+ *   messages,
+ * });
+ * ```
  */
-declare const vibezcheck: typeof createVibezCheck;
-/**
- * vibescheck alias (tolerates spelling difference)
- */
-declare const vibescheck: typeof createVibezCheck;
+declare function vibezcheck(modelOrId: any, options?: VibezCheckModelOptions): any;
+declare function vibezcheck(config?: VibezCheckConfig): VibezCheckClient;
+declare namespace vibezcheck {
+    var calculateCost: typeof calculateCost;
+    var getModelPricing: typeof getModelPricing;
+    var registerModelPricing: typeof registerModelPricing;
+    var create: typeof createVibezCheck;
+    var withBilling: typeof withBilling;
+    var createMeter: typeof createMeter;
+}
 /**
- * Singleton instance initialized with process.env
+ * Singleton client instance
  */
 declare const vibez: VibezCheckClient;
 /**
- * Singleton alias
+ * Aliases for developer typing convenience
  */
+declare const vibescheck: typeof vibezcheck;
 declare const vibes: VibezCheckClient;
 
-export { ApiKeyAuth, BillingHelper, CustomerManager, CustomerParam, MeterOptions, StreamWrapOptions, UsageEvent, UsageSummary, VibezCheckClient, type VibezCheckConfig, VibezMeter, createVibezCheck, vibes, vibescheck, vibez, vibezcheck };
+export { ApiKeyAuth, BillingHelper, CustomerManager, CustomerParam, MeterOptions, StreamWrapOptions, UsageEvent, UsageSummary, VibezCheckClient, type VibezCheckConfig, VibezCheckModelOptions, VibezMeter, calculateCost, createMeter, createVibezCheck, getModelPricing, registerModelPricing, vibes, vibescheck, vibez, vibezcheck, withBilling };

package/dist/index.d.ts

@@ -1,16 +1,17 @@
-import { M as MeterOptions, S as StreamWrapOptions, C as CustomerParam, U as UsageEvent, a as UsageSummary } from './types-CSrSmsd1.js';
-export { I as InferenceCost, b as ModelPricingRates, R as RecordUsageOptions, T as TokenUsage } from './types-CSrSmsd1.js';
-import { V as VibezMeter } from './client-txrE0D_D.js';
-export { c as createMeter } from './client-txrE0D_D.js';
+export { AnthropicStreamAccumulator, ExtractedUsage, MeterBatcher, UsageDispatchedCallback, detectAndExtractUsage, extractAnthropicResponseUsage, extractGeminiResponseUsage, extractGenericResponseUsage, extractOpenAIResponseUsage, inspectOpenAIStreamChunk, trackTokens, wrapAnthropicStream, wrapGeminiStream, wrapOpenAIStream, wrapStream, wrapUniversalStream } from './meter/index.js';
+import { VibezCheckModelOptions, withBilling } from './ai-sdk/index.js';
+export { WithBillingOptions, createVibezModel, meteredModel } from './ai-sdk/index.js';
+import { calculateCost, getModelPricing, registerModelPricing } from './pricing/index.js';
+export { CalculateCostParams, MODEL_PRICING_TABLE, calculateUsageCost, normalizeModelKey } from './pricing/index.js';
+import { M as MeterOptions, S as StreamWrapOptions, C as CustomerParam, U as UsageEvent, a as UsageSummary } from './types-DCuzLVyc.js';
+export { B as BudgetExceededEvent, b as CircuitBreakerOptions, I as InferenceCost, K as KnownModel, c as ModelPricingRates, R as RecordUsageOptions, d as SupportedAnthropicModel, e as SupportedDeepSeekModel, f as SupportedGoogleModel, g as SupportedModel, h as SupportedOpenAIModel, T as TokenUsage, V as VibezCircuitBreakerError } from './types-DCuzLVyc.js';
+import { V as VibezMeter, c as createMeter } from './client-S3qmTXTR.js';
 import { CustomerManager } from './customers/index.js';
 export { CustomerCache, CustomerManagerOptions, GetOrCreateCustomerParams, createCustomerManager } from './customers/index.js';
 import { ApiKeyAuth } from './auth/index.js';
 export { ApiKeyRecord, CreateApiKeyParams, VerifyAuthResult, createApiKeyAuth, extractAuthToken } from './auth/index.js';
 import { BillingHelper } from './billing/index.js';
 export { CreateCheckoutSessionParams, CreatePortalSessionParams, CreateTopUpSessionParams, createBillingHelper } from './billing/index.js';
-export { AnthropicStreamAccumulator, ExtractedUsage, MeterBatcher, UsageDispatchedCallback, detectAndExtractUsage, extractAnthropicResponseUsage, extractGeminiResponseUsage, extractGenericResponseUsage, extractOpenAIResponseUsage, inspectOpenAIStreamChunk, trackTokens, wrapAnthropicStream, wrapGeminiStream, wrapOpenAIStream, wrapStream, wrapUniversalStream } from './meter/index.js';
-export { CalculateCostParams, MODEL_PRICING_TABLE, calculateCost, calculateUsageCost, getModelPricing, normalizeModelKey, registerModelPricing } from './pricing/index.js';
-export { WithBillingOptions, meteredModel, withBilling } from './ai-sdk/index.js';
 import 'stripe';
 
 /**
@@ -48,17 +49,9 @@ declare class VibezCheckClient {
         customer?: CustomerParam;
     }): T;
     /**
-     * 1-Line Universal Stream Responder for API Routes (Next.js, Express, Hono)
+     * Declarative model resolver for Vercel AI SDK
      */
-    stream(params: {
-        model: string;
-        messages: Array<{
-            role: string;
-            content: string;
-        }>;
-        customer?: CustomerParam;
-        temperature?: number;
-    }): Promise<Response>;
+    model(modelOrId: any, options?: VibezCheckModelOptions): any;
     /**
      * Flush pending meter events (vital for serverless runtimes)
      */
@@ -73,20 +66,50 @@ declare class VibezCheckClient {
  */
 declare function createVibezCheck(config?: VibezCheckConfig): VibezCheckClient;
 /**
- * vibezcheck factory alias
+ * Declarative 1-line factory & model resolver for Vercel AI SDK
+ *
+ * @example
+ * ```typescript
+ * import { generateText, streamText } from 'ai';
+ * import { vibezcheck } from 'vibezcheck';
+ *
+ * // 1. Use string model identifier:
+ * const { text } = await generateText({
+ *   model: vibezcheck('openai/gpt-4o-mini', { customer: 'alex@example.com' }),
+ *   prompt: 'What is love?',
+ * });
+ *
+ * // 2. Use with streamText:
+ * const result = streamText({
+ *   model: vibezcheck('gpt-4o-mini', { customer: 'alex@example.com' }),
+ *   messages,
+ * });
+ *
+ * // 3. Wrap existing model instance:
+ * const result = streamText({
+ *   model: vibezcheck(openai('gpt-4o-mini'), { customer: 'alex@example.com' }),
+ *   messages,
+ * });
+ * ```
  */
-declare const vibezcheck: typeof createVibezCheck;
-/**
- * vibescheck alias (tolerates spelling difference)
- */
-declare const vibescheck: typeof createVibezCheck;
+declare function vibezcheck(modelOrId: any, options?: VibezCheckModelOptions): any;
+declare function vibezcheck(config?: VibezCheckConfig): VibezCheckClient;
+declare namespace vibezcheck {
+    var calculateCost: typeof calculateCost;
+    var getModelPricing: typeof getModelPricing;
+    var registerModelPricing: typeof registerModelPricing;
+    var create: typeof createVibezCheck;
+    var withBilling: typeof withBilling;
+    var createMeter: typeof createMeter;
+}
 /**
- * Singleton instance initialized with process.env
+ * Singleton client instance
  */
 declare const vibez: VibezCheckClient;
 /**
- * Singleton alias
+ * Aliases for developer typing convenience
  */
+declare const vibescheck: typeof vibezcheck;
 declare const vibes: VibezCheckClient;
 
-export { ApiKeyAuth, BillingHelper, CustomerManager, CustomerParam, MeterOptions, StreamWrapOptions, UsageEvent, UsageSummary, VibezCheckClient, type VibezCheckConfig, VibezMeter, createVibezCheck, vibes, vibescheck, vibez, vibezcheck };
+export { ApiKeyAuth, BillingHelper, CustomerManager, CustomerParam, MeterOptions, StreamWrapOptions, UsageEvent, UsageSummary, VibezCheckClient, type VibezCheckConfig, VibezCheckModelOptions, VibezMeter, calculateCost, createMeter, createVibezCheck, getModelPricing, registerModelPricing, vibes, vibescheck, vibez, vibezcheck, withBilling };

package/dist/index.js

@@ -38,6 +38,7 @@ __export(src_exports, {
   MODEL_PRICING_TABLE: () => MODEL_PRICING_TABLE,
   MeterBatcher: () => MeterBatcher,
   VibezCheckClient: () => VibezCheckClient,
+  VibezCircuitBreakerError: () => VibezCircuitBreakerError,
   VibezMeter: () => VibezMeter,
   calculateCost: () => calculateCost,
   calculateUsageCost: () => calculateUsageCost,
@@ -46,6 +47,7 @@ __export(src_exports, {
   createCustomerManager: () => createCustomerManager,
   createMeter: () => createMeter,
   createVibezCheck: () => createVibezCheck,
+  createVibezModel: () => createVibezModel,
   detectAndExtractUsage: () => detectAndExtractUsage,
   extractAnthropicResponseUsage: () => extractAnthropicResponseUsage,
   extractAuthToken: () => extractAuthToken,
@@ -893,6 +895,16 @@ function createMeter(options = {}) {
   return new VibezMeter(options);
 }
 
+// src/types.ts
+var VibezCircuitBreakerError = class extends Error {
+  event;
+  constructor(event) {
+    super(event.message);
+    this.name = "VibezCircuitBreakerError";
+    this.event = event;
+  }
+};
+
 // src/ai-sdk/with-billing.ts
 function withBilling(model, options = {}) {
   if (!model || typeof model !== "object") {
@@ -920,6 +932,38 @@ function withBilling(model, options = {}) {
       cachedTokens: cachedTokens > 0 ? cachedTokens : void 0
     };
     const cost = calculateUsageCost(modelId, usage);
+    if (options.maxCostPerCallUSD && cost.totalUSD > options.maxCostPerCallUSD) {
+      const budgetEvent = {
+        reason: "cost_per_call_exceeded",
+        limit: options.maxCostPerCallUSD,
+        current: cost.totalUSD,
+        model: modelId,
+        customerId,
+        message: `[vibezcheck] Circuit Breaker tripped: Call cost ($${cost.totalUSD.toFixed(4)}) exceeded maxCostPerCallUSD threshold ($${options.maxCostPerCallUSD.toFixed(4)}).`
+      };
+      if (options.onBudgetExceeded) {
+        options.onBudgetExceeded(budgetEvent);
+      }
+      if (options.throwOnBudgetExceeded) {
+        throw new VibezCircuitBreakerError(budgetEvent);
+      }
+    }
+    if (options.maxTokensPerCall && usage.totalTokens > options.maxTokensPerCall) {
+      const budgetEvent = {
+        reason: "max_tokens_exceeded",
+        limit: options.maxTokensPerCall,
+        current: usage.totalTokens,
+        model: modelId,
+        customerId,
+        message: `[vibezcheck] Circuit Breaker tripped: Token count (${usage.totalTokens}) exceeded maxTokensPerCall limit (${options.maxTokensPerCall}).`
+      };
+      if (options.onBudgetExceeded) {
+        options.onBudgetExceeded(budgetEvent);
+      }
+      if (options.throwOnBudgetExceeded) {
+        throw new VibezCircuitBreakerError(budgetEvent);
+      }
+    }
     const event = {
       timestamp: (/* @__PURE__ */ new Date()).toISOString(),
       model: modelId,
@@ -943,18 +987,9 @@ function withBilling(model, options = {}) {
       options.onUsage(event);
     }
   };
-  return new Proxy(model, {
+  const handler = {
     get(target, prop, receiver) {
       const originalValue = Reflect.get(target, prop, receiver);
-      if (prop === "doGenerate" && typeof originalValue === "function") {
-        return async function(...args) {
-          const result = await originalValue.apply(target, args);
-          if (result && result.usage) {
-            handleUsage(result.usage);
-          }
-          return result;
-        };
-      }
       if (prop === "doStream" && typeof originalValue === "function") {
         return async function(...args) {
           const result = await originalValue.apply(target, args);
@@ -962,61 +997,86 @@ function withBilling(model, options = {}) {
             return result;
           }
           const originalStream = result.stream;
-          if (typeof originalStream.getReader === "function") {
-            const reader = originalStream.getReader();
-            const transformedStream = new ReadableStream({
-              async start(controller) {
-                try {
-                  while (true) {
-                    const { done, value } = await reader.read();
-                    if (done) {
-                      controller.close();
-                      break;
-                    }
-                    if (value && typeof value === "object") {
-                      if (value.type === "finish" && value.usage) {
-                        handleUsage(value.usage);
-                      }
-                    }
-                    controller.enqueue(value);
-                  }
-                } catch (err) {
-                  controller.error(err);
-                }
-              }
-            });
-            return {
-              ...result,
-              stream: transformedStream
-            };
-          }
-          if (Symbol.asyncIterator in originalStream) {
-            const wrappedAsyncIterable = {
-              async *[Symbol.asyncIterator]() {
-                for await (const chunk of originalStream) {
-                  if (chunk && typeof chunk === "object") {
-                    if (chunk.type === "finish" && chunk.usage) {
-                      handleUsage(chunk.usage);
-                    }
-                  }
-                  yield chunk;
-                }
+          const transformStream = new TransformStream({
+            transform(chunk, controller) {
+              controller.enqueue(chunk);
+              if (chunk.type === "finish" && chunk.usage) {
+                handleUsage(chunk.usage);
               }
-            };
-            return {
-              ...result,
-              stream: wrappedAsyncIterable
-            };
+            }
+          });
+          return {
+            ...result,
+            stream: originalStream.pipeThrough(transformStream)
+          };
+        };
+      }
+      if (prop === "doGenerate" && typeof originalValue === "function") {
+        return async function(...args) {
+          const result = await originalValue.apply(target, args);
+          if (result && result.usage) {
+            handleUsage(result.usage);
           }
           return result;
         };
       }
       return originalValue;
     }
-  });
+  };
+  return new Proxy(model, handler);
 }
 var meteredModel = withBilling;
 
+// 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("/");
+    }
+    const apiKey = options.apiKey || process.env.AI_GATEWAY_API_KEY || process.env.OPENAI_API_KEY || process.env.ANTHROPIC_API_KEY || process.env.GOOGLE_GENERATIVE_AI_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 { createOpenAI } = require("@ai-sdk/openai");
+        const openaiProvider = createOpenAI({ apiKey, baseURL });
+        baseModelInstance = openaiProvider(cleanModelId);
+      } else if (providerName === "anthropic") {
+        const { createAnthropic } = require("@ai-sdk/anthropic");
+        const anthropicProvider = createAnthropic({ apiKey, baseURL });
+        baseModelInstance = anthropicProvider(cleanModelId);
+      } else if (providerName === "google") {
+        const { createGoogleGenerativeAI } = require("@ai-sdk/google");
+        const googleProvider = createGoogleGenerativeAI({ apiKey, baseURL });
+        baseModelInstance = googleProvider(cleanModelId);
+      }
+    } catch {
+      baseModelInstance = {
+        specificationVersion: "v2",
+        provider: providerName,
+        modelId: cleanModelId
+      };
+    }
+    if (!baseModelInstance) {
+      baseModelInstance = {
+        specificationVersion: "v2",
+        provider: providerName,
+        modelId: cleanModelId
+      };
+    }
+    return withBilling(baseModelInstance, options);
+  }
+  return modelOrId;
+}
+
 // src/customers/manager.ts
 var import_stripe2 = __toESM(require("stripe"));
 
@@ -1355,43 +1415,12 @@ var VibezCheckClient = class {
     });
   }
   /**
-   * 1-Line Universal Stream Responder for API Routes (Next.js, Express, Hono)
+   * Declarative model resolver for Vercel AI SDK
    */
-  async stream(params) {
-    const OpenAI = (await import("openai")).default;
-    const openai = new OpenAI();
-    const responseStream = await openai.chat.completions.create({
-      model: params.model,
-      messages: params.messages,
-      stream: true,
-      stream_options: { include_usage: true },
-      temperature: params.temperature
-    });
-    const meteredStream = this.wrapStream(responseStream, {
-      customer: params.customer,
-      model: params.model
-    });
-    const encoder = new TextEncoder();
-    const readable = new ReadableStream({
-      async start(controller) {
-        try {
-          for await (const chunk of meteredStream) {
-            const text = chunk.choices?.[0]?.delta?.content || "";
-            if (text) {
-              controller.enqueue(encoder.encode(text));
-            }
-          }
-          controller.close();
-        } catch (err) {
-          controller.error(err);
-        }
-      }
-    });
-    return new Response(readable, {
-      headers: {
-        "Content-Type": "text/plain; charset=utf-8",
-        "Transfer-Encoding": "chunked"
-      }
+  model(modelOrId, options) {
+    return createVibezModel(modelOrId, {
+      ...options,
+      meter: this.meter
     });
   }
   /**
@@ -1410,9 +1439,20 @@ var VibezCheckClient = class {
 function createVibezCheck(config = {}) {
   return new VibezCheckClient(config);
 }
-var vibezcheck = createVibezCheck;
-var vibescheck = createVibezCheck;
-var vibez = createVibezCheck();
+function vibezcheck(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;
+var vibez = new VibezCheckClient();
+var vibescheck = vibezcheck;
 var vibes = vibez;
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
@@ -1424,6 +1464,7 @@ var vibes = vibez;
   MODEL_PRICING_TABLE,
   MeterBatcher,
   VibezCheckClient,
+  VibezCircuitBreakerError,
   VibezMeter,
   calculateCost,
   calculateUsageCost,
@@ -1432,6 +1473,7 @@ var vibes = vibez;
   createCustomerManager,
   createMeter,
   createVibezCheck,
+  createVibezModel,
   detectAndExtractUsage,
   extractAnthropicResponseUsage,
   extractAuthToken,