attocode 0.1.8 → 0.2.0

package/dist/src/integrations/swarm/request-throttle.js

@@ -0,0 +1,263 @@
+/**
+ * Swarm Request Throttle
+ *
+ * Token bucket + minimum spacing + FIFO queue to prevent 429 rate limiting
+ * across all swarm workers. Since all subagents share the parent's provider
+ * by reference (agent.ts:4398), wrapping the provider here automatically
+ * throttles ALL downstream LLM calls.
+ */
+export const FREE_TIER_THROTTLE = {
+    maxConcurrent: 2,
+    refillRatePerSecond: 0.5,
+    minSpacingMs: 1500,
+};
+export const PAID_TIER_THROTTLE = {
+    maxConcurrent: 5,
+    refillRatePerSecond: 2.0,
+    minSpacingMs: 200,
+};
+// ─── SwarmThrottle ──────────────────────────────────────────────────────────
+/**
+ * Async semaphore with token bucket rate limiting and minimum spacing.
+ *
+ * - `acquire()` waits in FIFO order, then consumes a token once available.
+ * - Tokens refill passively based on elapsed wall-clock time.
+ * - No explicit `release()` — long LLM calls naturally allow refill.
+ */
+export class SwarmThrottle {
+    config;
+    tokens;
+    lastAcquireTime = 0;
+    lastRefillTime;
+    queue = [];
+    totalAcquired = 0;
+    refillTimer = null;
+    // Adaptive backoff state
+    originalConfig;
+    _backoffLevel = 0;
+    lastBackoffTime = 0;
+    static MAX_BACKOFF_LEVEL = 3;
+    static RECOVER_COOLDOWN_MS = 10_000;
+    constructor(config) {
+        this.config = config;
+        this.tokens = config.maxConcurrent;
+        this.lastRefillTime = Date.now();
+        this.originalConfig = { ...config };
+    }
+    /**
+     * Wait for a token to become available (FIFO).
+     * Enforces minSpacing between consecutive acquisitions.
+     */
+    async acquire() {
+        // FIFO: wait in queue if there are already waiters or no tokens
+        if (this.queue.length > 0 || this.refill() < 1) {
+            await new Promise(resolve => {
+                this.queue.push(resolve);
+                this.ensureRefillTimer();
+            });
+        }
+        // Enforce minimum spacing
+        const now = Date.now();
+        const elapsed = now - this.lastAcquireTime;
+        if (elapsed < this.config.minSpacingMs && this.lastAcquireTime > 0) {
+            const waitMs = this.config.minSpacingMs - elapsed;
+            await sleep(waitMs);
+        }
+        // Consume token (refill first after possible wait)
+        this.refill();
+        this.tokens = Math.max(0, this.tokens - 1);
+        this.lastAcquireTime = Date.now();
+        this.totalAcquired++;
+    }
+    /**
+     * Get current throttle stats for observability.
+     */
+    getStats() {
+        this.refill();
+        return {
+            pendingCount: this.queue.length,
+            availableTokens: Math.floor(this.tokens),
+            totalAcquired: this.totalAcquired,
+            backoffLevel: this._backoffLevel,
+            currentMaxConcurrent: this.config.maxConcurrent,
+            currentMinSpacingMs: this.config.minSpacingMs,
+        };
+    }
+    /** Current backoff level (0 = normal, max = 3). */
+    get backoffLevel() {
+        return this._backoffLevel;
+    }
+    /**
+     * Increase backoff: halve maxConcurrent (min 1), double minSpacingMs (max 5000),
+     * halve refillRate (min 0.1). Caps at level 3.
+     */
+    backoff() {
+        if (this._backoffLevel >= SwarmThrottle.MAX_BACKOFF_LEVEL)
+            return;
+        this._backoffLevel++;
+        this.lastBackoffTime = Date.now();
+        this.config = {
+            maxConcurrent: Math.max(1, Math.floor(this.config.maxConcurrent / 2)),
+            minSpacingMs: Math.min(5000, this.config.minSpacingMs * 2),
+            refillRatePerSecond: Math.max(0.1, this.config.refillRatePerSecond / 2),
+        };
+        // Clamp tokens to new maxConcurrent
+        this.tokens = Math.min(this.tokens, this.config.maxConcurrent);
+    }
+    /**
+     * Step back toward original config after sustained success.
+     * Only recovers if 10s+ have passed since last backoff.
+     */
+    recover() {
+        if (this._backoffLevel <= 0)
+            return;
+        if (Date.now() - this.lastBackoffTime < SwarmThrottle.RECOVER_COOLDOWN_MS)
+            return;
+        this._backoffLevel--;
+        // Interpolate back toward original config
+        const ratio = this._backoffLevel / SwarmThrottle.MAX_BACKOFF_LEVEL;
+        this.config = {
+            maxConcurrent: Math.max(1, Math.round(this.originalConfig.maxConcurrent * (1 - ratio * 0.5))),
+            minSpacingMs: Math.round(this.originalConfig.minSpacingMs * (1 + ratio)),
+            refillRatePerSecond: this.originalConfig.refillRatePerSecond * (1 - ratio * 0.5),
+        };
+    }
+    /**
+     * Feed rate limit info from response headers to proactively adjust throttle.
+     * If remaining requests or tokens are low, preemptively back off.
+     */
+    feedRateLimitInfo(info) {
+        // Proactive backoff: if < 5 remaining requests, increase spacing
+        if (info.remainingRequests !== undefined && info.remainingRequests < 5) {
+            if (this._backoffLevel < SwarmThrottle.MAX_BACKOFF_LEVEL) {
+                this.backoff();
+            }
+        }
+        // If reset is imminent (< 2s), briefly pause
+        if (info.resetSeconds !== undefined && info.resetSeconds < 2 && info.remainingRequests !== undefined && info.remainingRequests <= 1) {
+            this.config = {
+                ...this.config,
+                minSpacingMs: Math.max(this.config.minSpacingMs, info.resetSeconds * 1000),
+            };
+        }
+    }
+    /**
+     * Refill tokens based on elapsed time. Returns current token count.
+     */
+    refill() {
+        const now = Date.now();
+        const elapsedSec = (now - this.lastRefillTime) / 1000;
+        if (elapsedSec > 0) {
+            this.tokens = Math.min(this.config.maxConcurrent, this.tokens + elapsedSec * this.config.refillRatePerSecond);
+            this.lastRefillTime = now;
+        }
+        return this.tokens;
+    }
+    /**
+     * Start a timer to periodically check for token refills and wake waiters.
+     * Automatically stops when the queue is empty.
+     */
+    ensureRefillTimer() {
+        if (this.refillTimer !== null)
+            return;
+        // Check every ~50ms or minSpacingMs, whichever is smaller
+        const intervalMs = Math.min(50, this.config.minSpacingMs || 50);
+        this.refillTimer = setInterval(() => {
+            this.refill();
+            // Wake queued waiters if tokens available
+            while (this.queue.length > 0 && this.tokens >= 1) {
+                const resolve = this.queue.shift();
+                if (resolve)
+                    resolve();
+            }
+            // Stop timer when queue is drained
+            if (this.queue.length === 0 && this.refillTimer !== null) {
+                clearInterval(this.refillTimer);
+                this.refillTimer = null;
+            }
+        }, intervalMs);
+    }
+}
+// ─── ThrottledProvider ──────────────────────────────────────────────────────
+/**
+ * Wraps an LLMProvider with throttle.acquire() before each chat call.
+ * Delegates name, defaultModel, isConfigured() to inner provider.
+ */
+export class ThrottledProvider {
+    inner;
+    throttle;
+    constructor(inner, throttle) {
+        this.inner = inner;
+        this.throttle = throttle;
+    }
+    get name() {
+        return this.inner.name;
+    }
+    get defaultModel() {
+        return this.inner.defaultModel;
+    }
+    isConfigured() {
+        return this.inner.isConfigured();
+    }
+    async chat(messages, options) {
+        await this.throttle.acquire();
+        try {
+            const result = await this.inner.chat(messages, options);
+            // V3: Feed rate limit info from response headers for proactive throttle
+            if (result.rateLimitInfo) {
+                this.throttle.feedRateLimitInfo(result.rateLimitInfo);
+            }
+            this.throttle.recover();
+            return result;
+        }
+        catch (error) {
+            if (isRateLimitError(error)) {
+                this.throttle.backoff();
+            }
+            throw error;
+        }
+    }
+    /**
+     * Forward chatWithTools if the inner provider supports it.
+     */
+    async chatWithTools(...args) {
+        await this.throttle.acquire();
+        const inner = this.inner;
+        if (typeof inner.chatWithTools === 'function') {
+            try {
+                const result = await inner.chatWithTools(...args);
+                this.throttle.recover();
+                return result;
+            }
+            catch (error) {
+                if (isRateLimitError(error)) {
+                    this.throttle.backoff();
+                }
+                throw error;
+            }
+        }
+        throw new Error('Inner provider does not support chatWithTools');
+    }
+    /** Get the underlying throttle for stats/inspection. */
+    getThrottle() {
+        return this.throttle;
+    }
+}
+// ─── Factory ────────────────────────────────────────────────────────────────
+/**
+ * Create a throttled wrapper around an LLM provider.
+ */
+export function createThrottledProvider(provider, config = FREE_TIER_THROTTLE) {
+    const throttle = new SwarmThrottle(config);
+    return new ThrottledProvider(provider, throttle);
+}
+// ─── Helpers ────────────────────────────────────────────────────────────────
+function sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms));
+}
+/** Check if an error is a rate limit (429) or spend limit (402) error. */
+function isRateLimitError(error) {
+    const msg = String(error instanceof Error ? error.message : error).toLowerCase();
+    return msg.includes('429') || msg.includes('rate') || msg.includes('too many') || msg.includes('402');
+}
+//# sourceMappingURL=request-throttle.js.map

