attocode 0.2.4 → 0.2.6

package/dist/src/integrations/budget/resources.js

@@ -0,0 +1,362 @@
+/**
+ * Resource Monitor Integration
+ *
+ * Provides lightweight resource usage tracking and limits for agent operations.
+ * Integrates with the economics system to check both token budget AND system resources.
+ * Adapted from tricks/resource-monitor.ts.
+ *
+ * Usage:
+ *   const resources = createResourceManager({ maxMemoryMB: 512 });
+ *   const check = resources.check();
+ *   if (check.status === 'exceeded') { ... }
+ */
+// =============================================================================
+// RESOURCE MANAGER
+// =============================================================================
+/**
+ * Manages resource monitoring and limits for the agent.
+ */
+export class ResourceManager {
+    config;
+    startTime;
+    concurrentOps = 0;
+    previousStatus = 'healthy';
+    eventListeners = new Set();
+    constructor(config = {}) {
+        this.config = {
+            enabled: config.enabled ?? true,
+            maxMemoryMB: config.maxMemoryMB ?? 512,
+            maxCpuTimeSec: config.maxCpuTimeSec ?? 300, // 5 minutes
+            maxConcurrentOps: config.maxConcurrentOps ?? 10,
+            warnThreshold: config.warnThreshold ?? 0.7,
+            criticalThreshold: config.criticalThreshold ?? 0.9,
+        };
+        this.startTime = Date.now();
+    }
+    /**
+     * Get current resource usage.
+     */
+    getUsage() {
+        const mem = process.memoryUsage();
+        const memoryMB = mem.heapUsed / 1024 / 1024;
+        const cpuTimeSec = (Date.now() - this.startTime) / 1000;
+        return {
+            memoryMB,
+            memoryPercent: memoryMB / this.config.maxMemoryMB,
+            cpuTimeSec,
+            cpuPercent: cpuTimeSec / this.config.maxCpuTimeSec,
+            concurrentOps: this.concurrentOps,
+            opsPercent: this.concurrentOps / this.config.maxConcurrentOps,
+            timestamp: new Date(),
+        };
+    }
+    /**
+     * Check resource status and determine if operations can continue.
+     */
+    check() {
+        if (!this.config.enabled) {
+            return {
+                status: 'healthy',
+                usage: this.getUsage(),
+                canContinue: true,
+                recommendation: 'continue',
+            };
+        }
+        const usage = this.getUsage();
+        let status = 'healthy';
+        let message;
+        let canContinue = true;
+        let recommendation = 'continue';
+        // Check memory
+        if (usage.memoryPercent >= 1) {
+            status = 'exceeded';
+            message = `Memory limit exceeded (${usage.memoryMB.toFixed(1)}MB / ${this.config.maxMemoryMB}MB)`;
+            canContinue = false;
+            recommendation = 'stop';
+            this.emit({
+                type: 'resource.exceeded',
+                metric: 'memory',
+                value: usage.memoryMB,
+                limit: this.config.maxMemoryMB,
+            });
+        }
+        else if (usage.memoryPercent >= this.config.criticalThreshold) {
+            status = 'critical';
+            message = `Memory critically high (${(usage.memoryPercent * 100).toFixed(1)}%)`;
+            recommendation = 'slow_down';
+            this.emit({
+                type: 'resource.critical',
+                metric: 'memory',
+                value: usage.memoryPercent,
+                threshold: this.config.criticalThreshold,
+            });
+        }
+        else if (usage.memoryPercent >= this.config.warnThreshold) {
+            status = 'warning';
+            message = `Memory usage elevated (${(usage.memoryPercent * 100).toFixed(1)}%)`;
+            this.emit({
+                type: 'resource.warning',
+                metric: 'memory',
+                value: usage.memoryPercent,
+                threshold: this.config.warnThreshold,
+            });
+        }
+        // Check CPU time (only override if worse)
+        if (usage.cpuPercent >= 1 && status !== 'exceeded') {
+            status = 'exceeded';
+            message = `CPU time limit exceeded (${usage.cpuTimeSec.toFixed(1)}s / ${this.config.maxCpuTimeSec}s)`;
+            canContinue = false;
+            recommendation = 'stop';
+            this.emit({
+                type: 'resource.exceeded',
+                metric: 'cpuTime',
+                value: usage.cpuTimeSec,
+                limit: this.config.maxCpuTimeSec,
+            });
+        }
+        else if (usage.cpuPercent >= this.config.criticalThreshold && status === 'healthy') {
+            status = 'critical';
+            message = `Approaching CPU time limit (${(usage.cpuPercent * 100).toFixed(1)}%)`;
+            recommendation = 'slow_down';
+            this.emit({
+                type: 'resource.critical',
+                metric: 'cpuTime',
+                value: usage.cpuPercent,
+                threshold: this.config.criticalThreshold,
+            });
+        }
+        else if (usage.cpuPercent >= this.config.warnThreshold && status === 'healthy') {
+            status = 'warning';
+            message = `CPU time usage elevated (${(usage.cpuPercent * 100).toFixed(1)}%)`;
+            this.emit({
+                type: 'resource.warning',
+                metric: 'cpuTime',
+                value: usage.cpuPercent,
+                threshold: this.config.warnThreshold,
+            });
+        }
+        // Check concurrent operations
+        if (usage.opsPercent >= 1 && status !== 'exceeded') {
+            status = 'exceeded';
+            message = `Max concurrent operations reached (${usage.concurrentOps}/${this.config.maxConcurrentOps})`;
+            canContinue = false;
+            recommendation = 'stop';
+            this.emit({
+                type: 'resource.exceeded',
+                metric: 'concurrentOps',
+                value: usage.concurrentOps,
+                limit: this.config.maxConcurrentOps,
+            });
+        }
+        // Check for recovery
+        if (this.previousStatus !== 'healthy' && status === 'healthy') {
+            this.emit({
+                type: 'resource.recovered',
+                metric: 'overall',
+                previousStatus: this.previousStatus,
+            });
+        }
+        this.previousStatus = status;
+        return { status, usage, canContinue, message, recommendation };
+    }
+    /**
+     * Check if we can start a new operation.
+     */
+    canStartOperation() {
+        if (!this.config.enabled)
+            return true;
+        const check = this.check();
+        return check.canContinue && this.concurrentOps < this.config.maxConcurrentOps;
+    }
+    /**
+     * Track the start of an operation.
+     * Returns a function to call when the operation completes.
+     */
+    startOperation() {
+        this.concurrentOps++;
+        return () => {
+            this.concurrentOps = Math.max(0, this.concurrentOps - 1);
+        };
+    }
+    /**
+     * Run a function with operation tracking.
+     * Throws ResourceLimitError if resources exceeded.
+     */
+    async runTracked(fn) {
+        const check = this.check();
+        if (!check.canContinue) {
+            throw new ResourceLimitError(check.message || 'Resource limit exceeded');
+        }
+        const end = this.startOperation();
+        try {
+            return await fn();
+        }
+        finally {
+            end();
+        }
+    }
+    /**
+     * Run only if resources are available.
+     * Returns fallback value if resources exceeded.
+     */
+    async runIfAvailable(fn, fallback) {
+        if (!this.canStartOperation()) {
+            return fallback;
+        }
+        const check = this.check();
+        if (!check.canContinue || check.status === 'critical') {
+            return fallback;
+        }
+        return this.runTracked(fn);
+    }
+    /**
+     * Update resource limits.
+     */
+    setLimits(limits) {
+        Object.assign(this.config, limits);
+    }
+    /**
+     * Get current limits.
+     */
+    getLimits() {
+        return { ...this.config };
+    }
+    /**
+     * Reset timing and operation count.
+     */
+    reset() {
+        this.startTime = Date.now();
+        this.concurrentOps = 0;
+        this.previousStatus = 'healthy';
+    }
+    /**
+     * Reset only the CPU time counter (per-prompt reset).
+     * This allows long sessions with multiple prompts without hitting the wall-clock limit.
+     */
+    resetCpuTime() {
+        this.startTime = Date.now();
+    }
+    /**
+     * Get formatted status string.
+     */
+    getStatusString() {
+        const check = this.check();
+        const usage = check.usage;
+        return [
+            `Memory: ${usage.memoryMB.toFixed(1)}MB / ${this.config.maxMemoryMB}MB (${(usage.memoryPercent * 100).toFixed(1)}%)`,
+            `CPU Time: ${usage.cpuTimeSec.toFixed(1)}s / ${this.config.maxCpuTimeSec}s (${(usage.cpuPercent * 100).toFixed(1)}%)`,
+            `Operations: ${usage.concurrentOps} / ${this.config.maxConcurrentOps}`,
+            `Status: ${check.status.toUpperCase()}`,
+        ].join('\n');
+    }
+    /**
+     * Subscribe to resource events.
+     */
+    subscribe(listener) {
+        this.eventListeners.add(listener);
+        return () => this.eventListeners.delete(listener);
+    }
+    /**
+     * Emit a resource event.
+     */
+    emit(event) {
+        for (const listener of this.eventListeners) {
+            try {
+                listener(event);
+            }
+            catch {
+                // Ignore listener errors
+            }
+        }
+    }
+    /**
+     * Cleanup resources.
+     */
+    cleanup() {
+        this.eventListeners.clear();
+    }
+}
+// =============================================================================
+// ERROR
+// =============================================================================
+/**
+ * Error thrown when resource limits are exceeded.
+ */
+export class ResourceLimitError extends Error {
+    isResourceLimit = true;
+    constructor(message) {
+        super(message);
+        this.name = 'ResourceLimitError';
+    }
+}
+/**
+ * Check if an error is a resource limit error.
+ */
+export function isResourceLimitError(error) {
+    return (error instanceof ResourceLimitError ||
+        (error instanceof Error &&
+            'isResourceLimit' in error &&
+            error.isResourceLimit === true));
+}
+// =============================================================================
+// FACTORY FUNCTIONS
+// =============================================================================
+/**
+ * Create a resource manager with default limits.
+ */
+export function createResourceManager(config) {
+    return new ResourceManager(config);
+}
+/**
+ * Create a strict resource manager (lower limits).
+ */
+export function createStrictResourceManager() {
+    return new ResourceManager({
+        enabled: true,
+        maxMemoryMB: 128,
+        maxCpuTimeSec: 60,
+        maxConcurrentOps: 5,
+        warnThreshold: 0.5,
+        criticalThreshold: 0.8,
+    });
+}
+/**
+ * Create a lenient resource manager (higher limits).
+ */
+export function createLenientResourceManager() {
+    return new ResourceManager({
+        enabled: true,
+        maxMemoryMB: 1024,
+        maxCpuTimeSec: 1800, // 30 minutes for complex tasks
+        maxConcurrentOps: 50,
+        warnThreshold: 0.8,
+        criticalThreshold: 0.95,
+    });
+}
+// =============================================================================
+// INTEGRATION HELPERS
+// =============================================================================
+/**
+ * Combine resource check with economics budget check.
+ * Returns true only if BOTH resource AND budget checks pass.
+ */
+export function combinedShouldContinue(resourceManager, economicsShouldContinue) {
+    // If no resource manager, just use economics
+    if (!resourceManager) {
+        return {
+            canContinue: economicsShouldContinue,
+            reason: economicsShouldContinue ? undefined : 'Budget limit',
+        };
+    }
+    // Check resources
+    const resourceCheck = resourceManager.check();
+    // Both must pass
+    if (!economicsShouldContinue) {
+        return { canContinue: false, reason: 'Budget limit' };
+    }
+    if (!resourceCheck.canContinue) {
+        return { canContinue: false, reason: resourceCheck.message || 'Resource limit' };
+    }
+    return { canContinue: true };
+}
+//# sourceMappingURL=resources.js.map

