agentic-qe 2.6.3 → 2.6.5

package/dist/monitoring/ProviderHealthMonitor.js

@@ -0,0 +1,431 @@
+"use strict";
+/**
+ * ProviderHealthMonitor - Health monitoring and circuit breaker for LLM providers
+ *
+ * Monitors provider health with circuit breaker pattern to prevent cascading failures.
+ * Tracks response time, error rate, availability, and manages circuit state transitions.
+ *
+ * @module monitoring/ProviderHealthMonitor
+ * @version 1.0.0
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ProviderHealthMonitor = void 0;
+const events_1 = require("events");
+const Logger_1 = require("../utils/Logger");
+/**
+ * Default configuration values
+ */
+const DEFAULT_CONFIG = {
+    checkIntervalMs: 30000,
+    timeoutMs: 5000,
+    failureThreshold: 3,
+    recoveryTimeMs: 60000,
+    healthyLatencyThresholdMs: 3000
+};
+/**
+ * ProviderHealthMonitor - Monitors LLM provider health with circuit breaker pattern
+ *
+ * Features:
+ * - Automatic health checks at configurable intervals
+ * - Circuit breaker pattern (closed → open → half-open → closed)
+ * - Error rate and availability tracking with sliding window
+ * - Event emission for health changes and circuit state transitions
+ * - Concurrent health checks with timeout protection
+ * - Manual circuit control (force open, reset)
+ *
+ * @example
+ * ```typescript
+ * const monitor = new ProviderHealthMonitor({
+ *   checkIntervalMs: 30000,
+ *   failureThreshold: 3
+ * });
+ *
+ * monitor.registerProvider('ollama', async () => {
+ *   return await ollamaProvider.healthCheck();
+ * });
+ *
+ * monitor.on('health-change', (data) => {
+ *   console.log(`Provider ${data.providerId} health: ${data.healthy}`);
+ * });
+ *
+ * monitor.startMonitoring();
+ * ```
+ */
+class ProviderHealthMonitor extends events_1.EventEmitter {
+    constructor(config) {
+        super();
+        this.logger = Logger_1.Logger.getInstance();
+        this.config = { ...DEFAULT_CONFIG, ...config };
+        this.providers = new Map();
+        this.isMonitoring = false;
+        this.logger.debug('ProviderHealthMonitor initialized', { config: this.config });
+    }
+    /**
+     * Register a provider for health monitoring
+     *
+     * @param providerId - Unique identifier for the provider
+     * @param healthCheckFn - Function that returns provider health status
+     */
+    registerProvider(providerId, healthCheckFn) {
+        if (this.providers.has(providerId)) {
+            this.logger.warn(`Provider ${providerId} already registered, replacing`);
+        }
+        const state = {
+            providerId,
+            healthy: true,
+            latency: 0,
+            errorRate: 0,
+            availability: 1.0,
+            consecutiveFailures: 0,
+            circuitState: 'closed',
+            lastCheck: new Date(),
+            checkCount: 0,
+            successCount: 0
+        };
+        this.providers.set(providerId, {
+            providerId,
+            healthCheckFn,
+            state
+        });
+        this.logger.info(`Provider ${providerId} registered for health monitoring`);
+    }
+    /**
+     * Unregister a provider from health monitoring
+     *
+     * @param providerId - Provider to unregister
+     */
+    unregisterProvider(providerId) {
+        if (!this.providers.has(providerId)) {
+            this.logger.warn(`Provider ${providerId} not registered`);
+            return;
+        }
+        this.providers.delete(providerId);
+        this.logger.info(`Provider ${providerId} unregistered from health monitoring`);
+    }
+    /**
+     * Start automatic health monitoring
+     *
+     * Begins periodic health checks for all registered providers.
+     */
+    startMonitoring() {
+        if (this.isMonitoring) {
+            this.logger.warn('Health monitoring already started');
+            return;
+        }
+        this.isMonitoring = true;
+        // Perform immediate check on start
+        this.checkAllProviders().catch(error => {
+            this.logger.error('Initial health check failed', { error: error.message });
+        });
+        // Schedule periodic checks
+        this.monitoringInterval = setInterval(() => {
+            this.checkAllProviders().catch(error => {
+                this.logger.error('Periodic health check failed', { error: error.message });
+            });
+        }, this.config.checkIntervalMs);
+        this.logger.info('Health monitoring started', {
+            interval: this.config.checkIntervalMs,
+            providers: Array.from(this.providers.keys())
+        });
+    }
+    /**
+     * Stop automatic health monitoring
+     */
+    stopMonitoring() {
+        if (!this.isMonitoring) {
+            this.logger.warn('Health monitoring not started');
+            return;
+        }
+        if (this.monitoringInterval) {
+            clearInterval(this.monitoringInterval);
+            this.monitoringInterval = undefined;
+        }
+        this.isMonitoring = false;
+        this.logger.info('Health monitoring stopped');
+    }
+    /**
+     * Check health of a specific provider
+     *
+     * @param providerId - Provider to check
+     * @returns Health check result
+     */
+    async checkProviderHealth(providerId) {
+        const provider = this.providers.get(providerId);
+        if (!provider) {
+            throw new Error(`Provider ${providerId} not registered`);
+        }
+        const startTime = Date.now();
+        let result;
+        try {
+            // Check if circuit is open and recovery time has passed
+            if (provider.state.circuitState === 'open') {
+                const timeSinceOpen = Date.now() - (provider.circuitOpenedAt?.getTime() || 0);
+                if (timeSinceOpen >= this.config.recoveryTimeMs) {
+                    // Transition to half-open for retry
+                    this.transitionCircuitState(provider, 'half-open');
+                }
+                else {
+                    // Circuit still open, fail fast
+                    result = {
+                        providerId,
+                        healthy: false,
+                        latency: 0,
+                        timestamp: new Date(),
+                        error: 'Circuit breaker is open'
+                    };
+                    this.updateProviderState(provider, result);
+                    return result;
+                }
+            }
+            // Perform health check with timeout
+            const healthStatus = await this.withTimeout(provider.healthCheckFn(), this.config.timeoutMs, `Health check timeout for ${providerId}`);
+            const latency = Date.now() - startTime;
+            // Evaluate health based on response and latency
+            const healthy = healthStatus.healthy &&
+                latency < this.config.healthyLatencyThresholdMs;
+            result = {
+                providerId,
+                healthy,
+                latency,
+                timestamp: new Date(),
+                error: healthStatus.error
+            };
+            // Update state and handle circuit breaker logic
+            this.updateProviderState(provider, result);
+            if (result.healthy) {
+                this.handleSuccessfulCheck(provider);
+            }
+            else {
+                this.handleFailedCheck(provider, result.error);
+            }
+            return result;
+        }
+        catch (error) {
+            const latency = Date.now() - startTime;
+            const errorMessage = error.message;
+            result = {
+                providerId,
+                healthy: false,
+                latency,
+                timestamp: new Date(),
+                error: errorMessage
+            };
+            this.updateProviderState(provider, result);
+            this.handleFailedCheck(provider, errorMessage);
+            return result;
+        }
+    }
+    /**
+     * Check health of all registered providers concurrently
+     *
+     * @returns Array of health check results
+     */
+    async checkAllProviders() {
+        if (this.providers.size === 0) {
+            this.logger.warn('No providers registered for health check');
+            return [];
+        }
+        const providerIds = Array.from(this.providers.keys());
+        const results = await Promise.all(providerIds.map(id => this.checkProviderHealth(id)));
+        this.logger.debug('Completed health check for all providers', {
+            total: results.length,
+            healthy: results.filter(r => r.healthy).length,
+            unhealthy: results.filter(r => !r.healthy).length
+        });
+        return results;
+    }
+    /**
+     * Get current health state of a provider
+     *
+     * @param providerId - Provider to query
+     * @returns Current health state or undefined if not registered
+     */
+    getProviderHealth(providerId) {
+        const provider = this.providers.get(providerId);
+        return provider ? { ...provider.state } : undefined;
+    }
+    /**
+     * Get health state of all registered providers
+     *
+     * @returns Map of provider IDs to health states
+     */
+    getAllProviderHealth() {
+        const healthMap = new Map();
+        for (const [id, provider] of this.providers.entries()) {
+            healthMap.set(id, { ...provider.state });
+        }
+        return healthMap;
+    }
+    /**
+     * Get list of healthy providers
+     *
+     * @returns Array of provider IDs that are currently healthy
+     */
+    getHealthyProviders() {
+        return Array.from(this.providers.entries())
+            .filter(([_, provider]) => provider.state.healthy && provider.state.circuitState === 'closed')
+            .map(([id, _]) => id);
+    }
+    /**
+     * Check if a specific provider is healthy
+     *
+     * @param providerId - Provider to check
+     * @returns True if provider is healthy and circuit is closed
+     */
+    isProviderHealthy(providerId) {
+        const provider = this.providers.get(providerId);
+        if (!provider) {
+            return false;
+        }
+        return provider.state.healthy && provider.state.circuitState === 'closed';
+    }
+    /**
+     * Get current circuit breaker state for a provider
+     *
+     * @param providerId - Provider to query
+     * @returns Circuit state or 'closed' if not registered
+     */
+    getCircuitState(providerId) {
+        const provider = this.providers.get(providerId);
+        return provider?.state.circuitState || 'closed';
+    }
+    /**
+     * Manually force a circuit to open
+     *
+     * Useful for maintenance or emergency situations.
+     *
+     * @param providerId - Provider to open circuit for
+     */
+    forceCircuitOpen(providerId) {
+        const provider = this.providers.get(providerId);
+        if (!provider) {
+            throw new Error(`Provider ${providerId} not registered`);
+        }
+        const previousState = provider.state.circuitState;
+        this.transitionCircuitState(provider, 'open');
+        provider.circuitOpenedAt = new Date();
+        this.logger.warn(`Circuit manually forced open for provider ${providerId}`, {
+            previousState
+        });
+    }
+    /**
+     * Reset circuit breaker to closed state
+     *
+     * Useful for manual recovery or testing.
+     *
+     * @param providerId - Provider to reset
+     */
+    resetCircuit(providerId) {
+        const provider = this.providers.get(providerId);
+        if (!provider) {
+            throw new Error(`Provider ${providerId} not registered`);
+        }
+        const previousState = provider.state.circuitState;
+        provider.state.consecutiveFailures = 0;
+        provider.circuitOpenedAt = undefined;
+        this.transitionCircuitState(provider, 'closed');
+        this.logger.info(`Circuit reset for provider ${providerId}`, {
+            previousState
+        });
+    }
+    /**
+     * Update provider state based on health check result
+     */
+    updateProviderState(provider, result) {
+        const prevHealthy = provider.state.healthy;
+        provider.state.healthy = result.healthy;
+        provider.state.latency = result.latency;
+        provider.state.lastCheck = result.timestamp;
+        provider.state.lastError = result.error;
+        provider.state.checkCount++;
+        if (result.healthy) {
+            provider.state.successCount++;
+        }
+        // Calculate error rate (last 100 checks)
+        const recentWindow = Math.min(provider.state.checkCount, 100);
+        const recentSuccesses = Math.min(provider.state.successCount, recentWindow);
+        provider.state.errorRate = 1 - (recentSuccesses / recentWindow);
+        // Calculate availability (all-time)
+        provider.state.availability = provider.state.successCount / provider.state.checkCount;
+        // Emit health change event if status changed
+        if (prevHealthy !== result.healthy) {
+            this.emit('health-change', {
+                providerId: provider.providerId,
+                healthy: result.healthy,
+                previousHealthy: prevHealthy,
+                errorRate: provider.state.errorRate,
+                availability: provider.state.availability,
+                latency: result.latency,
+                timestamp: result.timestamp
+            });
+        }
+    }
+    /**
+     * Handle successful health check
+     */
+    handleSuccessfulCheck(provider) {
+        // Reset consecutive failures on success
+        provider.state.consecutiveFailures = 0;
+        // If circuit was half-open, close it
+        if (provider.state.circuitState === 'half-open') {
+            this.transitionCircuitState(provider, 'closed');
+            provider.circuitOpenedAt = undefined;
+            this.logger.info(`Circuit closed for provider ${provider.providerId} after successful recovery`);
+        }
+    }
+    /**
+     * Handle failed health check
+     */
+    handleFailedCheck(provider, error) {
+        provider.state.consecutiveFailures++;
+        this.logger.warn(`Health check failed for provider ${provider.providerId}`, {
+            consecutiveFailures: provider.state.consecutiveFailures,
+            failureThreshold: this.config.failureThreshold,
+            error
+        });
+        // Open circuit if failure threshold exceeded
+        if (provider.state.consecutiveFailures >= this.config.failureThreshold &&
+            provider.state.circuitState !== 'open') {
+            this.transitionCircuitState(provider, 'open');
+            provider.circuitOpenedAt = new Date();
+            this.logger.error(`Circuit opened for provider ${provider.providerId}`, {
+                consecutiveFailures: provider.state.consecutiveFailures,
+                failureThreshold: this.config.failureThreshold
+            });
+        }
+    }
+    /**
+     * Transition circuit breaker state
+     */
+    transitionCircuitState(provider, newState) {
+        const prevState = provider.state.circuitState;
+        provider.state.circuitState = newState;
+        // Emit circuit change event
+        this.emit('circuit-change', {
+            providerId: provider.providerId,
+            circuitState: newState,
+            previousState: prevState,
+            consecutiveFailures: provider.state.consecutiveFailures,
+            timestamp: new Date()
+        });
+    }
+    /**
+     * Execute a promise with timeout
+     */
+    async withTimeout(promise, timeoutMs, errorMessage) {
+        let timeoutHandle;
+        const timeoutPromise = new Promise((_, reject) => {
+            timeoutHandle = setTimeout(() => {
+                reject(new Error(errorMessage));
+            }, timeoutMs);
+        });
+        try {
+            return await Promise.race([promise, timeoutPromise]);
+        }
+        finally {
+            clearTimeout(timeoutHandle);
+        }
+    }
+}
+exports.ProviderHealthMonitor = ProviderHealthMonitor;
+//# sourceMappingURL=ProviderHealthMonitor.js.map