package/dist/src/integrations/swarm/request-throttle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-throttle.js","sourceRoot":"","sources":["../../../../src/integrations/swarm/request-throttle.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAeH,MAAM,CAAC,MAAM,kBAAkB,GAAmB;IAChD,aAAa,EAAE,CAAC;IAChB,mBAAmB,EAAE,GAAG;IACxB,YAAY,EAAE,IAAI;CACnB,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAmB;IAChD,aAAa,EAAE,CAAC;IAChB,mBAAmB,EAAE,GAAG;IACxB,YAAY,EAAE,GAAG;CAClB,CAAC;AAaF,+EAA+E;AAE/E;;;;;;GAMG;AACH,MAAM,OAAO,aAAa;IAeJ;IAdZ,MAAM,CAAS;IACf,eAAe,GAAG,CAAC,CAAC;IACpB,cAAc,CAAS;IACvB,KAAK,GAAsB,EAAE,CAAC;IAC9B,aAAa,GAAG,CAAC,CAAC;IAClB,WAAW,GAA0C,IAAI,CAAC;IAElE,yBAAyB;IACjB,cAAc,CAA2B;IACzC,aAAa,GAAG,CAAC,CAAC;IAClB,eAAe,GAAG,CAAC,CAAC;IACpB,MAAM,CAAU,iBAAiB,GAAG,CAAC,CAAC;IACtC,MAAM,CAAU,mBAAmB,GAAG,MAAM,CAAC;IAErD,YAAoB,MAAsB;QAAtB,WAAM,GAAN,MAAM,CAAgB;QACxC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,cAAc,GAAG,EAAE,GAAG,MAAM,EAAE,CAAC;IACtC,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,OAAO;QACX,gEAAgE;QAChE,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;gBAChC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzB,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC;QAED,0BAA0B;QAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,OAAO,GAAG,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC;QAC3C,IAAI,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,IAAI,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;YACnE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,OAAO,CAAC;YAClD,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC;QACtB,CAAC;QAED,mDAAmD;QACnD,IAAI,CAAC,MAAM,EAAE,CAAC;QACd,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAC3C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,IAAI,CAAC,MAAM,EAAE,CAAC;QACd,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;YAC/B,eAAe,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC;YACxC,aAAa,EAAE,IAAI,CAAC,aAAa;YACjC,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,oBAAoB,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;YAC/C,mBAAmB,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY;SAC9C,CAAC;IACJ,CAAC;IAED,mDAAmD;IACnD,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,iBAAiB;YAAE,OAAO;QAClE,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAElC,IAAI,CAAC,MAAM,GAAG;YACZ,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;YACrE,YAAY,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,GAAG,CAAC,CAAC;YAC1D,mBAAmB,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,GAAG,CAAC,CAAC;SACxE,CAAC;QAEF,oCAAoC;QACpC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;IACjE,CAAC;IAED;;;OAGG;IACH,OAAO;QACL,IAAI,IAAI,CAAC,aAAa,IAAI,CAAC;YAAE,OAAO;QACpC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,eAAe,GAAG,aAAa,CAAC,mBAAmB;YAAE,OAAO;QAElF,IAAI,CAAC,aAAa,EAAE,CAAC;QAErB,0CAA0C;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC,iBAAiB,CAAC;QACnE,IAAI,CAAC,MAAM,GAAG;YACZ,aAAa,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CACnC,IAAI,CAAC,cAAc,CAAC,aAAa,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,GAAG,CAAC,CACtD,CAAC;YACF,YAAY,EAAE,IAAI,CAAC,KAAK,CACtB,IAAI,CAAC,cAAc,CAAC,YAAY,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAC/C;YACD,mBAAmB,EAAE,IAAI,CAAC,cAAc,CAAC,mBAAmB,GAAG,CAAC,CAAC,GAAG,KAAK,GAAG,GAAG,CAAC;SACjF,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,iBAAiB,CAAC,IAAqF;QACrG,iEAAiE;QACjE,IAAI,IAAI,CAAC,iBAAiB,KAAK,SAAS,IAAI,IAAI,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;YACvE,IAAI,IAAI,CAAC,aAAa,GAAG,aAAa,CAAC,iBAAiB,EAAE,CAAC;gBACzD,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,CAAC;QACH,CAAC;QACD,6CAA6C;QAC7C,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,IAAI,IAAI,CAAC,YAAY,GAAG,CAAC,IAAI,IAAI,CAAC,iBAAiB,KAAK,SAAS,IAAI,IAAI,CAAC,iBAAiB,IAAI,CAAC,EAAE,CAAC;YACpI,IAAI,CAAC,MAAM,GAAG;gBACZ,GAAG,IAAI,CAAC,MAAM;gBACd,YAAY,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;aAC3E,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,MAAM;QACZ,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,UAAU,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC;QACtD,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,CACpB,IAAI,CAAC,MAAM,CAAC,aAAa,EACzB,IAAI,CAAC,MAAM,GAAG,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAC3D,CAAC;YACF,IAAI,CAAC,cAAc,GAAG,GAAG,CAAC;QAC5B,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;;OAGG;IACK,iBAAiB;QACvB,IAAI,IAAI,CAAC,WAAW,KAAK,IAAI;YAAE,OAAO;QAEtC,0DAA0D;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;QAChE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC,GAAG,EAAE;YAClC,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,0CAA0C;YAC1C,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;gBACjD,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACnC,IAAI,OAAO;oBAAE,OAAO,EAAE,CAAC;YACzB,CAAC;YACD,mCAAmC;YACnC,IAAI,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,IAAI,CAAC,WAAW,KAAK,IAAI,EAAE,CAAC;gBACzD,aAAa,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAChC,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;YAC1B,CAAC;QACH,CAAC,EAAE,UAAU,CAAC,CAAC;IACjB,CAAC;;AAGH,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,OAAO,iBAAiB;IAElB;IACA;IAFV,YACU,KAAkB,EAClB,QAAuB;QADvB,UAAK,GAAL,KAAK,CAAa;QAClB,aAAQ,GAAR,QAAQ,CAAe;IAC9B,CAAC;IAEJ,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC;IACjC,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,QAA0C,EAAE,OAAqB;QAC1E,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACxD,wEAAwE;YACxE,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACzB,IAAI,CAAC,QAAQ,CAAC,iBAAiB,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YACxD,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5B,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,GAAG,IAAe;QACpC,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAgE,CAAC;QACpF,IAAI,OAAO,KAAK,CAAC,aAAa,KAAK,UAAU,EAAE,CAAC;YAC9C,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,CAAC;gBAClD,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACxB,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC5B,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBAC1B,CAAC;gBACD,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,wDAAwD;IACxD,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;CACF;AAED,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,uBAAuB,CACrC,QAAqB,EACrB,SAAyB,kBAAkB;IAE3C,MAAM,QAAQ,GAAG,IAAI,aAAa,CAAC,MAAM,CAAC,CAAC;IAC3C,OAAO,IAAI,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;AACnD,CAAC;AAED,+EAA+E;AAE/E,SAAS,KAAK,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,0EAA0E;AAC1E,SAAS,gBAAgB,CAAC,KAAc;IACtC,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;IACjF,OAAO,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AACxG,CAAC"}

