societyai 0.0.1

package/dist/metrics.js

@@ -0,0 +1,271 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CommonCostConfigs = exports.PerformanceProfiler = exports.TokenCounter = exports.MetricsBuilder = exports.MetricsTracker = void 0;
+const logger_1 = require("./logger");
+class MetricsTracker {
+    constructor(options = {}) {
+        this.snapshots = new Map();
+        this.history = [];
+        this.costConfigs = new Map();
+        this.logger = (0, logger_1.getLogger)();
+        this.trackTokens = false;
+        this.trackCost = false;
+        this.trackTokens = options.trackTokens ?? false;
+        this.trackCost = options.trackCost ?? false;
+    }
+    start(id, metadata) {
+        const snapshot = {
+            id,
+            execution: {
+                startTime: Date.now(),
+                success: false,
+            },
+            metadata,
+        };
+        this.snapshots.set(id, snapshot);
+        this.logger.debug(`Started tracking: ${id}`);
+    }
+    end(id, options) {
+        const snapshot = this.snapshots.get(id);
+        if (!snapshot) {
+            throw new Error(`No tracking found for: ${id}`);
+        }
+        snapshot.execution.endTime = Date.now();
+        snapshot.execution.duration = snapshot.execution.endTime - snapshot.execution.startTime;
+        snapshot.execution.success = true;
+        if (options?.tokens && this.trackTokens) {
+            snapshot.tokens = options.tokens;
+            if (this.trackCost && options.tokens.model) {
+                const costConfig = this.costConfigs.get(options.tokens.model);
+                if (costConfig) {
+                    snapshot.cost = this.calculateCost(options.tokens, costConfig);
+                }
+            }
+        }
+        if (options?.custom) {
+            snapshot.custom = options.custom;
+        }
+        this.history.push(snapshot);
+        this.snapshots.delete(id);
+        this.logger.debug(`Ended tracking: ${id} (${snapshot.execution.duration}ms)`);
+        return snapshot;
+    }
+    fail(id, error) {
+        const snapshot = this.snapshots.get(id);
+        if (!snapshot) {
+            throw new Error(`No tracking found for: ${id}`);
+        }
+        snapshot.execution.endTime = Date.now();
+        snapshot.execution.duration = snapshot.execution.endTime - snapshot.execution.startTime;
+        snapshot.execution.success = false;
+        snapshot.execution.error = error;
+        this.history.push(snapshot);
+        this.snapshots.delete(id);
+        this.logger.debug(`Failed tracking: ${id} (${error.message})`);
+        return snapshot;
+    }
+    addCostConfig(config) {
+        this.costConfigs.set(config.model, config);
+        this.logger.debug(`Added cost config for: ${config.model}`);
+    }
+    calculateCost(tokens, config) {
+        const inputCost = (tokens.inputTokens / 1000) * config.inputCostPer1K;
+        const outputCost = (tokens.outputTokens / 1000) * config.outputCostPer1K;
+        return {
+            inputCost,
+            outputCost,
+            totalCost: inputCost + outputCost,
+            currency: config.currency ?? 'USD',
+        };
+    }
+    getCurrent(id) {
+        return this.snapshots.get(id);
+    }
+    getHistory(filter) {
+        if (!filter)
+            return this.history;
+        return this.history.filter((snapshot) => {
+            if (filter.id && snapshot.id !== filter.id)
+                return false;
+            if (filter.success !== undefined && snapshot.execution.success !== filter.success) {
+                return false;
+            }
+            return true;
+        });
+    }
+    getAggregated(filter) {
+        const snapshots = this.getHistory(filter);
+        const successful = snapshots.filter((s) => s.execution.success);
+        const failed = snapshots.filter((s) => !s.execution.success);
+        const totalDuration = snapshots.reduce((sum, s) => sum + (s.execution.duration ?? 0), 0);
+        const totalTokens = snapshots.reduce((sum, s) => sum + (s.tokens?.totalTokens ?? 0), 0);
+        const totalCost = snapshots.reduce((sum, s) => sum + (s.cost?.totalCost ?? 0), 0);
+        const byAgent = {};
+        for (const snapshot of snapshots) {
+            if (!byAgent[snapshot.id]) {
+                byAgent[snapshot.id] = [];
+            }
+            byAgent[snapshot.id].push(snapshot);
+        }
+        return {
+            totalExecutions: snapshots.length,
+            successfulExecutions: successful.length,
+            failedExecutions: failed.length,
+            totalDuration,
+            averageDuration: snapshots.length > 0 ? totalDuration / snapshots.length : 0,
+            totalTokens: totalTokens > 0 ? totalTokens : undefined,
+            totalCost: totalCost > 0 ? totalCost : undefined,
+            byAgent,
+        };
+    }
+    clearHistory() {
+        this.history = [];
+        this.logger.debug('Cleared metrics history');
+    }
+    export() {
+        return JSON.stringify({
+            current: Array.from(this.snapshots.values()),
+            history: this.history,
+            aggregated: this.getAggregated(),
+        }, null, 2);
+    }
+    exportOTel() {
+        return this.history.map((snapshot) => ({
+            traceId: this.generateTraceId(),
+            spanId: this.generateSpanId(),
+            name: snapshot.id,
+            kind: 'INTERNAL',
+            startTimeUnixNano: snapshot.execution.startTime * 1000000,
+            endTimeUnixNano: (snapshot.execution.endTime ?? Date.now()) * 1000000,
+            attributes: {
+                'societyai.success': snapshot.execution.success,
+                'societyai.duration_ms': snapshot.execution.duration,
+                'societyai.tokens.input': snapshot.tokens?.inputTokens,
+                'societyai.tokens.output': snapshot.tokens?.outputTokens,
+                'societyai.tokens.total': snapshot.tokens?.totalTokens,
+                'societyai.cost.total': snapshot.cost?.totalCost,
+                'societyai.cost.currency': snapshot.cost?.currency,
+                ...snapshot.custom,
+            },
+            status: {
+                code: snapshot.execution.success ? 'OK' : 'ERROR',
+                message: snapshot.execution.error?.message,
+            },
+        }));
+    }
+    generateTraceId() {
+        return Array.from({ length: 32 }, () => Math.floor(Math.random() * 16).toString(16)).join('');
+    }
+    generateSpanId() {
+        return Array.from({ length: 16 }, () => Math.floor(Math.random() * 16).toString(16)).join('');
+    }
+}
+exports.MetricsTracker = MetricsTracker;
+class MetricsBuilder {
+    constructor() {
+        this.trackTokens = false;
+        this.trackCost = false;
+        this.costConfigs = [];
+    }
+    static create() {
+        return new MetricsBuilder();
+    }
+    withTokenTracking() {
+        this.trackTokens = true;
+        return this;
+    }
+    withCostTracking(config) {
+        this.trackCost = true;
+        this.costConfigs.push(config);
+        return this;
+    }
+    build() {
+        const tracker = new MetricsTracker({
+            trackTokens: this.trackTokens,
+            trackCost: this.trackCost,
+        });
+        for (const config of this.costConfigs) {
+            tracker.addCostConfig(config);
+        }
+        return tracker;
+    }
+}
+exports.MetricsBuilder = MetricsBuilder;
+class TokenCounter {
+    static estimate(text) {
+        return Math.ceil(text.length / 4);
+    }
+    static count(prompt, response) {
+        const inputTokens = this.estimate(prompt);
+        const outputTokens = this.estimate(response);
+        return {
+            inputTokens,
+            outputTokens,
+            totalTokens: inputTokens + outputTokens,
+        };
+    }
+}
+exports.TokenCounter = TokenCounter;
+class PerformanceProfiler {
+    constructor() {
+        this.marks = new Map();
+        this.measures = new Map();
+    }
+    mark(name) {
+        this.marks.set(name, Date.now());
+    }
+    measure(name, startMark, endMark) {
+        const start = this.marks.get(startMark);
+        if (!start) {
+            throw new Error(`Start mark not found: ${startMark}`);
+        }
+        const end = endMark ? this.marks.get(endMark) : Date.now();
+        if (!end) {
+            throw new Error(`End mark not found: ${endMark}`);
+        }
+        const duration = end - start;
+        this.measures.set(name, duration);
+        return duration;
+    }
+    getMeasures() {
+        return Object.fromEntries(this.measures);
+    }
+    clear() {
+        this.marks.clear();
+        this.measures.clear();
+    }
+}
+exports.PerformanceProfiler = PerformanceProfiler;
+exports.CommonCostConfigs = {
+    'gpt-4': {
+        model: 'gpt-4',
+        inputCostPer1K: 0.03,
+        outputCostPer1K: 0.06,
+        currency: 'USD',
+    },
+    'gpt-4-turbo': {
+        model: 'gpt-4-turbo',
+        inputCostPer1K: 0.01,
+        outputCostPer1K: 0.03,
+        currency: 'USD',
+    },
+    'gpt-3.5-turbo': {
+        model: 'gpt-3.5-turbo',
+        inputCostPer1K: 0.0015,
+        outputCostPer1K: 0.002,
+        currency: 'USD',
+    },
+    'claude-3-opus': {
+        model: 'claude-3-opus',
+        inputCostPer1K: 0.015,
+        outputCostPer1K: 0.075,
+        currency: 'USD',
+    },
+    'claude-3-sonnet': {
+        model: 'claude-3-sonnet',
+        inputCostPer1K: 0.003,
+        outputCostPer1K: 0.015,
+        currency: 'USD',
+    },
+};
+//# sourceMappingURL=metrics.js.map

