@just-every/ensemble 0.2.87 → 0.2.88

package/dist/cjs/utils/cost_tracker.cjs

@@ -0,0 +1,226 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CostTracker = exports.costTracker = void 0;
+const model_data_js_1 = require("../data/model_data.cjs");
+const event_controller_js_1 = require("./event_controller.cjs");
+class CostTracker {
+    entries = [];
+    started = new Date();
+    onAddUsageCallbacks = [];
+    calculateCost(usage) {
+        if (typeof usage.cost === 'number') {
+            return usage;
+        }
+        if (usage.isFreeTierUsage) {
+            usage.cost = 0;
+            return usage;
+        }
+        const model = (0, model_data_js_1.findModel)(usage.model);
+        if (!model) {
+            console.error(`Model not found when recording usage: ${usage.model}`);
+            throw new Error(`Model not found when recording usage: ${usage.model}`);
+        }
+        if (usage.cost !== undefined && usage.cost !== null && usage.cost > 0) {
+            return usage;
+        }
+        usage.cost = 0;
+        const original_input_tokens = usage.input_tokens || 0;
+        const output_tokens = usage.output_tokens || 0;
+        const cached_tokens = usage.cached_tokens || 0;
+        const image_count = usage.image_count || 0;
+        const calculationTime = usage.timestamp || new Date();
+        const usesTimeBasedPricing = (typeof model.cost?.input_per_million === 'object' &&
+            model.cost.input_per_million !== null &&
+            'peak_price_per_million' in model.cost.input_per_million) ||
+            (typeof model.cost?.output_per_million === 'object' &&
+                model.cost.output_per_million !== null &&
+                'peak_price_per_million' in model.cost.output_per_million) ||
+            (typeof model.cost?.cached_input_per_million === 'object' &&
+                model.cost.cached_input_per_million !== null &&
+                'peak_price_per_million' in model.cost.cached_input_per_million);
+        if (!usage.timestamp && usesTimeBasedPricing) {
+            console.warn(`Timestamp missing for time-based pricing model '${usage.model}'. Defaulting to current time for calculation.`);
+        }
+        const getPrice = (tokensForTierCheck, costStructure, modality) => {
+            if (typeof costStructure === 'number') {
+                return costStructure;
+            }
+            if (typeof costStructure === 'object' && costStructure !== null) {
+                if ('text' in costStructure ||
+                    'audio' in costStructure ||
+                    'video' in costStructure ||
+                    'image' in costStructure) {
+                    const modalityPrice = costStructure;
+                    const selectedModality = modality || 'text';
+                    const modalityCost = modalityPrice[selectedModality];
+                    if (modalityCost !== undefined) {
+                        return getPrice(tokensForTierCheck, modalityCost);
+                    }
+                    return getPrice(tokensForTierCheck, modalityPrice.text || 0);
+                }
+                if ('peak_price_per_million' in costStructure) {
+                    const timeBasedCost = costStructure;
+                    const utcHour = calculationTime.getUTCHours();
+                    const utcMinute = calculationTime.getUTCMinutes();
+                    const currentTimeInMinutes = utcHour * 60 + utcMinute;
+                    const peakStartInMinutes = timeBasedCost.peak_utc_start_hour * 60 + timeBasedCost.peak_utc_start_minute;
+                    const peakEndInMinutes = timeBasedCost.peak_utc_end_hour * 60 + timeBasedCost.peak_utc_end_minute;
+                    let isPeakTime;
+                    if (peakStartInMinutes <= peakEndInMinutes) {
+                        isPeakTime =
+                            currentTimeInMinutes >= peakStartInMinutes && currentTimeInMinutes < peakEndInMinutes;
+                    }
+                    else {
+                        isPeakTime =
+                            currentTimeInMinutes >= peakStartInMinutes || currentTimeInMinutes < peakEndInMinutes;
+                    }
+                    return isPeakTime ? timeBasedCost.peak_price_per_million : timeBasedCost.off_peak_price_per_million;
+                }
+                else if ('threshold_tokens' in costStructure) {
+                    const tieredCost = costStructure;
+                    if (tokensForTierCheck <= tieredCost.threshold_tokens) {
+                        return tieredCost.price_below_threshold_per_million;
+                    }
+                    else {
+                        return tieredCost.price_above_threshold_per_million;
+                    }
+                }
+            }
+            return 0;
+        };
+        let nonCachedInputTokens = 0;
+        let actualCachedTokens = 0;
+        if (cached_tokens > 0 && model.cost?.cached_input_per_million !== undefined) {
+            actualCachedTokens = cached_tokens;
+            nonCachedInputTokens = Math.max(0, original_input_tokens - cached_tokens);
+        }
+        else {
+            nonCachedInputTokens = original_input_tokens;
+            actualCachedTokens = 0;
+        }
+        if (nonCachedInputTokens > 0 && model.cost?.input_per_million !== undefined) {
+            const inputPricePerMillion = getPrice(original_input_tokens, model.cost.input_per_million, usage.input_modality);
+            usage.cost += (nonCachedInputTokens / 1000000) * inputPricePerMillion;
+        }
+        if (actualCachedTokens > 0 && model.cost?.cached_input_per_million !== undefined) {
+            const cachedPricePerMillion = getPrice(actualCachedTokens, model.cost.cached_input_per_million);
+            usage.cost += (actualCachedTokens / 1000000) * cachedPricePerMillion;
+        }
+        if (output_tokens > 0 && model.cost?.output_per_million !== undefined) {
+            const outputPricePerMillion = getPrice(output_tokens, model.cost.output_per_million, usage.output_modality);
+            usage.cost += (output_tokens / 1000000) * outputPricePerMillion;
+        }
+        if (image_count > 0 && model.cost?.per_image) {
+            usage.cost += image_count * model.cost.per_image;
+        }
+        usage.cost = Math.max(0, usage.cost);
+        return usage;
+    }
+    onAddUsage(callback) {
+        this.onAddUsageCallbacks.push(callback);
+    }
+    addUsage(usage) {
+        try {
+            usage = this.calculateCost({ ...usage });
+            usage.timestamp = new Date();
+            this.entries.push(usage);
+            if ((0, event_controller_js_1.hasEventHandler)()) {
+                const costUpdateEvent = {
+                    type: 'cost_update',
+                    usage: {
+                        ...usage,
+                        total_tokens: (usage.input_tokens || 0) + (usage.output_tokens || 0),
+                    },
+                    timestamp: new Date().toISOString(),
+                };
+                (0, event_controller_js_1.emitEvent)(costUpdateEvent).catch(error => {
+                    console.error('Error emitting cost_update event:', error);
+                });
+            }
+            for (const callback of this.onAddUsageCallbacks) {
+                try {
+                    callback(usage);
+                }
+                catch (error) {
+                    console.error('Error in cost tracker callback:', error);
+                }
+            }
+        }
+        catch (err) {
+            console.error('Error recording usage:', err);
+        }
+    }
+    getTotalCost() {
+        return this.entries.reduce((sum, entry) => sum + (entry.cost || 0), 0);
+    }
+    getCostsByModel() {
+        const models = {};
+        for (const entry of this.entries) {
+            if (!models[entry.model]) {
+                models[entry.model] = {
+                    cost: 0,
+                    calls: 0,
+                };
+            }
+            models[entry.model].cost += entry.cost || 0;
+            models[entry.model].calls += 1;
+        }
+        return models;
+    }
+    printSummary() {
+        if (!this.entries.length) {
+            return;
+        }
+        const totalCost = this.getTotalCost();
+        const costsByModel = this.getCostsByModel();
+        const runtime = Math.round((new Date().getTime() - this.started.getTime()) / 1000);
+        console.log('\n\nCOST SUMMARY');
+        console.log(`Runtime: ${runtime} seconds`);
+        console.log(`Total API Cost: $${totalCost.toFixed(6)}`);
+        console.log('\nModels:');
+        for (const [model, modelData] of Object.entries(costsByModel)) {
+            console.log(`\t${model}:\t$${modelData.cost.toFixed(6)} (${modelData.calls} calls)`);
+        }
+        this.reset();
+    }
+    getCostInTimeWindow(seconds) {
+        const cutoffTime = new Date(Date.now() - seconds * 1000);
+        return this.entries
+            .filter(entry => entry.timestamp && entry.timestamp >= cutoffTime)
+            .reduce((sum, entry) => sum + (entry.cost || 0), 0);
+    }
+    getCostRate(windowSeconds = 60) {
+        const costInWindow = this.getCostInTimeWindow(windowSeconds);
+        return (costInWindow / windowSeconds) * 60;
+    }
+    getUsageInTimeWindow(seconds) {
+        const cutoffTime = new Date(Date.now() - seconds * 1000);
+        return this.entries.filter(entry => entry.timestamp && entry.timestamp >= cutoffTime);
+    }
+    getCostsByModelInTimeWindow(seconds) {
+        const models = {};
+        const entriesInWindow = this.getUsageInTimeWindow(seconds);
+        for (const entry of entriesInWindow) {
+            if (!models[entry.model]) {
+                models[entry.model] = {
+                    cost: 0,
+                    calls: 0,
+                };
+            }
+            models[entry.model].cost += entry.cost || 0;
+            models[entry.model].calls += 1;
+        }
+        return models;
+    }
+    reset() {
+        this.entries = [];
+        this.started = new Date();
+    }
+}
+exports.CostTracker = CostTracker;
+const globalObj = globalThis;
+if (!globalObj.__ENSEMBLE_COST_TRACKER__) {
+    globalObj.__ENSEMBLE_COST_TRACKER__ = new CostTracker();
+}
+exports.costTracker = globalObj.__ENSEMBLE_COST_TRACKER__;
+//# sourceMappingURL=cost_tracker.js.map

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