package/dist/src/integrations/swarm/swarm-budget.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Swarm Budget Pool
+ *
+ * Wraps SharedBudgetPool with swarm-specific defaults for managing
+ * budgets across 10-20+ tiny workers.
+ */
+import { type SharedBudgetPool, type BudgetPoolStats } from '../budget-pool.js';
+import type { SwarmConfig } from './types.js';
+export interface SwarmBudgetPool {
+    /** Underlying budget pool */
+    pool: SharedBudgetPool;
+    /** Reserve budget for the orchestrator (quality gates, synthesis) */
+    orchestratorReserve: number;
+    /** Max tokens per individual worker */
+    maxPerWorker: number;
+    /** Max cost per worker */
+    maxCostPerWorker: number;
+    /** Get pool stats */
+    getStats(): BudgetPoolStats;
+    /** Check if there's budget remaining for more workers */
+    hasCapacity(): boolean;
+}
+/**
+ * Create a swarm budget pool from SwarmConfig.
+ *
+ * Splits the total budget between:
+ * - Orchestrator reserve (default 15%): for decomposition, quality gates, synthesis
+ * - Worker pool (remaining 85%): shared among all workers with per-worker caps
+ */
+export declare function createSwarmBudgetPool(config: SwarmConfig): SwarmBudgetPool;
+//# sourceMappingURL=swarm-budget.d.ts.map

