attocode 0.1.0

package/dist/src/integrations/auto-compaction.js

@@ -0,0 +1,428 @@
+/**
+ * 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
+ */
+// =============================================================================
+// DEFAULT CONFIG
+// =============================================================================
+const DEFAULT_CONFIG = {
+    mode: 'auto',
+    warningThreshold: 0.80,
+    autoCompactThreshold: 0.90,
+    hardLimitThreshold: 0.98,
+    preserveRecentUserMessages: 5,
+    preserveRecentAssistantMessages: 5,
+    cooldownMs: 60000, // 1 minute
+    maxContextTokens: 200000, // Claude's context window
+};
+// =============================================================================
+// AUTO COMPACTION MANAGER
+// =============================================================================
+/**
+ * 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 class AutoCompactionManager {
+    compactor;
+    config;
+    lastCompactionTime = 0;
+    listeners = [];
+    constructor(compactor, config) {
+        this.compactor = compactor;
+        this.config = { ...DEFAULT_CONFIG, ...config };
+        // Validate thresholds are in correct order
+        if (this.config.warningThreshold >= this.config.autoCompactThreshold) {
+            throw new Error('warningThreshold must be less than autoCompactThreshold');
+        }
+        if (this.config.autoCompactThreshold >= this.config.hardLimitThreshold) {
+            throw new Error('autoCompactThreshold must be less than hardLimitThreshold');
+        }
+    }
+    /**
+     * Check token usage and maybe perform compaction.
+     *
+     * @param options - Current tokens and messages
+     * @returns Result indicating status and any compaction performed
+     */
+    async checkAndMaybeCompact(options) {
+        const { currentTokens, messages } = options;
+        const ratio = this.getTokenRatio(currentTokens);
+        const maxTokens = this.config.maxContextTokens;
+        // Determine which zone we're in
+        const zone = this.determineZone(ratio);
+        this.emit({
+            type: 'autocompaction.check',
+            currentTokens,
+            ratio,
+            threshold: zone,
+        });
+        // Zone: OK (< warningThreshold)
+        if (zone === 'ok') {
+            return {
+                status: 'ok',
+                currentTokens,
+                maxTokens,
+                ratio,
+            };
+        }
+        // Zone: Hard Limit (>= hardLimitThreshold)
+        if (zone === 'hard_limit') {
+            this.emit({
+                type: 'autocompaction.hard_limit',
+                currentTokens,
+                ratio,
+            });
+            return {
+                status: 'hard_limit',
+                currentTokens,
+                maxTokens,
+                ratio,
+            };
+        }
+        // Zone: Warning (warningThreshold <= ratio < autoCompactThreshold)
+        if (zone === 'warning') {
+            this.emit({
+                type: 'autocompaction.warning',
+                currentTokens,
+                ratio,
+            });
+            return {
+                status: 'warning',
+                currentTokens,
+                maxTokens,
+                ratio,
+            };
+        }
+        // Zone: Auto-Compact (autoCompactThreshold <= ratio < hardLimitThreshold)
+        // Check cooldown
+        if (this.isInCooldown()) {
+            const remaining = this.getRemainingCooldown();
+            this.emit({
+                type: 'autocompaction.cooldown',
+                remainingMs: remaining,
+            });
+            // During cooldown, treat as warning
+            return {
+                status: 'warning',
+                currentTokens,
+                maxTokens,
+                ratio,
+            };
+        }
+        // Handle based on mode
+        switch (this.config.mode) {
+            case 'auto':
+                return this.performCompaction(messages, currentTokens, maxTokens, ratio);
+            case 'approval':
+                this.emit({
+                    type: 'autocompaction.needs_approval',
+                    currentTokens,
+                    ratio,
+                });
+                return {
+                    status: 'needs_approval',
+                    currentTokens,
+                    maxTokens,
+                    ratio,
+                };
+            case 'manual':
+                // In manual mode, auto-compact zone is treated as warning
+                this.emit({
+                    type: 'autocompaction.warning',
+                    currentTokens,
+                    ratio,
+                });
+                return {
+                    status: 'warning',
+                    currentTokens,
+                    maxTokens,
+                    ratio,
+                };
+            default:
+                return {
+                    status: 'warning',
+                    currentTokens,
+                    maxTokens,
+                    ratio,
+                };
+        }
+    }
+    /**
+     * Get the current token ratio.
+     */
+    getTokenRatio(currentTokens) {
+        return currentTokens / this.config.maxContextTokens;
+    }
+    /**
+     * Check if currently in cooldown period.
+     */
+    isInCooldown() {
+        if (this.lastCompactionTime === 0) {
+            return false;
+        }
+        return Date.now() - this.lastCompactionTime < this.config.cooldownMs;
+    }
+    /**
+     * Get remaining cooldown time in milliseconds.
+     */
+    getRemainingCooldown() {
+        if (!this.isInCooldown()) {
+            return 0;
+        }
+        return this.config.cooldownMs - (Date.now() - this.lastCompactionTime);
+    }
+    /**
+     * Reset cooldown (e.g., after manual compaction request).
+     */
+    resetCooldown() {
+        this.lastCompactionTime = 0;
+    }
+    /**
+     * Force compaction regardless of mode or cooldown.
+     * Useful for manual compaction requests.
+     */
+    async forceCompact(messages) {
+        const currentTokens = this.compactor.estimateTokens(messages);
+        const ratio = this.getTokenRatio(currentTokens);
+        const maxTokens = this.config.maxContextTokens;
+        return this.performCompaction(messages, currentTokens, maxTokens, ratio);
+    }
+    /**
+     * Get current configuration.
+     */
+    getConfig() {
+        return { ...this.config };
+    }
+    /**
+     * Update configuration.
+     */
+    updateConfig(updates) {
+        this.config = { ...this.config, ...updates };
+        // Re-validate thresholds
+        if (this.config.warningThreshold >= this.config.autoCompactThreshold) {
+            throw new Error('warningThreshold must be less than autoCompactThreshold');
+        }
+        if (this.config.autoCompactThreshold >= this.config.hardLimitThreshold) {
+            throw new Error('autoCompactThreshold must be less than hardLimitThreshold');
+        }
+    }
+    /**
+     * Subscribe to events.
+     */
+    on(listener) {
+        this.listeners.push(listener);
+        return () => {
+            const idx = this.listeners.indexOf(listener);
+            if (idx >= 0)
+                this.listeners.splice(idx, 1);
+        };
+    }
+    // ===========================================================================
+    // PRIVATE METHODS
+    // ===========================================================================
+    /**
+     * Determine which zone the current ratio falls into.
+     */
+    determineZone(ratio) {
+        if (ratio >= this.config.hardLimitThreshold) {
+            return 'hard_limit';
+        }
+        if (ratio >= this.config.autoCompactThreshold) {
+            return 'auto_compact';
+        }
+        if (ratio >= this.config.warningThreshold) {
+            return 'warning';
+        }
+        return 'ok';
+    }
+    /**
+     * Perform the actual compaction.
+     */
+    async performCompaction(messages, currentTokens, maxTokens, ratio) {
+        this.emit({
+            type: 'autocompaction.triggered',
+            mode: this.config.mode,
+            currentTokens,
+        });
+        // Configure compactor to preserve the specified number of messages
+        // The compactor already handles preserving recent messages, but we can
+        // update its config if needed based on our preservation settings
+        const totalPreserve = this.config.preserveRecentUserMessages + this.config.preserveRecentAssistantMessages;
+        const currentCompactorConfig = this.compactor.getConfig();
+        // Temporarily adjust if our preserve count differs
+        const needsRestore = currentCompactorConfig.preserveRecentCount !== totalPreserve;
+        if (needsRestore) {
+            this.compactor.updateConfig({ preserveRecentCount: totalPreserve });
+        }
+        let result;
+        try {
+            result = await this.compactor.compact(messages);
+        }
+        catch (compactionError) {
+            // LLM-based compaction failed - use emergency truncation as fallback
+            console.error('[AutoCompaction] LLM compaction failed, using emergency truncation:', compactionError);
+            const emergencyResult = this.emergencyTruncate(messages, totalPreserve);
+            // Restore original config
+            if (needsRestore) {
+                this.compactor.updateConfig({ preserveRecentCount: currentCompactorConfig.preserveRecentCount });
+            }
+            // Update cooldown even for emergency truncation
+            this.lastCompactionTime = Date.now();
+            return emergencyResult;
+        }
+        // Restore original config
+        if (needsRestore) {
+            this.compactor.updateConfig({ preserveRecentCount: currentCompactorConfig.preserveRecentCount });
+        }
+        // Update cooldown
+        this.lastCompactionTime = Date.now();
+        const reduction = result.tokensBefore > 0
+            ? Math.round((1 - result.tokensAfter / result.tokensBefore) * 100)
+            : 0;
+        this.emit({
+            type: 'autocompaction.completed',
+            tokensBefore: result.tokensBefore,
+            tokensAfter: result.tokensAfter,
+            reduction,
+        });
+        return {
+            status: 'compacted',
+            currentTokens: result.tokensAfter,
+            maxTokens,
+            ratio: result.tokensAfter / maxTokens,
+            compactedMessages: result.preservedMessages,
+            summary: result.summary,
+        };
+    }
+    /**
+     * Emergency truncation fallback when LLM-based compaction fails.
+     * Simply keeps system message and last N messages without summarization.
+     */
+    emergencyTruncate(messages, preserveCount) {
+        const messagesBefore = messages.length;
+        // Find and keep the system message
+        const systemMessage = messages.find(m => m.role === 'system');
+        // Get the last N messages (excluding system message from count)
+        const nonSystemMessages = messages.filter(m => m.role !== 'system');
+        const preservedMessages = nonSystemMessages.slice(-preserveCount);
+        // Build the truncated message array
+        const truncatedMessages = [];
+        if (systemMessage) {
+            truncatedMessages.push(systemMessage);
+        }
+        // Add a marker message about the truncation
+        truncatedMessages.push({
+            role: 'system',
+            content: `[CONTEXT TRUNCATED: ${messagesBefore - preservedMessages.length - (systemMessage ? 1 : 0)} messages were removed due to compaction failure. The conversation continues from the most recent ${preservedMessages.length} messages.]`,
+        });
+        truncatedMessages.push(...preservedMessages);
+        const messagesAfter = truncatedMessages.length;
+        const tokensAfter = this.compactor.estimateTokens(truncatedMessages);
+        this.emit({
+            type: 'autocompaction.emergency_truncate',
+            reason: 'LLM compaction failed',
+            messagesBefore,
+            messagesAfter,
+        });
+        return {
+            status: 'compacted',
+            currentTokens: tokensAfter,
+            maxTokens: this.config.maxContextTokens,
+            ratio: tokensAfter / this.config.maxContextTokens,
+            compactedMessages: truncatedMessages,
+            summary: '[Emergency truncation - no summary available]',
+        };
+    }
+    /**
+     * Emit an event to all listeners.
+     */
+    emit(event) {
+        for (const listener of this.listeners) {
+            try {
+                listener(event);
+            }
+            catch {
+                // Ignore listener errors
+            }
+        }
+    }
+}
+// =============================================================================
+// FACTORY
+// =============================================================================
+/**
+ * Create an AutoCompactionManager.
+ */
+export function createAutoCompactionManager(compactor, config) {
+    return new AutoCompactionManager(compactor, config);
+}
+// =============================================================================
+// UTILITIES
+// =============================================================================
+/**
+ * Format a CompactionCheckResult for display.
+ */
+export function formatCompactionCheckResult(result) {
+    const percentUsed = Math.round(result.ratio * 100);
+    const lines = [];
+    switch (result.status) {
+        case 'ok':
+            lines.push(`Context usage: ${percentUsed}% (${result.currentTokens.toLocaleString()} / ${result.maxTokens.toLocaleString()} tokens)`);
+            lines.push('Status: OK');
+            break;
+        case 'warning':
+            lines.push(`Context usage: ${percentUsed}% (${result.currentTokens.toLocaleString()} / ${result.maxTokens.toLocaleString()} tokens)`);
+            lines.push('Status: WARNING - Context usage is high');
+            lines.push('Consider compacting the conversation to free up space.');
+            break;
+        case 'needs_approval':
+            lines.push(`Context usage: ${percentUsed}% (${result.currentTokens.toLocaleString()} / ${result.maxTokens.toLocaleString()} tokens)`);
+            lines.push('Status: NEEDS APPROVAL');
+            lines.push('Context is nearing capacity. Approve compaction to continue.');
+            break;
+        case 'compacted':
+            lines.push(`Context usage: ${percentUsed}% (${result.currentTokens.toLocaleString()} / ${result.maxTokens.toLocaleString()} tokens)`);
+            lines.push('Status: COMPACTED');
+            if (result.summary) {
+                lines.push('Summary preserved: Yes');
+            }
+            break;
+        case 'hard_limit':
+            lines.push(`Context usage: ${percentUsed}% (${result.currentTokens.toLocaleString()} / ${result.maxTokens.toLocaleString()} tokens)`);
+            lines.push('Status: HARD LIMIT REACHED');
+            lines.push('Cannot continue - context window is too full.');
+            lines.push('Manual intervention required.');
+            break;
+    }
+    return lines.join('\n');
+}
+/**
+ * Get suggested action based on check result.
+ */
+export function getSuggestedAction(result) {
+    switch (result.status) {
+        case 'ok':
+            return null;
+        case 'warning':
+            return 'Consider running /compact to free up context space';
+        case 'needs_approval':
+            return 'Approve context compaction to continue';
+        case 'compacted':
+            return null;
+        case 'hard_limit':
+            return 'Start a new conversation or manually clear context';
+        default:
+            return null;
+    }
+}
+//# sourceMappingURL=auto-compaction.js.map

