attocode 0.2.4 → 0.2.5

package/dist/src/integrations/budget/budget-pool.js

@@ -0,0 +1,205 @@
+/**
+ * Shared Budget Pool
+ *
+ * Enables parent-child budget sharing for multi-agent workflows.
+ * Instead of each subagent getting an independent budget (leading to 250%+ exposure),
+ * subagents draw from a shared pool that the parent allocates from its own budget.
+ *
+ * Model:
+ *   Parent budget: 200K tokens
+ *   Parent reserves: 50K (for synthesis after subagents complete)
+ *   Subagent pool: 150K (shared among all children)
+ *
+ * Each child can draw up to `maxPerChild` tokens from the pool, but the combined
+ * consumption never exceeds the pool total. This ensures total tree cost stays
+ * bounded by the parent's budget.
+ */
+// =============================================================================
+// SHARED BUDGET POOL
+// =============================================================================
+export class SharedBudgetPool {
+    config;
+    originalMaxPerChild;
+    allocations = new Map();
+    totalTokensUsed = 0;
+    totalCostUsed = 0;
+    /** Tokens reserved by active allocations (pessimistic accounting) */
+    totalTokensReserved = 0;
+    totalCostReserved = 0;
+    /** Optional trace collector for emitting budget.pool events */
+    traceCollector;
+    constructor(config) {
+        this.config = config;
+        this.originalMaxPerChild = config.maxPerChild;
+    }
+    /**
+     * Get pool remaining tokens (accounts for both used and reserved).
+     */
+    getPoolRemaining() {
+        const committed = Math.max(this.totalTokensUsed, this.totalTokensReserved);
+        return Math.max(0, this.config.totalTokens - committed);
+    }
+    /**
+     * Reserve a budget allocation for a child agent.
+     * Uses pessimistic accounting: reserved tokens count against pool capacity
+     * until actual usage is recorded via recordUsage() + release().
+     * Returns the allocation if there's enough room, or null if the pool is exhausted.
+     */
+    reserve(childId) {
+        // Account for both consumed AND reserved-but-not-yet-consumed tokens
+        const committed = Math.max(this.totalTokensUsed, this.totalTokensReserved);
+        const remaining = this.config.totalTokens - committed;
+        if (remaining <= 0) {
+            return null;
+        }
+        // Allocate up to maxPerChild or whatever remains, whichever is smaller
+        const tokenBudget = Math.min(this.config.maxPerChild, remaining);
+        const committedCost = Math.max(this.totalCostUsed, this.totalCostReserved);
+        const costRemaining = this.config.totalCost
+            ? this.config.totalCost - committedCost
+            : Infinity;
+        const costBudget = Math.min(this.config.maxCostPerChild ?? Infinity, costRemaining > 0 ? costRemaining : 0);
+        if (tokenBudget <= 0 || (this.config.totalCost && costBudget <= 0)) {
+            return null;
+        }
+        const allocation = {
+            id: childId,
+            tokenBudget,
+            costBudget,
+            tokensUsed: 0,
+            costUsed: 0,
+        };
+        // Track reservation pessimistically
+        this.totalTokensReserved += tokenBudget;
+        this.totalCostReserved += costBudget;
+        this.allocations.set(childId, allocation);
+        // Emit trace event for budget allocation
+        this.traceCollector?.record({
+            type: 'budget.pool',
+            data: {
+                action: 'allocate',
+                agentId: childId,
+                tokensAllocated: tokenBudget,
+                poolRemaining: this.getPoolRemaining(),
+                poolTotal: this.config.totalTokens,
+            },
+        });
+        return allocation;
+    }
+    /**
+     * Record token consumption for a child agent.
+     * Returns false if the child has exceeded its allocation.
+     */
+    recordUsage(childId, tokens, cost) {
+        const allocation = this.allocations.get(childId);
+        if (!allocation) {
+            return false;
+        }
+        allocation.tokensUsed += tokens;
+        allocation.costUsed += cost;
+        this.totalTokensUsed += tokens;
+        this.totalCostUsed += cost;
+        // Emit trace event for budget consumption
+        this.traceCollector?.record({
+            type: 'budget.pool',
+            data: {
+                action: 'consume',
+                agentId: childId,
+                tokensUsed: tokens,
+                poolRemaining: this.getPoolRemaining(),
+                poolTotal: this.config.totalTokens,
+            },
+        });
+        return allocation.tokensUsed <= allocation.tokenBudget;
+    }
+    /**
+     * Release an allocation, adjusting reserved totals to reflect actual usage.
+     * Must be called after recordUsage() to return unused budget to the pool.
+     */
+    release(childId) {
+        const allocation = this.allocations.get(childId);
+        if (allocation) {
+            // Release the pessimistic reservation, keeping only actual usage
+            this.totalTokensReserved -= allocation.tokenBudget;
+            this.totalCostReserved -= allocation.costBudget;
+            this.allocations.delete(childId);
+            // Emit trace event for budget release
+            this.traceCollector?.record({
+                type: 'budget.pool',
+                data: {
+                    action: 'release',
+                    agentId: childId,
+                    tokensAllocated: allocation.tokenBudget,
+                    tokensUsed: allocation.tokensUsed,
+                    poolRemaining: this.getPoolRemaining(),
+                    poolTotal: this.config.totalTokens,
+                },
+            });
+        }
+    }
+    /**
+     * Get remaining tokens for a specific child allocation.
+     */
+    getRemainingForChild(childId) {
+        const allocation = this.allocations.get(childId);
+        if (!allocation)
+            return 0;
+        return Math.max(0, allocation.tokenBudget - allocation.tokensUsed);
+    }
+    /**
+     * Get overall pool statistics.
+     */
+    getStats() {
+        const committed = Math.max(this.totalTokensUsed, this.totalTokensReserved);
+        return {
+            totalTokens: this.config.totalTokens,
+            tokensUsed: this.totalTokensUsed,
+            tokensRemaining: Math.max(0, this.config.totalTokens - committed),
+            activeAllocations: this.allocations.size,
+            utilization: this.config.totalTokens > 0
+                ? committed / this.config.totalTokens
+                : 0,
+        };
+    }
+    /**
+     * Check if the pool has enough budget for at least one more child.
+     */
+    hasCapacity() {
+        const committed = Math.max(this.totalTokensUsed, this.totalTokensReserved);
+        return (this.config.totalTokens - committed) > 10000; // Minimum 10K tokens
+    }
+    /**
+     * Temporarily override maxPerChild for parallel batch spawning.
+     * Used by spawnAgentsParallel to divide the pool equally among children
+     * instead of each child racing for the full maxPerChild allocation.
+     * Must be followed by resetMaxPerChild() in a finally block.
+     */
+    setMaxPerChild(maxPerChild) {
+        this.config.maxPerChild = maxPerChild;
+    }
+    /**
+     * Restore maxPerChild to the original value set at construction time.
+     */
+    resetMaxPerChild() {
+        this.config.maxPerChild = this.originalMaxPerChild;
+    }
+}
+// =============================================================================
+// FACTORY
+// =============================================================================
+/**
+ * Create a budget pool from a parent's total budget.
+ * Reserves a portion for the parent's own synthesis work.
+ */
+export function createBudgetPool(parentBudgetTokens, parentReserveRatio = 0.25, // Reserve 25% for parent synthesis
+maxPerChild = 100000) {
+    const parentReserve = Math.floor(parentBudgetTokens * parentReserveRatio);
+    const poolTokens = parentBudgetTokens - parentReserve;
+    return new SharedBudgetPool({
+        totalTokens: poolTokens,
+        maxPerChild: Math.min(maxPerChild, poolTokens),
+        totalCost: 0.50, // Default cost cap
+        maxCostPerChild: 0.25,
+    });
+}
+//# sourceMappingURL=budget-pool.js.map

