agentic-qe 2.6.4 → 2.6.5

package/dist/monitoring/QuotaManager.js

@@ -0,0 +1,351 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QuotaManager = void 0;
+exports.createQuotaManager = createQuotaManager;
+const events_1 = require("events");
+/**
+ * Manages quota tracking and enforcement across LLM providers
+ */
+class QuotaManager extends events_1.EventEmitter {
+    constructor(config) {
+        super();
+        this.config = config;
+        this.quotas = new Map();
+        this.usage = new Map();
+        this.resetTimers = new Map();
+        // Register all providers
+        config.providers.forEach(provider => this.registerProvider(provider));
+        // Load persisted state if enabled
+        if (config.persistState) {
+            this.loadState();
+        }
+        // Start cleanup interval for minute windows
+        this.cleanupInterval = setInterval(() => this.cleanupMinuteWindows(), 60000);
+    }
+    /**
+     * Register a new provider with quota limits
+     */
+    registerProvider(quota) {
+        this.quotas.set(quota.providerId, quota);
+        if (!this.usage.has(quota.providerId)) {
+            this.usage.set(quota.providerId, {
+                dailyCount: 0,
+                minuteWindow: [],
+                lastResetTime: new Date(),
+                triggeredThresholds: new Set()
+            });
+        }
+    }
+    /**
+     * Update quota configuration for a provider
+     */
+    updateQuota(providerId, updates) {
+        const existing = this.quotas.get(providerId);
+        if (!existing) {
+            throw new Error(`Provider ${providerId} not registered`);
+        }
+        const updated = { ...existing, ...updates };
+        this.quotas.set(providerId, updated);
+    }
+    /**
+     * Record a single request for a provider
+     */
+    recordRequest(providerId) {
+        this.recordRequests(providerId, 1);
+    }
+    /**
+     * Record multiple requests for a provider
+     */
+    recordRequests(providerId, count) {
+        const quota = this.quotas.get(providerId);
+        const record = this.usage.get(providerId);
+        if (!quota || !record) {
+            throw new Error(`Provider ${providerId} not registered`);
+        }
+        const now = Date.now();
+        // Update daily count
+        record.dailyCount += count;
+        // Update minute window
+        record.minuteWindow.push({ timestamp: now, count });
+        this.cleanupOldMinuteEntries(providerId);
+        // Check quota status and emit events
+        const status = this.getQuotaStatus(providerId);
+        if (status) {
+            this.checkThresholds(status, record);
+            if (this.config.alertCallback) {
+                this.config.alertCallback(status);
+            }
+        }
+        // Persist state if enabled
+        if (this.config.persistState) {
+            this.saveState();
+        }
+    }
+    /**
+     * Check if a request can be made without exceeding quota
+     */
+    canMakeRequest(providerId) {
+        const status = this.getQuotaStatus(providerId);
+        if (!status) {
+            return false;
+        }
+        const canMake = !status.isExhausted;
+        if (!canMake && this.config.enforcementMode === 'block') {
+            return false;
+        }
+        return canMake;
+    }
+    /**
+     * Get remaining quota for a provider
+     */
+    getRemainingQuota(providerId) {
+        const status = this.getQuotaStatus(providerId);
+        if (!status) {
+            return { daily: 0, minute: 0 };
+        }
+        return {
+            daily: status.dailyRemaining,
+            minute: status.minuteRemaining
+        };
+    }
+    /**
+     * Get current quota status for a provider
+     */
+    getQuotaStatus(providerId) {
+        const quota = this.quotas.get(providerId);
+        const record = this.usage.get(providerId);
+        if (!quota || !record) {
+            return undefined;
+        }
+        const minuteUsed = this.getMinuteUsage(providerId);
+        const minuteLimit = quota.minuteLimit || Infinity;
+        const percentageUsed = quota.dailyLimit === Infinity
+            ? 0
+            : (record.dailyCount / quota.dailyLimit) * 100;
+        const isExhausted = (quota.dailyLimit !== Infinity && record.dailyCount >= quota.dailyLimit) ||
+            (quota.minuteLimit !== undefined && minuteUsed >= quota.minuteLimit);
+        const warningLevel = this.determineWarningLevel(percentageUsed, isExhausted);
+        return {
+            providerId,
+            dailyUsed: record.dailyCount,
+            dailyRemaining: Math.max(0, quota.dailyLimit - record.dailyCount),
+            dailyLimit: quota.dailyLimit,
+            minuteUsed,
+            minuteRemaining: Math.max(0, minuteLimit - minuteUsed),
+            minuteLimit,
+            percentageUsed,
+            nextResetTime: this.calculateNextResetTime(quota.resetTimeUtc),
+            isExhausted,
+            warningLevel
+        };
+    }
+    /**
+     * Get quota status for all providers
+     */
+    getAllQuotaStatus() {
+        const statusMap = new Map();
+        for (const providerId of Array.from(this.quotas.keys())) {
+            const status = this.getQuotaStatus(providerId);
+            if (status) {
+                statusMap.set(providerId, status);
+            }
+        }
+        return statusMap;
+    }
+    /**
+     * Reset daily quota for a provider
+     */
+    resetDailyQuota(providerId) {
+        const record = this.usage.get(providerId);
+        if (!record) {
+            throw new Error(`Provider ${providerId} not registered`);
+        }
+        record.dailyCount = 0;
+        record.lastResetTime = new Date();
+        record.triggeredThresholds.clear();
+        const status = this.getQuotaStatus(providerId);
+        if (status) {
+            this.emit('quota-reset', status);
+        }
+        if (this.config.persistState) {
+            this.saveState();
+        }
+    }
+    /**
+     * Reset daily quotas for all providers
+     */
+    resetAllDailyQuotas() {
+        for (const providerId of Array.from(this.quotas.keys())) {
+            this.resetDailyQuota(providerId);
+        }
+    }
+    /**
+     * Start automatic quota reset scheduling
+     */
+    startAutoReset() {
+        // Schedule reset for each provider
+        for (const [providerId, quota] of Array.from(this.quotas.entries())) {
+            const nextReset = this.calculateNextResetTime(quota.resetTimeUtc);
+            const timeUntilReset = nextReset.getTime() - Date.now();
+            const timer = setTimeout(() => {
+                this.resetDailyQuota(providerId);
+                // Schedule next reset
+                this.scheduleNextReset(providerId);
+            }, timeUntilReset);
+            this.resetTimers.set(providerId, timer);
+        }
+    }
+    /**
+     * Stop automatic quota reset scheduling
+     */
+    stopAutoReset() {
+        for (const timer of Array.from(this.resetTimers.values())) {
+            clearTimeout(timer);
+        }
+        this.resetTimers.clear();
+        if (this.cleanupInterval) {
+            clearInterval(this.cleanupInterval);
+            this.cleanupInterval = undefined;
+        }
+    }
+    /**
+     * Stop cleanup interval (alias for use in CLI commands)
+     */
+    stopCleanup() {
+        this.stopAutoReset();
+    }
+    /**
+     * Save current state for persistence
+     */
+    saveState() {
+        // This is a placeholder - implementation would depend on storage mechanism
+        // Could save to file, database, or memory cache
+        const state = {
+            usage: Array.from(this.usage.entries()).map(([id, record]) => ({
+                providerId: id,
+                dailyCount: record.dailyCount,
+                lastResetTime: record.lastResetTime.toISOString(),
+                triggeredThresholds: Array.from(record.triggeredThresholds)
+            }))
+        };
+        // Store state (implementation specific)
+        // For now, this is a no-op
+    }
+    /**
+     * Load persisted state
+     */
+    loadState() {
+        // This is a placeholder - implementation would depend on storage mechanism
+        // Would restore usage counts and reset times from storage
+    }
+    scheduleNextReset(providerId) {
+        const quota = this.quotas.get(providerId);
+        if (!quota)
+            return;
+        const nextReset = this.calculateNextResetTime(quota.resetTimeUtc);
+        const timeUntilReset = nextReset.getTime() - Date.now();
+        const timer = setTimeout(() => {
+            this.resetDailyQuota(providerId);
+            this.scheduleNextReset(providerId);
+        }, timeUntilReset);
+        this.resetTimers.set(providerId, timer);
+    }
+    getMinuteUsage(providerId) {
+        const record = this.usage.get(providerId);
+        if (!record)
+            return 0;
+        const oneMinuteAgo = Date.now() - 60000;
+        return record.minuteWindow
+            .filter(entry => entry.timestamp > oneMinuteAgo)
+            .reduce((sum, entry) => sum + entry.count, 0);
+    }
+    cleanupOldMinuteEntries(providerId) {
+        const record = this.usage.get(providerId);
+        if (!record)
+            return;
+        const oneMinuteAgo = Date.now() - 60000;
+        record.minuteWindow = record.minuteWindow.filter(entry => entry.timestamp > oneMinuteAgo);
+    }
+    cleanupMinuteWindows() {
+        for (const providerId of Array.from(this.usage.keys())) {
+            this.cleanupOldMinuteEntries(providerId);
+        }
+    }
+    determineWarningLevel(percentageUsed, isExhausted) {
+        if (isExhausted)
+            return 'exhausted';
+        if (percentageUsed >= 90)
+            return 'critical';
+        if (percentageUsed >= 80)
+            return 'warning';
+        return 'none';
+    }
+    checkThresholds(status, record) {
+        const quota = this.quotas.get(status.providerId);
+        if (!quota)
+            return;
+        // Check if we've crossed any new thresholds
+        for (const threshold of quota.warningThresholds) {
+            if (status.percentageUsed >= threshold &&
+                !record.triggeredThresholds.has(threshold)) {
+                record.triggeredThresholds.add(threshold);
+                this.emit('quota-warning', status);
+            }
+        }
+        // Check for exhaustion
+        if (status.isExhausted && status.warningLevel === 'exhausted') {
+            this.emit('quota-exhausted', status);
+        }
+    }
+    calculateNextResetTime(resetTimeUtc) {
+        const [hours, minutes] = resetTimeUtc.split(':').map(Number);
+        const now = new Date();
+        const reset = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate(), hours, minutes, 0, 0));
+        // If reset time has passed today, schedule for tomorrow
+        if (reset.getTime() <= now.getTime()) {
+            reset.setUTCDate(reset.getUTCDate() + 1);
+        }
+        return reset;
+    }
+}
+exports.QuotaManager = QuotaManager;
+/**
+ * Factory function to create QuotaManager with common provider presets
+ */
+function createQuotaManager(config = {}) {
+    const defaultProviders = [
+        {
+            providerId: 'groq',
+            dailyLimit: 14400,
+            minuteLimit: 10,
+            resetTimeUtc: '00:00',
+            warningThresholds: [80, 90]
+        },
+        {
+            providerId: 'openrouter',
+            dailyLimit: 50,
+            resetTimeUtc: '00:00',
+            warningThresholds: [80, 90]
+        },
+        {
+            providerId: 'github-models',
+            dailyLimit: Infinity,
+            resetTimeUtc: '00:00',
+            warningThresholds: []
+        },
+        {
+            providerId: 'ollama',
+            dailyLimit: Infinity,
+            resetTimeUtc: '00:00',
+            warningThresholds: []
+        }
+    ];
+    const fullConfig = {
+        providers: defaultProviders,
+        enforcementMode: 'warn',
+        persistState: false,
+        ...config
+    };
+    return new QuotaManager(fullConfig);
+}
+//# sourceMappingURL=QuotaManager.js.map