package/dist/src/integrations/auto-compaction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"auto-compaction.js","sourceRoot":"","sources":["../../../src/integrations/auto-compaction.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAgEH,gFAAgF;AAChF,iBAAiB;AACjB,gFAAgF;AAEhF,MAAM,cAAc,GAAyB;IAC3C,IAAI,EAAE,MAAM;IACZ,gBAAgB,EAAE,IAAI;IACtB,oBAAoB,EAAE,IAAI;IAC1B,kBAAkB,EAAE,IAAI;IACxB,0BAA0B,EAAE,CAAC;IAC7B,+BAA+B,EAAE,CAAC;IAClC,UAAU,EAAE,KAAK,EAAE,WAAW;IAC9B,gBAAgB,EAAE,MAAM,EAAE,0BAA0B;CACrD,CAAC;AAEF,gFAAgF;AAChF,0BAA0B;AAC1B,gFAAgF;AAEhF;;;;;;;;GAQG;AACH,MAAM,OAAO,qBAAqB;IACxB,SAAS,CAAY;IACrB,MAAM,CAAuB;IAC7B,kBAAkB,GAAW,CAAC,CAAC;IAC/B,SAAS,GAAkC,EAAE,CAAC;IAEtD,YAAY,SAAoB,EAAE,MAAsC;QACtE,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM,EAAE,CAAC;QAE/C,2CAA2C;QAC3C,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACrE,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,oBAAoB,CAAC,OAG1B;QACC,MAAM,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;QAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;QAE/C,gCAAgC;QAChC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAEvC,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,sBAAsB;YAC5B,aAAa;YACb,KAAK;YACL,SAAS,EAAE,IAAI;SAChB,CAAC,CAAC;QAEH,gCAAgC;QAChC,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAClB,OAAO;gBACL,MAAM,EAAE,IAAI;gBACZ,aAAa;gBACb,SAAS;gBACT,KAAK;aACN,CAAC;QACJ,CAAC;QAED,2CAA2C;QAC3C,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC1B,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,2BAA2B;gBACjC,aAAa;gBACb,KAAK;aACN,CAAC,CAAC;YACH,OAAO;gBACL,MAAM,EAAE,YAAY;gBACpB,aAAa;gBACb,SAAS;gBACT,KAAK;aACN,CAAC;QACJ,CAAC;QAED,mEAAmE;QACnE,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,wBAAwB;gBAC9B,aAAa;gBACb,KAAK;aACN,CAAC,CAAC;YACH,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,aAAa;gBACb,SAAS;gBACT,KAAK;aACN,CAAC;QACJ,CAAC;QAED,0EAA0E;QAC1E,iBAAiB;QACjB,IAAI,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9C,IAAI,CAAC,IAAI,CAAC;gBACR,IAAI,EAAE,yBAAyB;gBAC/B,WAAW,EAAE,SAAS;aACvB,CAAC,CAAC;YAEH,oCAAoC;YACpC,OAAO;gBACL,MAAM,EAAE,SAAS;gBACjB,aAAa;gBACb,SAAS;gBACT,KAAK;aACN,CAAC;QACJ,CAAC;QAED,uBAAuB;QACvB,QAAQ,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACzB,KAAK,MAAM;gBACT,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YAE3E,KAAK,UAAU;gBACb,IAAI,CAAC,IAAI,CAAC;oBACR,IAAI,EAAE,+BAA+B;oBACrC,aAAa;oBACb,KAAK;iBACN,CAAC,CAAC;gBACH,OAAO;oBACL,MAAM,EAAE,gBAAgB;oBACxB,aAAa;oBACb,SAAS;oBACT,KAAK;iBACN,CAAC;YAEJ,KAAK,QAAQ;gBACX,0DAA0D;gBAC1D,IAAI,CAAC,IAAI,CAAC;oBACR,IAAI,EAAE,wBAAwB;oBAC9B,aAAa;oBACb,KAAK;iBACN,CAAC,CAAC;gBACH,OAAO;oBACL,MAAM,EAAE,SAAS;oBACjB,aAAa;oBACb,SAAS;oBACT,KAAK;iBACN,CAAC;YAEJ;gBACE,OAAO;oBACL,MAAM,EAAE,SAAS;oBACjB,aAAa;oBACb,SAAS;oBACT,KAAK;iBACN,CAAC;QACN,CAAC;IACH,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,aAAqB;QACjC,OAAO,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,YAAY;QACV,IAAI,IAAI,CAAC,kBAAkB,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO,KAAK,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;IACvE,CAAC;IAED;;OAEG;IACH,oBAAoB;QAClB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;YACzB,OAAO,CAAC,CAAC;QACX,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACzE,CAAC;IAED;;OAEG;IACH,aAAa;QACX,IAAI,CAAC,kBAAkB,GAAG,CAAC,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,YAAY,CAAC,QAAmB;QACpC,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC;QAChD,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC;QAE/C,OAAO,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IAC3E,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,OAAsC;QACjD,IAAI,CAAC,MAAM,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC;QAE7C,yBAAyB;QACzB,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACrE,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACvE,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IAED;;OAEG;IACH,EAAE,CAAC,QAAqC;QACtC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9B,OAAO,GAAG,EAAE;YACV,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAC7C,IAAI,GAAG,IAAI,CAAC;gBAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC,CAAC;IACJ,CAAC;IAED,8EAA8E;IAC9E,kBAAkB;IAClB,8EAA8E;IAE9E;;OAEG;IACK,aAAa,CAAC,KAAa;QACjC,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC5C,OAAO,YAAY,CAAC;QACtB,CAAC;QACD,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YAC9C,OAAO,cAAc,CAAC;QACxB,CAAC;QACD,IAAI,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,iBAAiB,CAC7B,QAAmB,EACnB,aAAqB,EACrB,SAAiB,EACjB,KAAa;QAEb,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,0BAA0B;YAChC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI;YACtB,aAAa;SACd,CAAC,CAAC;QAEH,mEAAmE;QACnE,uEAAuE;QACvE,iEAAiE;QACjE,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,0BAA0B,GAAG,IAAI,CAAC,MAAM,CAAC,+BAA+B,CAAC;QAC3G,MAAM,sBAAsB,GAAG,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;QAE1D,mDAAmD;QACnD,MAAM,YAAY,GAAG,sBAAsB,CAAC,mBAAmB,KAAK,aAAa,CAAC;QAClF,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,mBAAmB,EAAE,aAAa,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,MAAwB,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAClD,CAAC;QAAC,OAAO,eAAe,EAAE,CAAC;YACzB,qEAAqE;YACrE,OAAO,CAAC,KAAK,CAAC,qEAAqE,EAAE,eAAe,CAAC,CAAC;YAEtG,MAAM,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;YAExE,0BAA0B;YAC1B,IAAI,YAAY,EAAE,CAAC;gBACjB,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,mBAAmB,EAAE,sBAAsB,CAAC,mBAAmB,EAAE,CAAC,CAAC;YACnG,CAAC;YAED,gDAAgD;YAChD,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAErC,OAAO,eAAe,CAAC;QACzB,CAAC;QAED,0BAA0B;QAC1B,IAAI,YAAY,EAAE,CAAC;YACjB,IAAI,CAAC,SAAS,CAAC,YAAY,CAAC,EAAE,mBAAmB,EAAE,sBAAsB,CAAC,mBAAmB,EAAE,CAAC,CAAC;QACnG,CAAC;QAED,kBAAkB;QAClB,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAErC,MAAM,SAAS,GAAG,MAAM,CAAC,YAAY,GAAG,CAAC;YACvC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,YAAY,CAAC,GAAG,GAAG,CAAC;YAClE,CAAC,CAAC,CAAC,CAAC;QAEN,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,0BAA0B;YAChC,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,SAAS;SACV,CAAC,CAAC;QAEH,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,aAAa,EAAE,MAAM,CAAC,WAAW;YACjC,SAAS;YACT,KAAK,EAAE,MAAM,CAAC,WAAW,GAAG,SAAS;YACrC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;YAC3C,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC;IACJ,CAAC;IAED;;;OAGG;IACK,iBAAiB,CAAC,QAAmB,EAAE,aAAqB;QAClE,MAAM,cAAc,GAAG,QAAQ,CAAC,MAAM,CAAC;QAEvC,mCAAmC;QACnC,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAE9D,gEAAgE;QAChE,MAAM,iBAAiB,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACpE,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC,aAAa,CAAC,CAAC;QAElE,oCAAoC;QACpC,MAAM,iBAAiB,GAAc,EAAE,CAAC;QAExC,IAAI,aAAa,EAAE,CAAC;YAClB,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACxC,CAAC;QAED,4CAA4C;QAC5C,iBAAiB,CAAC,IAAI,CAAC;YACrB,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,uBAAuB,cAAc,GAAG,iBAAiB,CAAC,MAAM,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,qGAAqG,iBAAiB,CAAC,MAAM,aAAa;SAC9O,CAAC,CAAC;QAEH,iBAAiB,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC,CAAC;QAE7C,MAAM,aAAa,GAAG,iBAAiB,CAAC,MAAM,CAAC;QAC/C,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;QAErE,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,mCAAmC;YACzC,MAAM,EAAE,uBAAuB;YAC/B,cAAc;YACd,aAAa;SACd,CAAC,CAAC;QAEH,OAAO;YACL,MAAM,EAAE,WAAW;YACnB,aAAa,EAAE,WAAW;YAC1B,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,gBAAgB;YACvC,KAAK,EAAE,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB;YACjD,iBAAiB,EAAE,iBAAiB;YACpC,OAAO,EAAE,+CAA+C;SACzD,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,IAAI,CAAC,KAA0B;QACrC,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACtC,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,yBAAyB;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;CACF;AAED,gFAAgF;AAChF,UAAU;AACV,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,2BAA2B,CACzC,SAAoB,EACpB,MAAsC;IAEtC,OAAO,IAAI,qBAAqB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;AACtD,CAAC;AAED,gFAAgF;AAChF,YAAY;AACZ,gFAAgF;AAEhF;;GAEG;AACH,MAAM,UAAU,2BAA2B,CAAC,MAA6B;IACvE,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,GAAG,CAAC,CAAC;IACnD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;QACtB,KAAK,IAAI;YACP,KAAK,CAAC,IAAI,CAAC,kBAAkB,WAAW,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;YACtI,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzB,MAAM;QAER,KAAK,SAAS;YACZ,KAAK,CAAC,IAAI,CAAC,kBAAkB,WAAW,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;YACtI,KAAK,CAAC,IAAI,CAAC,yCAAyC,CAAC,CAAC;YACtD,KAAK,CAAC,IAAI,CAAC,wDAAwD,CAAC,CAAC;YACrE,MAAM;QAER,KAAK,gBAAgB;YACnB,KAAK,CAAC,IAAI,CAAC,kBAAkB,WAAW,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;YACtI,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACrC,KAAK,CAAC,IAAI,CAAC,8DAA8D,CAAC,CAAC;YAC3E,MAAM;QAER,KAAK,WAAW;YACd,KAAK,CAAC,IAAI,CAAC,kBAAkB,WAAW,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;YACtI,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YAChC,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACvC,CAAC;YACD,MAAM;QAER,KAAK,YAAY;YACf,KAAK,CAAC,IAAI,CAAC,kBAAkB,WAAW,MAAM,MAAM,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,MAAM,CAAC,SAAS,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;YACtI,KAAK,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;YACzC,KAAK,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;YAC5D,KAAK,CAAC,IAAI,CAAC,+BAA+B,CAAC,CAAC;YAC5C,MAAM;IACV,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,MAA6B;IAC9D,QAAQ,MAAM,CAAC,MAAM,EAAE,CAAC;QACtB,KAAK,IAAI;YACP,OAAO,IAAI,CAAC;QACd,KAAK,SAAS;YACZ,OAAO,oDAAoD,CAAC;QAC9D,KAAK,gBAAgB;YACnB,OAAO,wCAAwC,CAAC;QAClD,KAAK,WAAW;YACd,OAAO,IAAI,CAAC;QACd,KAAK,YAAY;YACf,OAAO,oDAAoD,CAAC;QAC9D;YACE,OAAO,IAAI,CAAC;IAChB,CAAC;AACH,CAAC"}

