@just-every/ensemble 0.2.87 → 0.2.88

package/dist/cjs/utils/quota_tracker.cjs

@@ -0,0 +1,311 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.quotaTracker = exports.QuotaTracker = void 0;
+class QuotaTracker {
+    quotas = {};
+    updateCallback;
+    openAIFreeQuota = {
+        gpt4Family: {
+            limit: 1000000,
+            used: 0,
+        },
+        gptMiniFamily: {
+            limit: 10000000,
+            used: 0,
+        },
+        gpt4Models: ['gpt-4.5-preview', 'gpt-4.1', 'gpt-4o', 'o1', 'o3'],
+        gptMiniModels: ['gpt-4.1-mini', 'gpt-4.1-nano', 'gpt-4o-mini', 'o1-mini', 'o4-mini'],
+    };
+    constructor(updateCallback) {
+        this.updateCallback = updateCallback;
+        this.initializeProviderQuotas();
+    }
+    setUpdateCallback(callback) {
+        this.updateCallback = callback;
+    }
+    initializeProviderQuotas() {
+        this.quotas['google'] = {
+            provider: 'google',
+            creditBalance: 0,
+            creditLimit: 0,
+            lastResetDate: new Date(),
+            models: {
+                'gemini-2.5-pro-exp-03-25': {
+                    model: 'gemini-2.5-pro-exp-03-25',
+                    dailyTokenLimit: 1000000,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 25,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 5,
+                        tokensPerMinute: 250000,
+                    },
+                    lastResetDate: new Date(),
+                },
+                'gemini-2.5-flash-preview-04-17': {
+                    model: 'gemini-2.5-flash-preview-04-17',
+                    dailyTokenLimit: 0,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 500,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 10,
+                        tokensPerMinute: 250000,
+                    },
+                    lastResetDate: new Date(),
+                },
+                'gemini-2.0-flash': {
+                    model: 'gemini-2.0-flash',
+                    dailyTokenLimit: 0,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 1500,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 15,
+                        tokensPerMinute: 1000000,
+                    },
+                    lastResetDate: new Date(),
+                },
+                'gemini-2.0-flash-lite': {
+                    model: 'gemini-2.0-flash-lite',
+                    dailyTokenLimit: 0,
+                    dailyTokensUsed: 0,
+                    dailyRequestLimit: 1500,
+                    dailyRequestsUsed: 0,
+                    rateLimit: {
+                        requestsPerMinute: 30,
+                        tokensPerMinute: 1000000,
+                    },
+                    lastResetDate: new Date(),
+                },
+            },
+        };
+        this.quotas['openai'] = {
+            provider: 'openai',
+            creditBalance: 0,
+            creditLimit: 0,
+            lastResetDate: new Date(),
+            info: {
+                freeQuota: this.openAIFreeQuota,
+            },
+            models: {},
+        };
+        this.quotas['xai'] = {
+            provider: 'xai',
+            creditBalance: 150,
+            creditLimit: 150,
+            lastResetDate: new Date(),
+            models: {},
+        };
+    }
+    getProviderQuota(provider) {
+        return (this.quotas[provider] || {
+            provider,
+            models: {},
+        });
+    }
+    getModelQuota(provider, model) {
+        const providerQuota = this.getProviderQuota(provider);
+        if (providerQuota.models[model]) {
+            return providerQuota.models[model];
+        }
+        return null;
+    }
+    trackUsage(provider, model, inputTokens, outputTokens) {
+        if (!this.quotas[provider]) {
+            return true;
+        }
+        const providerQuota = this.quotas[provider];
+        let significantChange = false;
+        const totalTokens = inputTokens + outputTokens;
+        const today = new Date();
+        if (providerQuota.lastResetDate &&
+            (today.getUTCDate() !== providerQuota.lastResetDate.getUTCDate() ||
+                today.getUTCMonth() !== providerQuota.lastResetDate.getUTCMonth() ||
+                today.getUTCFullYear() !== providerQuota.lastResetDate.getUTCFullYear())) {
+            for (const modelKey in providerQuota.models) {
+                const modelQuota = providerQuota.models[modelKey];
+                modelQuota.dailyTokensUsed = 0;
+                modelQuota.dailyRequestsUsed = 0;
+                modelQuota.lastResetDate = today;
+            }
+            if (provider === 'openai' && providerQuota.info?.freeQuota) {
+                const freeQuota = providerQuota.info.freeQuota;
+                freeQuota.gpt4Family.used = 0;
+                freeQuota.gptMiniFamily.used = 0;
+            }
+            providerQuota.lastResetDate = today;
+            significantChange = true;
+        }
+        if (provider === 'openai' && providerQuota.info?.freeQuota) {
+            const freeQuota = providerQuota.info.freeQuota;
+            if (freeQuota.gpt4Models.includes(model)) {
+                const prevUsed = freeQuota.gpt4Family.used;
+                freeQuota.gpt4Family.used += totalTokens;
+                console.log(`[QuotaTracker] OpenAI GPT-4 family usage: ${freeQuota.gpt4Family.used}/${freeQuota.gpt4Family.limit}`);
+                const prevPercent = Math.floor((prevUsed / freeQuota.gpt4Family.limit) * 10);
+                const currPercent = Math.floor((freeQuota.gpt4Family.used / freeQuota.gpt4Family.limit) * 10);
+                if (prevPercent !== currPercent) {
+                    significantChange = true;
+                }
+                if (freeQuota.gpt4Family.used >= freeQuota.gpt4Family.limit) {
+                    console.log(`[QuotaTracker] OpenAI GPT-4 family daily limit reached: ${freeQuota.gpt4Family.used} > ${freeQuota.gpt4Family.limit}`);
+                    significantChange = true;
+                    this.sendQuotaUpdate();
+                    return false;
+                }
+            }
+            if (freeQuota.gptMiniModels.includes(model)) {
+                const prevUsed = freeQuota.gptMiniFamily.used;
+                freeQuota.gptMiniFamily.used += totalTokens;
+                console.log(`[QuotaTracker] OpenAI GPT-Mini family usage: ${freeQuota.gptMiniFamily.used}/${freeQuota.gptMiniFamily.limit}`);
+                const prevPercent = Math.floor((prevUsed / freeQuota.gptMiniFamily.limit) * 10);
+                const currPercent = Math.floor((freeQuota.gptMiniFamily.used / freeQuota.gptMiniFamily.limit) * 10);
+                if (prevPercent !== currPercent) {
+                    significantChange = true;
+                }
+                if (freeQuota.gptMiniFamily.used >= freeQuota.gptMiniFamily.limit) {
+                    console.log(`[QuotaTracker] OpenAI GPT-Mini family daily limit reached: ${freeQuota.gptMiniFamily.used} > ${freeQuota.gptMiniFamily.limit}`);
+                    significantChange = true;
+                    this.sendQuotaUpdate();
+                    return false;
+                }
+            }
+        }
+        const modelQuota = this.getModelQuota(provider, model);
+        if (!modelQuota && provider !== 'openai') {
+            return true;
+        }
+        if (modelQuota) {
+            const previousDailyTokensUsed = modelQuota.dailyTokensUsed;
+            modelQuota.dailyTokensUsed += totalTokens;
+            const previousDailyRequestsUsed = modelQuota.dailyRequestsUsed;
+            modelQuota.dailyRequestsUsed += 1;
+            if (modelQuota.dailyTokenLimit > 0) {
+                const previousPercent = Math.floor((previousDailyTokensUsed / modelQuota.dailyTokenLimit) * 10);
+                const currentPercent = Math.floor((modelQuota.dailyTokensUsed / modelQuota.dailyTokenLimit) * 10);
+                if (previousPercent !== currentPercent) {
+                    significantChange = true;
+                }
+            }
+            if (modelQuota.dailyRequestLimit > 0) {
+                const previousPercent = Math.floor((previousDailyRequestsUsed / modelQuota.dailyRequestLimit) * 10);
+                const currentPercent = Math.floor((modelQuota.dailyRequestsUsed / modelQuota.dailyRequestLimit) * 10);
+                if (previousPercent !== currentPercent) {
+                    significantChange = true;
+                }
+            }
+            if (modelQuota.dailyTokenLimit > 0 && modelQuota.dailyTokensUsed >= modelQuota.dailyTokenLimit) {
+                console.log(`[QuotaTracker] ${provider} model ${model} daily token limit reached: ${modelQuota.dailyTokensUsed} > ${modelQuota.dailyTokenLimit}`);
+                significantChange = true;
+                this.sendQuotaUpdate();
+                return false;
+            }
+            if (modelQuota.dailyRequestLimit > 0 && modelQuota.dailyRequestsUsed >= modelQuota.dailyRequestLimit) {
+                console.log(`[QuotaTracker] ${provider} model ${model} daily request limit reached: ${modelQuota.dailyRequestsUsed} > ${modelQuota.dailyRequestLimit}`);
+                significantChange = true;
+                this.sendQuotaUpdate();
+                return false;
+            }
+        }
+        if (significantChange) {
+            this.sendQuotaUpdate();
+        }
+        return true;
+    }
+    hasQuota(provider, model) {
+        if (provider === 'openai') {
+            return this.hasOpenAIFreeQuota(model);
+        }
+        const modelQuota = this.getModelQuota(provider, model);
+        if (modelQuota) {
+            return ((modelQuota.dailyTokenLimit === 0 || modelQuota.dailyTokensUsed < modelQuota.dailyTokenLimit) &&
+                (modelQuota.dailyRequestLimit === 0 || modelQuota.dailyRequestsUsed < modelQuota.dailyRequestLimit));
+        }
+        return true;
+    }
+    hasOpenAIFreeQuota(model) {
+        const providerQuota = this.getProviderQuota('openai');
+        if (!providerQuota || !providerQuota.info?.freeQuota)
+            return true;
+        const freeQuota = providerQuota.info.freeQuota;
+        if (freeQuota.gpt4Models.includes(model)) {
+            return freeQuota.gpt4Family.used < freeQuota.gpt4Family.limit;
+        }
+        if (freeQuota.gptMiniModels.includes(model)) {
+            return freeQuota.gptMiniFamily.used < freeQuota.gptMiniFamily.limit;
+        }
+        return true;
+    }
+    trackCreditUsage(provider, creditAmount) {
+        const providerQuota = this.getProviderQuota(provider);
+        if (!providerQuota || providerQuota.creditBalance === undefined) {
+            return;
+        }
+        if (providerQuota.creditBalance !== undefined) {
+            providerQuota.creditBalance = Math.max(0, providerQuota.creditBalance - creditAmount);
+        }
+    }
+    getCreditBalance(provider) {
+        return this.getProviderQuota(provider)?.creditBalance || 0;
+    }
+    getSummary() {
+        const summary = {};
+        for (const [provider, providerQuota] of Object.entries(this.quotas)) {
+            summary[provider] = {
+                creditBalance: providerQuota.creditBalance,
+                creditLimit: providerQuota.creditLimit,
+                models: {},
+            };
+            for (const [modelName, modelQuota] of Object.entries(providerQuota.models)) {
+                summary[provider].models[modelName] = {
+                    tokens: {
+                        used: modelQuota.dailyTokensUsed,
+                        limit: modelQuota.dailyTokenLimit,
+                        percent: modelQuota.dailyTokenLimit > 0
+                            ? (modelQuota.dailyTokensUsed / modelQuota.dailyTokenLimit) * 100
+                            : 0,
+                    },
+                    requests: {
+                        used: modelQuota.dailyRequestsUsed,
+                        limit: modelQuota.dailyRequestLimit,
+                        percent: modelQuota.dailyRequestLimit > 0
+                            ? (modelQuota.dailyRequestsUsed / modelQuota.dailyRequestLimit) * 100
+                            : 0,
+                    },
+                    lastReset: modelQuota.lastResetDate,
+                };
+            }
+            if (provider === 'openai' && providerQuota.info?.freeQuota) {
+                const freeQuota = providerQuota.info.freeQuota;
+                summary[provider].freeTier = {
+                    gpt4Family: {
+                        used: freeQuota.gpt4Family.used,
+                        limit: freeQuota.gpt4Family.limit,
+                        percent: (freeQuota.gpt4Family.used / freeQuota.gpt4Family.limit) * 100,
+                    },
+                    gptMiniFamily: {
+                        used: freeQuota.gptMiniFamily.used,
+                        limit: freeQuota.gptMiniFamily.limit,
+                        percent: (freeQuota.gptMiniFamily.used / freeQuota.gptMiniFamily.limit) * 100,
+                    },
+                };
+            }
+        }
+        return summary;
+    }
+    sendQuotaUpdate() {
+        try {
+            if (this.updateCallback) {
+                const quotas = this.getSummary();
+                this.updateCallback(quotas);
+            }
+        }
+        catch (error) {
+            console.error('Error sending quota update:', error);
+        }
+    }
+}
+exports.QuotaTracker = QuotaTracker;
+exports.quotaTracker = new QuotaTracker();
+//# sourceMappingURL=quota_tracker.js.map

