universal-agent-protocol 0.5.0

package/dist/models/planner.js

@@ -0,0 +1,344 @@
+/**
+ * Task Planner
+ *
+ * Tier 1 of the Multi-Model Architecture.
+ * Responsible for:
+ * - Task decomposition into subtasks
+ * - Dependency analysis
+ * - Model assignment for each subtask
+ * - Quality assurance routing
+ */
+import { randomUUID } from 'crypto';
+// UUID generator
+const uuidv4 = () => randomUUID();
+const DEFAULT_OPTIONS = {
+    maxSubtasks: 10,
+    maxDepth: 3,
+    enableParallelization: true,
+    estimateTokenBudget: true,
+};
+/**
+ * Task Planner - decomposes tasks and creates execution plans
+ */
+export class TaskPlanner {
+    router;
+    config;
+    options;
+    constructor(router, config, options = {}) {
+        this.router = router;
+        this.config = config;
+        this.options = { ...DEFAULT_OPTIONS, ...options };
+    }
+    /**
+     * Create an execution plan for a task
+     */
+    createPlan(taskDescription) {
+        const planId = uuidv4();
+        // Classify the overall task
+        const classification = this.router.classifyTask(taskDescription);
+        // Decompose if needed
+        const subtasks = classification.requiresPlanning
+            ? this.decomposeTask(taskDescription, classification.complexity)
+            : [this.createSingleSubtask(taskDescription, classification)];
+        // Build dependency graph
+        const dependencies = this.analyzeDependencies(subtasks);
+        // Assign models to subtasks
+        const modelAssignments = this.assignModels(subtasks);
+        // Estimate cost and duration
+        const estimatedCost = this.estimateTotalCost(subtasks, modelAssignments);
+        const estimatedDuration = this.estimateDuration(subtasks, dependencies);
+        return {
+            id: planId,
+            originalTask: taskDescription,
+            subtasks,
+            dependencies,
+            modelAssignments,
+            estimatedCost,
+            estimatedDuration,
+            created: new Date(),
+        };
+    }
+    /**
+     * Decompose a complex task into subtasks
+     */
+    decomposeTask(taskDescription, complexity) {
+        const subtasks = [];
+        const lowerTask = taskDescription.toLowerCase();
+        // Analysis/understanding phase
+        if (complexity === 'high' || complexity === 'critical') {
+            subtasks.push({
+                id: uuidv4(),
+                title: 'Analyze requirements',
+                description: `Understand the requirements and constraints for: ${taskDescription}`,
+                type: 'planning',
+                complexity: 'medium',
+                inputs: ['task_description'],
+                outputs: ['requirements_analysis', 'constraints_list'],
+                constraints: ['Must identify all edge cases', 'Must note dependencies'],
+            });
+        }
+        // Design phase for architectural tasks
+        if (lowerTask.includes('architecture') || lowerTask.includes('design') || lowerTask.includes('refactor')) {
+            subtasks.push({
+                id: uuidv4(),
+                title: 'Design solution',
+                description: 'Create detailed design for the solution including interfaces, patterns, and structure',
+                type: 'planning',
+                complexity: 'high',
+                inputs: ['requirements_analysis'],
+                outputs: ['design_document', 'interface_definitions'],
+                constraints: ['Must follow SOLID principles', 'Must be testable'],
+            });
+        }
+        // Implementation phase
+        subtasks.push({
+            id: uuidv4(),
+            title: 'Implement solution',
+            description: 'Write the code implementing the required functionality',
+            type: 'coding',
+            complexity: complexity === 'critical' ? 'high' : complexity,
+            inputs: subtasks.length > 0 ? ['design_document'] : ['task_description'],
+            outputs: ['source_code'],
+            constraints: ['Must have proper error handling', 'Must be well-typed'],
+        });
+        // Testing phase
+        if (complexity !== 'low') {
+            subtasks.push({
+                id: uuidv4(),
+                title: 'Write tests',
+                description: 'Create unit tests and integration tests for the implementation',
+                type: 'coding',
+                complexity: 'medium',
+                inputs: ['source_code'],
+                outputs: ['test_code'],
+                constraints: ['Must achieve >80% coverage', 'Must test edge cases'],
+            });
+        }
+        // Review phase for critical tasks
+        if (complexity === 'critical' || lowerTask.includes('security')) {
+            subtasks.push({
+                id: uuidv4(),
+                title: 'Security review',
+                description: 'Review code for security vulnerabilities and best practices',
+                type: 'review',
+                complexity: 'high',
+                inputs: ['source_code'],
+                outputs: ['security_report', 'recommendations'],
+                constraints: ['Must check OWASP top 10', 'Must verify input validation'],
+            });
+        }
+        // Documentation for significant changes
+        if (complexity === 'high' || complexity === 'critical') {
+            subtasks.push({
+                id: uuidv4(),
+                title: 'Update documentation',
+                description: 'Update documentation to reflect the changes',
+                type: 'documentation',
+                complexity: 'low',
+                inputs: ['source_code', 'design_document'],
+                outputs: ['documentation'],
+                constraints: ['Must include usage examples', 'Must document breaking changes'],
+            });
+        }
+        return subtasks;
+    }
+    /**
+     * Create a single subtask for simple tasks
+     */
+    createSingleSubtask(taskDescription, classification) {
+        return {
+            id: uuidv4(),
+            title: 'Execute task',
+            description: taskDescription,
+            type: classification.taskType,
+            complexity: classification.complexity,
+            inputs: ['task_description'],
+            outputs: ['result'],
+            constraints: [],
+        };
+    }
+    /**
+     * Analyze dependencies between subtasks
+     */
+    analyzeDependencies(subtasks) {
+        const dependencies = new Map();
+        // Build output-to-subtask mapping
+        const outputProducers = new Map();
+        for (const subtask of subtasks) {
+            for (const output of subtask.outputs) {
+                outputProducers.set(output, subtask.id);
+            }
+        }
+        // Find dependencies based on inputs
+        for (const subtask of subtasks) {
+            const deps = [];
+            for (const input of subtask.inputs) {
+                const producer = outputProducers.get(input);
+                if (producer && producer !== subtask.id) {
+                    deps.push(producer);
+                }
+            }
+            if (deps.length > 0) {
+                dependencies.set(subtask.id, [...new Set(deps)]);
+            }
+        }
+        return dependencies;
+    }
+    /**
+     * Assign models to subtasks based on routing
+     */
+    assignModels(subtasks) {
+        const assignments = new Map();
+        for (const subtask of subtasks) {
+            const selection = this.router.selectModel(subtask.complexity, subtask.type, [] // Could extract keywords from description
+            );
+            assignments.set(subtask.id, selection.model.id);
+        }
+        return assignments;
+    }
+    /**
+     * Estimate total cost for the plan
+     */
+    estimateTotalCost(subtasks, assignments) {
+        let totalCost = 0;
+        for (const subtask of subtasks) {
+            const modelId = assignments.get(subtask.id);
+            const model = modelId ? this.router.getModel(modelId) : undefined;
+            if (model) {
+                // Estimate tokens based on complexity
+                const tokenMultiplier = {
+                    low: 1,
+                    medium: 2,
+                    high: 4,
+                    critical: 6,
+                };
+                const baseTokens = 5000;
+                const estimatedTokens = baseTokens * tokenMultiplier[subtask.complexity];
+                totalCost += this.router.estimateCost(model, estimatedTokens, estimatedTokens / 2);
+            }
+        }
+        return totalCost;
+    }
+    /**
+     * Estimate duration considering parallelization
+     */
+    estimateDuration(subtasks, dependencies) {
+        if (!this.options.enableParallelization) {
+            // Sequential execution
+            return subtasks.reduce((total, subtask) => {
+                const baseTime = 30000; // 30 seconds base
+                const multiplier = {
+                    low: 1,
+                    medium: 2,
+                    high: 4,
+                    critical: 6,
+                };
+                return total + baseTime * multiplier[subtask.complexity];
+            }, 0);
+        }
+        // Parallel execution - find critical path
+        const subtaskDurations = new Map();
+        for (const subtask of subtasks) {
+            const baseTime = 30000;
+            const multiplier = {
+                low: 1,
+                medium: 2,
+                high: 4,
+                critical: 6,
+            };
+            subtaskDurations.set(subtask.id, baseTime * multiplier[subtask.complexity]);
+        }
+        // Calculate earliest completion time for each subtask
+        const completionTimes = new Map();
+        const queue = subtasks.filter(s => !dependencies.has(s.id)).map(s => s.id);
+        while (queue.length > 0) {
+            const subtaskId = queue.shift();
+            const deps = dependencies.get(subtaskId) || [];
+            const maxDepTime = deps.length > 0
+                ? Math.max(...deps.map(d => completionTimes.get(d) || 0))
+                : 0;
+            completionTimes.set(subtaskId, maxDepTime + (subtaskDurations.get(subtaskId) || 0));
+            // Add dependent subtasks to queue
+            for (const subtask of subtasks) {
+                const subtaskDeps = dependencies.get(subtask.id) || [];
+                if (subtaskDeps.includes(subtaskId) && !completionTimes.has(subtask.id)) {
+                    const allDepsComplete = subtaskDeps.every(d => completionTimes.has(d));
+                    if (allDepsComplete) {
+                        queue.push(subtask.id);
+                    }
+                }
+            }
+        }
+        return Math.max(...Array.from(completionTimes.values()), 0);
+    }
+    /**
+     * Get execution order respecting dependencies
+     */
+    getExecutionOrder(plan) {
+        const levels = [];
+        const completed = new Set();
+        const remaining = new Set(plan.subtasks.map(s => s.id));
+        while (remaining.size > 0) {
+            const currentLevel = [];
+            for (const subtaskId of remaining) {
+                const deps = plan.dependencies.get(subtaskId) || [];
+                const allDepsComplete = deps.every(d => completed.has(d));
+                if (allDepsComplete) {
+                    currentLevel.push(subtaskId);
+                }
+            }
+            if (currentLevel.length === 0 && remaining.size > 0) {
+                // Circular dependency or bug - just add remaining
+                currentLevel.push(...remaining);
+                remaining.clear();
+            }
+            for (const id of currentLevel) {
+                remaining.delete(id);
+                completed.add(id);
+            }
+            if (currentLevel.length > 0) {
+                levels.push(currentLevel);
+            }
+        }
+        return levels;
+    }
+    /**
+     * Visualize the plan as text
+     */
+    visualizePlan(plan) {
+        const lines = [
+            `=== Execution Plan: ${plan.id.slice(0, 8)} ===`,
+            `Task: ${plan.originalTask.slice(0, 80)}${plan.originalTask.length > 80 ? '...' : ''}`,
+            `Estimated Cost: $${plan.estimatedCost.toFixed(4)}`,
+            `Estimated Duration: ${Math.round(plan.estimatedDuration / 1000)}s`,
+            '',
+            'Subtasks:',
+        ];
+        const executionOrder = this.getExecutionOrder(plan);
+        let level = 1;
+        for (const levelTasks of executionOrder) {
+            lines.push(`  Level ${level} (parallel):`);
+            for (const taskId of levelTasks) {
+                const subtask = plan.subtasks.find(s => s.id === taskId);
+                const modelId = plan.modelAssignments.get(taskId);
+                if (subtask) {
+                    lines.push(`    - [${subtask.complexity.toUpperCase()}] ${subtask.title}`);
+                    lines.push(`      Type: ${subtask.type}, Model: ${modelId || 'default'}`);
+                    const deps = plan.dependencies.get(taskId);
+                    if (deps && deps.length > 0) {
+                        lines.push(`      Depends on: ${deps.map(d => plan.subtasks.find(s => s.id === d)?.title || d).join(', ')}`);
+                    }
+                }
+            }
+            level++;
+        }
+        return lines.join('\n');
+    }
+}
+/**
+ * Create a planner instance
+ */
+export function createPlanner(router, config, options) {
+    return new TaskPlanner(router, config, options);
+}
+//# sourceMappingURL=planner.js.map