package/dist/src/integrations/cancellation.d.ts

@@ -0,0 +1,162 @@
+/**
+ * Cancellation Token Integration
+ *
+ * Provides graceful cancellation support for long-running agent operations.
+ * Based on the .NET CancellationToken pattern, adapted from tricks/cancellation.ts.
+ *
+ * Usage:
+ *   const cts = createCancellationTokenSource();
+ *   agent.run(task, { cancellationToken: cts.token });
+ *   // Later: cts.cancel('User requested cancellation');
+ */
+/**
+ * Token that can be checked for cancellation.
+ */
+export interface CancellationToken {
+    /** Whether cancellation has been requested */
+    readonly isCancellationRequested: boolean;
+    /** The reason for cancellation (if cancelled) */
+    readonly cancellationReason?: string;
+    /** Promise that resolves when cancelled */
+    readonly onCancellationRequested: Promise<string | void>;
+    /** Register a callback for cancellation */
+    register(callback: (reason?: string) => void): {
+        dispose: () => void;
+    };
+    /** Throw if cancelled */
+    throwIfCancellationRequested(): void;
+}
+/**
+ * Source that controls a cancellation token.
+ */
+export interface CancellationTokenSource {
+    /** The token */
+    readonly token: CancellationToken;
+    /** Whether cancellation has been requested */
+    readonly isCancellationRequested: boolean;
+    /** Request cancellation */
+    cancel(reason?: string): void;
+    /** Cancel after timeout */
+    cancelAfter(ms: number): this;
+    /** Dispose resources */
+    dispose(): void;
+}
+/**
+ * Options for cancellable operations.
+ */
+export interface CancellableOptions {
+    /** Cancellation token */
+    token?: CancellationToken;
+    /** Timeout in milliseconds (creates auto-cancelling token) */
+    timeout?: number;
+}
+/**
+ * Cancellation event for agent integration.
+ */
+export type CancellationEvent = {
+    type: 'cancellation.requested';
+    reason?: string;
+    timestamp: Date;
+} | {
+    type: 'cancellation.handled';
+    cleanupDuration: number;
+};
+export type CancellationEventListener = (event: CancellationEvent) => void;
+/**
+ * Error thrown when an operation is cancelled.
+ */
+export declare class CancellationError extends Error {
+    readonly isCancellation = true;
+    readonly reason?: string;
+    constructor(reason?: string);
+}
+/**
+ * Check if an error is a cancellation error.
+ */
+export declare function isCancellationError(error: unknown): error is CancellationError;
+/**
+ * Create a new cancellation token source.
+ */
+export declare function createCancellationTokenSource(): CancellationTokenSource;
+/**
+ * Create a token source that auto-cancels after timeout.
+ */
+export declare function createTimeoutToken(ms: number): CancellationTokenSource;
+/**
+ * Create a linked token that cancels when any source cancels.
+ */
+export declare function createLinkedToken(...sources: CancellationTokenSource[]): CancellationTokenSource;
+/**
+ * Create a "none" token that is never cancelled.
+ * Useful as a default when no cancellation is needed.
+ */
+export declare const CancellationToken: {
+    None: CancellationToken;
+};
+/**
+ * Run a function with cancellation support.
+ */
+export declare function withCancellation<T>(fn: (token: CancellationToken) => Promise<T>, options?: CancellableOptions): Promise<T>;
+/**
+ * Sleep with cancellation support.
+ */
+export declare function sleep(ms: number, token?: CancellationToken): Promise<void>;
+/**
+ * Race a promise against cancellation.
+ */
+export declare function race<T>(promise: Promise<T>, token: CancellationToken): Promise<T>;
+/**
+ * Create an AbortSignal from a CancellationToken.
+ * Useful for integrating with fetch and other AbortSignal-based APIs.
+ */
+export declare function toAbortSignal(token: CancellationToken): AbortSignal;
+/**
+ * Manager that handles cancellation for agent operations.
+ * Provides a clean interface for the ProductionAgent to use.
+ */
+export declare class CancellationManager {
+    private currentSource;
+    private eventListeners;
+    /**
+     * Create a new cancellation context for an operation.
+     * Disposes the previous context if one exists.
+     */
+    createContext(timeout?: number): CancellationToken;
+    /**
+     * Get the current cancellation token.
+     */
+    get token(): CancellationToken;
+    /**
+     * Check if cancellation has been requested.
+     */
+    get isCancelled(): boolean;
+    /**
+     * Request cancellation of the current operation.
+     */
+    cancel(reason?: string): void;
+    /**
+     * Dispose the current cancellation context.
+     */
+    disposeContext(): void;
+    /**
+     * Check if there's an active cancellation context.
+     */
+    hasActiveContext(): boolean;
+    /**
+     * Subscribe to cancellation events.
+     */
+    subscribe(listener: CancellationEventListener): () => void;
+    /**
+     * Emit a cancellation event.
+     */
+    private emit;
+    /**
+     * Cleanup all resources.
+     */
+    cleanup(): void;
+}
+/**
+ * Create a cancellation manager.
+ */
+export declare function createCancellationManager(): CancellationManager;
+//# sourceMappingURL=cancellation.d.ts.map