@@ -0,0 +1,33 @@
+import { ModelUsage } from "../data/model_data.cjs";
+declare class CostTracker {
+    private entries;
+    private started;
+    private onAddUsageCallbacks;
+    calculateCost(usage: ModelUsage): ModelUsage;
+    onAddUsage(callback: (usage: ModelUsage) => void): void;
+    addUsage(usage: ModelUsage): void;
+    getTotalCost(): number;
+    getCostsByModel(): Record<string, {
+        cost: number;
+        calls: number;
+    }>;
+    printSummary(): void;
+    getCostInTimeWindow(seconds: number): number;
+    getCostRate(windowSeconds?: number): number;
+    getUsageInTimeWindow(seconds: number): ModelUsage[];
+    getCostsByModelInTimeWindow(seconds: number): Record<string, {
+        cost: number;
+        calls: number;
+    }>;
+    reset(): void;
+}
+export declare const costTracker: CostTracker;
+export { CostTracker };
+export interface UsageEntry {
+    model: string;
+    input_tokens?: number;
+    output_tokens?: number;
+    image_count?: number;
+    timestamp?: Date;
+}
+//# sourceMappingURL=cost_tracker.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cost_tracker.d.ts","sourceRoot":"","sources":["../../../utils/cost_tracker.ts"],"names":[],"mappings":"AAOA,OAAO,EAAa,UAAU,EAA8C,MAAM,uBAAuB,CAAC;AAO1G,cAAM,WAAW;IACb,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,OAAO,CAAoB;IACnC,OAAO,CAAC,mBAAmB,CAA0C;IAUrE,aAAa,CAAC,KAAK,EAAE,UAAU,GAAG,UAAU;IAuK5C,UAAU,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,UAAU,KAAK,IAAI,GAAG,IAAI;IASvD,QAAQ,CAAC,KAAK,EAAE,UAAU,GAAG,IAAI;IA0CjC,YAAY,IAAI,MAAM;IAOtB,eAAe,IAAI,MAAM,CAAC,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAqBlE,YAAY,IAAI,IAAI;IA0BpB,mBAAmB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAY5C,WAAW,CAAC,aAAa,GAAE,MAAW,GAAG,MAAM;IAW/C,oBAAoB,CAAC,OAAO,EAAE,MAAM,GAAG,UAAU,EAAE;IAUnD,2BAA2B,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;IAsB7F,KAAK,IAAI,IAAI;CAIhB;AASD,eAAO,MAAM,WAAW,EAAE,WAAiD,CAAC;AAG5E,OAAO,EAAE,WAAW,EAAE,CAAC;AAGvB,MAAM,WAAW,UAAU;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,IAAI,CAAC;CACpB"}

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