package/dist/models/planner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"planner.js","sourceRoot":"","sources":["../../src/models/planner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AASpC,iBAAiB;AACjB,MAAM,MAAM,GAAG,GAAW,EAAE,CAAC,UAAU,EAAE,CAAC;AAY1C,MAAM,eAAe,GAAmB;IACtC,WAAW,EAAE,EAAE;IACf,QAAQ,EAAE,CAAC;IACX,qBAAqB,EAAE,IAAI;IAC3B,mBAAmB,EAAE,IAAI;CAC1B,CAAC;AAEF;;GAEG;AACH,MAAM,OAAO,WAAW;IACd,MAAM,CAAc;IACnB,MAAM,CAAmB;IAC1B,OAAO,CAAiB;IAEhC,YACE,MAAmB,EACnB,MAAwB,EACxB,UAA0B,EAAE;QAE5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,EAAE,GAAG,eAAe,EAAE,GAAG,OAAO,EAAE,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,eAAuB;QAChC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC;QAExB,4BAA4B;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QAEjE,sBAAsB;QACtB,MAAM,QAAQ,GAAG,cAAc,CAAC,gBAAgB;YAC9C,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,EAAE,cAAc,CAAC,UAAU,CAAC;YAChE,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,eAAe,EAAE,cAAc,CAAC,CAAC,CAAC;QAEhE,yBAAyB;QACzB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAExD,4BAA4B;QAC5B,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QAErD,6BAA6B;QAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACzE,MAAM,iBAAiB,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAExE,OAAO;YACL,EAAE,EAAE,MAAM;YACV,YAAY,EAAE,eAAe;YAC7B,QAAQ;YACR,YAAY;YACZ,gBAAgB;YAChB,aAAa;YACb,iBAAiB;YACjB,OAAO,EAAE,IAAI,IAAI,EAAE;SACpB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,eAAuB,EAAE,UAA0B;QACvE,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAG,eAAe,CAAC,WAAW,EAAE,CAAC;QAEhD,+BAA+B;QAC/B,IAAI,UAAU,KAAK,MAAM,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,EAAE;gBACZ,KAAK,EAAE,sBAAsB;gBAC7B,WAAW,EAAE,oDAAoD,eAAe,EAAE;gBAClF,IAAI,EAAE,UAAU;gBAChB,UAAU,EAAE,QAAQ;gBACpB,MAAM,EAAE,CAAC,kBAAkB,CAAC;gBAC5B,OAAO,EAAE,CAAC,uBAAuB,EAAE,kBAAkB,CAAC;gBACtD,WAAW,EAAE,CAAC,8BAA8B,EAAE,wBAAwB,CAAC;aACxE,CAAC,CAAC;QACL,CAAC;QAED,uCAAuC;QACvC,IAAI,SAAS,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YACzG,QAAQ,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,EAAE;gBACZ,KAAK,EAAE,iBAAiB;gBACxB,WAAW,EAAE,uFAAuF;gBACpG,IAAI,EAAE,UAAU;gBAChB,UAAU,EAAE,MAAM;gBAClB,MAAM,EAAE,CAAC,uBAAuB,CAAC;gBACjC,OAAO,EAAE,CAAC,iBAAiB,EAAE,uBAAuB,CAAC;gBACrD,WAAW,EAAE,CAAC,8BAA8B,EAAE,kBAAkB,CAAC;aAClE,CAAC,CAAC;QACL,CAAC;QAED,uBAAuB;QACvB,QAAQ,CAAC,IAAI,CAAC;YACZ,EAAE,EAAE,MAAM,EAAE;YACZ,KAAK,EAAE,oBAAoB;YAC3B,WAAW,EAAE,wDAAwD;YACrE,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,UAAU,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU;YAC3D,MAAM,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC;YACxE,OAAO,EAAE,CAAC,aAAa,CAAC;YACxB,WAAW,EAAE,CAAC,iCAAiC,EAAE,oBAAoB,CAAC;SACvE,CAAC,CAAC;QAEH,gBAAgB;QAChB,IAAI,UAAU,KAAK,KAAK,EAAE,CAAC;YACzB,QAAQ,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,EAAE;gBACZ,KAAK,EAAE,aAAa;gBACpB,WAAW,EAAE,gEAAgE;gBAC7E,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,QAAQ;gBACpB,MAAM,EAAE,CAAC,aAAa,CAAC;gBACvB,OAAO,EAAE,CAAC,WAAW,CAAC;gBACtB,WAAW,EAAE,CAAC,4BAA4B,EAAE,sBAAsB,CAAC;aACpE,CAAC,CAAC;QACL,CAAC;QAED,kCAAkC;QAClC,IAAI,UAAU,KAAK,UAAU,IAAI,SAAS,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAChE,QAAQ,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,EAAE;gBACZ,KAAK,EAAE,iBAAiB;gBACxB,WAAW,EAAE,6DAA6D;gBAC1E,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,MAAM;gBAClB,MAAM,EAAE,CAAC,aAAa,CAAC;gBACvB,OAAO,EAAE,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;gBAC/C,WAAW,EAAE,CAAC,yBAAyB,EAAE,8BAA8B,CAAC;aACzE,CAAC,CAAC;QACL,CAAC;QAED,wCAAwC;QACxC,IAAI,UAAU,KAAK,MAAM,IAAI,UAAU,KAAK,UAAU,EAAE,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC;gBACZ,EAAE,EAAE,MAAM,EAAE;gBACZ,KAAK,EAAE,sBAAsB;gBAC7B,WAAW,EAAE,6CAA6C;gBAC1D,IAAI,EAAE,eAAe;gBACrB,UAAU,EAAE,KAAK;gBACjB,MAAM,EAAE,CAAC,aAAa,EAAE,iBAAiB,CAAC;gBAC1C,OAAO,EAAE,CAAC,eAAe,CAAC;gBAC1B,WAAW,EAAE,CAAC,6BAA6B,EAAE,gCAAgC,CAAC;aAC/E,CAAC,CAAC;QACL,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACK,mBAAmB,CACzB,eAAuB,EACvB,cAAgE;QAEhE,OAAO;YACL,EAAE,EAAE,MAAM,EAAE;YACZ,KAAK,EAAE,cAAc;YACrB,WAAW,EAAE,eAAe;YAC5B,IAAI,EAAE,cAAc,CAAC,QAA2B;YAChD,UAAU,EAAE,cAAc,CAAC,UAAU;YACrC,MAAM,EAAE,CAAC,kBAAkB,CAAC;YAC5B,OAAO,EAAE,CAAC,QAAQ,CAAC;YACnB,WAAW,EAAE,EAAE;SAChB,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,mBAAmB,CAAC,QAAmB;QAC7C,MAAM,YAAY,GAAG,IAAI,GAAG,EAAoB,CAAC;QAEjD,kCAAkC;QAClC,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;QAClD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,KAAK,MAAM,MAAM,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBACrC,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;YAC1C,CAAC;QACH,CAAC;QAED,oCAAoC;QACpC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,IAAI,GAAa,EAAE,CAAC;YAC1B,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBACnC,MAAM,QAAQ,GAAG,eAAe,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAC5C,IAAI,QAAQ,IAAI,QAAQ,KAAK,OAAO,CAAC,EAAE,EAAE,CAAC;oBACxC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YACD,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACpB,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACnD,CAAC;QACH,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,QAAmB;QACtC,MAAM,WAAW,GAAG,IAAI,GAAG,EAAkB,CAAC;QAE9C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,WAAW,CACvC,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,IAAI,EACZ,EAAE,CAAC,0CAA0C;aAC9C,CAAC;YACF,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED;;OAEG;IACK,iBAAiB,CACvB,QAAmB,EACnB,WAAgC;QAEhC,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAC5C,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAElE,IAAI,KAAK,EAAE,CAAC;gBACV,sCAAsC;gBACtC,MAAM,eAAe,GAAmC;oBACtD,GAAG,EAAE,CAAC;oBACN,MAAM,EAAE,CAAC;oBACT,IAAI,EAAE,CAAC;oBACP,QAAQ,EAAE,CAAC;iBACZ,CAAC;gBACF,MAAM,UAAU,GAAG,IAAI,CAAC;gBACxB,MAAM,eAAe,GAAG,UAAU,GAAG,eAAe,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBAEzE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,EAAE,eAAe,EAAE,eAAe,GAAG,CAAC,CAAC,CAAC;YACrF,CAAC;QACH,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAED;;OAEG;IACK,gBAAgB,CACtB,QAAmB,EACnB,YAAmC;QAEnC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,qBAAqB,EAAE,CAAC;YACxC,uBAAuB;YACvB,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;gBACxC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,kBAAkB;gBAC1C,MAAM,UAAU,GAAmC;oBACjD,GAAG,EAAE,CAAC;oBACN,MAAM,EAAE,CAAC;oBACT,IAAI,EAAE,CAAC;oBACP,QAAQ,EAAE,CAAC;iBACZ,CAAC;gBACF,OAAO,KAAK,GAAG,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC3D,CAAC,EAAE,CAAC,CAAC,CAAC;QACR,CAAC;QAED,0CAA0C;QAC1C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAkB,CAAC;QACnD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,MAAM,QAAQ,GAAG,KAAK,CAAC;YACvB,MAAM,UAAU,GAAmC;gBACjD,GAAG,EAAE,CAAC;gBACN,MAAM,EAAE,CAAC;gBACT,IAAI,EAAE,CAAC;gBACP,QAAQ,EAAE,CAAC;aACZ,CAAC;YACF,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;QAC9E,CAAC;QAED,sDAAsD;QACtD,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;QAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAE3E,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;YACjC,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;YAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,GAAG,CAAC;gBAChC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;gBACzD,CAAC,CAAC,CAAC,CAAC;YACN,eAAe,CAAC,GAAG,CAAC,SAAS,EAAE,UAAU,GAAG,CAAC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEpF,kCAAkC;YAClC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;gBACvD,IAAI,WAAW,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;oBACxE,MAAM,eAAe,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;oBACvE,IAAI,eAAe,EAAE,CAAC;wBACpB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBACzB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9D,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,IAAmB;QACnC,MAAM,MAAM,GAAe,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;QACpC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAExD,OAAO,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,YAAY,GAAa,EAAE,CAAC;YAElC,KAAK,MAAM,SAAS,IAAI,SAAS,EAAE,CAAC;gBAClC,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;gBACpD,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE1D,IAAI,eAAe,EAAE,CAAC;oBACpB,YAAY,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC/B,CAAC;YACH,CAAC;YAED,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC,IAAI,SAAS,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBACpD,kDAAkD;gBAClD,YAAY,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;gBAChC,SAAS,CAAC,KAAK,EAAE,CAAC;YACpB,CAAC;YAED,KAAK,MAAM,EAAE,IAAI,YAAY,EAAE,CAAC;gBAC9B,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACrB,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACpB,CAAC;YAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC5B,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,IAAmB;QAC/B,MAAM,KAAK,GAAa;YACtB,uBAAuB,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,MAAM;YAChD,SAAS,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE;YACtF,oBAAoB,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACnD,uBAAuB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,GAAG;YACnE,EAAE;YACF,WAAW;SACZ,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,UAAU,IAAI,cAAc,EAAE,CAAC;YACxC,KAAK,CAAC,IAAI,CAAC,WAAW,KAAK,cAAc,CAAC,CAAC;YAC3C,KAAK,MAAM,MAAM,IAAI,UAAU,EAAE,CAAC;gBAChC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,MAAM,CAAC,CAAC;gBACzD,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;gBAClD,IAAI,OAAO,EAAE,CAAC;oBACZ,KAAK,CAAC,IAAI,CAAC,UAAU,OAAO,CAAC,UAAU,CAAC,WAAW,EAAE,KAAK,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;oBAC3E,KAAK,CAAC,IAAI,CAAC,eAAe,OAAO,CAAC,IAAI,YAAY,OAAO,IAAI,SAAS,EAAE,CAAC,CAAC;oBAC1E,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;oBAC3C,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAC5B,KAAK,CAAC,IAAI,CAAC,qBAAqB,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBAC/G,CAAC;gBACH,CAAC;YACH,CAAC;YACD,KAAK,EAAE,CAAC;QACV,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,aAAa,CAC3B,MAAmB,EACnB,MAAwB,EACxB,OAAwB;IAExB,OAAO,IAAI,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AAClD,CAAC"}

