agentic-qe 1.0.4 → 1.0.5

package/dist/core/routing/CostTracker.js

@@ -0,0 +1,210 @@
+"use strict";
+/**
+ * Cost Tracker
+ * Tracks model usage costs and provides analytics
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CostTracker = void 0;
+const types_1 = require("./types");
+const ModelRules_1 = require("./ModelRules");
+class CostTracker {
+    constructor(memoryStore, eventBus) {
+        this.memoryStore = memoryStore;
+        this.eventBus = eventBus;
+        this.costs = new Map();
+        this.sessionStartTime = Date.now();
+        this.initializeCosts();
+    }
+    /**
+     * Initialize cost tracking for all models
+     */
+    initializeCosts() {
+        Object.values(types_1.AIModel).forEach((model) => {
+            this.costs.set(model, {
+                modelId: model,
+                tokensUsed: 0,
+                estimatedCost: 0,
+                requestCount: 0,
+                avgTokensPerRequest: 0,
+                timestamp: Date.now(),
+            });
+        });
+    }
+    /**
+     * Track cost for a model usage
+     */
+    async trackCost(modelId, tokens) {
+        const cost = this.costs.get(modelId);
+        if (!cost) {
+            throw new Error(`Unknown model: ${modelId}`);
+        }
+        const capability = ModelRules_1.MODEL_CAPABILITIES[modelId];
+        const estimatedCost = tokens * capability.costPerToken;
+        // Update local cost tracking
+        cost.tokensUsed += tokens;
+        cost.estimatedCost += estimatedCost;
+        cost.requestCount += 1;
+        cost.avgTokensPerRequest = cost.tokensUsed / cost.requestCount;
+        cost.timestamp = Date.now();
+        this.costs.set(modelId, cost);
+        // Persist to SwarmMemoryManager
+        await this.persistCosts();
+        // Emit cost tracking event
+        this.eventBus.emit('router:cost-tracked', {
+            model: modelId,
+            tokens,
+            cost: estimatedCost,
+            totalCost: cost.estimatedCost,
+        });
+    }
+    /**
+     * Persist costs to SwarmMemoryManager
+     */
+    async persistCosts() {
+        const costsArray = Array.from(this.costs.values());
+        await this.memoryStore.store('routing/costs', costsArray, {
+            partition: 'coordination',
+            ttl: 86400, // 24 hours
+        });
+    }
+    /**
+     * Load costs from SwarmMemoryManager
+     */
+    async loadCosts() {
+        try {
+            const stored = await this.memoryStore.retrieve('routing/costs', {
+                partition: 'coordination',
+            });
+            if (stored && Array.isArray(stored)) {
+                stored.forEach((cost) => {
+                    this.costs.set(cost.modelId, cost);
+                });
+            }
+        }
+        catch (error) {
+            // No stored costs, use initialized values
+        }
+    }
+    /**
+     * Get router statistics
+     */
+    async getStats() {
+        const costsArray = Array.from(this.costs.values());
+        const totalRequests = costsArray.reduce((sum, c) => sum + c.requestCount, 0);
+        const totalCost = costsArray.reduce((sum, c) => sum + c.estimatedCost, 0);
+        // Calculate cost savings vs single model (Claude Sonnet 4.5)
+        const baselineModel = ModelRules_1.MODEL_CAPABILITIES[types_1.AIModel.CLAUDE_SONNET_4_5];
+        const totalTokens = costsArray.reduce((sum, c) => sum + c.tokensUsed, 0);
+        const baselineCost = totalTokens * baselineModel.costPerToken;
+        const costSavings = baselineCost - totalCost;
+        // Model distribution
+        const modelDistribution = {};
+        Object.values(types_1.AIModel).forEach((model) => {
+            const cost = this.costs.get(model);
+            modelDistribution[model] = cost?.requestCount || 0;
+        });
+        // Average costs
+        const avgCostPerTask = totalRequests > 0 ? totalCost / totalRequests : 0;
+        const avgCostPerTest = await this.calculateAvgCostPerTest();
+        return {
+            totalRequests,
+            totalCost,
+            costSavings,
+            modelDistribution,
+            avgCostPerTask,
+            avgCostPerTest,
+        };
+    }
+    /**
+     * Calculate average cost per test generated
+     */
+    async calculateAvgCostPerTest() {
+        try {
+            // Get test count from memory
+            const testMetrics = await this.memoryStore.retrieve('aqe/test-metrics', {
+                partition: 'coordination',
+            });
+            if (testMetrics && typeof testMetrics === 'object' && 'totalTests' in testMetrics) {
+                const totalTests = testMetrics.totalTests;
+                if (totalTests > 0) {
+                    const stats = await this.getStats();
+                    return stats.totalCost / totalTests;
+                }
+            }
+        }
+        catch (error) {
+            // No test metrics available
+        }
+        return 0;
+    }
+    /**
+     * Export cost dashboard data
+     */
+    async exportCostDashboard() {
+        const stats = await this.getStats();
+        const costsArray = Array.from(this.costs.values());
+        return {
+            summary: {
+                totalCost: stats.totalCost.toFixed(4),
+                totalRequests: stats.totalRequests,
+                costSavings: stats.costSavings.toFixed(4),
+                savingsPercentage: stats.totalCost > 0
+                    ? ((stats.costSavings / (stats.totalCost + stats.costSavings)) * 100).toFixed(2)
+                    : '0.00',
+                avgCostPerTask: stats.avgCostPerTask.toFixed(4),
+                avgCostPerTest: stats.avgCostPerTest.toFixed(4),
+                sessionDuration: this.formatDuration(Date.now() - this.sessionStartTime),
+            },
+            models: costsArray.map((cost) => {
+                const capability = ModelRules_1.MODEL_CAPABILITIES[cost.modelId];
+                return {
+                    model: cost.modelId,
+                    requests: cost.requestCount,
+                    tokensUsed: cost.tokensUsed,
+                    cost: cost.estimatedCost.toFixed(4),
+                    avgTokensPerRequest: Math.round(cost.avgTokensPerRequest),
+                    costPerToken: capability.costPerToken,
+                    percentage: stats.totalCost > 0
+                        ? ((cost.estimatedCost / stats.totalCost) * 100).toFixed(2)
+                        : '0.00',
+                };
+            }),
+            distribution: stats.modelDistribution,
+            timestamp: new Date().toISOString(),
+        };
+    }
+    /**
+     * Format duration in human-readable format
+     */
+    formatDuration(ms) {
+        const seconds = Math.floor(ms / 1000);
+        const minutes = Math.floor(seconds / 60);
+        const hours = Math.floor(minutes / 60);
+        if (hours > 0) {
+            return `${hours}h ${minutes % 60}m`;
+        }
+        else if (minutes > 0) {
+            return `${minutes}m ${seconds % 60}s`;
+        }
+        else {
+            return `${seconds}s`;
+        }
+    }
+    /**
+     * Reset cost tracking (for testing)
+     */
+    async reset() {
+        this.costs.clear();
+        this.initializeCosts();
+        this.sessionStartTime = Date.now();
+        await this.persistCosts();
+    }
+    /**
+     * Get cost for specific model
+     */
+    getModelCost(modelId) {
+        return this.costs.get(modelId);
+    }
+}
+exports.CostTracker = CostTracker;
+//# sourceMappingURL=CostTracker.js.map