package/dist/src/integrations/swarm/swarm-budget.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"swarm-budget.d.ts","sourceRoot":"","sources":["../../../../src/integrations/swarm/swarm-budget.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAoB,KAAK,gBAAgB,EAAE,KAAK,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAClG,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAI9C,MAAM,WAAW,eAAe;IAC9B,6BAA6B;IAC7B,IAAI,EAAE,gBAAgB,CAAC;IAEvB,qEAAqE;IACrE,mBAAmB,EAAE,MAAM,CAAC;IAE5B,uCAAuC;IACvC,YAAY,EAAE,MAAM,CAAC;IAErB,0BAA0B;IAC1B,gBAAgB,EAAE,MAAM,CAAC;IAEzB,qBAAqB;IACrB,QAAQ,IAAI,eAAe,CAAC;IAE5B,yDAAyD;IACzD,WAAW,IAAI,OAAO,CAAC;CACxB;AAED;;;;;;GAMG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,WAAW,GAAG,eAAe,CA4B1E"}

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

@@ -0,0 +1,36 @@
+/**
+ * Swarm Budget Pool
+ *
+ * Wraps SharedBudgetPool with swarm-specific defaults for managing
+ * budgets across 10-20+ tiny workers.
+ */
+import { createBudgetPool } from '../budget-pool.js';
+/**
+ * Create a swarm budget pool from SwarmConfig.
+ *
+ * Splits the total budget between:
+ * - Orchestrator reserve (default 15%): for decomposition, quality gates, synthesis
+ * - Worker pool (remaining 85%): shared among all workers with per-worker caps
+ */
+export function createSwarmBudgetPool(config) {
+    const orchestratorReserve = Math.floor(config.totalBudget * config.orchestratorReserveRatio);
+    const workerPoolBudget = config.totalBudget - orchestratorReserve;
+    // Per-worker cost cap: total cost divided by expected workers, with safety margin
+    const estimatedWorkers = Math.max(5, config.maxConcurrency * 3);
+    const maxCostPerWorker = config.maxCost / estimatedWorkers;
+    const pool = createBudgetPool(workerPoolBudget, 0, // No additional reserve within the pool (already reserved at swarm level)
+    config.maxTokensPerWorker);
+    return {
+        pool,
+        orchestratorReserve,
+        maxPerWorker: config.maxTokensPerWorker,
+        maxCostPerWorker,
+        getStats() {
+            return pool.getStats();
+        },
+        hasCapacity() {
+            return pool.hasCapacity();
+        },
+    };
+}
+//# sourceMappingURL=swarm-budget.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"swarm-budget.js","sourceRoot":"","sources":["../../../../src/integrations/swarm/swarm-budget.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,gBAAgB,EAA+C,MAAM,mBAAmB,CAAC;AAyBlG;;;;;;GAMG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAAmB;IACvD,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,GAAG,MAAM,CAAC,wBAAwB,CAAC,CAAC;IAC7F,MAAM,gBAAgB,GAAG,MAAM,CAAC,WAAW,GAAG,mBAAmB,CAAC;IAElE,kFAAkF;IAClF,MAAM,gBAAgB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;IAChE,MAAM,gBAAgB,GAAG,MAAM,CAAC,OAAO,GAAG,gBAAgB,CAAC;IAE3D,MAAM,IAAI,GAAG,gBAAgB,CAC3B,gBAAgB,EAChB,CAAC,EAAE,0EAA0E;IAC7E,MAAM,CAAC,kBAAkB,CAC1B,CAAC;IAEF,OAAO;QACL,IAAI;QACJ,mBAAmB;QACnB,YAAY,EAAE,MAAM,CAAC,kBAAkB;QACvC,gBAAgB;QAEhB,QAAQ;YACN,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QAED,WAAW;YACT,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC;QAC5B,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/src/integrations/swarm/swarm-config-loader.d.ts