@@ -0,0 +1 @@
+{"version":3,"file":"cost_tracker.js","sourceRoot":"","sources":["../../../utils/cost_tracker.ts"],"names":[],"mappings":";;;AAOA,yDAA0G;AAC1G,+DAAmE;AAMnE,MAAM,WAAW;IACL,OAAO,GAAiB,EAAE,CAAC;IAC3B,OAAO,GAAS,IAAI,IAAI,EAAE,CAAC;IAC3B,mBAAmB,GAAuC,EAAE,CAAC;IAUrE,aAAa,CAAC,KAAiB;QAE3B,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACjC,OAAO,KAAK,CAAC;QACjB,CAAC;QAGD,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YACxB,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC;YACf,OAAO,KAAK,CAAC;QACjB,CAAC;QAED,MAAM,KAAK,GAAG,IAAA,yBAAS,EAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrC,IAAI,CAAC,KAAK,EAAE,CAAC;YACT,OAAO,CAAC,KAAK,CAAC,yCAAyC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;YACtE,MAAM,IAAI,KAAK,CAAC,yCAAyC,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;QAC5E,CAAC;QAGD,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,IAAI,KAAK,IAAI,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YACpE,OAAO,KAAK,CAAC;QACjB,CAAC;QAGD,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC;QAGf,MAAM,qBAAqB,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;QACtD,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;QAC/C,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;QAG3C,MAAM,eAAe,GAAG,KAAK,CAAC,SAAS,IAAI,IAAI,IAAI,EAAE,CAAC;QAGtD,MAAM,oBAAoB,GACtB,CAAC,OAAO,KAAK,CAAC,IAAI,EAAE,iBAAiB,KAAK,QAAQ;YAC9C,KAAK,CAAC,IAAI,CAAC,iBAAiB,KAAK,IAAI;YACrC,wBAAwB,IAAI,KAAK,CAAC,IAAI,CAAC,iBAAiB,CAAC;YAC7D,CAAC,OAAO,KAAK,CAAC,IAAI,EAAE,kBAAkB,KAAK,QAAQ;gBAC/C,KAAK,CAAC,IAAI,CAAC,kBAAkB,KAAK,IAAI;gBACtC,wBAAwB,IAAI,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC;YAC9D,CAAC,OAAO,KAAK,CAAC,IAAI,EAAE,wBAAwB,KAAK,QAAQ;gBACrD,KAAK,CAAC,IAAI,CAAC,wBAAwB,KAAK,IAAI;gBAC5C,wBAAwB,IAAI,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAEzE,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,oBAAoB,EAAE,CAAC;YAC3C,OAAO,CAAC,IAAI,CACR,mDAAmD,KAAK,CAAC,KAAK,gDAAgD,CACjH,CAAC;QACN,CAAC;QAGD,MAAM,QAAQ,GAAG,CACb,kBAA0B,EAC1B,aAAgF,EAChF,QAA+C,EACzC,EAAE;YACR,IAAI,OAAO,aAAa,KAAK,QAAQ,EAAE,CAAC;gBACpC,OAAO,aAAa,CAAC;YACzB,CAAC;YAED,IAAI,OAAO,aAAa,KAAK,QAAQ,IAAI,aAAa,KAAK,IAAI,EAAE,CAAC;gBAE9D,IACI,MAAM,IAAI,aAAa;oBACvB,OAAO,IAAI,aAAa;oBACxB,OAAO,IAAI,aAAa;oBACxB,OAAO,IAAI,aAAa,EAC1B,CAAC;oBACC,MAAM,aAAa,GAAG,aAA8B,CAAC;oBACrD,MAAM,gBAAgB,GAAG,QAAQ,IAAI,MAAM,CAAC;oBAC5C,MAAM,YAAY,GAAG,aAAa,CAAC,gBAAgB,CAAC,CAAC;oBAGrD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;wBAC7B,OAAO,QAAQ,CAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC;oBACtD,CAAC;oBAGD,OAAO,QAAQ,CAAC,kBAAkB,EAAE,aAAa,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC;gBACjE,CAAC;gBAED,IAAI,wBAAwB,IAAI,aAAa,EAAE,CAAC;oBAE5C,MAAM,aAAa,GAAG,aAA+B,CAAC;oBACtD,MAAM,OAAO,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;oBAC9C,MAAM,SAAS,GAAG,eAAe,CAAC,aAAa,EAAE,CAAC;oBAClD,MAAM,oBAAoB,GAAG,OAAO,GAAG,EAAE,GAAG,SAAS,CAAC;oBACtD,MAAM,kBAAkB,GACpB,aAAa,CAAC,mBAAmB,GAAG,EAAE,GAAG,aAAa,CAAC,qBAAqB,CAAC;oBACjF,MAAM,gBAAgB,GAAG,aAAa,CAAC,iBAAiB,GAAG,EAAE,GAAG,aAAa,CAAC,mBAAmB,CAAC;oBAElG,IAAI,UAAmB,CAAC;oBACxB,IAAI,kBAAkB,IAAI,gBAAgB,EAAE,CAAC;wBACzC,UAAU;4BACN,oBAAoB,IAAI,kBAAkB,IAAI,oBAAoB,GAAG,gBAAgB,CAAC;oBAC9F,CAAC;yBAAM,CAAC;wBACJ,UAAU;4BACN,oBAAoB,IAAI,kBAAkB,IAAI,oBAAoB,GAAG,gBAAgB,CAAC;oBAC9F,CAAC;oBAED,OAAO,UAAU,CAAC,CAAC,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAC,CAAC,aAAa,CAAC,0BAA0B,CAAC;gBACxG,CAAC;qBAAM,IAAI,kBAAkB,IAAI,aAAa,EAAE,CAAC;oBAE7C,MAAM,UAAU,GAAG,aAA4B,CAAC;oBAChD,IAAI,kBAAkB,IAAI,UAAU,CAAC,gBAAgB,EAAE,CAAC;wBACpD,OAAO,UAAU,CAAC,iCAAiC,CAAC;oBACxD,CAAC;yBAAM,CAAC;wBACJ,OAAO,UAAU,CAAC,iCAAiC,CAAC;oBACxD,CAAC;gBACL,CAAC;YACL,CAAC;YACD,OAAO,CAAC,CAAC;QACb,CAAC,CAAC;QAGF,IAAI,oBAAoB,GAAG,CAAC,CAAC;QAC7B,IAAI,kBAAkB,GAAG,CAAC,CAAC;QAE3B,IAAI,aAAa,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,wBAAwB,KAAK,SAAS,EAAE,CAAC;YAC1E,kBAAkB,GAAG,aAAa,CAAC;YACnC,oBAAoB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,qBAAqB,GAAG,aAAa,CAAC,CAAC;QAC9E,CAAC;aAAM,CAAC;YACJ,oBAAoB,GAAG,qBAAqB,CAAC;YAC7C,kBAAkB,GAAG,CAAC,CAAC;QAC3B,CAAC;QAGD,IAAI,oBAAoB,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,iBAAiB,KAAK,SAAS,EAAE,CAAC;YAC1E,MAAM,oBAAoB,GAAG,QAAQ,CACjC,qBAAqB,EACrB,KAAK,CAAC,IAAI,CAAC,iBAAiB,EAC5B,KAAK,CAAC,cAAc,CACvB,CAAC;YACF,KAAK,CAAC,IAAI,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,GAAG,oBAAoB,CAAC;QAC1E,CAAC;QAGD,IAAI,kBAAkB,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,wBAAwB,KAAK,SAAS,EAAE,CAAC;YAC/E,MAAM,qBAAqB,GAAG,QAAQ,CAAC,kBAAkB,EAAE,KAAK,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAChG,KAAK,CAAC,IAAI,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,GAAG,qBAAqB,CAAC;QACzE,CAAC;QAGD,IAAI,aAAa,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,kBAAkB,KAAK,SAAS,EAAE,CAAC;YACpE,MAAM,qBAAqB,GAAG,QAAQ,CAAC,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,kBAAkB,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;YAC5G,KAAK,CAAC,IAAI,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,GAAG,qBAAqB,CAAC;QACpE,CAAC;QAGD,IAAI,WAAW,GAAG,CAAC,IAAI,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC;YAC3C,KAAK,CAAC,IAAI,IAAI,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;QACrD,CAAC;QAGD,KAAK,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAErC,OAAO,KAAK,CAAC;IACjB,CAAC;IAOD,UAAU,CAAC,QAAqC;QAC5C,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;IAOD,QAAQ,CAAC,KAAiB;QACtB,IAAI,CAAC;YAED,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;YACzC,KAAK,CAAC,SAAS,GAAG,IAAI,IAAI,EAAE,CAAC;YAG7B,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAGzB,IAAI,IAAA,qCAAe,GAAE,EAAE,CAAC;gBACpB,MAAM,eAAe,GAAoB;oBACrC,IAAI,EAAE,aAAa;oBACnB,KAAK,EAAE;wBACH,GAAG,KAAK;wBACR,YAAY,EAAE,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;qBACvE;oBACD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;iBACtC,CAAC;gBAGF,IAAA,+BAAS,EAAC,eAAe,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE;oBACrC,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;gBAC9D,CAAC,CAAC,CAAC;YACP,CAAC;YAGD,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC9C,IAAI,CAAC;oBACD,QAAQ,CAAC,KAAK,CAAC,CAAC;gBACpB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;gBAC5D,CAAC;YACL,CAAC;QACL,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,OAAO,CAAC,KAAK,CAAC,wBAAwB,EAAE,GAAG,CAAC,CAAC;QACjD,CAAC;IACL,CAAC;IAKD,YAAY;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC3E,CAAC;IAKD,eAAe;QACX,MAAM,MAAM,GAAoD,EAAE,CAAC;QAEnE,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC/B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG;oBAClB,IAAI,EAAE,CAAC;oBACP,KAAK,EAAE,CAAC;iBACX,CAAC;YACN,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;QACnC,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAKD,YAAY;QACR,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACvB,OAAO;QACX,CAAC;QAED,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACtC,MAAM,YAAY,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QAEnF,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,YAAY,OAAO,UAAU,CAAC,CAAC;QAC3C,OAAO,CAAC,GAAG,CAAC,oBAAoB,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAExD,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACzB,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;YAC5D,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,OAAO,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,KAAK,SAAS,CAAC,CAAC;QACzF,CAAC;QAED,IAAI,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC;IAOD,mBAAmB,CAAC,OAAe;QAC/B,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,OAAO;aACd,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,UAAU,CAAC;aACjE,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5D,CAAC;IAOD,WAAW,CAAC,gBAAwB,EAAE;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;QAE7D,OAAO,CAAC,YAAY,GAAG,aAAa,CAAC,GAAG,EAAE,CAAC;IAC/C,CAAC;IAOD,oBAAoB,CAAC,OAAe;QAChC,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC;QACzD,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,UAAU,CAAC,CAAC;IAC1F,CAAC;IAOD,2BAA2B,CAAC,OAAe;QACvC,MAAM,MAAM,GAAoD,EAAE,CAAC;QACnE,MAAM,eAAe,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAE3D,KAAK,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;YAClC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvB,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG;oBAClB,IAAI,EAAE,CAAC;oBACP,KAAK,EAAE,CAAC;iBACX,CAAC;YACN,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;QACnC,CAAC;QAED,OAAO,MAAM,CAAC;IAClB,CAAC;IAKD,KAAK;QACD,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,OAAO,GAAG,IAAI,IAAI,EAAE,CAAC;IAC9B,CAAC;CACJ;AAYQ,kCAAW;AATpB,MAAM,SAAS,GAAG,UAEjB,CAAC;AACF,IAAI,CAAC,SAAS,CAAC,yBAAyB,EAAE,CAAC;IACvC,SAAS,CAAC,yBAAyB,GAAG,IAAI,WAAW,EAAE,CAAC;AAC5D,CAAC;AACY,QAAA,WAAW,GAAgB,SAAS,CAAC,yBAAyB,CAAC"}