package/dist/src/integrations/cancellation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cancellation.d.ts","sourceRoot":"","sources":["../../../src/integrations/cancellation.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAMH;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,8CAA8C;IAC9C,QAAQ,CAAC,uBAAuB,EAAE,OAAO,CAAC;IAC1C,iDAAiD;IACjD,QAAQ,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IACrC,2CAA2C;IAC3C,QAAQ,CAAC,uBAAuB,EAAE,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;IACzD,2CAA2C;IAC3C,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,EAAE,MAAM,KAAK,IAAI,GAAG;QAAE,OAAO,EAAE,MAAM,IAAI,CAAA;KAAE,CAAC;IACvE,yBAAyB;IACzB,4BAA4B,IAAI,IAAI,CAAC;CACtC;AAED;;GAEG;AACH,MAAM,WAAW,uBAAuB;IACtC,gBAAgB;IAChB,QAAQ,CAAC,KAAK,EAAE,iBAAiB,CAAC;IAClC,8CAA8C;IAC9C,QAAQ,CAAC,uBAAuB,EAAE,OAAO,CAAC;IAC1C,2BAA2B;IAC3B,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,2BAA2B;IAC3B,WAAW,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,wBAAwB;IACxB,OAAO,IAAI,IAAI,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,yBAAyB;IACzB,KAAK,CAAC,EAAE,iBAAiB,CAAC;IAC1B,8DAA8D;IAC9D,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,MAAM,iBAAiB,GACzB;IAAE,IAAI,EAAE,wBAAwB,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,IAAI,CAAA;CAAE,GACpE;IAAE,IAAI,EAAE,sBAAsB,CAAC;IAAC,eAAe,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9D,MAAM,MAAM,yBAAyB,GAAG,CAAC,KAAK,EAAE,iBAAiB,KAAK,IAAI,CAAC;AA2G3E;;GAEG;AACH,qBAAa,iBAAkB,SAAQ,KAAK;IAC1C,QAAQ,CAAC,cAAc,QAAQ;IAC/B,QAAQ,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC;gBAEb,MAAM,CAAC,EAAE,MAAM;CAK5B;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,iBAAiB,CAK9E;AAMD;;GAEG;AACH,wBAAgB,6BAA6B,IAAI,uBAAuB,CAEvE;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,EAAE,EAAE,MAAM,GAAG,uBAAuB,CAEtE;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,GAAG,OAAO,EAAE,uBAAuB,EAAE,GAAG,uBAAuB,CAYhG;AAED;;;GAGG;AACH,eAAO,MAAM,iBAAiB;UAOvB,iBAAiB;CACvB,CAAC;AAMF;;GAEG;AACH,wBAAsB,gBAAgB,CAAC,CAAC,EACtC,EAAE,EAAE,CAAC,KAAK,EAAE,iBAAiB,KAAK,OAAO,CAAC,CAAC,CAAC,EAC5C,OAAO,GAAE,kBAAuB,GAC/B,OAAO,CAAC,CAAC,CAAC,CA2BZ;AAED;;GAEG;AACH,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC,CAkB1E;AAED;;GAEG;AACH,wBAAgB,IAAI,CAAC,CAAC,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,iBAAiB,GAAG,OAAO,CAAC,CAAC,CAAC,CAWjF;AAED;;;GAGG;AACH,wBAAgB,aAAa,CAAC,KAAK,EAAE,iBAAiB,GAAG,WAAW,CAUnE;AAMD;;;GAGG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,aAAa,CAAwC;IAC7D,OAAO,CAAC,cAAc,CAA6C;IAEnE;;;OAGG;IACH,aAAa,CAAC,OAAO,CAAC,EAAE,MAAM,GAAG,iBAAiB;IAUlD;;OAEG;IACH,IAAI,KAAK,IAAI,iBAAiB,CAE7B;IAED;;OAEG;IACH,IAAI,WAAW,IAAI,OAAO,CAEzB;IAED;;OAEG;IACH,MAAM,CAAC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI;IAQ7B;;OAEG;IACH,cAAc,IAAI,IAAI;IAOtB;;OAEG;IACH,gBAAgB,IAAI,OAAO;IAI3B;;OAEG;IACH,SAAS,CAAC,QAAQ,EAAE,yBAAyB,GAAG,MAAM,IAAI;IAK1D;;OAEG;IACH,OAAO,CAAC,IAAI;IAUZ;;OAEG;IACH,OAAO,IAAI,IAAI;CAIhB;AAED;;GAEG;AACH,wBAAgB,yBAAyB,IAAI,mBAAmB,CAE/D"}