package/dist/cjs/utils/quota_tracker.d.ts

@@ -0,0 +1,22 @@
+import { ModelProviderID, ModelSpecificQuota, ProviderQuota } from "../types/types.cjs";
+export type { ModelSpecificQuota, ProviderQuota };
+export type QuotaUpdateCallback = (quotaSummary: Record<string, any>) => void;
+export declare class QuotaTracker {
+    private quotas;
+    private updateCallback?;
+    private openAIFreeQuota;
+    constructor(updateCallback?: QuotaUpdateCallback);
+    setUpdateCallback(callback: QuotaUpdateCallback | undefined): void;
+    private initializeProviderQuotas;
+    getProviderQuota(provider: ModelProviderID): ProviderQuota;
+    getModelQuota(provider: ModelProviderID, model: string): ModelSpecificQuota | null;
+    trackUsage(provider: ModelProviderID, model: string, inputTokens: number, outputTokens: number): boolean;
+    hasQuota(provider: ModelProviderID, model: string): boolean;
+    hasOpenAIFreeQuota(model: string): boolean;
+    trackCreditUsage(provider: ModelProviderID, creditAmount: number): void;
+    getCreditBalance(provider: ModelProviderID): number;
+    getSummary(): Record<string, any>;
+    sendQuotaUpdate(): void;
+}
+export declare const quotaTracker: QuotaTracker;
+//# sourceMappingURL=quota_tracker.d.ts.map