package/dist/src/integrations/budget/resources.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"resources.js","sourceRoot":"","sources":["../../../../src/integrations/budget/resources.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAgEH,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF;;GAEG;AACH,MAAM,OAAO,eAAe;IAClB,MAAM,CAAiC;IACvC,SAAS,CAAS;IAClB,aAAa,GAAW,CAAC,CAAC;IAC1B,cAAc,GAAmB,SAAS,CAAC;IAC3C,cAAc,GAA+B,IAAI,GAAG,EAAE,CAAC;IAE/D,YAAY,SAA+B,EAAE;QAC3C,IAAI,CAAC,MAAM,GAAG;YACZ,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,IAAI;YAC/B,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,GAAG;YACtC,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,GAAG,EAAE,YAAY;YACxD,gBAAgB,EAAE,MAAM,CAAC,gBAAgB,IAAI,EAAE;YAC/C,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,GAAG;YAC1C,iBAAiB,EAAE,MAAM,CAAC,iBAAiB,IAAI,GAAG;SACnD,CAAC;QACF,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,MAAM,GAAG,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,GAAG,CAAC,QAAQ,GAAG,IAAI,GAAG,IAAI,CAAC;QAC5C,MAAM,UAAU,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;QAExD,OAAO;YACL,QAAQ;YACR,aAAa,EAAE,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;YACjD,UAAU;YACV,UAAU,EAAE,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,aAAa;YAClD,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,UAAU,EAAE,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;YAC7D,SAAS,EAAE,IAAI,IAAI,EAAE;SACtB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACzB,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,IAAI,CAAC,QAAQ,EAAE;gBACtB,WAAW,EAAE,IAAI;gBACjB,cAAc,EAAE,UAAU;aAC3B,CAAC;QACJ,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC9B,IAAI,MAAM,GAAmB,SAAS,CAAC;QACvC,IAAI,OAA2B,CAAC;QAChC,IAAI,WAAW,GAAG,IAAI,CAAC;QACvB,IAAI,cAAc,GAAsC,UAAU,CAAC;QAEnE,eAAe;QACf,IAAI,KAAK,CAAC,aAAa,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,GAAG,UAAU,CAAC;YACpB,OAAO,GAAG,0BAA0B,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,WAAW,KAAK,CAAC;YAClG,WAAW,GAAG,KAAK,CAAC;YACpB,cAAc,GAAG,MAAM,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,mBAAmB;gBACzB,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,QAAQ;gBACrB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;aAC/B,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,EAAE,CAAC;YAChE,MAAM,GAAG,UAAU,CAAC;YACpB,OAAO,GAAG,2BAA2B,CAAC,KAAK,CAAC,aAAa,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;YAChF,cAAc,GAAG,WAAW,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,mBAAmB;gBACzB,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB;aACzC,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,CAAC;YAC5D,MAAM,GAAG,SAAS,CAAC;YACnB,OAAO,GAAG,0BAA0B,CAAC,KAAK,CAAC,aAAa,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/E,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,kBAAkB;gBACxB,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,KAAK,CAAC,aAAa;gBAC1B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;aACrC,CAAC,CAAC;QACL,CAAC;QAED,0CAA0C;QAC1C,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;YACnD,MAAM,GAAG,UAAU,CAAC;YACpB,OAAO,GAAG,4BAA4B,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC;YACtG,WAAW,GAAG,KAAK,CAAC;YACpB,cAAc,GAAG,MAAM,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,mBAAmB;gBACzB,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,KAAK,CAAC,UAAU;gBACvB,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;aACjC,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACrF,MAAM,GAAG,UAAU,CAAC;YACpB,OAAO,GAAG,+BAA+B,CAAC,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;YACjF,cAAc,GAAG,WAAW,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,mBAAmB;gBACzB,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,KAAK,CAAC,UAAU;gBACvB,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,iBAAiB;aACzC,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,KAAK,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YACjF,MAAM,GAAG,SAAS,CAAC;YACnB,OAAO,GAAG,4BAA4B,CAAC,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;YAC9E,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,kBAAkB;gBACxB,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,KAAK,CAAC,UAAU;gBACvB,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;aACrC,CAAC,CAAC;QACL,CAAC;QAED,8BAA8B;QAC9B,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;YACnD,MAAM,GAAG,UAAU,CAAC;YACpB,OAAO,GAAG,sCAAsC,KAAK,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,GAAG,CAAC;YACvG,WAAW,GAAG,KAAK,CAAC;YACpB,cAAc,GAAG,MAAM,CAAC;YACxB,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,mBAAmB;gBACzB,MAAM,EAAE,eAAe;gBACvB,KAAK,EAAE,KAAK,CAAC,aAAa;gBAC1B,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;aACpC,CAAC,CAAC;QACL,CAAC;QAED,qBAAqB;QACrB,IAAI,IAAI,CAAC,cAAc,KAAK,SAAS,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;YAC9D,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,oBAAoB;gBAC1B,MAAM,EAAE,SAAS;gBACjB,cAAc,EAAE,IAAI,CAAC,cAAc;aACpC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC;QAE7B,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;IACjE,CAAC;IAED;;OAEG;IACH,iBAAiB;QACf,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,IAAI,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;IAChF,CAAC;IAED;;;OAGG;IACH,cAAc;QACZ,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,OAAO,GAAG,EAAE;YACV,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,UAAU,CAAI,EAAoB;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YACvB,MAAM,IAAI,kBAAkB,CAAC,KAAK,CAAC,OAAO,IAAI,yBAAyB,CAAC,CAAC;QAC3E,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;QAClC,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,GAAG,EAAE,CAAC;QACR,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,cAAc,CAAI,EAAoB,EAAE,QAAY;QACxD,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC;YAC9B,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACtD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,MAAqC;QAC7C,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACrC,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC5B,IAAI,CAAC,aAAa,GAAG,CAAC,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;IAClC,CAAC;IAED;;;OAGG;IACH,YAAY;QACV,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC9B,CAAC;IAED;;OAEG;IACH,eAAe;QACb,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAE1B,OAAO;YACL,WAAW,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,IAAI,CAAC,MAAM,CAAC,WAAW,OAAO,CAAC,KAAK,CAAC,aAAa,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;YACpH,aAAa,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,MAAM,CAAC,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI;YACrH,eAAe,KAAK,CAAC,aAAa,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE;YACtE,WAAW,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE;SACxC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAA+B;QACvC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACK,IAAI,CAAC,KAAoB;QAC/B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YAC3C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;CACF;AAED,gFAAgF;AAChF,QAAQ;AACR,gFAAgF;AAEhF;;GAEG;AACH,MAAM,OAAO,kBAAmB,SAAQ,KAAK;IAClC,eAAe,GAAG,IAAI,CAAC;IAEhC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;IACnC,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,KAAc;IACjD,OAAO,CACL,KAAK,YAAY,kBAAkB;QACnC,CAAC,KAAK,YAAY,KAAK;YACrB,iBAAiB,IAAI,KAAK;YACzB,KAA4B,CAAC,eAAe,KAAK,IAAI,CAAC,CAC1D,CAAC;AACJ,CAAC;AAED,gFAAgF;AAChF,oBAAoB;AACpB,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAA6B;IACjE,OAAO,IAAI,eAAe,CAAC,MAAM,CAAC,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,2BAA2B;IACzC,OAAO,IAAI,eAAe,CAAC;QACzB,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,GAAG;QAChB,aAAa,EAAE,EAAE;QACjB,gBAAgB,EAAE,CAAC;QACnB,aAAa,EAAE,GAAG;QAClB,iBAAiB,EAAE,GAAG;KACvB,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,4BAA4B;IAC1C,OAAO,IAAI,eAAe,CAAC;QACzB,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,IAAI;QACjB,aAAa,EAAE,IAAI,EAAE,+BAA+B;QACpD,gBAAgB,EAAE,EAAE;QACpB,aAAa,EAAE,GAAG;QAClB,iBAAiB,EAAE,IAAI;KACxB,CAAC,CAAC;AACL,CAAC;AAED,gFAAgF;AAChF,sBAAsB;AACtB,gFAAgF;AAEhF;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CACpC,eAAuC,EACvC,uBAAgC;IAEhC,6CAA6C;IAC7C,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO;YACL,WAAW,EAAE,uBAAuB;YACpC,MAAM,EAAE,uBAAuB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;SAC7D,CAAC;IACJ,CAAC;IAED,kBAAkB;IAClB,MAAM,aAAa,GAAG,eAAe,CAAC,KAAK,EAAE,CAAC;IAE9C,iBAAiB;IACjB,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7B,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;IACxD,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;QAC/B,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,MAAM,EAAE,aAAa,CAAC,OAAO,IAAI,gBAAgB,EAAE,CAAC;IACnF,CAAC;IAED,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;AAC/B,CAAC"}