package/dist/cjs/utils/create_tool_function.cjs

@@ -0,0 +1,182 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createToolFunction = createToolFunction;
+const validToolParameterTypes = ['string', 'number', 'boolean', 'array', 'object', 'null'];
+function createToolFunction(func, description, paramMap, returns, functionName) {
+    const funcStr = func.toString();
+    const funcName = (functionName || '').replaceAll(' ', '_') || func.name;
+    if (!funcName) {
+        throw new Error('[createToolFunction] Function name is required');
+    }
+    let toolDescription = description || `Tool for ${funcName}`;
+    if (returns) {
+        toolDescription += ` Returns: ${returns}`;
+    }
+    const cleanFuncStr = funcStr.replaceAll(/\n\s*/g, ' ');
+    const paramMatch = cleanFuncStr.match(/\(([^)]*)\)/);
+    const properties = {};
+    const required = [];
+    let injectAgentId = false;
+    let injectAbortSignal = false;
+    const params = paramMap
+        ? Object.keys(paramMap)
+        : paramMatch && paramMatch[1]
+            ? paramMatch[1]
+                .split(',')
+                .map(p => p.trim())
+                .filter(Boolean)
+            : [];
+    for (const paramUnknown of params) {
+        if (typeof paramUnknown !== 'string') {
+            console.warn(`Skipping non-string parameter in function signature analysis: ${paramUnknown}`);
+            continue;
+        }
+        const param = paramUnknown;
+        const paramParts = param.split('=').map(p => p.trim());
+        let paramName = paramParts[0].trim();
+        const defaultValue = paramParts.length > 1 ? paramParts[1].trim() : undefined;
+        if (paramName.includes(':')) {
+            paramName = paramName.split(':')[0].trim();
+        }
+        const isRestParam = paramName.startsWith('...');
+        const cleanParamName = isRestParam ? paramName.substring(3) : paramName;
+        if (cleanParamName === 'inject_agent_id') {
+            injectAgentId = true;
+            continue;
+        }
+        if (cleanParamName === 'abort_signal') {
+            injectAbortSignal = true;
+            continue;
+        }
+        const paramInfoRaw = paramMap?.[cleanParamName];
+        let paramInfoObj = undefined;
+        let paramInfoDesc = undefined;
+        if (typeof paramInfoRaw === 'string') {
+            paramInfoDesc = paramInfoRaw;
+            paramInfoObj = { type: 'string', description: paramInfoRaw };
+        }
+        else if (typeof paramInfoRaw === 'object' && paramInfoRaw !== null) {
+            paramInfoObj = paramInfoRaw;
+            paramInfoDesc =
+                typeof paramInfoRaw.description === 'function' ? paramInfoRaw.description() : paramInfoRaw.description;
+        }
+        const apiParamName = cleanParamName;
+        let paramType = 'string';
+        if (paramInfoObj?.type && validToolParameterTypes.includes(paramInfoObj.type)) {
+            paramType = paramInfoObj.type;
+        }
+        else if (isRestParam) {
+            paramType = 'array';
+        }
+        else if (defaultValue !== undefined) {
+            if (defaultValue === 'false' || defaultValue === 'true') {
+                paramType = 'boolean';
+            }
+            else if (!isNaN(Number(defaultValue)) && !defaultValue.startsWith('"') && !defaultValue.startsWith("'")) {
+                paramType = 'number';
+            }
+            else if (defaultValue === '[]' || defaultValue.startsWith('[')) {
+                paramType = 'array';
+            }
+            else if (defaultValue === '{}' || defaultValue.startsWith('{')) {
+                paramType = 'object';
+            }
+        }
+        const finalDescription = paramInfoDesc || `The ${cleanParamName} parameter`;
+        const paramDef = {
+            type: paramType,
+            description: finalDescription,
+        };
+        if (paramType === 'array') {
+            if (paramInfoObj?.items) {
+                paramDef.items = paramInfoObj.items;
+            }
+            else {
+                paramDef.items = {
+                    type: 'string',
+                };
+            }
+        }
+        if (paramType === 'object') {
+            if (paramInfoObj?.properties) {
+                paramDef.properties = paramInfoObj.properties;
+            }
+            if (paramInfoObj?.required) {
+                paramDef.required = paramInfoObj.required;
+            }
+        }
+        if (paramInfoObj?.enum) {
+            if (typeof paramInfoObj.enum === 'function') {
+                const enumFn = paramInfoObj.enum;
+                const fnStr = enumFn.toString();
+                const isAsync = fnStr.includes('__awaiter') ||
+                    fnStr.startsWith('async ') ||
+                    enumFn.constructor.name === 'AsyncFunction';
+                if (isAsync) {
+                    paramDef.enum = enumFn;
+                }
+                else {
+                    paramDef.enum = enumFn();
+                }
+            }
+            else {
+                paramDef.enum = paramInfoObj.enum;
+            }
+        }
+        if (paramInfoObj?.minimum !== undefined) {
+            paramDef.minimum = paramInfoObj.minimum;
+        }
+        if (paramInfoObj?.maximum !== undefined) {
+            paramDef.maximum = paramInfoObj.maximum;
+        }
+        if (paramInfoObj?.default !== undefined) {
+            paramDef.default = paramInfoObj.default;
+        }
+        else if (defaultValue !== undefined) {
+            paramDef.default = defaultValue;
+        }
+        if (paramInfoObj?.minLength !== undefined) {
+            paramDef.minLength = paramInfoObj.minLength;
+        }
+        if (paramInfoObj?.maxLength !== undefined) {
+            paramDef.maxLength = paramInfoObj.maxLength;
+        }
+        if (paramInfoObj?.pattern !== undefined) {
+            paramDef.pattern = paramInfoObj.pattern;
+        }
+        if (paramInfoObj?.minItems !== undefined) {
+            paramDef.minItems = paramInfoObj.minItems;
+        }
+        if (paramInfoObj?.additionalProperties !== undefined) {
+            paramDef.additionalProperties = paramInfoObj.additionalProperties;
+        }
+        properties[apiParamName] = paramDef;
+        if (paramDef.default === undefined && !paramInfoObj?.optional) {
+            required.push(apiParamName);
+        }
+    }
+    if (!injectAgentId && /\(\s*[^)]*\binject_agent_id\b/.test(funcStr)) {
+        injectAgentId = true;
+    }
+    if (!injectAbortSignal && /\(\s*[^)]*\babort_signal\b/.test(funcStr)) {
+        injectAbortSignal = true;
+    }
+    return {
+        function: func,
+        definition: {
+            type: 'function',
+            function: {
+                name: funcName,
+                description: toolDescription,
+                parameters: {
+                    type: 'object',
+                    properties,
+                    required: required.length > 0 ? required : undefined,
+                },
+            },
+        },
+        ...(injectAgentId && { injectAgentId }),
+        ...(injectAbortSignal && { injectAbortSignal }),
+    };
+}
+//# sourceMappingURL=create_tool_function.js.map

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