package/dist/cjs/utils/quota_tracker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota_tracker.d.ts","sourceRoot":"","sources":["../../../utils/quota_tracker.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,eAAe,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAGvF,YAAY,EAAE,kBAAkB,EAAE,aAAa,EAAE,CAAC;AA4BlD,MAAM,MAAM,mBAAmB,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,KAAK,IAAI,CAAC;AAK9E,qBAAa,YAAY;IAErB,OAAO,CAAC,MAAM,CAAqC;IAGnD,OAAO,CAAC,cAAc,CAAC,CAAsB;IAG7C,OAAO,CAAC,eAAe,CAarB;gBAEU,cAAc,CAAC,EAAE,mBAAmB;IAQhD,iBAAiB,CAAC,QAAQ,EAAE,mBAAmB,GAAG,SAAS,GAAG,IAAI;IAOlE,OAAO,CAAC,wBAAwB;IA4FhC,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,aAAa;IAa1D,aAAa,CAAC,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,GAAG,kBAAkB,GAAG,IAAI;IAgBlF,UAAU,CAAC,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,OAAO;IAuKxG,QAAQ,CAAC,QAAQ,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO;IAuB3D,kBAAkB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO;IAsB1C,gBAAgB,CAAC,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,GAAG,IAAI;IAcvE,gBAAgB,CAAC,QAAQ,EAAE,eAAe,GAAG,MAAM;IAOnD,UAAU,IAAI,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC;IA2DjC,eAAe,IAAI,IAAI;CAU1B;AAGD,eAAO,MAAM,YAAY,cAAqB,CAAC"}