package/dist/monitoring/QuotaManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QuotaManager.js","sourceRoot":"","sources":["../../src/monitoring/QuotaManager.ts"],"names":[],"mappings":";;;AAgbA,gDAuCC;AAvdD,mCAAsC;AA+CtC;;GAEG;AACH,MAAa,YAAa,SAAQ,qBAAY;IAO5C,YAAY,MAA0B;QACpC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;QACxB,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,WAAW,GAAG,IAAI,GAAG,EAAE,CAAC;QAE7B,yBAAyB;QACzB,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEtE,kCAAkC;QAClC,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;YACxB,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,CAAC;QAED,4CAA4C;QAC5C,IAAI,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,EAAE,KAAK,CAAC,CAAC;IAC/E,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,KAAoB;QACnC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAEzC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC;YACtC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,EAAE;gBAC/B,UAAU,EAAE,CAAC;gBACb,YAAY,EAAE,EAAE;gBAChB,aAAa,EAAE,IAAI,IAAI,EAAE;gBACzB,mBAAmB,EAAE,IAAI,GAAG,EAAE;aAC/B,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,UAAkB,EAAE,OAA+B;QAC7D,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC7C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,OAAO,GAAG,EAAE,GAAG,QAAQ,EAAE,GAAG,OAAO,EAAE,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,UAAkB;QAC9B,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,UAAkB,EAAE,KAAa;QAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE1C,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;YACtB,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,qBAAqB;QACrB,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC;QAE3B,uBAAuB;QACvB,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;QAEzC,qCAAqC;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YAErC,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;gBAC9B,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,2BAA2B;QAC3B,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,UAAkB;QAC/B,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,CAAC,WAAW,CAAC;QAEpC,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,eAAe,KAAK,OAAO,EAAE,CAAC;YACxD,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,UAAkB;QAClC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;QACjC,CAAC;QAED,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,cAAc;YAC5B,MAAM,EAAE,MAAM,CAAC,eAAe;SAC/B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,UAAkB;QAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAE1C,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,EAAE,CAAC;YACtB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,QAAQ,CAAC;QAClD,MAAM,cAAc,GAAG,KAAK,CAAC,UAAU,KAAK,QAAQ;YAClD,CAAC,CAAC,CAAC;YACH,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC,GAAG,GAAG,CAAC;QAEjD,MAAM,WAAW,GACf,CAAC,KAAK,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC;YACxE,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,IAAI,UAAU,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;QAEvE,MAAM,YAAY,GAAG,IAAI,CAAC,qBAAqB,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QAE7E,OAAO;YACL,UAAU;YACV,SAAS,EAAE,MAAM,CAAC,UAAU;YAC5B,cAAc,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;YACjE,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,UAAU;YACV,eAAe,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,UAAU,CAAC;YACtD,WAAW;YACX,cAAc;YACd,aAAa,EAAE,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,YAAY,CAAC;YAC9D,WAAW;YACX,YAAY;SACb,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,MAAM,SAAS,GAAG,IAAI,GAAG,EAAuB,CAAC;QAEjD,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;YAC/C,IAAI,MAAM,EAAE,CAAC;gBACX,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,UAAkB;QAChC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,YAAY,UAAU,iBAAiB,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC;QACtB,MAAM,CAAC,aAAa,GAAG,IAAI,IAAI,EAAE,CAAC;QAClC,MAAM,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;QAEnC,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC/C,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,CAAC;QACnC,CAAC;QAED,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAC7B,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACxD,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,mCAAmC;QACnC,KAAK,MAAM,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;YACpE,MAAM,SAAS,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAClE,MAAM,cAAc,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAExD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBAC5B,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;gBACjC,sBAAsB;gBACtB,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACrC,CAAC,EAAE,cAAc,CAAC,CAAC;YAEnB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa;QACX,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC;YAC1D,YAAY,CAAC,KAAK,CAAC,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QAEzB,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACpC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACnC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,WAAW;QACT,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,SAAS;QACP,2EAA2E;QAC3E,gDAAgD;QAChD,MAAM,KAAK,GAAG;YACZ,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC7D,UAAU,EAAE,EAAE;gBACd,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,aAAa,EAAE,MAAM,CAAC,aAAa,CAAC,WAAW,EAAE;gBACjD,mBAAmB,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC;aAC5D,CAAC,CAAC;SACJ,CAAC;QAEF,wCAAwC;QACxC,2BAA2B;IAC7B,CAAC;IAED;;OAEG;IACH,SAAS;QACP,2EAA2E;QAC3E,0DAA0D;IAC5D,CAAC;IAEO,iBAAiB,CAAC,UAAkB;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,SAAS,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAClE,MAAM,cAAc,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAExD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;YAC5B,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;YACjC,IAAI,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QACrC,CAAC,EAAE,cAAc,CAAC,CAAC;QAEnB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IAC1C,CAAC;IAEO,cAAc,CAAC,UAAkB;QACvC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC;QAEtB,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;QACxC,OAAO,MAAM,CAAC,YAAY;aACvB,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,YAAY,CAAC;aAC/C,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAClD,CAAC;IAEO,uBAAuB,CAAC,UAAkB;QAChD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC,MAAM;YAAE,OAAO;QAEpB,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;QACxC,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC,MAAM,CAC9C,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,YAAY,CACxC,CAAC;IACJ,CAAC;IAEO,oBAAoB;QAC1B,KAAK,MAAM,UAAU,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC;YACvD,IAAI,CAAC,uBAAuB,CAAC,UAAU,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAEO,qBAAqB,CAC3B,cAAsB,EACtB,WAAoB;QAEpB,IAAI,WAAW;YAAE,OAAO,WAAW,CAAC;QACpC,IAAI,cAAc,IAAI,EAAE;YAAE,OAAO,UAAU,CAAC;QAC5C,IAAI,cAAc,IAAI,EAAE;YAAE,OAAO,SAAS,CAAC;QAC3C,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,eAAe,CAAC,MAAmB,EAAE,MAAmB;QAC9D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACjD,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,4CAA4C;QAC5C,KAAK,MAAM,SAAS,IAAI,KAAK,CAAC,iBAAiB,EAAE,CAAC;YAChD,IACE,MAAM,CAAC,cAAc,IAAI,SAAS;gBAClC,CAAC,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,EAC1C,CAAC;gBACD,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;gBAC1C,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,IAAI,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,YAAY,KAAK,WAAW,EAAE,CAAC;YAC9D,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,YAAoB;QACjD,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAC7B,GAAG,CAAC,cAAc,EAAE,EACpB,GAAG,CAAC,WAAW,EAAE,EACjB,GAAG,CAAC,UAAU,EAAE,EAChB,KAAK,EACL,OAAO,EACP,CAAC,EACD,CAAC,CACF,CAAC,CAAC;QAEH,wDAAwD;QACxD,IAAI,KAAK,CAAC,OAAO,EAAE,IAAI,GAAG,CAAC,OAAO,EAAE,EAAE,CAAC;YACrC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC,CAAC;QAC3C,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAzXD,oCAyXC;AAED;;GAEG;AACH,SAAgB,kBAAkB,CAChC,SAAsC,EAAE;IAExC,MAAM,gBAAgB,GAAoB;QACxC;YACE,UAAU,EAAE,MAAM;YAClB,UAAU,EAAE,KAAK;YACjB,WAAW,EAAE,EAAE;YACf,YAAY,EAAE,OAAO;YACrB,iBAAiB,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;SAC5B;QACD;YACE,UAAU,EAAE,YAAY;YACxB,UAAU,EAAE,EAAE;YACd,YAAY,EAAE,OAAO;YACrB,iBAAiB,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC;SAC5B;QACD;YACE,UAAU,EAAE,eAAe;YAC3B,UAAU,EAAE,QAAQ;YACpB,YAAY,EAAE,OAAO;YACrB,iBAAiB,EAAE,EAAE;SACtB;QACD;YACE,UAAU,EAAE,QAAQ;YACpB,UAAU,EAAE,QAAQ;YACpB,YAAY,EAAE,OAAO;YACrB,iBAAiB,EAAE,EAAE;SACtB;KACF,CAAC;IAEF,MAAM,UAAU,GAAuB;QACrC,SAAS,EAAE,gBAAgB;QAC3B,eAAe,EAAE,MAAM;QACvB,YAAY,EAAE,KAAK;QACnB,GAAG,MAAM;KACV,CAAC;IAEF,OAAO,IAAI,YAAY,CAAC,UAAU,CAAC,CAAC;AACtC,CAAC"}