@@ -0,0 +1,3 @@
+import { ToolFunction, ToolParameterMap } from "../types/types.cjs";
+export declare function createToolFunction(func: (...args: any[]) => Promise<any> | any, description?: string, paramMap?: ToolParameterMap, returns?: string, functionName?: string): ToolFunction;
+//# sourceMappingURL=create_tool_function.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"create_tool_function.d.ts","sourceRoot":"","sources":["../../../utils/create_tool_function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAoC,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAuErG,wBAAgB,kBAAkB,CAC9B,IAAI,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,OAAO,CAAC,GAAG,CAAC,GAAG,GAAG,EAC5C,WAAW,CAAC,EAAE,MAAM,EACpB,QAAQ,CAAC,EAAE,gBAAgB,EAC3B,OAAO,CAAC,EAAE,MAAM,EAChB,YAAY,CAAC,EAAE,MAAM,GACtB,YAAY,CAgOd"}

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

@@ -0,0 +1 @@
+{"version":3,"file":"create_tool_function.js","sourceRoot":"","sources":["../../../utils/create_tool_function.ts"],"names":[],"mappings":";;AAuEA,gDAsOC;AA3SD,MAAM,uBAAuB,GAAwB,CAAC,QAAQ,EAAE,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;AAqEhH,SAAgB,kBAAkB,CAC9B,IAA4C,EAC5C,WAAoB,EACpB,QAA2B,EAC3B,OAAgB,EAChB,YAAqB;IAErB,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;IAChC,MAAM,QAAQ,GAAG,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC;IAExE,IAAI,CAAC,QAAQ,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACtE,CAAC;IAED,IAAI,eAAe,GAAG,WAAW,IAAI,YAAY,QAAQ,EAAE,CAAC;IAC5D,IAAI,OAAO,EAAE,CAAC;QACV,eAAe,IAAI,aAAa,OAAO,EAAE,CAAC;IAC9C,CAAC;IAGD,MAAM,YAAY,GAAG,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IACvD,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IAErD,MAAM,UAAU,GAAkC,EAAE,CAAC;IACrD,MAAM,QAAQ,GAAa,EAAE,CAAC;IAE9B,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,iBAAiB,GAAG,KAAK,CAAC;IAE9B,MAAM,MAAM,GAAG,QAAQ;QACnB,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC;QACvB,CAAC,CAAC,UAAU,IAAI,UAAU,CAAC,CAAC,CAAC;YAC3B,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;iBACR,KAAK,CAAC,GAAG,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;iBAClB,MAAM,CAAC,OAAO,CAAC;YACtB,CAAC,CAAC,EAAE,CAAC;IAEX,KAAK,MAAM,YAAY,IAAI,MAAM,EAAE,CAAC;QAChC,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;YACnC,OAAO,CAAC,IAAI,CAAC,iEAAiE,YAAY,EAAE,CAAC,CAAC;YAC9F,SAAS;QACb,CAAC;QACD,MAAM,KAAK,GAAG,YAAsB,CAAC;QAGrC,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACvD,IAAI,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACrC,MAAM,YAAY,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAG9E,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,SAAS,GAAG,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,CAAC;QAGD,MAAM,WAAW,GAAG,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,cAAc,GAAG,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAGxE,IAAI,cAAc,KAAK,iBAAiB,EAAE,CAAC;YACvC,aAAa,GAAG,IAAI,CAAC;YACrB,SAAS;QACb,CAAC;QAED,IAAI,cAAc,KAAK,cAAc,EAAE,CAAC;YACpC,iBAAiB,GAAG,IAAI,CAAC;YACzB,SAAS;QACb,CAAC;QAGD,MAAM,YAAY,GAAuC,QAAQ,EAAE,CAAC,cAAc,CAAC,CAAC;QACpF,IAAI,YAAY,GAA8B,SAAS,CAAC;QACxD,IAAI,aAAa,GAAuB,SAAS,CAAC;QAElD,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;YACnC,aAAa,GAAG,YAAY,CAAC;YAC7B,YAAY,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;QACjE,CAAC;aAAM,IAAI,OAAO,YAAY,KAAK,QAAQ,IAAI,YAAY,KAAK,IAAI,EAAE,CAAC;YACnE,YAAY,GAAG,YAAY,CAAC;YAC5B,aAAa;gBACT,OAAO,YAAY,CAAC,WAAW,KAAK,UAAU,CAAC,CAAC,CAAC,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,YAAY,CAAC,WAAW,CAAC;QAC/G,CAAC;QAED,MAAM,YAAY,GAAG,cAAc,CAAC;QAGpC,IAAI,SAAS,GAAsB,QAAQ,CAAC;QAG5C,IAAI,YAAY,EAAE,IAAI,IAAI,uBAAuB,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5E,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC;QAClC,CAAC;aAAM,IAAI,WAAW,EAAE,CAAC;YAErB,SAAS,GAAG,OAAO,CAAC;QACxB,CAAC;aAAM,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YAEpC,IAAI,YAAY,KAAK,OAAO,IAAI,YAAY,KAAK,MAAM,EAAE,CAAC;gBACtD,SAAS,GAAG,SAAS,CAAC;YAC1B,CAAC;iBAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACxG,SAAS,GAAG,QAAQ,CAAC;YACzB,CAAC;iBAAM,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC/D,SAAS,GAAG,OAAO,CAAC;YACxB,CAAC;iBAAM,IAAI,YAAY,KAAK,IAAI,IAAI,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC/D,SAAS,GAAG,QAAQ,CAAC;YACzB,CAAC;QACL,CAAC;QAGD,MAAM,gBAAgB,GAAG,aAAa,IAAI,OAAO,cAAc,YAAY,CAAC;QAG5E,MAAM,QAAQ,GAAQ;YAClB,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,gBAAgB;SAChC,CAAC;QAGF,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YACxB,IAAI,YAAY,EAAE,KAAK,EAAE,CAAC;gBACtB,QAAQ,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC;YACxC,CAAC;iBAAM,CAAC;gBAEJ,QAAQ,CAAC,KAAK,GAAG;oBACb,IAAI,EAAE,QAAQ;iBACjB,CAAC;YACN,CAAC;QACL,CAAC;QAGD,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;YACzB,IAAI,YAAY,EAAE,UAAU,EAAE,CAAC;gBAC3B,QAAQ,CAAC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;YAClD,CAAC;YACD,IAAI,YAAY,EAAE,QAAQ,EAAE,CAAC;gBACzB,QAAQ,CAAC,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;YAC9C,CAAC;QACL,CAAC;QAGD,IAAI,YAAY,EAAE,IAAI,EAAE,CAAC;YACrB,IAAI,OAAO,YAAY,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAE1C,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC;gBACjC,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC;gBAChC,MAAM,OAAO,GACT,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC;oBAC3B,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC;oBAC1B,MAAM,CAAC,WAAW,CAAC,IAAI,KAAK,eAAe,CAAC;gBAEhD,IAAI,OAAO,EAAE,CAAC;oBAGV,QAAQ,CAAC,IAAI,GAAG,MAAM,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBAEJ,QAAQ,CAAC,IAAI,GAAG,MAAM,EAAE,CAAC;gBAC7B,CAAC;YACL,CAAC;iBAAM,CAAC;gBACJ,QAAQ,CAAC,IAAI,GAAG,YAAY,CAAC,IAAI,CAAC;YACtC,CAAC;QACL,CAAC;QAGD,IAAI,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAI,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAI,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;QAC5C,CAAC;aAAM,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;YACpC,QAAQ,CAAC,OAAO,GAAG,YAAY,CAAC;QACpC,CAAC;QACD,IAAI,YAAY,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;YACxC,QAAQ,CAAC,SAAS,GAAG,YAAY,CAAC,SAAS,CAAC;QAChD,CAAC;QACD,IAAI,YAAY,EAAE,SAAS,KAAK,SAAS,EAAE,CAAC;YACxC,QAAQ,CAAC,SAAS,GAAG,YAAY,CAAC,SAAS,CAAC;QAChD,CAAC;QACD,IAAI,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE,CAAC;YACtC,QAAQ,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;QAC5C,CAAC;QACD,IAAI,YAAY,EAAE,QAAQ,KAAK,SAAS,EAAE,CAAC;YACvC,QAAQ,CAAC,QAAQ,GAAG,YAAY,CAAC,QAAQ,CAAC;QAC9C,CAAC;QACD,IAAI,YAAY,EAAE,oBAAoB,KAAK,SAAS,EAAE,CAAC;YACnD,QAAQ,CAAC,oBAAoB,GAAG,YAAY,CAAC,oBAAoB,CAAC;QACtE,CAAC;QAED,UAAU,CAAC,YAAY,CAAC,GAAG,QAAQ,CAAC;QAGpC,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC;YAC5D,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChC,CAAC;IACL,CAAC;IAKD,IAAI,CAAC,aAAa,IAAI,+BAA+B,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QAClE,aAAa,GAAG,IAAI,CAAC;IACzB,CAAC;IAID,IAAI,CAAC,iBAAiB,IAAI,4BAA4B,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;QACnE,iBAAiB,GAAG,IAAI,CAAC;IAC7B,CAAC;IAGD,OAAO;QACH,QAAQ,EAAE,IAAI;QACd,UAAU,EAAE;YACR,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACN,IAAI,EAAE,QAAQ;gBACd,WAAW,EAAE,eAAe;gBAC5B,UAAU,EAAE;oBACR,IAAI,EAAE,QAAQ;oBACd,UAAU;oBACV,QAAQ,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS;iBACvD;aACJ;SACJ;QACD,GAAG,CAAC,aAAa,IAAI,EAAE,aAAa,EAAE,CAAC;QACvC,GAAG,CAAC,iBAAiB,IAAI,EAAE,iBAAiB,EAAE,CAAC;KAClD,CAAC;AACN,CAAC"}