package/dist/cjs/utils/quota_tracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"quota_tracker.js","sourceRoot":"","sources":["../../../utils/quota_tracker.ts"],"names":[],"mappings":";;;AA0CA,MAAa,YAAY;IAEb,MAAM,GAAkC,EAAE,CAAC;IAG3C,cAAc,CAAuB;IAGrC,eAAe,GAAoB;QACvC,UAAU,EAAE;YACR,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,CAAC;SACV;QACD,aAAa,EAAE;YACX,KAAK,EAAE,QAAQ;YACf,IAAI,EAAE,CAAC;SACV;QAED,UAAU,EAAE,CAAC,iBAAiB,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,CAAC;QAEhE,aAAa,EAAE,CAAC,cAAc,EAAE,cAAc,EAAE,aAAa,EAAE,SAAS,EAAE,SAAS,CAAC;KACvF,CAAC;IAEF,YAAY,cAAoC;QAC5C,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,wBAAwB,EAAE,CAAC;IACpC,CAAC;IAKD,iBAAiB,CAAC,QAAyC;QACvD,IAAI,CAAC,cAAc,GAAG,QAAQ,CAAC;IACnC,CAAC;IAKO,wBAAwB;QAE5B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG;YACpB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,IAAI,IAAI,EAAE;YACzB,MAAM,EAAE;gBAEJ,0BAA0B,EAAE;oBACxB,KAAK,EAAE,0BAA0B;oBACjC,eAAe,EAAE,OAAO;oBACxB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,EAAE;oBACrB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,CAAC;wBACpB,eAAe,EAAE,MAAM;qBAC1B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;gBAED,gCAAgC,EAAE;oBAC9B,KAAK,EAAE,gCAAgC;oBACvC,eAAe,EAAE,CAAC;oBAClB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,GAAG;oBACtB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,EAAE;wBACrB,eAAe,EAAE,MAAM;qBAC1B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;gBAED,kBAAkB,EAAE;oBAChB,KAAK,EAAE,kBAAkB;oBACzB,eAAe,EAAE,CAAC;oBAClB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,IAAI;oBACvB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,EAAE;wBACrB,eAAe,EAAE,OAAO;qBAC3B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;gBAED,uBAAuB,EAAE;oBACrB,KAAK,EAAE,uBAAuB;oBAC9B,eAAe,EAAE,CAAC;oBAClB,eAAe,EAAE,CAAC;oBAClB,iBAAiB,EAAE,IAAI;oBACvB,iBAAiB,EAAE,CAAC;oBACpB,SAAS,EAAE;wBACP,iBAAiB,EAAE,EAAE;wBACrB,eAAe,EAAE,OAAO;qBAC3B;oBACD,aAAa,EAAE,IAAI,IAAI,EAAE;iBAC5B;aACJ;SACJ,CAAC;QAGF,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG;YACpB,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,CAAC;YAChB,WAAW,EAAE,CAAC;YACd,aAAa,EAAE,IAAI,IAAI,EAAE;YACzB,IAAI,EAAE;gBACF,SAAS,EAAE,IAAI,CAAC,eAAe;aAClC;YACD,MAAM,EAAE,EAEP;SACJ,CAAC;QAGF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG;YACjB,QAAQ,EAAE,KAAK;YACf,aAAa,EAAE,GAAG;YAClB,WAAW,EAAE,GAAG;YAChB,aAAa,EAAE,IAAI,IAAI,EAAE;YACzB,MAAM,EAAE,EAEP;SACJ,CAAC;IACN,CAAC;IAKD,gBAAgB,CAAC,QAAyB;QACtC,OAAO,CACH,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI;YACrB,QAAQ;YACR,MAAM,EAAE,EAAE;SACb,CACJ,CAAC;IACN,CAAC;IAMD,aAAa,CAAC,QAAyB,EAAE,KAAa;QAClD,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAGtD,IAAI,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;YAC9B,OAAO,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;QAGD,OAAO,IAAI,CAAC;IAChB,CAAC;IAMD,UAAU,CAAC,QAAyB,EAAE,KAAa,EAAE,WAAmB,EAAE,YAAoB;QAE1F,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC;QAChB,CAAC;QAGD,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,IAAI,iBAAiB,GAAG,KAAK,CAAC;QAC9B,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,CAAC;QAG/C,MAAM,KAAK,GAAG,IAAI,IAAI,EAAE,CAAC;QACzB,IACI,aAAa,CAAC,aAAa;YAC3B,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,aAAa,CAAC,aAAa,CAAC,UAAU,EAAE;gBAC5D,KAAK,CAAC,WAAW,EAAE,KAAK,aAAa,CAAC,aAAa,CAAC,WAAW,EAAE;gBACjE,KAAK,CAAC,cAAc,EAAE,KAAK,aAAa,CAAC,aAAa,CAAC,cAAc,EAAE,CAAC,EAC9E,CAAC;YAEC,KAAK,MAAM,QAAQ,IAAI,aAAa,CAAC,MAAM,EAAE,CAAC;gBAC1C,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAClD,UAAU,CAAC,eAAe,GAAG,CAAC,CAAC;gBAC/B,UAAU,CAAC,iBAAiB,GAAG,CAAC,CAAC;gBACjC,UAAU,CAAC,aAAa,GAAG,KAAK,CAAC;YACrC,CAAC;YAGD,IAAI,QAAQ,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;gBACzD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAA4B,CAAC;gBAClE,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,CAAC,CAAC;gBAC9B,SAAS,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,CAAC;YACrC,CAAC;YAGD,aAAa,CAAC,aAAa,GAAG,KAAK,CAAC;YACpC,iBAAiB,GAAG,IAAI,CAAC;QAC7B,CAAC;QAGD,IAAI,QAAQ,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YACzD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAA4B,CAAC;YAGlE,IAAI,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvC,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC;gBAC3C,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,WAAW,CAAC;gBACzC,OAAO,CAAC,GAAG,CACP,6CAA6C,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,CACzG,CAAC;gBAGF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC7E,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;gBAC9F,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;oBAC9B,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;gBAGD,IAAI,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBAC1D,OAAO,CAAC,GAAG,CACP,2DAA2D,SAAS,CAAC,UAAU,CAAC,IAAI,MAAM,SAAS,CAAC,UAAU,CAAC,KAAK,EAAE,CACzH,CAAC;oBACF,iBAAiB,GAAG,IAAI,CAAC;oBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;oBACvB,OAAO,KAAK,CAAC;gBACjB,CAAC;YACL,CAAC;YAGD,IAAI,SAAS,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC1C,MAAM,QAAQ,GAAG,SAAS,CAAC,aAAa,CAAC,IAAI,CAAC;gBAC9C,SAAS,CAAC,aAAa,CAAC,IAAI,IAAI,WAAW,CAAC;gBAC5C,OAAO,CAAC,GAAG,CACP,gDAAgD,SAAS,CAAC,aAAa,CAAC,IAAI,IAAI,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,CAClH,CAAC;gBAGF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;gBAChF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpG,IAAI,WAAW,KAAK,WAAW,EAAE,CAAC;oBAC9B,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;gBAGD,IAAI,SAAS,CAAC,aAAa,CAAC,IAAI,IAAI,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;oBAChE,OAAO,CAAC,GAAG,CACP,8DAA8D,SAAS,CAAC,aAAa,CAAC,IAAI,MAAM,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,CAClI,CAAC;oBACF,iBAAiB,GAAG,IAAI,CAAC;oBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;oBACvB,OAAO,KAAK,CAAC;gBACjB,CAAC;YACL,CAAC;QACL,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAGvD,IAAI,CAAC,UAAU,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC;QAChB,CAAC;QAGD,IAAI,UAAU,EAAE,CAAC;YAEb,MAAM,uBAAuB,GAAG,UAAU,CAAC,eAAe,CAAC;YAC3D,UAAU,CAAC,eAAe,IAAI,WAAW,CAAC;YAG1C,MAAM,yBAAyB,GAAG,UAAU,CAAC,iBAAiB,CAAC;YAC/D,UAAU,CAAC,iBAAiB,IAAI,CAAC,CAAC;YAGlC,IAAI,UAAU,CAAC,eAAe,GAAG,CAAC,EAAE,CAAC;gBACjC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,uBAAuB,GAAG,UAAU,CAAC,eAAe,CAAC,GAAG,EAAE,CAAC,CAAC;gBAChG,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC,GAAG,EAAE,CAAC,CAAC;gBAElG,IAAI,eAAe,KAAK,cAAc,EAAE,CAAC;oBACrC,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;YACL,CAAC;YAGD,IAAI,UAAU,CAAC,iBAAiB,GAAG,CAAC,EAAE,CAAC;gBACnC,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,yBAAyB,GAAG,UAAU,CAAC,iBAAiB,CAAC,GAAG,EAAE,CAAC,CAAC;gBACpG,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC,iBAAiB,GAAG,UAAU,CAAC,iBAAiB,CAAC,GAAG,EAAE,CAAC,CAAC;gBAEtG,IAAI,eAAe,KAAK,cAAc,EAAE,CAAC;oBACrC,iBAAiB,GAAG,IAAI,CAAC;gBAC7B,CAAC;YACL,CAAC;YAGD,IAAI,UAAU,CAAC,eAAe,GAAG,CAAC,IAAI,UAAU,CAAC,eAAe,IAAI,UAAU,CAAC,eAAe,EAAE,CAAC;gBAC7F,OAAO,CAAC,GAAG,CACP,kBAAkB,QAAQ,UAAU,KAAK,+BAA+B,UAAU,CAAC,eAAe,MAAM,UAAU,CAAC,eAAe,EAAE,CACvI,CAAC;gBACF,iBAAiB,GAAG,IAAI,CAAC;gBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,KAAK,CAAC;YACjB,CAAC;YAGD,IAAI,UAAU,CAAC,iBAAiB,GAAG,CAAC,IAAI,UAAU,CAAC,iBAAiB,IAAI,UAAU,CAAC,iBAAiB,EAAE,CAAC;gBACnG,OAAO,CAAC,GAAG,CACP,kBAAkB,QAAQ,UAAU,KAAK,iCAAiC,UAAU,CAAC,iBAAiB,MAAM,UAAU,CAAC,iBAAiB,EAAE,CAC7I,CAAC;gBACF,iBAAiB,GAAG,IAAI,CAAC;gBACzB,IAAI,CAAC,eAAe,EAAE,CAAC;gBACvB,OAAO,KAAK,CAAC;YACjB,CAAC;QACL,CAAC;QAGD,IAAI,iBAAiB,EAAE,CAAC;YACpB,IAAI,CAAC,eAAe,EAAE,CAAC;QAC3B,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAMD,QAAQ,CAAC,QAAyB,EAAE,KAAa;QAE7C,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC1C,CAAC;QAGD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QACvD,IAAI,UAAU,EAAE,CAAC;YAEb,OAAO,CACH,CAAC,UAAU,CAAC,eAAe,KAAK,CAAC,IAAI,UAAU,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC;gBAC7F,CAAC,UAAU,CAAC,iBAAiB,KAAK,CAAC,IAAI,UAAU,CAAC,iBAAiB,GAAG,UAAU,CAAC,iBAAiB,CAAC,CACtG,CAAC;QACN,CAAC;QAGD,OAAO,IAAI,CAAC;IAChB,CAAC;IAKD,kBAAkB,CAAC,KAAa;QAC5B,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,SAAS;YAAE,OAAO,IAAI,CAAC;QAElE,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAA4B,CAAC;QAGlE,IAAI,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,OAAO,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC;QAClE,CAAC;QAGD,IAAI,SAAS,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1C,OAAO,SAAS,CAAC,aAAa,CAAC,IAAI,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC;QACxE,CAAC;QAED,OAAO,IAAI,CAAC;IAChB,CAAC;IAKD,gBAAgB,CAAC,QAAyB,EAAE,YAAoB;QAC5D,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YAC9D,OAAO;QACX,CAAC;QAED,IAAI,aAAa,CAAC,aAAa,KAAK,SAAS,EAAE,CAAC;YAC5C,aAAa,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,aAAa,CAAC,aAAa,GAAG,YAAY,CAAC,CAAC;QAC1F,CAAC;IACL,CAAC;IAKD,gBAAgB,CAAC,QAAyB;QACtC,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,EAAE,aAAa,IAAI,CAAC,CAAC;IAC/D,CAAC;IAKD,UAAU;QACN,MAAM,OAAO,GAAwB,EAAE,CAAC;QAExC,KAAK,MAAM,CAAC,QAAQ,EAAE,aAAa,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAElE,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAChB,aAAa,EAAE,aAAa,CAAC,aAAa;gBAC1C,WAAW,EAAE,aAAa,CAAC,WAAW;gBACtC,MAAM,EAAE,EAAE;aACb,CAAC;YAGF,KAAK,MAAM,CAAC,SAAS,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;gBAEzE,OAAO,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG;oBAClC,MAAM,EAAE;wBACJ,IAAI,EAAE,UAAU,CAAC,eAAe;wBAChC,KAAK,EAAE,UAAU,CAAC,eAAe;wBACjC,OAAO,EACH,UAAU,CAAC,eAAe,GAAG,CAAC;4BAC1B,CAAC,CAAC,CAAC,UAAU,CAAC,eAAe,GAAG,UAAU,CAAC,eAAe,CAAC,GAAG,GAAG;4BACjE,CAAC,CAAC,CAAC;qBACd;oBACD,QAAQ,EAAE;wBACN,IAAI,EAAE,UAAU,CAAC,iBAAiB;wBAClC,KAAK,EAAE,UAAU,CAAC,iBAAiB;wBACnC,OAAO,EACH,UAAU,CAAC,iBAAiB,GAAG,CAAC;4BAC5B,CAAC,CAAC,CAAC,UAAU,CAAC,iBAAiB,GAAG,UAAU,CAAC,iBAAiB,CAAC,GAAG,GAAG;4BACrE,CAAC,CAAC,CAAC;qBACd;oBACD,SAAS,EAAE,UAAU,CAAC,aAAa;iBACtC,CAAC;YACN,CAAC;YAGD,IAAI,QAAQ,KAAK,QAAQ,IAAI,aAAa,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;gBACzD,MAAM,SAAS,GAAG,aAAa,CAAC,IAAI,CAAC,SAA4B,CAAC;gBAClE,OAAO,CAAC,QAAQ,CAAC,CAAC,QAAQ,GAAG;oBACzB,UAAU,EAAE;wBACR,IAAI,EAAE,SAAS,CAAC,UAAU,CAAC,IAAI;wBAC/B,KAAK,EAAE,SAAS,CAAC,UAAU,CAAC,KAAK;wBACjC,OAAO,EAAE,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,GAAG,GAAG;qBAC1E;oBACD,aAAa,EAAE;wBACX,IAAI,EAAE,SAAS,CAAC,aAAa,CAAC,IAAI;wBAClC,KAAK,EAAE,SAAS,CAAC,aAAa,CAAC,KAAK;wBACpC,OAAO,EAAE,CAAC,SAAS,CAAC,aAAa,CAAC,IAAI,GAAG,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,GAAG;qBAChF;iBACJ,CAAC;YACN,CAAC;QACL,CAAC;QAED,OAAO,OAAO,CAAC;IACnB,CAAC;IAKD,eAAe;QACX,IAAI,CAAC;YACD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;gBACtB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBACjC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YAChC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;QACxD,CAAC;IACL,CAAC;CACJ;AA7cD,oCA6cC;AAGY,QAAA,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/cjs/utils/retry_handler.cjs