package/dist/monitoring/ProviderHealthMonitor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderHealthMonitor.js","sourceRoot":"","sources":["../../src/monitoring/ProviderHealthMonitor.ts"],"names":[],"mappings":";AAAA;;;;;;;;GAQG;;;AAEH,mCAAsC;AACtC,4CAAyC;AAyDzC;;GAEG;AACH,MAAM,cAAc,GAAyB;IAC3C,eAAe,EAAE,KAAK;IACtB,SAAS,EAAE,IAAI;IACf,gBAAgB,EAAE,CAAC;IACnB,cAAc,EAAE,KAAK;IACrB,yBAAyB,EAAE,IAAI;CAChC,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAa,qBAAsB,SAAQ,qBAAY;IAOrD,YAAY,MAAsC;QAChD,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,eAAM,CAAC,WAAW,EAAE,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAC/C,IAAI,CAAC,SAAS,GAAG,IAAI,GAAG,EAAE,CAAC;QAC3B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAE1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,CAAC;IAED;;;;;OAKG;IACH,gBAAgB,CACd,UAAkB,EAClB,aAA6C;QAE7C,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,UAAU,gCAAgC,CAAC,CAAC;QAC3E,CAAC;QAED,MAAM,KAAK,GAAwB;YACjC,UAAU;YACV,OAAO,EAAE,IAAI;YACb,OAAO,EAAE,CAAC;YACV,SAAS,EAAE,CAAC;YACZ,YAAY,EAAE,GAAG;YACjB,mBAAmB,EAAE,CAAC;YACtB,YAAY,EAAE,QAAQ;YACtB,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,UAAU,EAAE,CAAC;YACb,YAAY,EAAE,CAAC;SAChB,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE;YAC7B,UAAU;YACV,aAAa;YACb,KAAK;SACN,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,UAAU,mCAAmC,CAAC,CAAC;IAC9E,CAAC;IAED;;;;OAIG;IACH,kBAAkB,CAAC,UAAkB;QACnC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;YAC1D,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,UAAU,sCAAsC,CAAC,CAAC;IACjF,CAAC;IAED;;;;OAIG;IACH,eAAe;QACb,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;YACtD,OAAO;QACT,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAEzB,mCAAmC;QACnC,IAAI,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;YACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,2BAA2B;QAC3B,IAAI,CAAC,kBAAkB,GAAG,WAAW,CAAC,GAAG,EAAE;YACzC,IAAI,CAAC,iBAAiB,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;gBACrC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,8BAA8B,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAC9E,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAEhC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE;YAC5C,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,eAAe;YACrC,SAAS,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;SAC7C,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAClD,OAAO;QACT,CAAC;QAED,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,aAAa,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YACvC,IAAI,CAAC,kBAAkB,GAAG,SAAS,CAAC;QACtC,CAAC;QAED,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;IAChD,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,mBAAmB,CAAC,UAAkB;QAC1C,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC7B,IAAI,MAAyB,CAAC;QAE9B,IAAI,CAAC;YACH,wDAAwD;YACxD,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBAC3C,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,QAAQ,CAAC,eAAe,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC9E,IAAI,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;oBAChD,oCAAoC;oBACpC,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;gBACrD,CAAC;qBAAM,CAAC;oBACN,gCAAgC;oBAChC,MAAM,GAAG;wBACP,UAAU;wBACV,OAAO,EAAE,KAAK;wBACd,OAAO,EAAE,CAAC;wBACV,SAAS,EAAE,IAAI,IAAI,EAAE;wBACrB,KAAK,EAAE,yBAAyB;qBACjC,CAAC;oBACF,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;oBAC3C,OAAO,MAAM,CAAC;gBAChB,CAAC;YACH,CAAC;YAED,oCAAoC;YACpC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,WAAW,CACzC,QAAQ,CAAC,aAAa,EAAE,EACxB,IAAI,CAAC,MAAM,CAAC,SAAS,EACrB,4BAA4B,UAAU,EAAE,CACzC,CAAC;YAEF,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAEvC,gDAAgD;YAChD,MAAM,OAAO,GAAG,YAAY,CAAC,OAAO;gBAClC,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;YAElD,MAAM,GAAG;gBACP,UAAU;gBACV,OAAO;gBACP,OAAO;gBACP,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,KAAK,EAAE,YAAY,CAAC,KAAK;aAC1B,CAAC;YAEF,gDAAgD;YAChD,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAE3C,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;YACjD,CAAC;YAED,OAAO,MAAM,CAAC;QAEhB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACvC,MAAM,YAAY,GAAI,KAAe,CAAC,OAAO,CAAC;YAE9C,MAAM,GAAG;gBACP,UAAU;gBACV,OAAO,EAAE,KAAK;gBACd,OAAO;gBACP,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,KAAK,EAAE,YAAY;aACpB,CAAC;YAEF,IAAI,CAAC,mBAAmB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;YAE/C,OAAO,MAAM,CAAC;QAChB,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,iBAAiB;QACrB,IAAI,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,0CAA0C,CAAC,CAAC;YAC7D,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QAEtD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC,CACpD,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE;YAC5D,KAAK,EAAE,OAAO,CAAC,MAAM;YACrB,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM;YAC9C,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM;SAClD,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;;OAKG;IACH,iBAAiB,CAAC,UAAkB;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,OAAO,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IACtD,CAAC;IAED;;;;OAIG;IACH,oBAAoB;QAClB,MAAM,SAAS,GAAG,IAAI,GAAG,EAA+B,CAAC;QAEzD,KAAK,MAAM,CAAC,EAAE,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,EAAE,CAAC;YACtD,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;;;OAIG;IACH,mBAAmB;QACjB,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;aACxC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,EAAE,EAAE,CACxB,QAAQ,CAAC,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,KAAK,QAAQ,CACnE;aACA,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1B,CAAC;IAED;;;;;OAKG;IACH,iBAAiB,CAAC,UAAkB;QAClC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,QAAQ,CAAC,KAAK,CAAC,OAAO,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,KAAK,QAAQ,CAAC;IAC5E,CAAC;IAED;;;;;OAKG;IACH,eAAe,CAAC,UAAkB;QAChC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,OAAO,QAAQ,EAAE,KAAK,CAAC,YAAY,IAAI,QAAQ,CAAC;IAClD,CAAC;IAED;;;;;;OAMG;IACH,gBAAgB,CAAC,UAAkB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAClD,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAC9C,QAAQ,CAAC,eAAe,GAAG,IAAI,IAAI,EAAE,CAAC;QAEtC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,6CAA6C,UAAU,EAAE,EAAE;YAC1E,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAED;;;;;;OAMG;IACH,YAAY,CAAC,UAAkB;QAC7B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAClD,QAAQ,CAAC,KAAK,CAAC,mBAAmB,GAAG,CAAC,CAAC;QACvC,QAAQ,CAAC,eAAe,GAAG,SAAS,CAAC;QACrC,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAEhD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,8BAA8B,UAAU,EAAE,EAAE;YAC3D,aAAa;SACd,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,mBAAmB,CACzB,QAA4B,EAC5B,MAAyB;QAEzB,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,OAAO,CAAC;QAE3C,QAAQ,CAAC,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACxC,QAAQ,CAAC,KAAK,CAAC,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACxC,QAAQ,CAAC,KAAK,CAAC,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAC5C,QAAQ,CAAC,KAAK,CAAC,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC;QACxC,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;QAE5B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;QAChC,CAAC;QAED,yCAAyC;QACzC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAC9D,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;QAC5E,QAAQ,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,GAAG,CAAC,eAAe,GAAG,YAAY,CAAC,CAAC;QAEhE,oCAAoC;QACpC,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC;QAEtF,6CAA6C;QAC7C,IAAI,WAAW,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE;gBACzB,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,eAAe,EAAE,WAAW;gBAC5B,SAAS,EAAE,QAAQ,CAAC,KAAK,CAAC,SAAS;gBACnC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY;gBACzC,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,qBAAqB,CAAC,QAA4B;QACxD,wCAAwC;QACxC,QAAQ,CAAC,KAAK,CAAC,mBAAmB,GAAG,CAAC,CAAC;QAEvC,qCAAqC;QACrC,IAAI,QAAQ,CAAC,KAAK,CAAC,YAAY,KAAK,WAAW,EAAE,CAAC;YAChD,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAChD,QAAQ,CAAC,eAAe,GAAG,SAAS,CAAC;YACrC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,QAAQ,CAAC,UAAU,4BAA4B,CAAC,CAAC;QACnG,CAAC;IACH,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,QAA4B,EAAE,KAAc;QACpE,QAAQ,CAAC,KAAK,CAAC,mBAAmB,EAAE,CAAC;QAErC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,oCAAoC,QAAQ,CAAC,UAAU,EAAE,EAAE;YAC1E,mBAAmB,EAAE,QAAQ,CAAC,KAAK,CAAC,mBAAmB;YACvD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;YAC9C,KAAK;SACN,CAAC,CAAC;QAEH,6CAA6C;QAC7C,IACE,QAAQ,CAAC,KAAK,CAAC,mBAAmB,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB;YAClE,QAAQ,CAAC,KAAK,CAAC,YAAY,KAAK,MAAM,EACtC,CAAC;YACD,IAAI,CAAC,sBAAsB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC9C,QAAQ,CAAC,eAAe,GAAG,IAAI,IAAI,EAAE,CAAC;YAEtC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAA+B,QAAQ,CAAC,UAAU,EAAE,EAAE;gBACtE,mBAAmB,EAAE,QAAQ,CAAC,KAAK,CAAC,mBAAmB;gBACvD,gBAAgB,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;aAC/C,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACK,sBAAsB,CAC5B,QAA4B,EAC5B,QAAyC;QAEzC,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAC9C,QAAQ,CAAC,KAAK,CAAC,YAAY,GAAG,QAAQ,CAAC;QAEvC,4BAA4B;QAC5B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAC1B,UAAU,EAAE,QAAQ,CAAC,UAAU;YAC/B,YAAY,EAAE,QAAQ;YACtB,aAAa,EAAE,SAAS;YACxB,mBAAmB,EAAE,QAAQ,CAAC,KAAK,CAAC,mBAAmB;YACvD,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CACvB,OAAmB,EACnB,SAAiB,EACjB,YAAoB;QAEpB,IAAI,aAA6B,CAAC;QAElC,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YACtD,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC9B,MAAM,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;YAClC,CAAC,EAAE,SAAS,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,OAAO,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,cAAc,CAAC,CAAC,CAAC;QACvD,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,aAAc,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;CACF;AAjdD,sDAidC"}

