tokenmeter 0.9.0 → 0.9.1

package/dist/processor/TokenMeterProcessor.d.ts

@@ -1,8 +1,31 @@
 /**
  * TokenMeter Span Processor
  *
- * An OpenTelemetry SpanProcessor that calculates costs based on usage data
- * and adds cost attributes to spans before they are exported.
+ * An OpenTelemetry SpanProcessor that calculates and logs costs for spans
+ * with usage data. Useful for debugging and observability of AI costs.
+ *
+ * ## Important Limitation
+ *
+ * This processor CANNOT add cost attributes to spans after they end.
+ * OpenTelemetry's `ReadableSpan` interface doesn't allow attribute modification
+ * after `span.end()` is called. The processor can only:
+ * - Log calculated costs for debugging
+ * - Validate pricing configuration
+ * - Use with `pricingOverrides` for cost estimation
+ *
+ * ## When to Use
+ *
+ * 1. **Debugging**: Verify cost calculations during development
+ * 2. **Monitoring non-monitor() spans**: Log costs for spans from external
+ *    sources (e.g., Vercel AI SDK's experimental_telemetry) that already
+ *    include usage data
+ *
+ * ## For Production Cost Tracking
+ *
+ * Use `monitor()` to wrap your AI clients instead - it calculates and adds
+ * `tokenmeter.cost_usd` to spans BEFORE they end, which is the proper approach.
+ *
+ * @see monitor() for production cost tracking
  */
 import type { Context } from "@opentelemetry/api";
 import type { SpanProcessor, ReadableSpan, Span } from "@opentelemetry/sdk-trace-base";
@@ -10,13 +33,18 @@ import type { TokenMeterProcessorConfig } from "../types.js";
 /**
  * TokenMeter SpanProcessor
  *
- * Intercepts spans on end and calculates costs based on usage attributes.
- * The calculated cost is added as an attribute before the span is exported.
+ * Calculates costs for spans with usage data and logs them.
+ *
+ * **Note**: This processor cannot add cost attributes to spans after they end.
+ * For production cost tracking, use `monitor()` instead.
  *
  * @example
  * ```typescript
  * import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
- * import { TokenMeterProcessor } from 'tokenmeter';
+ * import { TokenMeterProcessor, configureLogger } from 'tokenmeter';
+ *
+ * // Enable logging to see calculated costs
+ * configureLogger({ level: 'debug' });
  *
  * const provider = new NodeTracerProvider();
  * provider.addSpanProcessor(new TokenMeterProcessor());
@@ -33,9 +61,11 @@ export declare class TokenMeterProcessor implements SpanProcessor {
     /**
      * Called when a span starts (no-op for TokenMeter)
      */
-    onStart(span: Span, parentContext: Context): void;
+    onStart(_span: Span, _parentContext: Context): void;
     /**
-     * Called when a span ends - this is where we calculate and add cost
+     * Called when a span ends - calculates and logs cost.
+     *
+     * Note: Cannot add cost attribute to span after end() - use monitor() for that.
      */
     onEnd(span: ReadableSpan): void;
     /**

package/dist/processor/TokenMeterProcessor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TokenMeterProcessor.d.ts","sourceRoot":"","sources":["../../src/processor/TokenMeterProcessor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,EACV,aAAa,EACb,YAAY,EACZ,IAAI,EACL,MAAM,+BAA+B,CAAC;AAOvC,OAAO,KAAK,EAAE,yBAAyB,EAAmB,MAAM,aAAa,CAAC;AAG9E;;;;;;;;;;;;;;;GAeG;AACH,qBAAa,mBAAoB,YAAW,aAAa;IACvD,OAAO,CAAC,QAAQ,CAAgC;IAChD,OAAO,CAAC,eAAe,CAA8B;IACrD,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,gBAAgB,CAA+B;gBAE3C,MAAM,GAAE,yBAA8B;YAQpC,iBAAiB;IAU/B;;OAEG;IACH,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,GAAG,IAAI;IAKjD;;OAEG;IACH,KAAK,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI;IAmD/B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAO/B;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAGlC;AAED,eAAe,mBAAmB,CAAC"}
+{"version":3,"file":"TokenMeterProcessor.d.ts","sourceRoot":"","sources":["../../src/processor/TokenMeterProcessor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,KAAK,EACV,aAAa,EACb,YAAY,EACZ,IAAI,EACL,MAAM,+BAA+B,CAAC;AAOvC,OAAO,KAAK,EAAE,yBAAyB,EAAmB,MAAM,aAAa,CAAC;AAI9E;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,qBAAa,mBAAoB,YAAW,aAAa;IACvD,OAAO,CAAC,QAAQ,CAAgC;IAChD,OAAO,CAAC,eAAe,CAA8B;IACrD,OAAO,CAAC,MAAM,CAA4B;IAC1C,OAAO,CAAC,gBAAgB,CAA+B;gBAE3C,MAAM,GAAE,yBAA8B;YAQpC,iBAAiB;IAU/B;;OAEG;IACH,OAAO,CAAC,KAAK,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,GAAG,IAAI;IAInD;;;;OAIG;IACH,KAAK,CAAC,IAAI,EAAE,YAAY,GAAG,IAAI;IAuC/B;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA0BzB;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAO/B;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;CAGlC;AAED,eAAe,mBAAmB,CAAC"}

package/dist/processor/TokenMeterProcessor.js

@@ -1,21 +1,50 @@
 /**
  * TokenMeter Span Processor
  *
- * An OpenTelemetry SpanProcessor that calculates costs based on usage data
- * and adds cost attributes to spans before they are exported.
+ * An OpenTelemetry SpanProcessor that calculates and logs costs for spans
+ * with usage data. Useful for debugging and observability of AI costs.
+ *
+ * ## Important Limitation
+ *
+ * This processor CANNOT add cost attributes to spans after they end.
+ * OpenTelemetry's `ReadableSpan` interface doesn't allow attribute modification
+ * after `span.end()` is called. The processor can only:
+ * - Log calculated costs for debugging
+ * - Validate pricing configuration
+ * - Use with `pricingOverrides` for cost estimation
+ *
+ * ## When to Use
+ *
+ * 1. **Debugging**: Verify cost calculations during development
+ * 2. **Monitoring non-monitor() spans**: Log costs for spans from external
+ *    sources (e.g., Vercel AI SDK's experimental_telemetry) that already
+ *    include usage data
+ *
+ * ## For Production Cost Tracking
+ *
+ * Use `monitor()` to wrap your AI clients instead - it calculates and adds
+ * `tokenmeter.cost_usd` to spans BEFORE they end, which is the proper approach.
+ *
+ * @see monitor() for production cost tracking
  */
 import { loadManifest, getModelPricing, calculateCost, getCachedManifest, } from "../pricing/manifest.js";
 import { TM_ATTRIBUTES, GEN_AI_ATTRIBUTES } from "../types.js";