package/dist/cjs/utils/delta_buffer.cjs

@@ -0,0 +1,65 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DeltaBuffer = void 0;
+exports.bufferDelta = bufferDelta;
+exports.flushBufferedDeltas = flushBufferedDeltas;
+class DeltaBuffer {
+    step;
+    max;
+    timeLimitMs;
+    buffer = '';
+    startTimestamp = null;
+    threshold;
+    constructor(step = 20, max = 400, initial = 20, timeLimitMs = 10_000) {
+        this.step = step;
+        this.max = max;
+        this.timeLimitMs = timeLimitMs;
+        this.threshold = initial;
+    }
+    add(chunk) {
+        this.buffer += chunk;
+        if (this.startTimestamp === null)
+            this.startTimestamp = Date.now();
+        const shouldFlushBySize = this.buffer.length >= this.threshold;
+        const shouldFlushByTime = this.startTimestamp !== null && Date.now() - this.startTimestamp >= this.timeLimitMs;
+        if (shouldFlushBySize || shouldFlushByTime) {
+            const out = this.buffer;
+            this.buffer = '';
+            this.startTimestamp = null;
+            if (shouldFlushBySize && this.threshold < this.max) {
+                this.threshold += this.step;
+            }
+            return out;
+        }
+        return null;
+    }
+    flush() {
+        if (!this.buffer)
+            return null;
+        const out = this.buffer;
+        this.buffer = '';
+        this.startTimestamp = null;
+        return out;
+    }
+}
+exports.DeltaBuffer = DeltaBuffer;
+function bufferDelta(store, messageId, chunk, makeEvent) {
+    let buf = store.get(messageId);
+    if (!buf) {
+        buf = new DeltaBuffer();
+        store.set(messageId, buf);
+    }
+    const out = buf.add(chunk);
+    return out !== null ? [makeEvent(out)] : [];
+}
+function flushBufferedDeltas(store, makeEvent) {
+    const events = [];
+    for (const [id, buf] of store) {
+        const out = buf.flush();
+        if (out !== null)
+            events.push(makeEvent(id, out));
+    }
+    store.clear();
+    return events;
+}
+//# sourceMappingURL=delta_buffer.js.map

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