package/dist/core/routing/CostTracker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CostTracker.js","sourceRoot":"","sources":["../../../src/core/routing/CostTracker.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,mCAA0D;AAC1D,6CAAkD;AAIlD,MAAa,WAAW;IAMtB,YAAY,WAA+B,EAAE,QAAkB;QAC7D,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACnC,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,eAAe;QACrB,MAAM,CAAC,MAAM,CAAC,eAAO,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACvC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,EAAE;gBACpB,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,CAAC;gBACb,aAAa,EAAE,CAAC;gBAChB,YAAY,EAAE,CAAC;gBACf,mBAAmB,EAAE,CAAC;gBACtB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CAAC,OAAgB,EAAE,MAAc;QAC9C,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,kBAAkB,OAAO,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,MAAM,UAAU,GAAG,+BAAkB,CAAC,OAAO,CAAC,CAAC;QAC/C,MAAM,aAAa,GAAG,MAAM,GAAG,UAAU,CAAC,YAAY,CAAC;QAEvD,6BAA6B;QAC7B,IAAI,CAAC,UAAU,IAAI,MAAM,CAAC;QAC1B,IAAI,CAAC,aAAa,IAAI,aAAa,CAAC;QACpC,IAAI,CAAC,YAAY,IAAI,CAAC,CAAC;QACvB,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC;QAC/D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE5B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAE9B,gCAAgC;QAChC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAE1B,2BAA2B;QAC3B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,qBAAqB,EAAE;YACxC,KAAK,EAAE,OAAO;YACd,MAAM;YACN,IAAI,EAAE,aAAa;YACnB,SAAS,EAAE,IAAI,CAAC,aAAa;SAC9B,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY;QACxB,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACnD,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,eAAe,EAAE,UAAU,EAAE;YACxD,SAAS,EAAE,cAAc;YACzB,GAAG,EAAE,KAAK,EAAE,WAAW;SACxB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS;QACb,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,eAAe,EAAE;gBAC9D,SAAS,EAAE,cAAc;aAC1B,CAAC,CAAC;YAEH,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAe,EAAE,EAAE;oBACjC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBACrC,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,0CAA0C;QAC5C,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QACnD,MAAM,aAAa,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QAC7E,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;QAE1E,6DAA6D;QAC7D,MAAM,aAAa,GAAG,+BAAkB,CAAC,eAAO,CAAC,iBAAiB,CAAC,CAAC;QACpE,MAAM,WAAW,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;QACzE,MAAM,YAAY,GAAG,WAAW,GAAG,aAAa,CAAC,YAAY,CAAC;QAC9D,MAAM,WAAW,GAAG,YAAY,GAAG,SAAS,CAAC;QAE7C,qBAAqB;QACrB,MAAM,iBAAiB,GAA4B,EAAS,CAAC;QAC7D,MAAM,CAAC,MAAM,CAAC,eAAO,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACvC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YACnC,iBAAiB,CAAC,KAAK,CAAC,GAAG,IAAI,EAAE,YAAY,IAAI,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,gBAAgB;QAChB,MAAM,cAAc,GAAG,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;QACzE,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAE5D,OAAO;YACL,aAAa;YACb,SAAS;YACT,WAAW;YACX,iBAAiB;YACjB,cAAc;YACd,cAAc;SACf,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC;YACH,6BAA6B;YAC7B,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,kBAAkB,EAAE;gBACtE,SAAS,EAAE,cAAc;aAC1B,CAAC,CAAC;YAEH,IAAI,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,IAAI,YAAY,IAAI,WAAW,EAAE,CAAC;gBAClF,MAAM,UAAU,GAAI,WAAmB,CAAC,UAAU,CAAC;gBACnD,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;oBACnB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACpC,OAAO,KAAK,CAAC,SAAS,GAAG,UAAU,CAAC;gBACtC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,4BAA4B;QAC9B,CAAC;QAED,OAAO,CAAC,CAAC;IACX,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB;QACvB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;QAEnD,OAAO;YACL,OAAO,EAAE;gBACP,SAAS,EAAE,KAAK,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC;gBACrC,aAAa,EAAE,KAAK,CAAC,aAAa;gBAClC,WAAW,EAAE,KAAK,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC;gBACzC,iBAAiB,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;oBACpC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,WAAW,GAAG,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;oBAChF,CAAC,CAAC,MAAM;gBACV,cAAc,EAAE,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC/C,cAAc,EAAE,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC/C,eAAe,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC;aACzE;YACD,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC9B,MAAM,UAAU,GAAG,+BAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACpD,OAAO;oBACL,KAAK,EAAE,IAAI,CAAC,OAAO;oBACnB,QAAQ,EAAE,IAAI,CAAC,YAAY;oBAC3B,UAAU,EAAE,IAAI,CAAC,UAAU;oBAC3B,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC;oBACnC,mBAAmB,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,mBAAmB,CAAC;oBACzD,YAAY,EAAE,UAAU,CAAC,YAAY;oBACrC,UAAU,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;wBAC7B,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,SAAS,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;wBAC3D,CAAC,CAAC,MAAM;iBACX,CAAC;YACJ,CAAC,CAAC;YACF,YAAY,EAAE,KAAK,CAAC,iBAAiB;YACrC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACpC,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,EAAU;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;QACzC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;QAEvC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,OAAO,GAAG,KAAK,KAAK,OAAO,GAAG,EAAE,GAAG,CAAC;QACtC,CAAC;aAAM,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YACvB,OAAO,GAAG,OAAO,KAAK,OAAO,GAAG,EAAE,GAAG,CAAC;QACxC,CAAC;aAAM,CAAC;YACN,OAAO,GAAG,OAAO,GAAG,CAAC;QACvB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACnB,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACnC,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,OAAgB;QAC3B,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;CACF;AA/ND,kCA+NC"}