+import { logger } from "../logger.js";
 /**
  * TokenMeter SpanProcessor
  *
- * Intercepts spans on end and calculates costs based on usage attributes.
- * The calculated cost is added as an attribute before the span is exported.
+ * Calculates costs for spans with usage data and logs them.
+ *
+ * **Note**: This processor cannot add cost attributes to spans after they end.
+ * For production cost tracking, use `monitor()` instead.
  *
  * @example
  * ```typescript
  * import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
- * import { TokenMeterProcessor } from 'tokenmeter';
+ * import { TokenMeterProcessor, configureLogger } from 'tokenmeter';
+ *
+ * // Enable logging to see calculated costs
+ * configureLogger({ level: 'debug' });
  *
  * const provider = new NodeTracerProvider();
  * provider.addSpanProcessor(new TokenMeterProcessor());
@@ -40,21 +69,21 @@ export class TokenMeterProcessor {
             });
         }
         catch (error) {
-            console.error("[tokenmeter] Failed to load pricing manifest:", error);
+            logger.error("Failed to load pricing manifest:", error);
         }
     }
     /**
      * Called when a span starts (no-op for TokenMeter)
      */
-    onStart(span, parentContext) {
-        // We don't need to do anything when spans start
-        // Cost calculation happens on end
+    onStart(_span, _parentContext) {
+        // No-op: Cost calculation happens in onEnd
     }
     /**
-     * Called when a span ends - this is where we calculate and add cost
+     * Called when a span ends - calculates and logs cost.
+     *
+     * Note: Cannot add cost attribute to span after end() - use monitor() for that.
      */
     onEnd(span) {
-        // Get usage attributes from span
         const attrs = span.attributes;
         // Check if this span has usage data
         const inputUnits = attrs[TM_ATTRIBUTES.INPUT_UNITS] ||
@@ -77,18 +106,10 @@ export class TokenMeterProcessor {
             inputUnits,
             outputUnits,
         });
-        // Add cost attribute to span
-        // Note: OTel allows modifying span attributes after end but before export
-        // We use a workaround by storing cost in the span's resource or via events
-        // For now, we'll set it directly (works with most exporters)
+        // Log calculated cost for debugging
+        // Note: We cannot add this to the span - ReadableSpan is immutable after end()
         if (cost !== null) {
-            // Unfortunately, ReadableSpan doesn't allow setting attributes after end
-            // We need to use a custom approach - storing in a side channel or using events
-            // For this implementation, we'll log a warning and add via span events
-            // A production implementation would use a custom exporter or modify before end
-            // The proper way is to calculate cost BEFORE span.end() in the proxy
-            // This processor is for catching spans from other sources (like Vercel AI SDK)
-            console.debug(`[tokenmeter] Calculated cost for ${provider}/${model}: $${cost.toFixed(6)}`);
+            logger.debug(`Calculated cost for ${provider}/${model}: $${cost.toFixed(6)}`);
         }
     }
     /**
@@ -102,12 +123,12 @@ export class TokenMeterProcessor {
         // Fall back to manifest
         const manifest = this.manifest || getCachedManifest();
         if (!manifest) {
-            console.warn(`[tokenmeter] Pricing manifest not loaded, cannot calculate cost`);
+            logger.warn("Pricing manifest not loaded, cannot calculate cost");
             return null;
         }
         const pricing = getModelPricing(provider, model, manifest);
         if (!pricing) {
-            console.warn(`[tokenmeter] No pricing found for ${provider}/${model}`);
+            logger.warn(`No pricing found for ${provider}/${model}`);
             return 0; // Return 0, not null, to indicate we tried but found no pricing
         }
         return calculateCost(usage, pricing);

package/dist/processor/TokenMeterProcessor.js.map

@@ -1 +1 @@
-{"version":3,"file":"TokenMeterProcessor.js","sourceRoot":"","sources":["../../src/processor/TokenMeterProcessor.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAQH,OAAO,EACL,YAAY,EACZ,eAAe,EACf,aAAa,EACb,iBAAiB,GAClB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAE/D;;;;;;;;;;;;;;;GAeG;AACH,MAAM,OAAO,mBAAmB;IACtB,QAAQ,GAA2B,IAAI,CAAC;IACxC,eAAe,GAAyB,IAAI,CAAC;IAC7C,MAAM,CAA4B;IAClC,gBAAgB,CAA+B;IAEvD,YAAY,SAAoC,EAAE;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC;QAEtD,uCAAuC;QACvC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,GAAG,MAAM,YAAY,CAAC;gBACjC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;aACrC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+CAA+C,EAAE,KAAK,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,IAAU,EAAE,aAAsB;QACxC,gDAAgD;QAChD,kCAAkC;IACpC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAkB;QACtB,iCAAiC;QACjC,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC;QAE9B,oCAAoC;QACpC,MAAM,UAAU,GACb,KAAK,CAAC,aAAa,CAAC,WAAW,CAAY;YAC3C,KAAK,CAAC,iBAAiB,CAAC,YAAY,CAAY,CAAC;QACpD,MAAM,WAAW,GACd,KAAK,CAAC,aAAa,CAAC,YAAY,CAAY;YAC5C,KAAK,CAAC,iBAAiB,CAAC,aAAa,CAAY,CAAC;QAErD,yBAAyB;QACzB,IAAI,UAAU,KAAK,SAAS,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,yBAAyB;QACzB,MAAM,QAAQ,GACX,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAY;YACxC,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAY;YAC3C,SAAS,CAAC;QACZ,MAAM,KAAK,GACR,KAAK,CAAC,aAAa,CAAC,KAAK,CAAY;YACrC,KAAK,CAAC,iBAAiB,CAAC,KAAK,CAAY;YAC1C,SAAS,CAAC;QAEZ,iBAAiB;QACjB,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,KAAK,EAAE;YACnD,UAAU;YACV,WAAW;SACZ,CAAC,CAAC;QAEH,6BAA6B;QAC7B,0EAA0E;QAC1E,2EAA2E;QAC3E,6DAA6D;QAC7D,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,yEAAyE;YACzE,+EAA+E;YAC/E,uEAAuE;YACvE,+EAA+E;YAE/E,qEAAqE;YACrE,+EAA+E;YAC/E,OAAO,CAAC,KAAK,CACX,oCAAoC,QAAQ,IAAI,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAC7E,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CACvB,QAAgB,EAChB,KAAa,EACb,KAAoD;QAEpD,wBAAwB;QACxB,IAAI,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACtE,CAAC;QAED,wBAAwB;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,iBAAiB,EAAE,CAAC;QACtD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,CAAC,IAAI,CAAC,iEAAiE,CAAC,CAAC;YAChF,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,CAAC,IAAI,CAAC,qCAAqC,QAAQ,IAAI,KAAK,EAAE,CAAC,CAAC;YACvE,OAAO,CAAC,CAAC,CAAC,gEAAgE;QAC5E,CAAC;QAED,OAAO,aAAa,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,sCAAsC;QACtC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,eAAe,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,iCAAiC;IACnC,CAAC;CACF;AAED,eAAe,mBAAmB,CAAC"}
+{"version":3,"file":"TokenMeterProcessor.js","sourceRoot":"","sources":["../../src/processor/TokenMeterProcessor.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AAQH,OAAO,EACL,YAAY,EACZ,eAAe,EACf,aAAa,EACb,iBAAiB,GAClB,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AAEtC;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAO,mBAAmB;IACtB,QAAQ,GAA2B,IAAI,CAAC;IACxC,eAAe,GAAyB,IAAI,CAAC;IAC7C,MAAM,CAA4B;IAClC,gBAAgB,CAA+B;IAEvD,YAAY,SAAoC,EAAE;QAChD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,IAAI,EAAE,CAAC;QAEtD,uCAAuC;QACvC,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAClD,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,GAAG,MAAM,YAAY,CAAC;gBACjC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;aACrC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,KAAW,EAAE,cAAuB;QAC1C,2CAA2C;IAC7C,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,IAAkB;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC;QAE9B,oCAAoC;QACpC,MAAM,UAAU,GACb,KAAK,CAAC,aAAa,CAAC,WAAW,CAAY;YAC3C,KAAK,CAAC,iBAAiB,CAAC,YAAY,CAAY,CAAC;QACpD,MAAM,WAAW,GACd,KAAK,CAAC,aAAa,CAAC,YAAY,CAAY;YAC5C,KAAK,CAAC,iBAAiB,CAAC,aAAa,CAAY,CAAC;QAErD,yBAAyB;QACzB,IAAI,UAAU,KAAK,SAAS,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,yBAAyB;QACzB,MAAM,QAAQ,GACX,KAAK,CAAC,aAAa,CAAC,QAAQ,CAAY;YACxC,KAAK,CAAC,iBAAiB,CAAC,MAAM,CAAY;YAC3C,SAAS,CAAC;QACZ,MAAM,KAAK,GACR,KAAK,CAAC,aAAa,CAAC,KAAK,CAAY;YACrC,KAAK,CAAC,iBAAiB,CAAC,KAAK,CAAY;YAC1C,SAAS,CAAC;QAEZ,iBAAiB;QACjB,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,KAAK,EAAE;YACnD,UAAU;YACV,WAAW;SACZ,CAAC,CAAC;QAEH,oCAAoC;QACpC,+EAA+E;QAC/E,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,MAAM,CAAC,KAAK,CAAC,uBAAuB,QAAQ,IAAI,KAAK,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAChF,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CACvB,QAAgB,EAChB,KAAa,EACb,KAAoD;QAEpD,wBAAwB;QACxB,IAAI,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7C,OAAO,aAAa,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACtE,CAAC;QAED,wBAAwB;QACxB,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,iBAAiB,EAAE,CAAC;QACtD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,CAAC,IAAI,CAAC,oDAAoD,CAAC,CAAC;YAClE,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,OAAO,GAAG,eAAe,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,MAAM,CAAC,IAAI,CAAC,wBAAwB,QAAQ,IAAI,KAAK,EAAE,CAAC,CAAC;YACzD,OAAO,CAAC,CAAC,CAAC,gEAAgE;QAC5E,CAAC;QAED,OAAO,aAAa,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,sCAAsC;QACtC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,eAAe,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,iCAAiC;IACnC,CAAC;CACF;AAED,eAAe,mBAAmB,CAAC"}

package/dist/query/client.d.ts

@@ -32,30 +32,6 @@ export interface QueryClient {
     /** Close the database connection */
     close(): Promise<void>;
 }