package/dist/models/router.d.ts

@@ -0,0 +1,75 @@
+/**
+ * Model Router
+ *
+ * Routes tasks to appropriate models based on:
+ * - Task complexity and type
+ * - Routing rules configuration
+ * - Cost optimization strategy
+ * - Available model capabilities
+ */
+import { ModelConfig, MultiModelConfig, TaskClassificationResult, TaskComplexity, ModelSelection, RoutingRule, ModelRole } from './types.js';
+/**
+ * Model Router - routes tasks to appropriate models
+ */
+export declare class ModelRouter {
+    private config;
+    private models;
+    private routingRules;
+    private roleAssignments;
+    constructor(config: MultiModelConfig);
+    private initializeModels;
+    private initializeRoles;
+    /**
+     * Classify a task to determine complexity and type
+     */
+    classifyTask(taskDescription: string): TaskClassificationResult;
+    /**
+     * Select the appropriate model for a task
+     */
+    selectModel(complexity: TaskComplexity, taskType: string, keywords: string[]): ModelSelection;
+    private buildReasoning;
+    /**
+     * Estimate cost for a model invocation
+     */
+    estimateCost(model: ModelConfig, inputTokens: number, outputTokens: number): number;
+    /**
+     * Get model by ID
+     */
+    getModel(modelId: string): ModelConfig | undefined;
+    /**
+     * Get model for a specific role
+     */
+    getModelForRole(role: ModelRole): ModelConfig | undefined;
+    /**
+     * Get all configured models
+     */
+    getAllModels(): ModelConfig[];
+    /**
+     * Get routing analysis for debugging
+     */
+    analyzeRouting(taskDescription: string): {
+        classification: TaskClassificationResult;
+        matchedRules: Array<{
+            rule: RoutingRule;
+            matched: boolean;
+            reason: string;
+        }>;
+        costComparison: Array<{
+            model: string;
+            cost: number;
+        }>;
+    };
+}
+/**
+ * Create a router instance from config
+ */
+export declare function createRouter(config: MultiModelConfig): ModelRouter;
+/**
+ * Default router with cost-optimized settings
+ */
+export declare function createCostOptimizedRouter(): ModelRouter;
+/**
+ * Default router with performance-first settings
+ */
+export declare function createPerformanceRouter(): ModelRouter;
+//# sourceMappingURL=router.d.ts.map