@@ -0,0 +1,131 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_RETRY_OPTIONS = void 0;
+exports.isRetryableError = isRetryableError;
+exports.calculateDelay = calculateDelay;
+exports.retryWithBackoff = retryWithBackoff;
+exports.retryStreamWithBackoff = retryStreamWithBackoff;
+exports.DEFAULT_RETRY_OPTIONS = {
+    maxRetries: 3,
+    initialDelay: 1000,
+    maxDelay: 30000,
+    backoffMultiplier: 2,
+    retryableErrors: new Set([
+        'ECONNRESET',
+        'ETIMEDOUT',
+        'ENOTFOUND',
+        'ECONNREFUSED',
+        'EPIPE',
+        'EHOSTUNREACH',
+        'EAI_AGAIN',
+        'ENETUNREACH',
+        'ECONNABORTED',
+        'ESOCKETTIMEDOUT',
+    ]),
+    retryableStatusCodes: new Set([
+        408,
+        429,
+        500,
+        502,
+        503,
+        504,
+        522,
+        524,
+    ]),
+};
+function isRetryableError(error, options = {}) {
+    const opts = { ...exports.DEFAULT_RETRY_OPTIONS, ...options };
+    if (error.code && opts.retryableErrors.has(error.code)) {
+        return true;
+    }
+    if (error.status && opts.retryableStatusCodes.has(error.status)) {
+        return true;
+    }
+    if (error.message &&
+        (error.message.includes('fetch failed') ||
+            error.message.includes('network error') ||
+            error.message.includes('ECONNRESET') ||
+            error.message.includes('ETIMEDOUT'))) {
+        return true;
+    }
+    if (error.message &&
+        (error.message.includes('Incomplete JSON segment') ||
+            error.message.includes('Connection error') ||
+            error.message.includes('Request timeout'))) {
+        return true;
+    }
+    return false;
+}
+function calculateDelay(attempt, options = {}) {
+    const opts = { ...exports.DEFAULT_RETRY_OPTIONS, ...options };
+    const baseDelay = opts.initialDelay * Math.pow(opts.backoffMultiplier, attempt - 1);
+    const delay = Math.min(baseDelay, opts.maxDelay);
+    const jitter = delay * 0.1 * (Math.random() * 2 - 1);
+    return Math.round(delay + jitter);
+}
+async function retryWithBackoff(fn, options = {}) {
+    const opts = { ...exports.DEFAULT_RETRY_OPTIONS, ...options };
+    let lastError;
+    for (let attempt = 1; attempt <= opts.maxRetries + 1; attempt++) {
+        try {
+            return await fn();
+        }
+        catch (error) {
+            lastError = error;
+            if (!isRetryableError(error, opts) || attempt > opts.maxRetries) {
+                throw error;
+            }
+            const delay = calculateDelay(attempt, opts);
+            if (opts.onRetry) {
+                opts.onRetry(error, attempt);
+            }
+            await new Promise(resolve => setTimeout(resolve, delay));
+        }
+    }
+    throw lastError;
+}
+async function* retryStreamWithBackoff(createStream, options = {}) {
+    const opts = { ...exports.DEFAULT_RETRY_OPTIONS, ...options };
+    let lastError;
+    const buffer = [];
+    let hasStartedYielding = false;
+    for (let attempt = 1; attempt <= opts.maxRetries + 1; attempt++) {
+        try {
+            const stream = createStream();
+            if (hasStartedYielding) {
+                let skipCount = buffer.length;
+                for await (const item of stream) {
+                    if (skipCount > 0) {
+                        skipCount--;
+                        continue;
+                    }
+                    yield item;
+                }
+            }
+            else {
+                for await (const item of stream) {
+                    buffer.push(item);
+                    hasStartedYielding = true;
+                    yield item;
+                }
+            }
+            return;
+        }
+        catch (error) {
+            lastError = error;
+            if (hasStartedYielding) {
+                throw error;
+            }
+            if (!isRetryableError(error, opts) || attempt > opts.maxRetries) {
+                throw error;
+            }
+            const delay = calculateDelay(attempt, opts);
+            if (opts.onRetry) {
+                opts.onRetry(error, attempt);
+            }
+            await new Promise(resolve => setTimeout(resolve, delay));
+        }
+    }
+    throw lastError;
+}
+//# sourceMappingURL=retry_handler.js.map