-/**
- * Create a query client for cost aggregation
- *
- * @example
- * ```typescript
- * import { createQueryClient } from 'tokenmeter/query';
- *
- * const client = createQueryClient({
- *   connectionString: process.env.DATABASE_URL,
- * });
- *
- * // Get total costs for an organization
- * const orgCosts = await client.getCostByOrg('org_123', {
- *   from: '2024-01-01',
- *   to: '2024-01-31',
- *   groupBy: ['model'],
- * });
- *
- * console.log(`Total: $${orgCosts.totalCost}`);
- * for (const group of orgCosts.groups ?? []) {
- *   console.log(`  ${group.key.model}: $${group.cost}`);
- * }
- * ```
- */
 export declare function createQueryClient(config: QueryClientConfig): Promise<QueryClient>;
 export default createQueryClient;
 //# sourceMappingURL=client.d.ts.map

package/dist/query/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/query/client.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAUhE;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,mCAAmC;IACnC,gBAAgB,EAAE,MAAM,CAAC;IACzB,8CAA8C;IAC9C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,qDAAqD;IACrD,QAAQ,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IAC1D,oCAAoC;IACpC,aAAa,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,QAAQ,CAAC,GACzC,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,4CAA4C;IAC5C,YAAY,CACV,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,GACjD,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,qCAAqC;IACrC,cAAc,CACZ,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,OAAO,CAAC,GACxC,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,wCAAwC;IACxC,iBAAiB,CACf,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,GAC3C,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,0CAA0C;IAC1C,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IACzD,oCAAoC;IACpC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,wBAAsB,iBAAiB,CACrC,MAAM,EAAE,iBAAiB,GACxB,OAAO,CAAC,WAAW,CAAC,CAqOtB;AAED,eAAe,iBAAiB,CAAC"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../src/query/client.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAUhE;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,mCAAmC;IACnC,gBAAgB,EAAE,MAAM,CAAC;IACzB,8CAA8C;IAC9C,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,qDAAqD;IACrD,QAAQ,CAAC,OAAO,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IAC1D,oCAAoC;IACpC,aAAa,CACX,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,QAAQ,CAAC,GACzC,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,4CAA4C;IAC5C,YAAY,CACV,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,gBAAgB,CAAC,GACjD,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,qCAAqC;IACrC,cAAc,CACZ,KAAK,EAAE,MAAM,EACb,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,OAAO,CAAC,GACxC,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,wCAAwC;IACxC,iBAAiB,CACf,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,IAAI,CAAC,gBAAgB,EAAE,UAAU,CAAC,GAC3C,OAAO,CAAC,UAAU,CAAC,CAAC;IACvB,0CAA0C;IAC1C,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;IACzD,oCAAoC;IACpC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AA4CD,wBAAsB,iBAAiB,CACrC,MAAM,EAAE,iBAAiB,GACxB,OAAO,CAAC,WAAW,CAAC,CA+PtB;AAED,eAAe,iBAAiB,CAAC"}