package/dist/monitoring/QuotaManager.d.ts

@@ -0,0 +1,122 @@
+import { EventEmitter } from 'events';
+/**
+ * Quota configuration for a specific LLM provider
+ */
+export interface ProviderQuota {
+    providerId: string;
+    dailyLimit: number;
+    minuteLimit?: number;
+    resetTimeUtc: string;
+    warningThresholds: number[];
+}
+/**
+ * Current quota status for a provider
+ */
+export interface QuotaStatus {
+    providerId: string;
+    dailyUsed: number;
+    dailyRemaining: number;
+    dailyLimit: number;
+    minuteUsed: number;
+    minuteRemaining: number;
+    minuteLimit: number;
+    percentageUsed: number;
+    nextResetTime: Date;
+    isExhausted: boolean;
+    warningLevel: 'none' | 'warning' | 'critical' | 'exhausted';
+}
+/**
+ * Configuration for QuotaManager
+ */
+export interface QuotaManagerConfig {
+    providers: ProviderQuota[];
+    enforcementMode: 'warn' | 'block' | 'none';
+    alertCallback?: (status: QuotaStatus) => void;
+    persistState?: boolean;
+}
+/**
+ * Manages quota tracking and enforcement across LLM providers
+ */
+export declare class QuotaManager extends EventEmitter {
+    private config;
+    private quotas;
+    private usage;
+    private resetTimers;
+    private cleanupInterval?;
+    constructor(config: QuotaManagerConfig);
+    /**
+     * Register a new provider with quota limits
+     */
+    registerProvider(quota: ProviderQuota): void;
+    /**
+     * Update quota configuration for a provider
+     */
+    updateQuota(providerId: string, updates: Partial<ProviderQuota>): void;
+    /**
+     * Record a single request for a provider
+     */
+    recordRequest(providerId: string): void;
+    /**
+     * Record multiple requests for a provider
+     */
+    recordRequests(providerId: string, count: number): void;
+    /**
+     * Check if a request can be made without exceeding quota
+     */
+    canMakeRequest(providerId: string): boolean;
+    /**
+     * Get remaining quota for a provider
+     */
+    getRemainingQuota(providerId: string): {
+        daily: number;
+        minute: number;
+    };
+    /**
+     * Get current quota status for a provider
+     */
+    getQuotaStatus(providerId: string): QuotaStatus | undefined;
+    /**
+     * Get quota status for all providers
+     */
+    getAllQuotaStatus(): Map<string, QuotaStatus>;
+    /**
+     * Reset daily quota for a provider
+     */
+    resetDailyQuota(providerId: string): void;
+    /**
+     * Reset daily quotas for all providers
+     */
+    resetAllDailyQuotas(): void;
+    /**
+     * Start automatic quota reset scheduling
+     */
+    startAutoReset(): void;
+    /**
+     * Stop automatic quota reset scheduling
+     */
+    stopAutoReset(): void;
+    /**
+     * Stop cleanup interval (alias for use in CLI commands)
+     */
+    stopCleanup(): void;
+    /**
+     * Save current state for persistence
+     */
+    saveState(): void;
+    /**
+     * Load persisted state
+     */
+    loadState(): void;
+    private scheduleNextReset;
+    private getMinuteUsage;
+    private cleanupOldMinuteEntries;
+    private cleanupMinuteWindows;
+    private determineWarningLevel;
+    private checkThresholds;
+    private calculateNextResetTime;
+}
+/**
+ * Factory function to create QuotaManager with common provider presets
+ */
+export declare function createQuotaManager(config?: Partial<QuotaManagerConfig>): QuotaManager;
+//# sourceMappingURL=QuotaManager.d.ts.map