package/dist/core/routing/FleetManagerIntegration.d.ts

@@ -0,0 +1,68 @@
+/**
+ * FleetManager Integration for Multi-Model Router
+ * Integrates the AdaptiveModelRouter with FleetManager
+ */
+import { FleetManager } from '../FleetManager';
+import { AdaptiveModelRouter } from './AdaptiveModelRouter';
+import { SwarmMemoryManager } from '../memory/SwarmMemoryManager';
+import { EventBus } from '../EventBus';
+import { RouterConfig } from './types';
+/**
+ * Extended FleetManager with routing capabilities
+ */
+export declare class RoutingEnabledFleetManager {
+    private fleetManager;
+    private router;
+    private memoryStore;
+    private eventBus;
+    constructor(fleetManager: FleetManager, memoryStore: SwarmMemoryManager, eventBus: EventBus, routerConfig?: Partial<RouterConfig>);
+    /**
+     * Setup event handlers for router integration
+     */
+    private setupEventHandlers;
+    /**
+     * Handle task submission - select optimal model
+     */
+    private handleTaskSubmitted;
+    /**
+     * Handle task completion - track actual costs
+     */
+    private handleTaskCompleted;
+    /**
+     * Handle model failure - select fallback
+     */
+    private handleModelFailure;
+    /**
+     * Estimate tokens from result (fallback when not provided)
+     */
+    private estimateTokensFromResult;
+    /**
+     * Get router statistics
+     */
+    getRouterStats(): Promise<import("./types").RouterStats>;
+    /**
+     * Export cost dashboard
+     */
+    exportCostDashboard(): Promise<any>;
+    /**
+     * Enable or disable routing
+     */
+    setRoutingEnabled(enabled: boolean): void;
+    /**
+     * Update router configuration
+     */
+    updateRouterConfig(config: Partial<RouterConfig>): void;
+    /**
+     * Get underlying router instance
+     */
+    getRouter(): AdaptiveModelRouter;
+    /**
+     * Get underlying fleet manager instance
+     */
+    getFleetManager(): FleetManager;
+}
+/**
+ * Factory function to create routing-enabled fleet manager
+ */
+export declare function createRoutingEnabledFleetManager(fleetManager: FleetManager, memoryStore: SwarmMemoryManager, eventBus: EventBus, routerConfig?: Partial<RouterConfig>): RoutingEnabledFleetManager;
+//# sourceMappingURL=FleetManagerIntegration.d.ts.map