package/dist/metrics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.js","sourceRoot":"","sources":["../src/metrics.ts"],"names":[],"mappings":";;;AA4BA,qCAAqC;AA+GrC,MAAa,cAAc;IAQzB,YACE,UAGI,EAAE;QAXA,cAAS,GAAiC,IAAI,GAAG,EAAE,CAAC;QACpD,YAAO,GAAsB,EAAE,CAAC;QAChC,gBAAW,GAA4B,IAAI,GAAG,EAAE,CAAC;QACjD,WAAM,GAAG,IAAA,kBAAS,GAAE,CAAC;QACrB,gBAAW,GAAY,KAAK,CAAC;QAC7B,cAAS,GAAY,KAAK,CAAC;QAQjC,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,KAAK,CAAC;QAChD,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;IAC9C,CAAC;IAKD,KAAK,CAAC,EAAU,EAAE,QAAkC;QAClD,MAAM,QAAQ,GAAoB;YAChC,EAAE;YACF,SAAS,EAAE;gBACT,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,OAAO,EAAE,KAAK;aACf;YACD,QAAQ;SACT,CAAC;QAEF,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;QACjC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;IAC/C,CAAC;IAKD,GAAG,CAAC,EAAU,EAAE,OAAoE;QAClF,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACxC,QAAQ,CAAC,SAAS,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC;QACxF,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC;QAElC,IAAI,OAAO,EAAE,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACxC,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAEjC,IAAI,IAAI,CAAC,SAAS,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAC9D,IAAI,UAAU,EAAE,CAAC;oBACf,QAAQ,CAAC,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;YACpB,QAAQ,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QACnC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAE1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,KAAK,QAAQ,CAAC,SAAS,CAAC,QAAQ,KAAK,CAAC,CAAC;QAC9E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,IAAI,CAAC,EAAU,EAAE,KAAY;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACxC,QAAQ,CAAC,SAAS,CAAC,QAAQ,GAAG,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,QAAQ,CAAC,SAAS,CAAC,SAAS,CAAC;QACxF,QAAQ,CAAC,SAAS,CAAC,OAAO,GAAG,KAAK,CAAC;QACnC,QAAQ,CAAC,SAAS,CAAC,KAAK,GAAG,KAAK,CAAC;QAEjC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5B,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAE1B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC;QAC/D,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,aAAa,CAAC,MAAkB;QAC9B,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;IAC9D,CAAC;IAKO,aAAa,CAAC,MAAoB,EAAE,MAAkB;QAC5D,MAAM,SAAS,GAAG,CAAC,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,MAAM,CAAC,cAAc,CAAC;QACtE,MAAM,UAAU,GAAG,CAAC,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,MAAM,CAAC,eAAe,CAAC;QAEzE,OAAO;YACL,SAAS;YACT,UAAU;YACV,SAAS,EAAE,SAAS,GAAG,UAAU;YACjC,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,KAAK;SACnC,CAAC;IACJ,CAAC;IAKD,UAAU,CAAC,EAAU;QACnB,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAChC,CAAC;IAKD,UAAU,CAAC,MAA2C;QACpD,IAAI,CAAC,MAAM;YAAE,OAAO,IAAI,CAAC,OAAO,CAAC;QAEjC,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE;YACtC,IAAI,MAAM,CAAC,EAAE,IAAI,QAAQ,CAAC,EAAE,KAAK,MAAM,CAAC,EAAE;gBAAE,OAAO,KAAK,CAAC;YACzD,IAAI,MAAM,CAAC,OAAO,KAAK,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,OAAO,KAAK,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClF,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC;IAKD,aAAa,CAAC,MAAwB;QACpC,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAE1C,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChE,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAE7D,MAAM,aAAa,GAAG,SAAS,CAAC,MAAM,CACpC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,IAAI,CAAC,CAAC,EAC7C,CAAC,CACF,CAAC;QAEF,MAAM,WAAW,GAAG,SAAS,CAAC,MAAM,CAClC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,WAAW,IAAI,CAAC,CAAC,EAC9C,CAAC,CACF,CAAC;QAEF,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAChC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,SAAS,IAAI,CAAC,CAAC,EAC1C,CAAC,CACF,CAAC;QAGF,MAAM,OAAO,GAAsC,EAAE,CAAC;QACtD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YACjC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;gBAC1B,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC;YAC5B,CAAC;YACD,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,CAAC;QAED,OAAO;YACL,eAAe,EAAE,SAAS,CAAC,MAAM;YACjC,oBAAoB,EAAE,UAAU,CAAC,MAAM;YACvC,gBAAgB,EAAE,MAAM,CAAC,MAAM;YAC/B,aAAa;YACb,eAAe,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC5E,WAAW,EAAE,WAAW,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;YACtD,SAAS,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YAChD,OAAO;SACR,CAAC;IACJ,CAAC;IAKD,YAAY;QACV,IAAI,CAAC,OAAO,GAAG,EAAE,CAAC;QAClB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;IAC/C,CAAC;IAKD,MAAM;QACJ,OAAO,IAAI,CAAC,SAAS,CAAC;YACpB,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;YAC5C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,UAAU,EAAE,IAAI,CAAC,aAAa,EAAE;SACjC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACd,CAAC;IAKD,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACrC,OAAO,EAAE,IAAI,CAAC,eAAe,EAAE;YAC/B,MAAM,EAAE,IAAI,CAAC,cAAc,EAAE;YAC7B,IAAI,EAAE,QAAQ,CAAC,EAAE;YACjB,IAAI,EAAE,UAAU;YAChB,iBAAiB,EAAE,QAAQ,CAAC,SAAS,CAAC,SAAS,GAAG,OAAS;YAC3D,eAAe,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,OAAS;YACvE,UAAU,EAAE;gBACV,mBAAmB,EAAE,QAAQ,CAAC,SAAS,CAAC,OAAO;gBAC/C,uBAAuB,EAAE,QAAQ,CAAC,SAAS,CAAC,QAAQ;gBACpD,wBAAwB,EAAE,QAAQ,CAAC,MAAM,EAAE,WAAW;gBACtD,yBAAyB,EAAE,QAAQ,CAAC,MAAM,EAAE,YAAY;gBACxD,wBAAwB,EAAE,QAAQ,CAAC,MAAM,EAAE,WAAW;gBACtD,sBAAsB,EAAE,QAAQ,CAAC,IAAI,EAAE,SAAS;gBAChD,yBAAyB,EAAE,QAAQ,CAAC,IAAI,EAAE,QAAQ;gBAClD,GAAG,QAAQ,CAAC,MAAM;aACnB;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO;gBACjD,OAAO,EAAE,QAAQ,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO;aAC3C;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAKO,eAAe;QACrB,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,CACrC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAC5C,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACb,CAAC;IAKO,cAAc;QACpB,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,CACrC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAC5C,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACb,CAAC;CACF;AArPD,wCAqPC;AA0BD,MAAa,cAAc;IAA3B;QACU,gBAAW,GAAY,KAAK,CAAC;QAC7B,cAAS,GAAY,KAAK,CAAC;QAC3B,gBAAW,GAAiB,EAAE,CAAC;IAsCzC,CAAC;IApCC,MAAM,CAAC,MAAM;QACX,OAAO,IAAI,cAAc,EAAE,CAAC;IAC9B,CAAC;IAKD,iBAAiB;QACf,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,gBAAgB,CAAC,MAAkB;QACjC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAKD,KAAK;QACH,MAAM,OAAO,GAAG,IAAI,cAAc,CAAC;YACjC,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;SAC1B,CAAC,CAAC;QAEH,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACtC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAzCD,wCAyCC;AASD,MAAa,YAAY;IAKvB,MAAM,CAAC,QAAQ,CAAC,IAAY;QAC1B,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACpC,CAAC;IAKD,MAAM,CAAC,KAAK,CAAC,MAAc,EAAE,QAAgB;QAC3C,MAAM,WAAW,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAE7C,OAAO;YACL,WAAW;YACX,YAAY;YACZ,WAAW,EAAE,WAAW,GAAG,YAAY;SACxC,CAAC;IACJ,CAAC;CACF;AAtBD,oCAsBC;AASD,MAAa,mBAAmB;IAAhC;QACU,UAAK,GAAwB,IAAI,GAAG,EAAE,CAAC;QACvC,aAAQ,GAAwB,IAAI,GAAG,EAAE,CAAC;IA0CpD,CAAC;IArCC,IAAI,CAAC,IAAY;QACf,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IACnC,CAAC;IAKD,OAAO,CAAC,IAAY,EAAE,SAAiB,EAAE,OAAgB;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,yBAAyB,SAAS,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3D,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,uBAAuB,OAAO,EAAE,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,QAAQ,GAAG,GAAG,GAAG,KAAK,CAAC;QAC7B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAClC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAKD,WAAW;QACT,OAAO,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC3C,CAAC;IAKD,KAAK;QACH,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACxB,CAAC;CACF;AA5CD,kDA4CC;AASY,QAAA,iBAAiB,GAAG;IAC/B,OAAO,EAAE;QACP,KAAK,EAAE,OAAO;QACd,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;QACrB,QAAQ,EAAE,KAAK;KAChB;IACD,aAAa,EAAE;QACb,KAAK,EAAE,aAAa;QACpB,cAAc,EAAE,IAAI;QACpB,eAAe,EAAE,IAAI;QACrB,QAAQ,EAAE,KAAK;KAChB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,eAAe;QACtB,cAAc,EAAE,MAAM;QACtB,eAAe,EAAE,KAAK;QACtB,QAAQ,EAAE,KAAK;KAChB;IACD,eAAe,EAAE;QACf,KAAK,EAAE,eAAe;QACtB,cAAc,EAAE,KAAK;QACrB,eAAe,EAAE,KAAK;QACtB,QAAQ,EAAE,KAAK;KAChB;IACD,iBAAiB,EAAE;QACjB,KAAK,EAAE,iBAAiB;QACxB,cAAc,EAAE,KAAK;QACrB,eAAe,EAAE,KAAK;QACtB,QAAQ,EAAE,KAAK;KAChB;CACF,CAAC"}