package/dist/cjs/utils/retry_handler.d.ts

@@ -0,0 +1,15 @@
+export interface RetryOptions {
+    maxRetries?: number;
+    initialDelay?: number;
+    maxDelay?: number;
+    backoffMultiplier?: number;
+    retryableErrors?: Set<string>;
+    retryableStatusCodes?: Set<number>;
+    onRetry?: (error: any, attempt: number) => void;
+}
+export declare const DEFAULT_RETRY_OPTIONS: Required<Omit<RetryOptions, 'onRetry'>>;
+export declare function isRetryableError(error: any, options?: RetryOptions): boolean;
+export declare function calculateDelay(attempt: number, options?: RetryOptions): number;
+export declare function retryWithBackoff<T>(fn: () => Promise<T>, options?: RetryOptions): Promise<T>;
+export declare function retryStreamWithBackoff<T>(createStream: () => AsyncGenerator<T>, options?: RetryOptions): AsyncGenerator<T>;
+//# sourceMappingURL=retry_handler.d.ts.map

package/dist/cjs/utils/retry_handler.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry_handler.d.ts","sourceRoot":"","sources":["../../../utils/retry_handler.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,YAAY;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,eAAe,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,oBAAoB,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACnC,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,MAAM,KAAK,IAAI,CAAC;CACnD;AAED,eAAO,MAAM,qBAAqB,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC,CA2BzE,CAAC;AAEF,wBAAgB,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAmChF;AAED,wBAAgB,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,GAAE,YAAiB,GAAG,MAAM,CAQlF;AAED,wBAAsB,gBAAgB,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,GAAE,YAAiB,GAAG,OAAO,CAAC,CAAC,CAAC,CA2BtG;AAED,wBAAuB,sBAAsB,CAAC,CAAC,EAC3C,YAAY,EAAE,MAAM,cAAc,CAAC,CAAC,CAAC,EACrC,OAAO,GAAE,YAAiB,GAC3B,cAAc,CAAC,CAAC,CAAC,CAwDnB"}