@@ -0,0 +1,51 @@
+/**
+ * Swarm Config Loader
+ *
+ * YAML parser + config loader + merge logic for swarm.yaml files.
+ * Supports nested objects, block arrays, multiline strings, and type coercion.
+ *
+ * Search order:
+ *   1. {cwd}/.attocode/swarm.yaml
+ *   2. {cwd}/.attocode/swarm.yml
+ *   3. {cwd}/.attocode/swarm.json (fallback)
+ *   4. {homedir}/.attocode/swarm.yaml
+ *   5. null if none found
+ *
+ * Merge order: DEFAULT < yaml < CLI overrides
+ */
+import type { SwarmConfig } from './types.js';
+import { DEFAULT_SWARM_CONFIG } from './types.js';
+/** Raw parsed YAML — untyped nested structure */
+export type SwarmYamlConfig = Record<string, unknown>;
+/**
+ * Parse a simple YAML string into a nested object.
+ *
+ * Supports:
+ * - Key-value pairs (key: value)
+ * - Nested objects via indentation (2-space indent)
+ * - Block arrays (- item, including objects within arrays)
+ * - Multiline strings (| indicator)
+ * - Type coercion (bool, number, null)
+ * - Comments (# ...) and blank lines
+ *
+ * NOT a general-purpose YAML parser — handles the swarm.yaml schema.
+ */
+export declare function parseSwarmYaml(content: string): SwarmYamlConfig;
+/**
+ * Load swarm config from filesystem.
+ * Returns null if no config file found.
+ */
+export declare function loadSwarmYamlConfig(cwd?: string): SwarmYamlConfig | null;
+/**
+ * Map parsed YAML config to SwarmConfig fields.
+ */
+export declare function yamlToSwarmConfig(yaml: SwarmYamlConfig, orchestratorModel: string): Partial<SwarmConfig>;
+/**
+ * Merge configs: DEFAULT < yaml < CLI overrides.
+ */
+export declare function mergeSwarmConfigs(defaults: typeof DEFAULT_SWARM_CONFIG, yamlConfig: Partial<SwarmConfig> | null, cliOverrides: {
+    paidOnly?: boolean;
+    orchestratorModel: string;
+    resumeSessionId?: string;
+}): SwarmConfig;
+//# sourceMappingURL=swarm-config-loader.d.ts.map