package/dist/middleware.d.ts

@@ -0,0 +1,147 @@
+import { AIModel, StepResult, WorkflowContext } from './types';
+export interface MiddlewareContext {
+    input: unknown;
+    processedInput: unknown;
+    metadata: Map<string, unknown>;
+    startTime: number;
+    agentId?: string;
+    stepId?: string;
+    signal?: AbortSignal;
+}
+export interface MiddlewareResult {
+    output: string;
+    continue: boolean;
+    metadata?: Record<string, unknown>;
+}
+export type NextFunction = (ctx: MiddlewareContext) => Promise<MiddlewareResult>;
+export type MiddlewareFn = (ctx: MiddlewareContext, next: NextFunction) => Promise<MiddlewareResult>;
+export interface Middleware {
+    name: string;
+    description?: string;
+    priority?: number;
+    fn: MiddlewareFn;
+}
+export declare class MiddlewareChain {
+    private middlewares;
+    private model?;
+    static create(): MiddlewareChain;
+    use(middleware: Middleware | MiddlewareFn): this;
+    useAt(index: number, middleware: Middleware): this;
+    useBefore(name: string, middleware: Middleware): this;
+    useAfter(name: string, middleware: Middleware): this;
+    remove(name: string): this;
+    forModel(model: AIModel): this;
+    getMiddlewares(): readonly Middleware[];
+    sortByPriority(): this;
+    build(): ComposedMiddleware;
+    wrap(model: AIModel): MiddlewareWrappedModel;
+}
+export declare class ComposedMiddleware {
+    private middlewares;
+    private model?;
+    constructor(middlewares: Middleware[], model?: AIModel | undefined);
+    execute(input: unknown, signal?: AbortSignal): Promise<MiddlewareResult>;
+}
+export declare class MiddlewareWrappedModel implements AIModel {
+    private model;
+    private chain;
+    constructor(model: AIModel, middlewares: Middleware[]);
+    name(): string;
+    process(prompt: unknown, signal?: AbortSignal): Promise<string>;
+    supportsPromptType(promptType: string): boolean;
+}
+export declare const Middlewares: {
+    readonly logging: (options?: {
+        prefix?: string;
+        logInput?: boolean;
+        logOutput?: boolean;
+    }) => Middleware;
+    readonly timing: (options?: {
+        onComplete?: (durationMs: number) => void;
+    }) => Middleware;
+    readonly retry: (options?: {
+        maxAttempts?: number;
+        delay?: number;
+        backoffFactor?: number;
+        retryOn?: (error: Error) => boolean;
+    }) => Middleware;
+    readonly cache: (options?: {
+        ttl?: number;
+        keyGenerator?: (input: unknown) => string;
+        storage?: Map<string, {
+            value: string;
+            expires: number;
+        }>;
+    }) => Middleware;
+    readonly rateLimit: (options: {
+        maxRequests: number;
+        windowMs: number;
+        onLimitReached?: () => void;
+    }) => Middleware;
+    readonly timeout: (ms: number) => Middleware;
+    readonly transformInput: (transformer: (input: unknown) => unknown) => Middleware;
+    readonly transformOutput: (transformer: (output: string) => string) => Middleware;
+    readonly validation: (options: {
+        validateInput?: (input: unknown) => boolean | string;
+        validateOutput?: (output: string) => boolean | string;
+    }) => Middleware;
+    readonly fallback: (fallbackValue: string | ((error: Error) => string)) => Middleware;
+    readonly metrics: (collector: MetricsCollector) => Middleware;
+    readonly circuitBreaker: (options: {
+        threshold: number;
+        timeout: number;
+        onOpen?: () => void;
+        onClose?: () => void;
+        onHalfOpen?: () => void;
+    }) => Middleware;
+    readonly dedupe: () => Middleware;
+};
+export interface MetricsCollector {
+    incrementCounter(name: string, value?: number): void;
+    recordHistogram(name: string, value: number): void;
+    recordGauge(name: string, value: number): void;
+}
+export declare class InMemoryMetricsCollector implements MetricsCollector {
+    private counters;
+    private histograms;
+    private gauges;
+    incrementCounter(name: string, value?: number): void;
+    recordHistogram(name: string, value: number): void;
+    recordGauge(name: string, value: number): void;
+    getCounter(name: string): number;
+    getHistogram(name: string): {
+        count: number;
+        sum: number;
+        avg: number;
+        min: number;
+        max: number;
+    };
+    getGauge(name: string): number;
+    getAll(): {
+        counters: Record<string, number>;
+        histograms: Record<string, number[]>;
+        gauges: Record<string, number>;
+    };
+    reset(): void;
+}
+export interface StepMiddlewareContext {
+    stepId: string;
+    stepName: string;
+    agentIds: string[];
+    input: string;
+    workflowContext: WorkflowContext;
+    metadata: Map<string, unknown>;
+}
+export type StepMiddlewareFn = (ctx: StepMiddlewareContext, next: () => Promise<StepResult[]>) => Promise<StepResult[]>;
+export interface StepMiddleware {
+    name: string;
+    description?: string;
+    fn: StepMiddlewareFn;
+}
+export declare const StepMiddlewares: {
+    readonly logging: () => StepMiddleware;
+    readonly timing: (onComplete?: (stepId: string, durationMs: number) => void) => StepMiddleware;
+    readonly filterResults: (predicate: (result: StepResult) => boolean) => StepMiddleware;
+    readonly enrichResults: (enricher: (result: StepResult, ctx: StepMiddlewareContext) => StepResult) => StepMiddleware;
+};
+//# sourceMappingURL=middleware.d.ts.map