package/dist/models/router.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"router.d.ts","sourceRoot":"","sources":["../../src/models/router.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EACL,WAAW,EAEX,gBAAgB,EAChB,wBAAwB,EACxB,cAAc,EACd,cAAc,EACd,WAAW,EAEX,SAAS,EACV,MAAM,YAAY,CAAC;AAgCpB;;GAEG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,MAAM,CAAmB;IACjC,OAAO,CAAC,MAAM,CAA2B;IACzC,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,eAAe,CAAyB;gBAEpC,MAAM,EAAE,gBAAgB;IAgBpC,OAAO,CAAC,gBAAgB;IAuBxB,OAAO,CAAC,eAAe;IAavB;;OAEG;IACH,YAAY,CAAC,eAAe,EAAE,MAAM,GAAG,wBAAwB;IAmE/D;;OAEG;IACH,WAAW,CACT,UAAU,EAAE,cAAc,EAC1B,QAAQ,EAAE,MAAM,EAChB,QAAQ,EAAE,MAAM,EAAE,GACjB,cAAc;IA8DjB,OAAO,CAAC,cAAc;IA0BtB;;OAEG;IACH,YAAY,CAAC,KAAK,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,GAAG,MAAM;IAMnF;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,SAAS;IAIlD;;OAEG;IACH,eAAe,CAAC,IAAI,EAAE,SAAS,GAAG,WAAW,GAAG,SAAS;IAKzD;;OAEG;IACH,YAAY,IAAI,WAAW,EAAE;IAI7B;;OAEG;IACH,cAAc,CAAC,eAAe,EAAE,MAAM,GAAG;QACvC,cAAc,EAAE,wBAAwB,CAAC;QACzC,YAAY,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,WAAW,CAAC;YAAC,OAAO,EAAE,OAAO,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QAC7E,cAAc,EAAE,KAAK,CAAC;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KACxD;CAuDF;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,MAAM,EAAE,gBAAgB,GAAG,WAAW,CAElE;AAED;;GAEG;AACH,wBAAgB,yBAAyB,IAAI,WAAW,CAkBvD;AAED;;GAEG;AACH,wBAAgB,uBAAuB,IAAI,WAAW,CAYrD"}