@@ -0,0 +1,14 @@
+export declare class DeltaBuffer {
+    private readonly step;
+    private readonly max;
+    private readonly timeLimitMs;
+    private buffer;
+    private startTimestamp;
+    private threshold;
+    constructor(step?: number, max?: number, initial?: number, timeLimitMs?: number);
+    add(chunk: string): string | null;
+    flush(): string | null;
+}
+export declare function bufferDelta<T>(store: Map<string, DeltaBuffer>, messageId: string, chunk: string, makeEvent: (content: string) => T): T[];
+export declare function flushBufferedDeltas<T>(store: Map<string, DeltaBuffer>, makeEvent: (id: string, content: string) => T): T[];
+//# sourceMappingURL=delta_buffer.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"delta_buffer.d.ts","sourceRoot":"","sources":["../../../utils/delta_buffer.ts"],"names":[],"mappings":"AAQA,qBAAa,WAAW;IAMhB,OAAO,CAAC,QAAQ,CAAC,IAAI;IACrB,OAAO,CAAC,QAAQ,CAAC,GAAG;IAEpB,OAAO,CAAC,QAAQ,CAAC,WAAW;IARhC,OAAO,CAAC,MAAM,CAAM;IACpB,OAAO,CAAC,cAAc,CAAuB;IAC7C,OAAO,CAAC,SAAS,CAAS;gBAGL,IAAI,SAAK,EACT,GAAG,SAAM,EAC1B,OAAO,SAAK,EACK,WAAW,SAAS;IAKzC,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI;IAsBjC,KAAK,IAAI,MAAM,GAAG,IAAI;CAOzB;AAMD,wBAAgB,WAAW,CAAC,CAAC,EACzB,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,EAC/B,SAAS,EAAE,MAAM,EACjB,KAAK,EAAE,MAAM,EACb,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,CAAC,GAClC,CAAC,EAAE,CAQL;AAMD,wBAAgB,mBAAmB,CAAC,CAAC,EACjC,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,EAC/B,SAAS,EAAE,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,CAAC,GAC9C,CAAC,EAAE,CAQL"}

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