package/dist/src/integrations/budget/budget-pool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget-pool.js","sourceRoot":"","sources":["../../../../src/integrations/budget/budget-pool.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AA6CH,gFAAgF;AAChF,qBAAqB;AACrB,gFAAgF;AAEhF,MAAM,OAAO,gBAAgB;IACnB,MAAM,CAAmB;IAChB,mBAAmB,CAAS;IACrC,WAAW,GAAG,IAAI,GAAG,EAA4B,CAAC;IAClD,eAAe,GAAG,CAAC,CAAC;IACpB,aAAa,GAAG,CAAC,CAAC;IAC1B,qEAAqE;IAC7D,mBAAmB,GAAG,CAAC,CAAC;IACxB,iBAAiB,GAAG,CAAC,CAAC;IAE9B,+DAA+D;IAC/D,cAAc,CAAkB;IAEhC,YAAY,MAAwB;QAClC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,WAAW,CAAC;IAChD,CAAC;IAED;;OAEG;IACK,gBAAgB;QACtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC3E,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,SAAS,CAAC,CAAC;IAC1D,CAAC;IAED;;;;;OAKG;IACH,OAAO,CAAC,OAAe;QACrB,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC3E,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,SAAS,CAAC;QACtD,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;YACnB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,uEAAuE;QACvE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACjE,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC3E,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS;YACzC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,aAAa;YACvC,CAAC,CAAC,QAAQ,CAAC;QACb,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CACzB,IAAI,CAAC,MAAM,CAAC,eAAe,IAAI,QAAQ,EACvC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CACtC,CAAC;QAEF,IAAI,WAAW,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,IAAI,UAAU,IAAI,CAAC,CAAC,EAAE,CAAC;YACnE,OAAO,IAAI,CAAC;QACd,CAAC;QAED,MAAM,UAAU,GAAqB;YACnC,EAAE,EAAE,OAAO;YACX,WAAW;YACX,UAAU;YACV,UAAU,EAAE,CAAC;YACb,QAAQ,EAAE,CAAC;SACZ,CAAC;QAEF,oCAAoC;QACpC,IAAI,CAAC,mBAAmB,IAAI,WAAW,CAAC;QACxC,IAAI,CAAC,iBAAiB,IAAI,UAAU,CAAC;QACrC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;QAE1C,yCAAyC;QACzC,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC;YAC1B,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE;gBACJ,MAAM,EAAE,UAAU;gBAClB,OAAO,EAAE,OAAO;gBAChB,eAAe,EAAE,WAAW;gBAC5B,aAAa,EAAE,IAAI,CAAC,gBAAgB,EAAE;gBACtC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;aACnC;SACF,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;;OAGG;IACH,WAAW,CAAC,OAAe,EAAE,MAAc,EAAE,IAAY;QACvD,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO,KAAK,CAAC;QACf,CAAC;QAED,UAAU,CAAC,UAAU,IAAI,MAAM,CAAC;QAChC,UAAU,CAAC,QAAQ,IAAI,IAAI,CAAC;QAC5B,IAAI,CAAC,eAAe,IAAI,MAAM,CAAC;QAC/B,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC;QAE3B,0CAA0C;QAC1C,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC;YAC1B,IAAI,EAAE,aAAa;YACnB,IAAI,EAAE;gBACJ,MAAM,EAAE,SAAS;gBACjB,OAAO,EAAE,OAAO;gBAChB,UAAU,EAAE,MAAM;gBAClB,aAAa,EAAE,IAAI,CAAC,gBAAgB,EAAE;gBACtC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;aACnC;SACF,CAAC,CAAC;QAEH,OAAO,UAAU,CAAC,UAAU,IAAI,UAAU,CAAC,WAAW,CAAC;IACzD,CAAC;IAED;;;OAGG;IACH,OAAO,CAAC,OAAe;QACrB,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,UAAU,EAAE,CAAC;YACf,iEAAiE;YACjE,IAAI,CAAC,mBAAmB,IAAI,UAAU,CAAC,WAAW,CAAC;YACnD,IAAI,CAAC,iBAAiB,IAAI,UAAU,CAAC,UAAU,CAAC;YAChD,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAEjC,sCAAsC;YACtC,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC;gBAC1B,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE;oBACJ,MAAM,EAAE,SAAS;oBACjB,OAAO,EAAE,OAAO;oBAChB,eAAe,EAAE,UAAU,CAAC,WAAW;oBACvC,UAAU,EAAE,UAAU,CAAC,UAAU;oBACjC,aAAa,EAAE,IAAI,CAAC,gBAAgB,EAAE;oBACtC,SAAS,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;iBACnC;aACF,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED;;OAEG;IACH,oBAAoB,CAAC,OAAe;QAClC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,UAAU;YAAE,OAAO,CAAC,CAAC;QAC1B,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC3E,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;YACpC,UAAU,EAAE,IAAI,CAAC,eAAe;YAChC,eAAe,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,SAAS,CAAC;YACjE,iBAAiB,EAAE,IAAI,CAAC,WAAW,CAAC,IAAI;YACxC,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,CAAC;gBACtC,CAAC,CAAC,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW;gBACrC,CAAC,CAAC,CAAC;SACN,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,WAAW;QACT,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC3E,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,KAAK,CAAC,CAAC,qBAAqB;IAC7E,CAAC;IAED;;;;;OAKG;IACH,cAAc,CAAC,WAAmB;QAChC,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,WAAW,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,IAAI,CAAC,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,mBAAmB,CAAC;IACrD,CAAC;CACF;AAED,gFAAgF;AAChF,UAAU;AACV,gFAAgF;AAEhF;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAC9B,kBAA0B,EAC1B,qBAA6B,IAAI,EAAE,mCAAmC;AACtE,cAAsB,MAAM;IAE5B,MAAM,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,GAAG,kBAAkB,CAAC,CAAC;IAC1E,MAAM,UAAU,GAAG,kBAAkB,GAAG,aAAa,CAAC;IAEtD,OAAO,IAAI,gBAAgB,CAAC;QAC1B,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,UAAU,CAAC;QAC9C,SAAS,EAAE,IAAI,EAAE,mBAAmB;QACpC,eAAe,EAAE,IAAI;KACtB,CAAC,CAAC;AACL,CAAC"}

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