package/dist/src/integrations/context/auto-compaction.d.ts

@@ -0,0 +1,210 @@
+/**
+ * AutoCompactionManager - Monitors token usage and triggers compaction
+ *
+ * Implements a three-tier threshold system:
+ * - OK (< 80%): Continue normally
+ * - Warning (80-89%): Alert user, no automatic action
+ * - Auto-Compact (90-97%): Depending on mode, auto-compact or request approval
+ * - Hard Limit (>= 98%): Agent cannot continue - context too full
+ */
+import type { Message } from '../../types.js';
+import type { Compactor } from './compaction.js';
+import type { Reference } from '../../tricks/reversible-compaction.js';
+/**
+ * Result from a custom compaction handler (supports reversible compaction).
+ */
+export interface CustomCompactionResult {
+    /** The summarized content */
+    summary: string;
+    /** Token count before compaction */
+    tokensBefore: number;
+    /** Token count after compaction */
+    tokensAfter: number;
+    /** Messages to preserve in context */
+    preservedMessages: Message[];
+    /** Preserved references for retrieval (from reversible compaction) */
+    references?: Reference[];
+}
+/**
+ * Custom compaction handler function type.
+ * Allows integration with reversible compaction or other compaction strategies.
+ */
+export type CompactHandler = (messages: Message[]) => Promise<CustomCompactionResult>;
+/**
+ * Configuration for automatic compaction behavior.
+ */
+export interface AutoCompactionConfig {
+    /** Compaction mode: auto (no approval), approval (asks user), manual (never auto) */
+    mode: 'auto' | 'approval' | 'manual';
+    /** Warning threshold as ratio (default: 0.80) */
+    warningThreshold: number;
+    /** Auto-compact threshold as ratio (default: 0.90) */
+    autoCompactThreshold: number;
+    /** Hard limit threshold as ratio (default: 0.98) */
+    hardLimitThreshold: number;
+    /** Number of recent user messages to preserve (default: 5) */
+    preserveRecentUserMessages: number;
+    /** Number of recent assistant messages to preserve (default: 5) */
+    preserveRecentAssistantMessages: number;
+    /** Cooldown between compactions in ms (default: 60000) */
+    cooldownMs: number;
+    /** Max context window tokens (default: 200000 for Claude) */
+    maxContextTokens: number;
+    /** Optional custom compaction handler (e.g., for reversible compaction with reference preservation) */
+    compactHandler?: CompactHandler;
+}
+/**
+ * Result of checking token usage and potentially compacting.
+ */
+export interface CompactionCheckResult {
+    /** Current status after check */
+    status: 'ok' | 'warning' | 'compacted' | 'needs_approval' | 'hard_limit';
+    /** Current token count */
+    currentTokens: number;
+    /** Maximum allowed tokens */
+    maxTokens: number;
+    /** Ratio of current to max (0-1) */
+    ratio: number;
+    /** Compacted messages if compaction occurred */
+    compactedMessages?: Message[];
+    /** Summary generated during compaction */
+    summary?: string;
+    /** Preserved references from reversible compaction (for later retrieval) */
+    references?: Reference[];
+}
+/**
+ * Events emitted by AutoCompactionManager.
+ */
+export type AutoCompactionEvent = {
+    type: 'autocompaction.check';
+    currentTokens: number;
+    ratio: number;
+    threshold: string;
+} | {
+    type: 'autocompaction.warning';
+    currentTokens: number;
+    ratio: number;
+} | {
+    type: 'autocompaction.triggered';
+    mode: string;
+    currentTokens: number;
+} | {
+    type: 'autocompaction.completed';
+    tokensBefore: number;
+    tokensAfter: number;
+    reduction: number;
+} | {
+    type: 'autocompaction.cooldown';
+    remainingMs: number;
+} | {
+    type: 'autocompaction.hard_limit';
+    currentTokens: number;
+    ratio: number;
+} | {
+    type: 'autocompaction.needs_approval';
+    currentTokens: number;
+    ratio: number;
+} | {
+    type: 'autocompaction.emergency_truncate';
+    reason: string;
+    messagesBefore: number;
+    messagesAfter: number;
+};
+export type AutoCompactionEventListener = (event: AutoCompactionEvent) => void;
+/**
+ * Monitors token usage and triggers compaction based on thresholds.
+ *
+ * Three-tier threshold design:
+ * - OK (< 80%): Normal operation
+ * - Warning (80-89%): User alerted but no action
+ * - Auto-Compact (90-97%): Automatic compaction (in auto mode) or approval request
+ * - Hard Limit (>= 98%): Cannot continue - too close to context limit
+ */
+export declare class AutoCompactionManager {
+    private compactor;
+    private config;
+    private lastCompactionTime;
+    private listeners;
+    constructor(compactor: Compactor, config?: Partial<AutoCompactionConfig>);
+    /**
+     * Check token usage and maybe perform compaction.
+     *
+     * @param options - Current tokens and messages
+     * @returns Result indicating status and any compaction performed
+     */
+    checkAndMaybeCompact(options: {
+        currentTokens: number;
+        messages: Message[];
+    }): Promise<CompactionCheckResult>;
+    /**
+     * Get the current token ratio.
+     */
+    getTokenRatio(currentTokens: number): number;
+    /**
+     * Check if currently in cooldown period.
+     */
+    isInCooldown(): boolean;
+    /**
+     * Get remaining cooldown time in milliseconds.
+     */
+    getRemainingCooldown(): number;
+    /**
+     * Reset cooldown (e.g., after manual compaction request).
+     */
+    resetCooldown(): void;
+    /**
+     * Force compaction regardless of mode or cooldown.
+     * Useful for manual compaction requests.
+     */
+    forceCompact(messages: Message[]): Promise<CompactionCheckResult>;
+    /**
+     * Get current configuration.
+     */
+    getConfig(): AutoCompactionConfig;
+    /**
+     * Update configuration.
+     */
+    updateConfig(updates: Partial<AutoCompactionConfig>): void;
+    /**
+     * Subscribe to events.
+     */
+    on(listener: AutoCompactionEventListener): () => void;
+    /**
+     * Determine which zone the current ratio falls into.
+     */
+    private determineZone;
+    /**
+     * Perform the actual compaction.
+     */
+    private performCompaction;
+    /**
+     * Perform compaction using custom handler (supports reversible compaction with references).
+     */
+    private performCustomCompaction;
+    /**
+     * Perform compaction using default compactor.
+     */
+    private performDefaultCompaction;
+    /**
+     * Emergency truncation fallback when LLM-based compaction fails.
+     * Simply keeps system message and last N messages without summarization.
+     */
+    private emergencyTruncate;
+    /**
+     * Emit an event to all listeners.
+     */
+    private emit;
+}
+/**
+ * Create an AutoCompactionManager.
+ */
+export declare function createAutoCompactionManager(compactor: Compactor, config?: Partial<AutoCompactionConfig>): AutoCompactionManager;
+/**
+ * Format a CompactionCheckResult for display.
+ */
+export declare function formatCompactionCheckResult(result: CompactionCheckResult): string;
+/**
+ * Get suggested action based on check result.
+ */
+export declare function getSuggestedAction(result: CompactionCheckResult): string | null;
+//# sourceMappingURL=auto-compaction.d.ts.map