package/dist/providers/GitHubModelsProvider.d.ts

@@ -0,0 +1,117 @@
+/**
+ * GitHubModelsProvider - GitHub Models API Integration
+ *
+ * Provides LLM capabilities through GitHub Models with support for:
+ * - Unlimited free usage in GitHub Codespaces
+ * - OpenAI-compatible API format
+ * - Multiple model families (GPT-4, Phi, LLaMA)
+ * - Streaming responses
+ * - Automatic Codespaces detection
+ *
+ * @module providers/GitHubModelsProvider
+ * @version 1.0.0
+ */
+import { ILLMProvider, LLMProviderConfig, LLMCompletionOptions, LLMCompletionResponse, LLMStreamEvent, LLMEmbeddingOptions, LLMEmbeddingResponse, LLMTokenCountOptions, LLMHealthStatus, LLMProviderMetadata } from './ILLMProvider';
+/**
+ * GitHub Models-specific configuration
+ */
+export interface GitHubModelsProviderConfig extends LLMProviderConfig {
+    /** GitHub token (defaults to process.env.GITHUB_TOKEN) */
+    token?: string;
+    /** Base URL for GitHub Models API */
+    baseUrl?: string;
+    /** Default model name */
+    defaultModel?: string;
+    /** Force Codespaces mode (auto-detected if not set) */
+    inCodespaces?: boolean;
+}
+/**
+ * GitHubModelsProvider - GitHub Models API implementation
+ *
+ * This provider enables access to GitHub Models with unlimited free usage
+ * in GitHub Codespaces. Supports OpenAI-compatible API format.
+ */
+export declare class GitHubModelsProvider implements ILLMProvider {
+    private readonly logger;
+    private config;
+    private isInitialized;
+    private baseUrl;
+    private token;
+    private inCodespaces;
+    private requestCount;
+    private currentModel;
+    constructor(config?: GitHubModelsProviderConfig);
+    /**
+     * Check if running in GitHub Codespaces
+     */
+    isInCodespaces(): boolean;
+    /**
+     * Initialize the GitHub Models provider
+     */
+    initialize(): Promise<void>;
+    /**
+     * Complete a prompt using GitHub Models
+     */
+    complete(options: LLMCompletionOptions): Promise<LLMCompletionResponse>;
+    /**
+     * Stream a completion using GitHub Models
+     */
+    streamComplete(options: LLMCompletionOptions): AsyncIterableIterator<LLMStreamEvent>;
+    /**
+     * Generate embeddings (not supported by GitHub Models)
+     */
+    embed(options: LLMEmbeddingOptions): Promise<LLMEmbeddingResponse>;
+    /**
+     * Count tokens in text (approximate)
+     */
+    countTokens(options: LLMTokenCountOptions): Promise<number>;
+    /**
+     * Health check - verify API access
+     */
+    healthCheck(): Promise<LLMHealthStatus>;
+    /**
+     * Get provider metadata
+     */
+    getMetadata(): LLMProviderMetadata;
+    /**
+     * Shutdown the provider
+     */
+    shutdown(): Promise<void>;
+    /**
+     * Track cost for a request
+     */
+    trackCost(usage: LLMCompletionResponse['usage']): number;
+    /**
+     * Set the current model for subsequent requests
+     */
+    setModel(model: string): Promise<void>;
+    /**
+     * Get current model
+     */
+    getCurrentModel(): string;
+    /**
+     * Get list of available models
+     */
+    getAvailableModels(): string[];
+    /**
+     * Validate GitHub token
+     */
+    private validateToken;
+    /**
+     * Build messages array from options
+     */
+    private buildMessages;
+    /**
+     * Map OpenAI finish reason to Claude format
+     */
+    private mapFinishReason;
+    /**
+     * Estimate token count (approximate: 1 token ≈ 4 characters)
+     */
+    private estimateTokens;
+    /**
+     * Ensure provider is initialized
+     */
+    private ensureInitialized;
+}
+//# sourceMappingURL=GitHubModelsProvider.d.ts.map