package/dist/src/integrations/swarm/swarm-config-loader.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"swarm-config-loader.d.ts","sourceRoot":"","sources":["../../../../src/integrations/swarm/swarm-config-loader.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,OAAO,KAAK,EAAE,WAAW,EAAqC,MAAM,YAAY,CAAC;AACjF,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAIlD,iDAAiD;AACjD,MAAM,MAAM,eAAe,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAItD;;;;;;;;;;;;GAYG;AACH,wBAAgB,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,eAAe,CA0J/D;AAID;;;GAGG;AACH,wBAAgB,mBAAmB,CAAC,GAAG,CAAC,EAAE,MAAM,GAAG,eAAe,GAAG,IAAI,CA0BxE;AAID;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,eAAe,EAAE,iBAAiB,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,CAAC,CAqIxG;AAID;;GAEG;AACH,wBAAgB,iBAAiB,CAC/B,QAAQ,EAAE,OAAO,oBAAoB,EACrC,UAAU,EAAE,OAAO,CAAC,WAAW,CAAC,GAAG,IAAI,EACvC,YAAY,EAAE;IAAE,QAAQ,CAAC,EAAE,OAAO,CAAC;IAAC,iBAAiB,EAAE,MAAM,CAAC;IAAC,eAAe,CAAC,EAAE,MAAM,CAAA;CAAE,GACxF,WAAW,CA+Bb"}