package/dist/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../src/middleware.ts"],"names":[],"mappings":"AA0BA,OAAO,EAAE,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAU/D,MAAM,WAAW,iBAAiB;IAEhC,KAAK,EAAE,OAAO,CAAC;IAEf,cAAc,EAAE,OAAO,CAAC;IAExB,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAE/B,SAAS,EAAE,MAAM,CAAC;IAElB,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB,MAAM,CAAC,EAAE,WAAW,CAAC;CACtB;AAKD,MAAM,WAAW,gBAAgB;IAE/B,MAAM,EAAE,MAAM,CAAC;IAEf,QAAQ,EAAE,OAAO,CAAC;IAElB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACpC;AAKD,MAAM,MAAM,YAAY,GAAG,CAAC,GAAG,EAAE,iBAAiB,KAAK,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAKjF,MAAM,MAAM,YAAY,GAAG,CACzB,GAAG,EAAE,iBAAiB,EACtB,IAAI,EAAE,YAAY,KACf,OAAO,CAAC,gBAAgB,CAAC,CAAC;AAK/B,MAAM,WAAW,UAAU;IAEzB,IAAI,EAAE,MAAM,CAAC;IAEb,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAElB,EAAE,EAAE,YAAY,CAAC;CAClB;AASD,qBAAa,eAAe;IAC1B,OAAO,CAAC,WAAW,CAAoB;IACvC,OAAO,CAAC,KAAK,CAAC,CAAU;IAKxB,MAAM,CAAC,MAAM,IAAI,eAAe;IAOhC,GAAG,CAAC,UAAU,EAAE,UAAU,GAAG,YAAY,GAAG,IAAI;IAehD,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI;IAQlD,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI;IAarD,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,UAAU,GAAG,IAAI;IAapD,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;IAQ1B,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,IAAI;IAQ9B,cAAc,IAAI,SAAS,UAAU,EAAE;IAOvC,cAAc,IAAI,IAAI;IAQtB,KAAK,IAAI,kBAAkB;IAO3B,IAAI,CAAC,KAAK,EAAE,OAAO,GAAG,sBAAsB;CAG7C;AAKD,qBAAa,kBAAkB;IAE3B,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,KAAK,CAAC;gBADN,WAAW,EAAE,UAAU,EAAE,EACzB,KAAK,CAAC,EAAE,OAAO,YAAA;IAMnB,OAAO,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC;CA2B/E;AAKD,qBAAa,sBAAuB,YAAW,OAAO;IAIlD,OAAO,CAAC,KAAK;IAHf,OAAO,CAAC,KAAK,CAAqB;gBAGxB,KAAK,EAAE,OAAO,EACtB,WAAW,EAAE,UAAU,EAAE;IAK3B,IAAI,IAAI,MAAM;IAIR,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC;IAKrE,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO;CAGhD;AASD,eAAO,MAAM,WAAW;iCAIF;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,OAAO,CAAC;QAAC,SAAS,CAAC,EAAE,OAAO,CAAA;KAAE,KAAG,UAAU;gCA2B1E;QAAE,UAAU,CAAC,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAA;KAAE,KAAG,UAAU;+BAsB3D;QAChB,WAAW,CAAC,EAAE,MAAM,CAAC;QACrB,KAAK,CAAC,EAAE,MAAM,CAAC;QACf,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,KAAK,OAAO,CAAC;KACrC,KAAG,UAAU;+BAuCI;QAChB,GAAG,CAAC,EAAE,MAAM,CAAC;QACb,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,MAAM,CAAC;QAC1C,OAAO,CAAC,EAAE,GAAG,CAAC,MAAM,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,OAAO,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC3D,KAAG,UAAU;kCAkCO;QACnB,WAAW,EAAE,MAAM,CAAC;QACpB,QAAQ,EAAE,MAAM,CAAC;QACjB,cAAc,CAAC,EAAE,MAAM,IAAI,CAAC;KAC7B,KAAG,UAAU;2BA8BA,MAAM,KAAG,UAAU;2CAgCH,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,KAAG,UAAU;4CAavC,CAAC,MAAM,EAAE,MAAM,KAAK,MAAM,KAAG,UAAU;mCAgBhD;QACpB,aAAa,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,KAAK,OAAO,GAAG,MAAM,CAAC;QACrD,cAAc,CAAC,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,GAAG,MAAM,CAAC;KACvD,KAAG,UAAU;uCAgCY,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,KAAK,MAAM,CAAC,KAAG,UAAU;kCAmBrD,gBAAgB,KAAG,UAAU;uCAwBxB;QACxB,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,MAAM,CAAC,EAAE,MAAM,IAAI,CAAC;QACpB,OAAO,CAAC,EAAE,MAAM,IAAI,CAAC;QACrB,UAAU,CAAC,EAAE,MAAM,IAAI,CAAC;KACzB,KAAG,UAAU;2BAgDF,UAAU;CA0Bd,CAAC;AASX,MAAM,WAAW,gBAAgB;IAC/B,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrD,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACnD,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;CAChD;AAKD,qBAAa,wBAAyB,YAAW,gBAAgB;IAC/D,OAAO,CAAC,QAAQ,CAA6B;IAC7C,OAAO,CAAC,UAAU,CAA+B;IACjD,OAAO,CAAC,MAAM,CAA6B;IAE3C,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,GAAE,MAAU,GAAG,IAAI;IAIvD,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAMlD,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAI9C,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIhC,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE;IAejG,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAI9B,MAAM,IAAI;QAAE,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;KAAE;IAQpH,KAAK,IAAI,IAAI;CAKd;AASD,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;IACd,eAAe,EAAE,eAAe,CAAC;IACjC,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAKD,MAAM,MAAM,gBAAgB,GAAG,CAC7B,GAAG,EAAE,qBAAqB,EAC1B,IAAI,EAAE,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC,KAC9B,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;AAK3B,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,EAAE,EAAE,gBAAgB,CAAC;CACtB;AAKD,eAAO,MAAM,eAAe;4BAIb,cAAc;mCAcL,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,KAAK,IAAI,KAAG,cAAc;wCAexD,CAAC,MAAM,EAAE,UAAU,KAAK,OAAO,KAAG,cAAc;uCAWjD,CAAC,MAAM,EAAE,UAAU,EAAE,GAAG,EAAE,qBAAqB,KAAK,UAAU,KAAG,cAAc;CAOjG,CAAC"}