package/dist/providers/GitHubModelsProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"GitHubModelsProvider.d.ts","sourceRoot":"","sources":["../../src/providers/GitHubModelsProvider.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAGH,OAAO,EACL,YAAY,EACZ,iBAAiB,EACjB,oBAAoB,EACpB,qBAAqB,EACrB,cAAc,EACd,mBAAmB,EACnB,oBAAoB,EACpB,oBAAoB,EACpB,eAAe,EACf,mBAAmB,EAEpB,MAAM,gBAAgB,CAAC;AAGxB;;GAEG;AACH,MAAM,WAAW,0BAA2B,SAAQ,iBAAiB;IACnE,0DAA0D;IAC1D,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qCAAqC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yBAAyB;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,uDAAuD;IACvD,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AA6ED;;;;;GAKG;AACH,qBAAa,oBAAqB,YAAW,YAAY;IACvD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,OAAO,CAAC,MAAM,CAA6B;IAC3C,OAAO,CAAC,aAAa,CAAU;IAC/B,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAqB;IAClC,OAAO,CAAC,YAAY,CAAU;IAC9B,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,YAAY,CAAS;gBAEjB,MAAM,GAAE,0BAA+B;IAoBnD;;OAEG;IACH,cAAc,IAAI,OAAO;IAIzB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAgDjC;;OAEG;IACG,QAAQ,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAsE7E;;OAEG;IACI,cAAc,CAAC,OAAO,EAAE,oBAAoB,GAAG,qBAAqB,CAAC,cAAc,CAAC;IA4F3F;;OAEG;IACG,KAAK,CAAC,OAAO,EAAE,mBAAmB,GAAG,OAAO,CAAC,oBAAoB,CAAC;IASxE;;OAEG;IACG,WAAW,CAAC,OAAO,EAAE,oBAAoB,GAAG,OAAO,CAAC,MAAM,CAAC;IAIjE;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,eAAe,CAAC;IAqD7C;;OAEG;IACH,WAAW,IAAI,mBAAmB;IAmBlC;;OAEG;IACG,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAO/B;;OAEG;IACH,SAAS,CAAC,KAAK,EAAE,qBAAqB,CAAC,OAAO,CAAC,GAAG,MAAM;IAkBxD;;OAEG;IACG,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAW5C;;OAEG;IACH,eAAe,IAAI,MAAM;IAIzB;;OAEG;IACH,kBAAkB,IAAI,MAAM,EAAE;IAI9B;;OAEG;YACW,aAAa;IAgD3B;;OAEG;IACH,OAAO,CAAC,aAAa;IA2BrB;;OAEG;IACH,OAAO,CAAC,eAAe;IAWvB;;OAEG;IACH,OAAO,CAAC,cAAc;IAItB;;OAEG;IACH,OAAO,CAAC,iBAAiB;CAU1B"}