package/dist/cjs/utils/retry_handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"retry_handler.js","sourceRoot":"","sources":["../../../utils/retry_handler.ts"],"names":[],"mappings":";;;AAuCA,4CAmCC;AAED,wCAQC;AAED,4CA2BC;AAED,wDA2DC;AApKY,QAAA,qBAAqB,GAA4C;IAC1E,UAAU,EAAE,CAAC;IACb,YAAY,EAAE,IAAI;IAClB,QAAQ,EAAE,KAAK;IACf,iBAAiB,EAAE,CAAC;IACpB,eAAe,EAAE,IAAI,GAAG,CAAC;QACrB,YAAY;QACZ,WAAW;QACX,WAAW;QACX,cAAc;QACd,OAAO;QACP,cAAc;QACd,WAAW;QACX,aAAa;QACb,cAAc;QACd,iBAAiB;KACpB,CAAC;IACF,oBAAoB,EAAE,IAAI,GAAG,CAAC;QAC1B,GAAG;QACH,GAAG;QACH,GAAG;QACH,GAAG;QACH,GAAG;QACH,GAAG;QACH,GAAG;QACH,GAAG;KACN,CAAC;CACL,CAAC;AAEF,SAAgB,gBAAgB,CAAC,KAAU,EAAE,UAAwB,EAAE;IACnE,MAAM,IAAI,GAAG,EAAE,GAAG,6BAAqB,EAAE,GAAG,OAAO,EAAE,CAAC;IAGtD,IAAI,KAAK,CAAC,IAAI,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACrD,OAAO,IAAI,CAAC;IAChB,CAAC;IAGD,IAAI,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9D,OAAO,IAAI,CAAC;IAChB,CAAC;IAGD,IACI,KAAK,CAAC,OAAO;QACb,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;YACnC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAC;YACvC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;YACpC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,EAC1C,CAAC;QACC,OAAO,IAAI,CAAC;IAChB,CAAC;IAGD,IACI,KAAK,CAAC,OAAO;QACb,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,yBAAyB,CAAC;YAC9C,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC;YAC1C,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,CAAC,EAChD,CAAC;QACC,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,OAAO,KAAK,CAAC;AACjB,CAAC;AAED,SAAgB,cAAc,CAAC,OAAe,EAAE,UAAwB,EAAE;IACtE,MAAM,IAAI,GAAG,EAAE,GAAG,6BAAqB,EAAE,GAAG,OAAO,EAAE,CAAC;IACtD,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,iBAAiB,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;IACpF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IAGjD,MAAM,MAAM,GAAG,KAAK,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IACrD,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,CAAC;AACtC,CAAC;AAEM,KAAK,UAAU,gBAAgB,CAAI,EAAoB,EAAE,UAAwB,EAAE;IACtF,MAAM,IAAI,GAAG,EAAE,GAAG,6BAAqB,EAAE,GAAG,OAAO,EAAE,CAAC;IACtD,IAAI,SAAc,CAAC;IAEnB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,UAAU,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC9D,IAAI,CAAC;YACD,OAAO,MAAM,EAAE,EAAE,CAAC;QACtB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,SAAS,GAAG,KAAK,CAAC;YAGlB,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9D,MAAM,KAAK,CAAC;YAChB,CAAC;YAGD,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YACjC,CAAC;YAGD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;IAED,MAAM,SAAS,CAAC;AACpB,CAAC;AAEM,KAAK,SAAS,CAAC,CAAC,sBAAsB,CACzC,YAAqC,EACrC,UAAwB,EAAE;IAE1B,MAAM,IAAI,GAAG,EAAE,GAAG,6BAAqB,EAAE,GAAG,OAAO,EAAE,CAAC;IACtD,IAAI,SAAc,CAAC;IACnB,MAAM,MAAM,GAAQ,EAAE,CAAC;IACvB,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAE/B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,IAAI,CAAC,UAAU,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC9D,IAAI,CAAC;YACD,MAAM,MAAM,GAAG,YAAY,EAAE,CAAC;YAG9B,IAAI,kBAAkB,EAAE,CAAC;gBACrB,IAAI,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC;gBAC9B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;oBAC9B,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;wBAChB,SAAS,EAAE,CAAC;wBACZ,SAAS;oBACb,CAAC;oBACD,MAAM,IAAI,CAAC;gBACf,CAAC;YACL,CAAC;iBAAM,CAAC;gBAEJ,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;oBAC9B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBAClB,kBAAkB,GAAG,IAAI,CAAC;oBAC1B,MAAM,IAAI,CAAC;gBACf,CAAC;YACL,CAAC;YAGD,OAAO;QACX,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,SAAS,GAAG,KAAK,CAAC;YAGlB,IAAI,kBAAkB,EAAE,CAAC;gBACrB,MAAM,KAAK,CAAC;YAChB,CAAC;YAGD,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9D,MAAM,KAAK,CAAC;YAChB,CAAC;YAGD,MAAM,KAAK,GAAG,cAAc,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;YACjC,CAAC;YAGD,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QAC7D,CAAC;IACL,CAAC;IAED,MAAM,SAAS,CAAC;AACpB,CAAC"}