package/dist/core/routing/FleetManagerIntegration.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"FleetManagerIntegration.d.ts","sourceRoot":"","sources":["../../../src/core/routing/FleetManagerIntegration.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,8BAA8B,CAAC;AAClE,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAA2B,YAAY,EAAE,MAAM,SAAS,CAAC;AAGhE;;GAEG;AACH,qBAAa,0BAA0B;IACrC,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,MAAM,CAAsB;IACpC,OAAO,CAAC,WAAW,CAAqB;IACxC,OAAO,CAAC,QAAQ,CAAW;gBAGzB,YAAY,EAAE,YAAY,EAC1B,WAAW,EAAE,kBAAkB,EAC/B,QAAQ,EAAE,QAAQ,EAClB,YAAY,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC;IAUtC;;OAEG;IACH,OAAO,CAAC,kBAAkB;IAiB1B;;OAEG;YACW,mBAAmB;IAwBjC;;OAEG;YACW,mBAAmB;IA6BjC;;OAEG;YACW,kBAAkB;IAwBhC;;OAEG;IACH,OAAO,CAAC,wBAAwB;IAOhC;;OAEG;IACG,cAAc;IAIpB;;OAEG;IACG,mBAAmB;IAIzB;;OAEG;IACH,iBAAiB,CAAC,OAAO,EAAE,OAAO,GAAG,IAAI;IAIzC;;OAEG;IACH,kBAAkB,CAAC,MAAM,EAAE,OAAO,CAAC,YAAY,CAAC,GAAG,IAAI;IAIvD;;OAEG;IACH,SAAS,IAAI,mBAAmB;IAIhC;;OAEG;IACH,eAAe,IAAI,YAAY;CAGhC;AAED;;GAEG;AACH,wBAAgB,gCAAgC,CAC9C,YAAY,EAAE,YAAY,EAC1B,WAAW,EAAE,kBAAkB,EAC/B,QAAQ,EAAE,QAAQ,EAClB,YAAY,CAAC,EAAE,OAAO,CAAC,YAAY,CAAC,GACnC,0BAA0B,CAO5B"}