package/dist/src/integrations/context/auto-compaction.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-compaction.d.ts","sourceRoot":"","sources":["../../../../src/integrations/context/auto-compaction.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,KAAK,EAAE,SAAS,EAAoB,MAAM,iBAAiB,CAAC;AACnE,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,uCAAuC,CAAC;AAOvE;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,6BAA6B;IAC7B,OAAO,EAAE,MAAM,CAAC;IAChB,oCAAoC;IACpC,YAAY,EAAE,MAAM,CAAC;IACrB,mCAAmC;IACnC,WAAW,EAAE,MAAM,CAAC;IACpB,sCAAsC;IACtC,iBAAiB,EAAE,OAAO,EAAE,CAAC;IAC7B,sEAAsE;IACtE,UAAU,CAAC,EAAE,SAAS,EAAE,CAAC;CAC1B;AAED;;;GAGG;AACH,MAAM,MAAM,cAAc,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,sBAAsB,CAAC,CAAC;AAEtF;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,qFAAqF;IACrF,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,QAAQ,CAAC;IACrC,iDAAiD;IACjD,gBAAgB,EAAE,MAAM,CAAC;IACzB,sDAAsD;IACtD,oBAAoB,EAAE,MAAM,CAAC;IAC7B,oDAAoD;IACpD,kBAAkB,EAAE,MAAM,CAAC;IAC3B,8DAA8D;IAC9D,0BAA0B,EAAE,MAAM,CAAC;IACnC,mEAAmE;IACnE,+BAA+B,EAAE,MAAM,CAAC;IACxC,0DAA0D;IAC1D,UAAU,EAAE,MAAM,CAAC;IACnB,6DAA6D;IAC7D,gBAAgB,EAAE,MAAM,CAAC;IACzB,uGAAuG;IACvG,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,iCAAiC;IACjC,MAAM,EAAE,IAAI,GAAG,SAAS,GAAG,WAAW,GAAG,gBAAgB,GAAG,YAAY,CAAC;IACzE,0BAA0B;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,6BAA6B;IAC7B,SAAS,EAAE,MAAM,CAAC;IAClB,oCAAoC;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,gDAAgD;IAChD,iBAAiB,CAAC,EAAE,OAAO,EAAE,CAAC;IAC9B,0CAA0C;IAC1C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,4EAA4E;IAC5E,UAAU,CAAC,EAAE,SAAS,EAAE,CAAC;CAC1B;AAED;;GAEG;AACH,MAAM,MAAM,mBAAmB,GAC3B;IAAE,IAAI,EAAE,sBAAsB,CAAC;IAAC,aAAa,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACzF;IAAE,IAAI,EAAE,wBAAwB,CAAC;IAAC,aAAa,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GACxE;IAAE,IAAI,EAAE,0BAA0B,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,CAAA;CAAE,GACzE;IACE,IAAI,EAAE,0BAA0B,CAAC;IACjC,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,yBAAyB,CAAC;IAAC,WAAW,EAAE,MAAM,CAAA;CAAE,GACxD;IAAE,IAAI,EAAE,2BAA2B,CAAC;IAAC,aAAa,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAC3E;IAAE,IAAI,EAAE,+BAA+B,CAAC;IAAC,aAAa,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAC/E;IACE,IAAI,EAAE,mCAAmC,CAAC;IAC1C,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;CACvB,CAAC;AAEN,MAAM,MAAM,2BAA2B,GAAG,CAAC,KAAK,EAAE,mBAAmB,KAAK,IAAI,CAAC;AAqB/E;;;;;;;;GAQG;AACH,qBAAa,qBAAqB;IAChC,OAAO,CAAC,SAAS,CAAY;IAC7B,OAAO,CAAC,MAAM,CAAuB;IACrC,OAAO,CAAC,kBAAkB,CAAa;IACvC,OAAO,CAAC,SAAS,CAAqC;gBAE1C,SAAS,EAAE,SAAS,EAAE,MAAM,CAAC,EAAE,OAAO,CAAC,oBAAoB,CAAC;IAaxE;;;;;OAKG;IACG,oBAAoB,CAAC,OAAO,EAAE;QAClC,aAAa,EAAE,MAAM,CAAC;QACtB,QAAQ,EAAE,OAAO,EAAE,CAAC;KACrB,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAmHlC;;OAEG;IACH,aAAa,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM;IAI5C;;OAEG;IACH,YAAY,IAAI,OAAO;IAOvB;;OAEG;IACH,oBAAoB,IAAI,MAAM;IAO9B;;OAEG;IACH,aAAa,IAAI,IAAI;IAIrB;;;OAGG;IACG,YAAY,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,qBAAqB,CAAC;IAQvE;;OAEG;IACH,SAAS,IAAI,oBAAoB;IAIjC;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,oBAAoB,CAAC,GAAG,IAAI;IAY1D;;OAEG;IACH,EAAE,CAAC,QAAQ,EAAE,2BAA2B,GAAG,MAAM,IAAI;IAYrD;;OAEG;IACH,OAAO,CAAC,aAAa;IAarB;;OAEG;YACW,iBAAiB;IAqB/B;;OAEG;YACW,uBAAuB;IA4CrC;;OAEG;YACW,wBAAwB;IAuEtC;;;OAGG;IACH,OAAO,CAAC,iBAAiB;IAkDzB;;OAEG;IACH,OAAO,CAAC,IAAI;CASb;AAMD;;GAEG;AACH,wBAAgB,2BAA2B,CACzC,SAAS,EAAE,SAAS,EACpB,MAAM,CAAC,EAAE,OAAO,CAAC,oBAAoB,CAAC,GACrC,qBAAqB,CAEvB;AAMD;;GAEG;AACH,wBAAgB,2BAA2B,CAAC,MAAM,EAAE,qBAAqB,GAAG,MAAM,CAiDjF;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,qBAAqB,GAAG,MAAM,GAAG,IAAI,CAe/E"}