package/dist/cjs/utils/running_tool_tracker.cjs

@@ -0,0 +1,133 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.runningToolTracker = exports.RunningToolTracker = void 0;
+class RunningToolTracker {
+    runningTools = new Map();
+    completionHandlers = [];
+    addRunningTool(id, toolName, agentName, args) {
+        const abortController = new AbortController();
+        const runningTool = {
+            id,
+            toolName,
+            agentName,
+            args,
+            startTime: Date.now(),
+            abortController,
+        };
+        this.runningTools.set(id, runningTool);
+        return runningTool;
+    }
+    markTimedOut(id) {
+        const tool = this.runningTools.get(id);
+        if (tool) {
+            tool.timedOut = true;
+        }
+    }
+    async completeRunningTool(id, result, _agent) {
+        const tool = this.runningTools.get(id);
+        if (!tool)
+            return;
+        tool.completed = true;
+        tool.result = result;
+        const duration = Date.now() - tool.startTime;
+        if (tool.timedOut) {
+            const event = {
+                id,
+                toolName: tool.toolName,
+                agentName: tool.agentName,
+                duration,
+                timedOut: true,
+                result,
+            };
+            this.completionHandlers.forEach(handler => handler(event));
+        }
+        this.runningTools.delete(id);
+    }
+    async failRunningTool(id, error, _agent) {
+        const tool = this.runningTools.get(id);
+        if (!tool)
+            return;
+        tool.failed = true;
+        tool.error = error;
+        const duration = Date.now() - tool.startTime;
+        if (tool.timedOut) {
+            const event = {
+                id,
+                toolName: tool.toolName,
+                agentName: tool.agentName,
+                duration,
+                timedOut: true,
+                error,
+            };
+            this.completionHandlers.forEach(handler => handler(event));
+        }
+        this.runningTools.delete(id);
+    }
+    getRunningTool(id) {
+        return this.runningTools.get(id);
+    }
+    getAllRunningTools() {
+        return Array.from(this.runningTools.values());
+    }
+    getRunningToolsForAgent(agentName) {
+        return this.getAllRunningTools().filter(tool => tool.agentName === agentName);
+    }
+    abortRunningTool(id) {
+        const tool = this.runningTools.get(id);
+        if (tool && tool.abortController) {
+            tool.abortController.abort();
+        }
+    }
+    onCompletion(handler) {
+        this.completionHandlers.push(handler);
+    }
+    clear() {
+        this.runningTools.forEach(tool => {
+            if (tool.abortController) {
+                tool.abortController.abort();
+            }
+        });
+        this.runningTools.clear();
+        this.completionHandlers = [];
+    }
+    isToolRunning(agentName, toolName) {
+        return this.getAllRunningTools().some(tool => tool.agentName === agentName && tool.toolName === toolName);
+    }
+    getRunningToolCount() {
+        return this.runningTools.size;
+    }
+    async waitForTool(id, timeout) {
+        return new Promise((resolve, reject) => {
+            const tool = this.runningTools.get(id);
+            if (!tool) {
+                resolve(null);
+                return;
+            }
+            let timeoutId;
+            const cleanup = () => {
+                if (timeoutId)
+                    clearTimeout(timeoutId);
+                const index = this.completionHandlers.indexOf(handler);
+                if (index > -1) {
+                    this.completionHandlers.splice(index, 1);
+                }
+            };
+            const handler = (event) => {
+                if (event.id === id) {
+                    cleanup();
+                    resolve(event);
+                }
+            };
+            this.onCompletion(handler);
+            if (timeout) {
+                timeoutId = setTimeout(() => {
+                    cleanup();
+                    reject(new Error(`Timeout waiting for tool ${id}`));
+                }, timeout);
+            }
+        });
+    }
+}
+exports.RunningToolTracker = RunningToolTracker;
+exports.runningToolTracker = new RunningToolTracker();
+//# sourceMappingURL=running_tool_tracker.js.map