@x12i/ai-gateway 7.9.1

package/dist/types.js

@@ -0,0 +1,6 @@
+/**
+ * Type definitions for AI Gateway
+ *
+ * Extends router types with gateway-specific enhancements
+ */
+export {};

package/dist/usage-tracker.d.ts

@@ -0,0 +1,78 @@
+/**
+ * Usage Tracker
+ *
+ * Manages usage tracking for LLM requests including:
+ * - Token counting
+ * - Consumption calculation (RPM/TPM)
+ * - Usage recording
+ *
+ * Note: x-models dependency removed - usage tracking functions are permanently disabled
+ */
+import type { Logxer } from '@x12i/logxer';
+import type { LLMResponse } from './types.js';
+export interface UsageTrackerConfig {
+    enableUsageTracking: boolean;
+    usageTier?: string;
+    logger: Logxer;
+}
+/**
+ * Manages usage tracking for LLM requests
+ */
+export declare class UsageTracker {
+    private enabled;
+    private logger;
+    constructor(config: UsageTrackerConfig);
+    /**
+     * Extracts token usage from router response
+     *
+     * @param response - Router response object
+     * @returns Token usage breakdown
+     */
+    extractTokens(response: LLMResponse): {
+        prompt: number;
+        completion: number;
+        total: number;
+    };
+    /**
+     * Records usage for a request
+     *
+     * @param params - Usage recording parameters
+     */
+    recordUsage(params: {
+        timestamp: number;
+        duration: number;
+        tokens: {
+            prompt: number;
+            completion: number;
+            total: number;
+        };
+        provider?: string;
+        modelId?: string;
+    }): Promise<void>;
+    /**
+     * Calculates consumption for a request
+     *
+     * @param params - Consumption calculation parameters
+     * @returns Consumption data or null if tracking is disabled
+     */
+    calculateConsumption(params: {
+        latencyMs: number;
+        promptTokens: number;
+        completionTokens: number;
+        provider?: string;
+    }): Promise<{
+        rpmConsumption: number;
+        tpmConsumption: number;
+    } | null>;
+    /**
+     * Logs consumption data for debugging
+     *
+     * @param jobId - Job ID for logging context
+     * @param provider - Provider name
+     * @param consumption - Consumption data
+     */
+    logConsumption(jobId: string, provider: string, consumption: {
+        rpmConsumption: number;
+        tpmConsumption: number;
+    } | null): void;
+}

package/dist/usage-tracker.js

@@ -0,0 +1,79 @@
+/**
+ * Usage Tracker
+ *
+ * Manages usage tracking for LLM requests including:
+ * - Token counting
+ * - Consumption calculation (RPM/TPM)
+ * - Usage recording
+ *
+ * Note: x-models dependency removed - usage tracking functions are permanently disabled
+ */
+/**
+ * Manages usage tracking for LLM requests
+ */
+export class UsageTracker {
+    enabled;
+    logger;
+    constructor(config) {
+        this.enabled = config.enableUsageTracking;
+        this.logger = config.logger;
+        // x-models removed - usage tracking initialization disabled
+    }
+    /**
+     * Extracts token usage from router response
+     *
+     * @param response - Router response object
+     * @returns Token usage breakdown
+     */
+    extractTokens(response) {
+        return {
+            prompt: response.usage?.promptTokens || 0,
+            completion: response.usage?.completionTokens || 0,
+            total: response.usage?.totalTokens || 0
+        };
+    }
+    /**
+     * Records usage for a request
+     *
+     * @param params - Usage recording parameters
+     */
+    async recordUsage(params) {
+        // x-models removed - usage recording disabled
+        if (!this.enabled || !params.provider) {
+            return;
+        }
+        // No-op: usage tracking permanently disabled
+    }
+    /**
+     * Calculates consumption for a request
+     *
+     * @param params - Consumption calculation parameters
+     * @returns Consumption data or null if tracking is disabled
+     */
+    async calculateConsumption(params) {
+        // x-models removed - consumption calculation disabled
+        if (!this.enabled || !params.provider) {
+            return null;
+        }
+        // Always return null - usage tracking permanently disabled
+        return null;
+    }
+    /**
+     * Logs consumption data for debugging
+     *
+     * @param jobId - Job ID for logging context
+     * @param provider - Provider name
+     * @param consumption - Consumption data
+     */
+    logConsumption(jobId, provider, consumption) {
+        if (!consumption) {
+            return;
+        }
+        this.logger.debug('Usage consumption calculated', {
+            jobId,
+            provider,
+            rpmConsumption: consumption.rpmConsumption.toFixed(2),
+            tpmConsumption: consumption.tpmConsumption.toFixed(2)
+        });
+    }
+}