package/dist/core/routing/FleetManagerIntegration.js

@@ -0,0 +1,170 @@
+"use strict";
+/**
+ * FleetManager Integration for Multi-Model Router
+ * Integrates the AdaptiveModelRouter with FleetManager
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RoutingEnabledFleetManager = void 0;
+exports.createRoutingEnabledFleetManager = createRoutingEnabledFleetManager;
+const AdaptiveModelRouter_1 = require("./AdaptiveModelRouter");
+const QETask_1 = require("./QETask");
+/**
+ * Extended FleetManager with routing capabilities
+ */
+class RoutingEnabledFleetManager {
+    constructor(fleetManager, memoryStore, eventBus, routerConfig) {
+        this.fleetManager = fleetManager;
+        this.memoryStore = memoryStore;
+        this.eventBus = eventBus;
+        this.router = new AdaptiveModelRouter_1.AdaptiveModelRouter(memoryStore, eventBus, routerConfig);
+        this.setupEventHandlers();
+    }
+    /**
+     * Setup event handlers for router integration
+     */
+    setupEventHandlers() {
+        // Listen for task submissions to perform model selection
+        this.eventBus.on('task:submitted', async (data) => {
+            await this.handleTaskSubmitted(data.task);
+        });
+        // Listen for task completions to track costs
+        this.eventBus.on('task:completed', async (data) => {
+            await this.handleTaskCompleted(data.task, data.result);
+        });
+        // Listen for model failures to trigger fallback
+        this.eventBus.on('model:failed', async (data) => {
+            await this.handleModelFailure(data.task, data.model, data.error);
+        });
+    }
+    /**
+     * Handle task submission - select optimal model
+     */
+    async handleTaskSubmitted(task) {
+        try {
+            const qeTask = (0, QETask_1.taskToQETask)(task);
+            const selection = await this.router.selectModel(qeTask);
+            // Store model selection in task metadata
+            task.metadata = {
+                ...task.metadata,
+                modelSelection: selection,
+            };
+            // Emit model selection event
+            this.eventBus.emit('router:model-assigned', {
+                taskId: task.getId(),
+                model: selection.model,
+                complexity: selection.complexity,
+                estimatedCost: selection.estimatedCost,
+            });
+        }
+        catch (error) {
+            console.error('Failed to select model for task:', error);
+            // Continue with default model
+        }
+    }
+    /**
+     * Handle task completion - track actual costs
+     */
+    async handleTaskCompleted(task, result) {
+        try {
+            const metadata = task.metadata;
+            if (!metadata || !metadata.modelSelection) {
+                return;
+            }
+            const selection = metadata.modelSelection;
+            const actualTokens = result.tokensUsed || this.estimateTokensFromResult(result);
+            // Track actual cost
+            await this.router.trackCost(selection.model, actualTokens);
+            // Calculate accuracy of cost estimation
+            const estimatedTokens = selection.estimatedCost / 0.00005; // Rough reverse calculation
+            const accuracy = actualTokens > 0 ? Math.abs(1 - estimatedTokens / actualTokens) : 0;
+            this.eventBus.emit('router:cost-accuracy', {
+                taskId: task.getId(),
+                model: selection.model,
+                estimatedTokens,
+                actualTokens,
+                accuracy,
+            });
+        }
+        catch (error) {
+            console.error('Failed to track task cost:', error);
+        }
+    }
+    /**
+     * Handle model failure - select fallback
+     */
+    async handleModelFailure(task, failedModel, error) {
+        try {
+            const qeTask = (0, QETask_1.taskToQETask)(task);
+            const fallbackModel = this.router.getFallbackModel(failedModel, qeTask);
+            // Update task metadata with fallback model
+            task.metadata = {
+                ...task.metadata,
+                fallbackModel,
+                originalModel: failedModel,
+                failureReason: error.message,
+            };
+            this.eventBus.emit('router:fallback-triggered', {
+                taskId: task.getId(),
+                failedModel,
+                fallbackModel,
+                error: error.message,
+            });
+        }
+        catch (error) {
+            console.error('Failed to handle model failure:', error);
+        }
+    }
+    /**
+     * Estimate tokens from result (fallback when not provided)
+     */
+    estimateTokensFromResult(result) {
+        if (!result)
+            return 0;
+        const resultStr = JSON.stringify(result);
+        return Math.round(resultStr.length / 4); // Rough estimate
+    }
+    /**
+     * Get router statistics
+     */
+    async getRouterStats() {
+        return await this.router.getStats();
+    }
+    /**
+     * Export cost dashboard
+     */
+    async exportCostDashboard() {
+        return await this.router.exportCostDashboard();
+    }
+    /**
+     * Enable or disable routing
+     */
+    setRoutingEnabled(enabled) {
+        this.router.setEnabled(enabled);
+    }
+    /**
+     * Update router configuration
+     */
+    updateRouterConfig(config) {
+        this.router.updateConfig(config);
+    }
+    /**
+     * Get underlying router instance
+     */
+    getRouter() {
+        return this.router;
+    }
+    /**
+     * Get underlying fleet manager instance
+     */
+    getFleetManager() {
+        return this.fleetManager;
+    }
+}
+exports.RoutingEnabledFleetManager = RoutingEnabledFleetManager;
+/**
+ * Factory function to create routing-enabled fleet manager
+ */
+function createRoutingEnabledFleetManager(fleetManager, memoryStore, eventBus, routerConfig) {
+    return new RoutingEnabledFleetManager(fleetManager, memoryStore, eventBus, routerConfig);
+}
+//# sourceMappingURL=FleetManagerIntegration.js.map