@@ -0,0 +1,229 @@
+/**
+ * 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;
+/**
+ * Progress-aware timeout token.
+ * Extends CancellationTokenSource with the ability to reset the timeout when progress is detected.
+ * Only triggers timeout if there's been no progress for `idleTimeout` milliseconds.
+ */
+export interface ProgressAwareTimeoutSource extends CancellationTokenSource {
+    /** Call this when progress is detected (e.g., tool call, token update) */
+    reportProgress(): void;
+    /** Get time since last progress in ms */
+    getIdleTime(): number;
+    /** Get total elapsed time in ms */
+    getElapsedTime(): number;
+}
+/**
+ * Create a progress-aware timeout token.
+ *
+ * @param maxTimeout - Maximum total execution time (hard limit)
+ * @param idleTimeout - Time without progress before cancellation (soft limit)
+ * @param checkInterval - How often to check for idle timeout (default: 5000ms)
+ *
+ * The token will be cancelled when either:
+ * 1. maxTimeout is reached (regardless of progress)
+ * 2. idleTimeout passes without any progress reported
+ *
+ * @example
+ * ```ts
+ * const timeout = createProgressAwareTimeout(300000, 60000); // 5 min max, 1 min idle
+ *
+ * // In event handler:
+ * subAgent.subscribe(event => {
+ *   if (['tool.start', 'tool.complete', 'llm.start', 'llm.complete'].includes(event.type)) {
+ *     timeout.reportProgress();
+ *   }
+ * });
+ * ```
+ */
+export declare function createProgressAwareTimeout(maxTimeout: number, idleTimeout: number, checkInterval?: number): ProgressAwareTimeoutSource;
+/**
+ * Graceful timeout source that supports a wrapup phase before hard cancellation.
+ * Extends ProgressAwareTimeoutSource with wrapup callbacks.
+ */
+export interface GracefulTimeoutSource extends ProgressAwareTimeoutSource {
+    /** Register a callback that fires `wrapupWindow` ms before the hard cancel */
+    onWrapupWarning(callback: () => void): void;
+    /** Whether the timeout is currently in the wrapup phase */
+    isInWrapupPhase(): boolean;
+}
+/**
+ * Create a graceful timeout that fires a wrapup warning before hard cancellation.
+ *
+ * The lifecycle is:
+ * 1. Normal operation — progress extends the idle timer (same as createProgressAwareTimeout)
+ * 2. Wrapup phase — `wrapupWindow` ms before hard kill, wrapup callbacks fire
+ *    - Progress during wrapup does NOT extend the deadline (wrapping up, not working)
+ * 3. Hard kill — `race()` throws CancellationError after wrapup window expires
+ *
+ * @param maxTimeout - Maximum total execution time (hard limit)
+ * @param idleTimeout - Time without progress before cancellation (soft limit)
+ * @param wrapupWindow - Time between wrapup warning and hard kill (default: 30000ms)
+ * @param checkInterval - How often to check for idle timeout (default: 5000ms)
+ */
+export declare function createGracefulTimeout(maxTimeout: number, idleTimeout: number, wrapupWindow?: number, checkInterval?: number): GracefulTimeoutSource;
+/**
+ * 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;
+    /**
+     * Get the current cancellation source for linking with other tokens.
+     * Returns null if no active context.
+     */
+    getSource(): CancellationTokenSource | null;
+    /**
+     * 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/budget/cancellation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cancellation.d.ts","sourceRoot":"","sources":["../../../../src/integrations/budget/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;;;;GAIG;AACH,MAAM,WAAW,0BAA2B,SAAQ,uBAAuB;IACzE,0EAA0E;IAC1E,cAAc,IAAI,IAAI,CAAC;IACvB,yCAAyC;IACzC,WAAW,IAAI,MAAM,CAAC;IACtB,mCAAmC;IACnC,cAAc,IAAI,MAAM,CAAC;CAC1B;AAED;;;;;;;;;;;;;;;;;;;;;;GAsBG;AACH,wBAAgB,0BAA0B,CACxC,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,MAAM,EACnB,aAAa,SAAO,GACnB,0BAA0B,CA0C5B;AAED;;;GAGG;AACH,MAAM,WAAW,qBAAsB,SAAQ,0BAA0B;IACvE,8EAA8E;IAC9E,eAAe,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,IAAI,CAAC;IAC5C,2DAA2D;IAC3D,eAAe,IAAI,OAAO,CAAC;CAC5B;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,qBAAqB,CACnC,UAAU,EAAE,MAAM,EAClB,WAAW,EAAE,MAAM,EACnB,YAAY,SAAQ,EACpB,aAAa,SAAO,GACnB,qBAAqB,CAgGvB;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;;;OAGG;IACH,SAAS,IAAI,uBAAuB,GAAG,IAAI;IAI3C;;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"}