@@ -0,0 +1 @@
+{"version":3,"file":"delta_buffer.js","sourceRoot":"","sources":["../../../utils/delta_buffer.ts"],"names":[],"mappings":";;;AAyDA,kCAaC;AAMD,kDAWC;AA/ED,MAAa,WAAW;IAMC;IACA;IAEA;IARb,MAAM,GAAG,EAAE,CAAC;IACZ,cAAc,GAAkB,IAAI,CAAC;IACrC,SAAS,CAAS;IAE1B,YACqB,OAAO,EAAE,EACT,MAAM,GAAG,EAC1B,OAAO,GAAG,EAAE,EACK,cAAc,MAAM;QAHpB,SAAI,GAAJ,IAAI,CAAK;QACT,QAAG,GAAH,GAAG,CAAM;QAET,gBAAW,GAAX,WAAW,CAAS;QAErC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC;IAC7B,CAAC;IAED,GAAG,CAAC,KAAa;QACb,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC;QAGrB,IAAI,IAAI,CAAC,cAAc,KAAK,IAAI;YAAE,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEnE,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,CAAC;QAC/D,MAAM,iBAAiB,GAAG,IAAI,CAAC,cAAc,KAAK,IAAI,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,WAAW,CAAC;QAE/G,IAAI,iBAAiB,IAAI,iBAAiB,EAAE,CAAC;YACzC,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC;YACxB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;YACjB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAC3B,IAAI,iBAAiB,IAAI,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;gBAEjD,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC;YAChC,CAAC;YACD,OAAO,GAAG,CAAC;QACf,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,KAAK;QACD,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC;QACxB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,OAAO,GAAG,CAAC;IACf,CAAC;CACJ;AA3CD,kCA2CC;AAMD,SAAgB,WAAW,CACvB,KAA+B,EAC/B,SAAiB,EACjB,KAAa,EACb,SAAiC;IAEjC,IAAI,GAAG,GAAG,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC/B,IAAI,CAAC,GAAG,EAAE,CAAC;QACP,GAAG,GAAG,IAAI,WAAW,EAAE,CAAC;QACxB,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IAC9B,CAAC;IACD,MAAM,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IAC3B,OAAO,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AAChD,CAAC;AAMD,SAAgB,mBAAmB,CAC/B,KAA+B,EAC/B,SAA6C;IAE7C,MAAM,MAAM,GAAQ,EAAE,CAAC;IACvB,KAAK,MAAM,CAAC,EAAE,EAAE,GAAG,CAAC,IAAI,KAAK,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,KAAK,EAAE,CAAC;QACxB,IAAI,GAAG,KAAK,IAAI;YAAE,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;IACtD,CAAC;IACD,KAAK,CAAC,KAAK,EAAE,CAAC;IACd,OAAO,MAAM,CAAC;AAClB,CAAC"}