package/dist/core/routing/FleetManagerIntegration.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FleetManagerIntegration.js","sourceRoot":"","sources":["../../../src/core/routing/FleetManagerIntegration.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAmMH,4EAYC;AA3MD,+DAA4D;AAI5D,qCAAwC;AAExC;;GAEG;AACH,MAAa,0BAA0B;IAMrC,YACE,YAA0B,EAC1B,WAA+B,EAC/B,QAAkB,EAClB,YAAoC;QAEpC,IAAI,CAAC,YAAY,GAAG,YAAY,CAAC;QACjC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,IAAI,yCAAmB,CAAC,WAAW,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;QAE3E,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,kBAAkB;QACxB,yDAAyD;QACzD,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,gBAAgB,EAAE,KAAK,EAAE,IAAoB,EAAE,EAAE;YAChE,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,6CAA6C;QAC7C,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,gBAAgB,EAAE,KAAK,EAAE,IAAiC,EAAE,EAAE;YAC7E,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;QAEH,gDAAgD;QAChD,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,cAAc,EAAE,KAAK,EAAE,IAAkD,EAAE,EAAE;YAC5F,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACnE,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB,CAAC,IAAU;QAC1C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,qBAAY,EAAC,IAAI,CAAC,CAAC;YAClC,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAExD,yCAAyC;YACxC,IAAY,CAAC,QAAQ,GAAG;gBACvB,GAAI,IAAY,CAAC,QAAQ;gBACzB,cAAc,EAAE,SAAS;aAC1B,CAAC;YAEF,6BAA6B;YAC7B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,uBAAuB,EAAE;gBAC1C,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE;gBACpB,KAAK,EAAE,SAAS,CAAC,KAAK;gBACtB,UAAU,EAAE,SAAS,CAAC,UAAU;gBAChC,aAAa,EAAE,SAAS,CAAC,aAAa;aACvC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;YACzD,8BAA8B;QAChC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,mBAAmB,CAAC,IAAU,EAAE,MAAW;QACvD,IAAI,CAAC;YACH,MAAM,QAAQ,GAAI,IAAY,CAAC,QAAQ,CAAC;YACxC,IAAI,CAAC,QAAQ,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;gBAC1C,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,QAAQ,CAAC,cAAgC,CAAC;YAC5D,MAAM,YAAY,GAAG,MAAM,CAAC,UAAU,IAAI,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;YAEhF,oBAAoB;YACpB,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;YAE3D,wCAAwC;YACxC,MAAM,eAAe,GAAG,SAAS,CAAC,aAAa,GAAG,OAAO,CAAC,CAAC,4BAA4B;YACvF,MAAM,QAAQ,GAAG,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,eAAe,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAErF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,sBAAsB,EAAE;gBACzC,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE;gBACpB,KAAK,EAAE,SAAS,CAAC,KAAK;gBACtB,eAAe;gBACf,YAAY;gBACZ,QAAQ;aACT,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,kBAAkB,CAAC,IAAU,EAAE,WAAoB,EAAE,KAAY;QAC7E,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,qBAAY,EAAC,IAAI,CAAC,CAAC;YAClC,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;YAExE,2CAA2C;YAC1C,IAAY,CAAC,QAAQ,GAAG;gBACvB,GAAI,IAAY,CAAC,QAAQ;gBACzB,aAAa;gBACb,aAAa,EAAE,WAAW;gBAC1B,aAAa,EAAE,KAAK,CAAC,OAAO;aAC7B,CAAC;YAEF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,2BAA2B,EAAE;gBAC9C,MAAM,EAAE,IAAI,CAAC,KAAK,EAAE;gBACpB,WAAW;gBACX,aAAa;gBACb,KAAK,EAAE,KAAK,CAAC,OAAO;aACrB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED;;OAEG;IACK,wBAAwB,CAAC,MAAW;QAC1C,IAAI,CAAC,MAAM;YAAE,OAAO,CAAC,CAAC;QAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB;IAC5D,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc;QAClB,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB;QACvB,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,OAAgB;QAChC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAClC,CAAC;IAED;;OAEG;IACH,kBAAkB,CAAC,MAA6B;QAC9C,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED;;OAEG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;CACF;AAjLD,gEAiLC;AAED;;GAEG;AACH,SAAgB,gCAAgC,CAC9C,YAA0B,EAC1B,WAA+B,EAC/B,QAAkB,EAClB,YAAoC;IAEpC,OAAO,IAAI,0BAA0B,CACnC,YAAY,EACZ,WAAW,EACX,QAAQ,EACR,YAAY,CACb,CAAC;AACJ,CAAC"}