package/dist/query/client.js

@@ -27,13 +27,36 @@
  * }
  * ```
  */
+/**
+ * Validate table name to prevent SQL injection
+ */
+function validateTableName(name) {
+    // Only allow alphanumeric characters and underscores
+    if (!/^[a-zA-Z_][a-zA-Z0-9_]*$/.test(name)) {
+        throw new Error(`Invalid table name: "${name}". Table names must start with a letter or underscore and contain only alphanumeric characters and underscores.`);
+    }
+    // Limit length to prevent abuse
+    if (name.length > 63) {
+        throw new Error(`Table name too long: "${name}". Maximum length is 63 characters.`);
+    }
+}
 export async function createQueryClient(config) {
     const tableName = config.tableName ?? "tokenmeter_events";
+    // Validate table name to prevent SQL injection
+    validateTableName(tableName);
     // Dynamic import to avoid requiring pg at load time
     const { Pool } = await import("pg");
     const pool = new Pool({
         connectionString: config.connectionString,
     });
+    // Test connection on initialization
+    try {
+        await pool.query("SELECT 1");
+    }
+    catch (error) {
+        await pool.end();
+        throw new Error(`Failed to connect to database: ${error instanceof Error ? error.message : "Unknown error"}`);
+    }
     /**
      * Build WHERE clause from options
      */
@@ -75,17 +98,27 @@ export async function createQueryClient(config) {
         return { clause, values, nextIndex: paramIndex };
     }
     /**
-     * Map groupBy field names to column names
+     * Allowed groupBy fields (whitelist for SQL injection prevention)
+     */
+    const ALLOWED_GROUP_BY_FIELDS = {
+        provider: "provider",
+        model: "model",
+        organizationId: "organization_id",
+        userId: "user_id",
+        workflowId: "workflow_id",
+    };
+    /**
+     * Map groupBy field names to column names.
+     * Throws an error for unknown fields to prevent SQL injection.
+     *
+     * @throws {Error} If field is not in the allowed whitelist
      */
     function mapGroupByField(field) {
-        const fieldMap = {
-            provider: "provider",
-            model: "model",
-            organizationId: "organization_id",
-            userId: "user_id",
-            workflowId: "workflow_id",
-        };
-        return fieldMap[field] ?? field;
+        const column = ALLOWED_GROUP_BY_FIELDS[field];
+        if (!column) {
+            throw new Error(`Invalid groupBy field: "${field}". Allowed fields: ${Object.keys(ALLOWED_GROUP_BY_FIELDS).join(", ")}`);
+        }
+        return column;
     }
     /**
      * Execute a cost query

package/dist/query/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/query/client.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAsDH;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,MAAyB;IAEzB,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,mBAAmB,CAAC;IAE1D,oDAAoD;IACpD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;IACpC,MAAM,IAAI,GAAS,IAAI,IAAI,CAAC;QAC1B,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;KAC1C,CAAC,CAAC;IAEH;;OAEG;IACH,SAAS,gBAAgB,CACvB,OAAyB,EACzB,kBAA0B,CAAC;QAE3B,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,IAAI,UAAU,GAAG,eAAe,CAAC;QAEjC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACjB,UAAU,CAAC,IAAI,CAAC,kBAAkB,UAAU,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,IAAI,CACT,OAAO,CAAC,IAAI,YAAY,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CACrE,CAAC;YACF,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CAAC,kBAAkB,UAAU,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,IAAI,CACT,OAAO,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAC/D,CAAC;YACF,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,UAAU,CAAC,IAAI,CAAC,eAAe,UAAU,EAAE,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC9B,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,UAAU,CAAC,IAAI,CAAC,YAAY,UAAU,EAAE,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3B,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,UAAU,CAAC,IAAI,CAAC,sBAAsB,UAAU,EAAE,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACpC,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,UAAU,CAAC,IAAI,CAAC,cAAc,UAAU,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC5B,UAAU,EAAE,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GACV,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAEnE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,SAAS,eAAe,CAAC,KAAa;QACpC,MAAM,QAAQ,GAA2B;YACvC,QAAQ,EAAE,UAAU;YACpB,KAAK,EAAE,OAAO;YACd,cAAc,EAAE,iBAAiB;YACjC,MAAM,EAAE,SAAS;YACjB,UAAU,EAAE,aAAa;SAC1B,CAAC;QACF,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,KAAK,UAAU,QAAQ,CAAC,UAA4B,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QAE5D,IAAI,KAAa,CAAC;QAClB,IAAI,WAAW,GAAG,MAAM,CAAC;QAEzB,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,sBAAsB;YACtB,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,YAAY,GAAG,OAAO;iBACzB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBACX,uCAAuC;gBACvC,MAAM,OAAO,GACX,GAAG,KAAK,iBAAiB;oBACvB,CAAC,CAAC,gBAAgB;oBAClB,CAAC,CAAC,GAAG,KAAK,SAAS;wBACjB,CAAC,CAAC,QAAQ;wBACV,CAAC,CAAC,GAAG,KAAK,aAAa;4BACrB,CAAC,CAAC,YAAY;4BACd,CAAC,CAAC,GAAG,CAAC;gBACd,OAAO,GAAG,GAAG,QAAQ,OAAO,GAAG,CAAC;YAClC,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,KAAK,GAAG;;YAEF,YAAY;;;eAGT,SAAS;UACd,MAAM;mBACG,aAAa;;UAEtB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;OAC7C,CAAC;YAEF,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,WAAW,GAAG,CAAC,GAAG,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAEpD,mBAAmB;YACnB,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,MAAM,MAAM,GAAyB,EAAE,CAAC;YAExC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAsC,EAAE,CAAC;gBAChE,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,IAAc,CAAC,CAAC;gBAC5C,MAAM,KAAK,GAAG,GAAG,CAAC,KAAe,CAAC;gBAElC,SAAS,IAAI,IAAI,CAAC;gBAClB,UAAU,IAAI,KAAK,CAAC;gBAEpB,qCAAqC;gBACrC,MAAM,GAAG,GAA2B,EAAE,CAAC;gBACvC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,MAAM,OAAO,GACX,KAAK,KAAK,iBAAiB;wBACzB,CAAC,CAAC,gBAAgB;wBAClB,CAAC,CAAC,KAAK,KAAK,SAAS;4BACnB,CAAC,CAAC,QAAQ;4BACV,CAAC,CAAC,KAAK,KAAK,aAAa;gCACvB,CAAC,CAAC,YAAY;gCACd,CAAC,CAAC,KAAK,CAAC;oBAChB,GAAG,CAAC,OAAO,CAAC,GAAI,GAAG,CAAC,OAAO,CAAY,IAAI,EAAE,CAAC;gBAChD,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACpC,CAAC;YAED,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,yBAAyB;YACzB,KAAK,GAAG;;;;eAIC,SAAS;UACd,MAAM;OACT,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC/C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAA0C,CAAC;YAEpE,OAAO;gBACL,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC;gBACrC,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ;QAER,KAAK,CAAC,aAAa,CACjB,MAAc,EACd,UAA4C,EAAE;YAE9C,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC;QAED,KAAK,CAAC,YAAY,CAChB,KAAa,EACb,UAAoD,EAAE;YAEtD,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,KAAK,CAAC,cAAc,CAClB,KAAa,EACb,UAA2C,EAAE;YAE7C,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,CAAC;QAED,KAAK,CAAC,iBAAiB,CACrB,QAAgB,EAChB,UAA8C,EAAE;YAEhD,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,KAAK,CAAC,eAAe,CAAC,UAAkB;YACtC,MAAM,KAAK,GAAG;;;;eAIL,SAAS;;OAEjB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACrD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAA0C,CAAC;YAEpE,OAAO;gBACL,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC;gBACrC,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,KAAK;YACT,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,eAAe,iBAAiB,CAAC"}
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/query/client.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAsDH;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH;;GAEG;AACH,SAAS,iBAAiB,CAAC,IAAY;IACrC,qDAAqD;IACrD,IAAI,CAAC,0BAA0B,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3C,MAAM,IAAI,KAAK,CACb,wBAAwB,IAAI,iHAAiH,CAC9I,CAAC;IACJ,CAAC;IACD,gCAAgC;IAChC,IAAI,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CACb,yBAAyB,IAAI,qCAAqC,CACnE,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,MAAyB;IAEzB,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,mBAAmB,CAAC;IAE1D,+CAA+C;IAC/C,iBAAiB,CAAC,SAAS,CAAC,CAAC;IAE7B,oDAAoD;IACpD,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;IACpC,MAAM,IAAI,GAAS,IAAI,IAAI,CAAC;QAC1B,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;KAC1C,CAAC,CAAC;IAEH,oCAAoC;IACpC,IAAI,CAAC;QACH,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;IAC/B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CACb,kCAAkC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,EAAE,CAC7F,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,SAAS,gBAAgB,CACvB,OAAyB,EACzB,kBAA0B,CAAC;QAE3B,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,MAAM,MAAM,GAAc,EAAE,CAAC;QAC7B,IAAI,UAAU,GAAG,eAAe,CAAC;QAEjC,IAAI,OAAO,CAAC,IAAI,EAAE,CAAC;YACjB,UAAU,CAAC,IAAI,CAAC,kBAAkB,UAAU,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,IAAI,CACT,OAAO,CAAC,IAAI,YAAY,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CACrE,CAAC;YACF,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;YACf,UAAU,CAAC,IAAI,CAAC,kBAAkB,UAAU,EAAE,CAAC,CAAC;YAChD,MAAM,CAAC,IAAI,CACT,OAAO,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAC/D,CAAC;YACF,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,UAAU,CAAC,IAAI,CAAC,eAAe,UAAU,EAAE,CAAC,CAAC;YAC7C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC9B,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YAClB,UAAU,CAAC,IAAI,CAAC,YAAY,UAAU,EAAE,CAAC,CAAC;YAC1C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3B,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,cAAc,EAAE,CAAC;YAC3B,UAAU,CAAC,IAAI,CAAC,sBAAsB,UAAU,EAAE,CAAC,CAAC;YACpD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;YACpC,UAAU,EAAE,CAAC;QACf,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,UAAU,CAAC,IAAI,CAAC,cAAc,UAAU,EAAE,CAAC,CAAC;YAC5C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC5B,UAAU,EAAE,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GACV,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAEnE,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,MAAM,uBAAuB,GAA2B;QACtD,QAAQ,EAAE,UAAU;QACpB,KAAK,EAAE,OAAO;QACd,cAAc,EAAE,iBAAiB;QACjC,MAAM,EAAE,SAAS;QACjB,UAAU,EAAE,aAAa;KAC1B,CAAC;IAEF;;;;;OAKG;IACH,SAAS,eAAe,CAAC,KAAa;QACpC,MAAM,MAAM,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,2BAA2B,KAAK,sBAAsB,MAAM,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACxG,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,KAAK,UAAU,QAAQ,CAAC,UAA4B,EAAE;QACpD,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAChE,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC;QAE5D,IAAI,KAAa,CAAC;QAClB,IAAI,WAAW,GAAG,MAAM,CAAC;QAEzB,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,sBAAsB;YACtB,MAAM,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,YAAY,GAAG,OAAO;iBACzB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;gBACX,uCAAuC;gBACvC,MAAM,OAAO,GACX,GAAG,KAAK,iBAAiB;oBACvB,CAAC,CAAC,gBAAgB;oBAClB,CAAC,CAAC,GAAG,KAAK,SAAS;wBACjB,CAAC,CAAC,QAAQ;wBACV,CAAC,CAAC,GAAG,KAAK,aAAa;4BACrB,CAAC,CAAC,YAAY;4BACd,CAAC,CAAC,GAAG,CAAC;gBACd,OAAO,GAAG,GAAG,QAAQ,OAAO,GAAG,CAAC;YAClC,CAAC,CAAC;iBACD,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,KAAK,GAAG;;YAEF,YAAY;;;eAGT,SAAS;UACd,MAAM;mBACG,aAAa;;UAEtB,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,UAAU,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE;OAC7C,CAAC;YAEF,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,WAAW,GAAG,CAAC,GAAG,MAAM,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAEpD,mBAAmB;YACnB,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,MAAM,MAAM,GAAyB,EAAE,CAAC;YAExC,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAsC,EAAE,CAAC;gBAChE,MAAM,IAAI,GAAG,UAAU,CAAC,GAAG,CAAC,IAAc,CAAC,CAAC;gBAC5C,MAAM,KAAK,GAAG,GAAG,CAAC,KAAe,CAAC;gBAElC,SAAS,IAAI,IAAI,CAAC;gBAClB,UAAU,IAAI,KAAK,CAAC;gBAEpB,qCAAqC;gBACrC,MAAM,GAAG,GAA2B,EAAE,CAAC;gBACvC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,MAAM,OAAO,GACX,KAAK,KAAK,iBAAiB;wBACzB,CAAC,CAAC,gBAAgB;wBAClB,CAAC,CAAC,KAAK,KAAK,SAAS;4BACnB,CAAC,CAAC,QAAQ;4BACV,CAAC,CAAC,KAAK,KAAK,aAAa;gCACvB,CAAC,CAAC,YAAY;gCACd,CAAC,CAAC,KAAK,CAAC;oBAChB,GAAG,CAAC,OAAO,CAAC,GAAI,GAAG,CAAC,OAAO,CAAY,IAAI,EAAE,CAAC;gBAChD,CAAC;gBAED,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YACpC,CAAC;YAED,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;QAClD,CAAC;aAAM,CAAC;YACN,yBAAyB;YACzB,KAAK,GAAG;;;;eAIC,SAAS;UACd,MAAM;OACT,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC/C,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAA0C,CAAC;YAEpE,OAAO;gBACL,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC;gBACrC,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO;QACL,QAAQ;QAER,KAAK,CAAC,aAAa,CACjB,MAAc,EACd,UAA4C,EAAE;YAE9C,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC;QAED,KAAK,CAAC,YAAY,CAChB,KAAa,EACb,UAAoD,EAAE;YAEtD,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC,CAAC;QACzD,CAAC;QAED,KAAK,CAAC,cAAc,CAClB,KAAa,EACb,UAA2C,EAAE;YAE7C,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;QACzC,CAAC;QAED,KAAK,CAAC,iBAAiB,CACrB,QAAgB,EAChB,UAA8C,EAAE;YAEhD,OAAO,QAAQ,CAAC,EAAE,GAAG,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,KAAK,CAAC,eAAe,CAAC,UAAkB;YACtC,MAAM,KAAK,GAAG;;;;eAIL,SAAS;;OAEjB,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YACrD,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAA0C,CAAC;YAEpE,OAAO;gBACL,SAAS,EAAE,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC;gBACrC,KAAK,EAAE,GAAG,CAAC,KAAK;aACjB,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,KAAK;YACT,MAAM,IAAI,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,eAAe,iBAAiB,CAAC"}

package/dist/registry.d.ts

@@ -0,0 +1,127 @@
+/**
+ * Provider Registry
+ *
+ * Extensible system for registering custom AI providers.
+ * Allows users to add support for new providers without modifying core code.
+ */
+import type { ExtractionStrategy, UsageData } from "./types.js";
+/**
+ * Symbol used to mark objects with their provider.
+ * Allows explicit provider identification without duck-typing.
+ *
+ * @example
+ * ```typescript
+ * import { TOKENMETER_PROVIDER } from 'tokenmeter';
+ *
+ * const client = new CustomAIClient();
+ * client[TOKENMETER_PROVIDER] = 'custom-ai';
+ *
+ * const monitored = monitor(client); // Will use 'custom-ai' as provider
+ * ```
+ */
+export declare const TOKENMETER_PROVIDER: unique symbol;
+/**
+ * Configuration for registering a custom provider
+ */
+export interface ProviderConfig {
+    /** Unique provider identifier (e.g., 'google-ai', 'custom-llm') */
+    name: string;
+    /**
+     * Function to detect if a client belongs to this provider.
+     * Called during monitor() if no explicit provider is set.
+     *
+     * @param client - The client instance being monitored
+     * @returns true if this provider should handle the client
+     */
+    detect?: (client: unknown) => boolean;
+    /**
+     * Function to extract usage data from API responses.
+     *
+     * @param response - The API response
+     * @param args - The original arguments passed to the API call
+     * @returns Usage data or null if extraction failed
+     */
+    extractUsage?: (response: unknown, args: unknown[]) => Partial<UsageData> | null;
+    /**
+     * Function to extract the model name from request args or response.
+     *
+     * @param args - The original arguments passed to the API call
+     * @param response - The API response (if available)
+     * @returns Model identifier string
+     */
+    extractModel?: (args: unknown[], response?: unknown) => string;
+    /**
+     * List of method names that are factory methods returning objects to be proxied.
+     * These methods won't create spans but their return values will be wrapped.
+     *
+     * @example ['getGenerativeModel', 'createClient', 'getModel']
+     */
+    factoryMethods?: string[];
+    /**
+     * Full extraction strategy (alternative to extractUsage/extractModel).
+     * If provided, this takes precedence over extractUsage and extractModel.
+     */
+    strategy?: ExtractionStrategy;
+}
+/**
+ * Register a custom provider with tokenmeter.
+ *
+ * This is the recommended way to add support for new AI providers
+ * without modifying the core library code.
+ *
+ * @example
+ * ```typescript
+ * import { registerProvider } from 'tokenmeter';
+ *
+ * registerProvider({
+ *   name: 'my-ai-provider',
+ *   detect: (client) => 'generateText' in client && 'myProvider' in client,
+ *   extractUsage: (response) => ({
+ *     inputUnits: response.usage?.inputTokens,
+ *     outputUnits: response.usage?.outputTokens,
+ *   }),
+ *   extractModel: (args) => args[0]?.model || 'unknown',
+ *   factoryMethods: ['createModel'],
+ * });
+ *
+ * // Now you can monitor your custom client
+ * const client = monitor(new MyAIProvider());
+ * ```
+ */
+export declare function registerProvider(config: ProviderConfig): void;
+/**
+ * Unregister a provider
+ */
+export declare function unregisterProvider(name: string): boolean;
+/**
+ * Get a registered provider by name
+ */
+export declare function getProvider(name: string): ProviderConfig | undefined;
+/**
+ * Get all registered providers
+ */
+export declare function getRegisteredProviders(): ProviderConfig[];
+/**
+ * Clear all registered providers (useful for testing)
+ */
+export declare function clearProviderRegistry(): void;
+/**
+ * Detect provider from a client using the registry.
+ * Returns the first matching provider or undefined.
+ *
+ * @internal
+ */
+export declare function detectProviderFromRegistry(client: unknown): string | undefined;
+/**
+ * Get factory methods for a provider
+ *
+ * @internal
+ */
+export declare function getFactoryMethods(provider: string): string[];
+/**
+ * Get extraction strategy for a provider from the registry
+ *
+ * @internal
+ */
+export declare function getRegisteredStrategy(provider: string): ExtractionStrategy | undefined;
+//# sourceMappingURL=registry.d.ts.map

package/dist/registry.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.d.ts","sourceRoot":"","sources":["../src/registry.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,kBAAkB,EAAE,SAAS,EAAE,MAAM,YAAY,CAAC;AAEhE;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,mBAAmB,eAAoC,CAAC;AAErE;;GAEG;AACH,MAAM,WAAW,cAAc;IAC7B,mEAAmE;IACnE,IAAI,EAAE,MAAM,CAAC;IAEb;;;;;;OAMG;IACH,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,OAAO,KAAK,OAAO,CAAC;IAEtC;;;;;;OAMG;IACH,YAAY,CAAC,EAAE,CACb,QAAQ,EAAE,OAAO,EACjB,IAAI,EAAE,OAAO,EAAE,KACZ,OAAO,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;IAE/B;;;;;;OAMG;IACH,YAAY,CAAC,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,QAAQ,CAAC,EAAE,OAAO,KAAK,MAAM,CAAC;IAE/D;;;;;OAKG;IACH,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAE1B;;;OAGG;IACH,QAAQ,CAAC,EAAE,kBAAkB,CAAC;CAC/B;AAOD;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,wBAAgB,gBAAgB,CAAC,MAAM,EAAE,cAAc,GAAG,IAAI,CAoC7D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAExD;AAED;;GAEG;AACH,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS,CAEpE;AAED;;GAEG;AACH,wBAAgB,sBAAsB,IAAI,cAAc,EAAE,CAEzD;AAED;;GAEG;AACH,wBAAgB,qBAAqB,IAAI,IAAI,CAE5C;AAED;;;;;GAKG;AACH,wBAAgB,0BAA0B,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,GAAG,SAAS,CAkB9E;AAED;;;;GAIG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,CAG5D;AAED;;;;GAIG;AACH,wBAAgB,qBAAqB,CACnC,QAAQ,EAAE,MAAM,GACf,kBAAkB,GAAG,SAAS,CAGhC"}

package/dist/registry.js

@@ -0,0 +1,144 @@
+/**
+ * Provider Registry
+ *
+ * Extensible system for registering custom AI providers.
+ * Allows users to add support for new providers without modifying core code.
+ */
+/**
+ * Symbol used to mark objects with their provider.
+ * Allows explicit provider identification without duck-typing.
+ *
+ * @example
+ * ```typescript
+ * import { TOKENMETER_PROVIDER } from 'tokenmeter';
+ *
+ * const client = new CustomAIClient();
+ * client[TOKENMETER_PROVIDER] = 'custom-ai';
+ *
+ * const monitored = monitor(client); // Will use 'custom-ai' as provider
+ * ```
+ */
+export const TOKENMETER_PROVIDER = Symbol.for("tokenmeter.provider");
+/**
+ * Internal registry storage
+ */
+const providerRegistry = new Map();
+/**
+ * Register a custom provider with tokenmeter.
+ *
+ * This is the recommended way to add support for new AI providers
+ * without modifying the core library code.
+ *
+ * @example
+ * ```typescript
+ * import { registerProvider } from 'tokenmeter';
+ *
+ * registerProvider({
+ *   name: 'my-ai-provider',
+ *   detect: (client) => 'generateText' in client && 'myProvider' in client,
+ *   extractUsage: (response) => ({
+ *     inputUnits: response.usage?.inputTokens,
+ *     outputUnits: response.usage?.outputTokens,
+ *   }),
+ *   extractModel: (args) => args[0]?.model || 'unknown',
+ *   factoryMethods: ['createModel'],
+ * });
+ *
+ * // Now you can monitor your custom client
+ * const client = monitor(new MyAIProvider());
+ * ```
+ */
+export function registerProvider(config) {
+    if (!config.name) {
+        throw new Error("Provider name is required");
+    }
+    // If strategy is not provided, create one from extractUsage/extractModel
+    if (!config.strategy && config.extractUsage) {
+        config.strategy = {
+            provider: config.name,
+            canHandle: (_methodPath, result) => {
+                // Try to extract usage - if it succeeds, we can handle this result
+                const usage = config.extractUsage(result, []);
+                return usage !== null;
+            },
+            extract: (_methodPath, result, args) => {
+                const partialUsage = config.extractUsage(result, args);
+                if (!partialUsage)
+                    return null;
+                const model = config.extractModel
+                    ? config.extractModel(args, result)
+                    : "unknown";
+                return {
+                    provider: config.name,
+                    model,
+                    ...partialUsage,
+                };
+            },
+        };
+    }
+    providerRegistry.set(config.name, config);
+}
+/**
+ * Unregister a provider
+ */
+export function unregisterProvider(name) {
+    return providerRegistry.delete(name);
+}
+/**
+ * Get a registered provider by name
+ */
+export function getProvider(name) {
+    return providerRegistry.get(name);
+}
+/**
+ * Get all registered providers
+ */
+export function getRegisteredProviders() {
+    return Array.from(providerRegistry.values());
+}
+/**
+ * Clear all registered providers (useful for testing)
+ */
+export function clearProviderRegistry() {
+    providerRegistry.clear();
+}
+/**
+ * Detect provider from a client using the registry.
+ * Returns the first matching provider or undefined.
+ *
+ * @internal
+ */
+export function detectProviderFromRegistry(client) {
+    // First, check for explicit Symbol-based provider
+    if (client &&
+        typeof client === "object" &&
+        TOKENMETER_PROVIDER in client) {
+        return client[TOKENMETER_PROVIDER];
+    }
+    // Then check registered providers
+    for (const [name, config] of providerRegistry) {
+        if (config.detect && config.detect(client)) {
+            return name;
+        }
+    }
+    return undefined;
+}
+/**
+ * Get factory methods for a provider
+ *
+ * @internal
+ */
+export function getFactoryMethods(provider) {
+    const config = providerRegistry.get(provider);
+    return config?.factoryMethods || [];
+}
+/**
+ * Get extraction strategy for a provider from the registry
+ *
+ * @internal
+ */
+export function getRegisteredStrategy(provider) {
+    const config = providerRegistry.get(provider);
+    return config?.strategy;
+}
+//# sourceMappingURL=registry.js.map

package/dist/registry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"registry.js","sourceRoot":"","sources":["../src/registry.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,MAAM,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;AAsDrE;;GAEG;AACH,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA0B,CAAC;AAE3D;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,UAAU,gBAAgB,CAAC,MAAsB;IACrD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,yEAAyE;IACzE,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QAC5C,MAAM,CAAC,QAAQ,GAAG;YAChB,QAAQ,EAAE,MAAM,CAAC,IAAI;YACrB,SAAS,EAAE,CAAC,WAAqB,EAAE,MAAe,EAAW,EAAE;gBAC7D,mEAAmE;gBACnE,MAAM,KAAK,GAAG,MAAM,CAAC,YAAa,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBAC/C,OAAO,KAAK,KAAK,IAAI,CAAC;YACxB,CAAC;YACD,OAAO,EAAE,CACP,WAAqB,EACrB,MAAe,EACf,IAAe,EACG,EAAE;gBACpB,MAAM,YAAY,GAAG,MAAM,CAAC,YAAa,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;gBACxD,IAAI,CAAC,YAAY;oBAAE,OAAO,IAAI,CAAC;gBAE/B,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY;oBAC/B,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC;oBACnC,CAAC,CAAC,SAAS,CAAC;gBAEd,OAAO;oBACL,QAAQ,EAAE,MAAM,CAAC,IAAI;oBACrB,KAAK;oBACL,GAAG,YAAY;iBAChB,CAAC;YACJ,CAAC;SACF,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;AAC5C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,IAAY;IAC7C,OAAO,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACvC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,IAAY;IACtC,OAAO,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;AACpC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB;IACpC,OAAO,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,qBAAqB;IACnC,gBAAgB,CAAC,KAAK,EAAE,CAAC;AAC3B,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,0BAA0B,CAAC,MAAe;IACxD,kDAAkD;IAClD,IACE,MAAM;QACN,OAAO,MAAM,KAAK,QAAQ;QAC1B,mBAAmB,IAAI,MAAM,EAC7B,CAAC;QACD,OAAQ,MAAiC,CAAC,mBAAmB,CAAC,CAAC;IACjE,CAAC;IAED,kCAAkC;IAClC,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,gBAAgB,EAAE,CAAC;QAC9C,IAAI,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YAC3C,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,iBAAiB,CAAC,QAAgB;IAChD,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9C,OAAO,MAAM,EAAE,cAAc,IAAI,EAAE,CAAC;AACtC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACnC,QAAgB;IAEhB,MAAM,MAAM,GAAG,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IAC9C,OAAO,MAAM,EAAE,QAAQ,CAAC;AAC1B,CAAC"}