package/dist/monitoring/QuotaManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"QuotaManager.d.ts","sourceRoot":"","sources":["../../src/monitoring/QuotaManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,QAAQ,CAAC;AAEtC;;GAEG;AACH,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,iBAAiB,EAAE,MAAM,EAAE,CAAC;CAC7B;AAED;;GAEG;AACH,MAAM,WAAW,WAAW;IAC1B,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,MAAM,CAAC;IACxB,WAAW,EAAE,MAAM,CAAC;IACpB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,IAAI,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,YAAY,EAAE,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,WAAW,CAAC;CAC7D;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,SAAS,EAAE,aAAa,EAAE,CAAC;IAC3B,eAAe,EAAE,MAAM,GAAG,OAAO,GAAG,MAAM,CAAC;IAC3C,aAAa,CAAC,EAAE,CAAC,MAAM,EAAE,WAAW,KAAK,IAAI,CAAC;IAC9C,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AASD;;GAEG;AACH,qBAAa,YAAa,SAAQ,YAAY;IAC5C,OAAO,CAAC,MAAM,CAAqB;IACnC,OAAO,CAAC,MAAM,CAA6B;IAC3C,OAAO,CAAC,KAAK,CAA2B;IACxC,OAAO,CAAC,WAAW,CAA8B;IACjD,OAAO,CAAC,eAAe,CAAC,CAAiB;gBAE7B,MAAM,EAAE,kBAAkB;IAmBtC;;OAEG;IACH,gBAAgB,CAAC,KAAK,EAAE,aAAa,GAAG,IAAI;IAa5C;;OAEG;IACH,WAAW,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,aAAa,CAAC,GAAG,IAAI;IAUtE;;OAEG;IACH,aAAa,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAIvC;;OAEG;IACH,cAAc,CAAC,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAiCvD;;OAEG;IACH,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;IAe3C;;OAEG;IACH,iBAAiB,CAAC,UAAU,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE;IAYxE;;OAEG;IACH,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAmC3D;;OAEG;IACH,iBAAiB,IAAI,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC;IAa7C;;OAEG;IACH,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAoBzC;;OAEG;IACH,mBAAmB,IAAI,IAAI;IAM3B;;OAEG;IACH,cAAc,IAAI,IAAI;IAgBtB;;OAEG;IACH,aAAa,IAAI,IAAI;IAYrB;;OAEG;IACH,WAAW,IAAI,IAAI;IAInB;;OAEG;IACH,SAAS,IAAI,IAAI;IAgBjB;;OAEG;IACH,SAAS,IAAI,IAAI;IAKjB,OAAO,CAAC,iBAAiB;IAezB,OAAO,CAAC,cAAc;IAUtB,OAAO,CAAC,uBAAuB;IAU/B,OAAO,CAAC,oBAAoB;IAM5B,OAAO,CAAC,qBAAqB;IAU7B,OAAO,CAAC,eAAe;IAqBvB,OAAO,CAAC,sBAAsB;CAoB/B;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,MAAM,GAAE,OAAO,CAAC,kBAAkB,CAAM,GACvC,YAAY,CAqCd"}