package/dist/core/routing/ModelRules.d.ts

@@ -0,0 +1,38 @@
+/**
+ * Model Selection Rules
+ * Task-specific and complexity-based model selection rules
+ */
+import { AIModel, TaskComplexity, ModelCapability } from './types';
+/**
+ * Model capabilities and pricing
+ */
+export declare const MODEL_CAPABILITIES: Record<AIModel, ModelCapability>;
+/**
+ * Task-specific model selection rules by agent type and complexity
+ */
+export declare const MODEL_RULES: Record<string, Record<TaskComplexity, AIModel>>;
+/**
+ * Fallback chain for each model (ordered by capability)
+ */
+export declare const FALLBACK_CHAINS: Record<AIModel, AIModel[]>;
+/**
+ * Complexity detection keywords
+ */
+export declare const COMPLEXITY_KEYWORDS: {
+    simple: string[];
+    moderate: string[];
+    complex: string[];
+    critical: string[];
+};
+/**
+ * Default router configuration
+ */
+export declare const DEFAULT_ROUTER_CONFIG: {
+    enabled: boolean;
+    defaultModel: AIModel;
+    enableCostTracking: boolean;
+    enableFallback: boolean;
+    maxRetries: number;
+    costThreshold: number;
+};
+//# sourceMappingURL=ModelRules.d.ts.map

package/dist/core/routing/ModelRules.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelRules.d.ts","sourceRoot":"","sources":["../../../src/core/routing/ModelRules.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AAEnE;;GAEG;AACH,eAAO,MAAM,kBAAkB,EAAE,MAAM,CAAC,OAAO,EAAE,eAAe,CAiC/D,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,cAAc,EAAE,OAAO,CAAC,CA2CvE,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,eAAe,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,EAAE,CAqBtD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,mBAAmB;;;;;CAqC/B,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,qBAAqB;;;;;;;CAOjC,CAAC"}