@framers/agentos 0.1.126 → 0.1.128

package/dist/orchestration/planning/MissionPlanner.js

@@ -0,0 +1,334 @@
+/**
+ * @file MissionPlanner.ts
+ * @description Tree of Thought mission planner.
+ *
+ * Three-phase planning pipeline:
+ *   Phase 1 — Divergent Exploration: generate N candidate graphs (Yao et al. 2023)
+ *   Phase 2 — Evaluation and Selection: score and pick the best (Xie et al. 2023)
+ *   Phase 3 — Refinement / Reflexion: fix gaps before execution (Shinn et al. 2023)
+ *
+ * Each phase uses a separate prompt template loaded from `prompts/`.
+ */
+import { readFileSync } from 'node:fs';
+import { resolve, dirname } from 'node:path';
+import { fileURLToPath } from 'node:url';
+import { START, END } from '../ir/types.js';
+import { ProviderAssignmentEngine } from './ProviderAssignmentEngine.js';
+const __dirname = dirname(fileURLToPath(import.meta.url));
+/** Load a prompt template from the prompts/ directory. */
+function loadPrompt(name) {
+    return readFileSync(resolve(__dirname, 'prompts', `${name}.md`), 'utf-8');
+}
+/** Replace {{PLACEHOLDER}} tokens in a template string. */
+function fillPrompt(template, vars) {
+    let result = template;
+    for (const [key, value] of Object.entries(vars)) {
+        result = result.replaceAll(`{{${key}}}`, value);
+    }
+    return result;
+}
+/** Extract first JSON object from LLM output that may be wrapped in prose. */
+function extractJson(text) {
+    const match = text.match(/\{[\s\S]*\}/);
+    if (!match)
+        throw new Error(`No JSON found in LLM response: ${text.slice(0, 200)}`);
+    return JSON.parse(match[0]);
+}
+const STRATEGIES = ['linear', 'parallel', 'hierarchical'];
+/**
+ * Tree of Thought mission planner.
+ *
+ * Generates N candidate execution graphs, evaluates them on four dimensions,
+ * selects the best (or synthesizes a hybrid), and refines it before
+ * compiling to `CompiledExecutionGraph`.
+ */
+export class MissionPlanner {
+    constructor(config) {
+        this.config = config;
+        this.planLlm = config.plannerLlmCaller ?? config.llmCaller;
+        this.decompositionPrompt = loadPrompt('decomposition');
+        this.evaluationPrompt = loadPrompt('evaluation');
+        this.refinementPrompt = loadPrompt('refinement');
+    }
+    /**
+     * Run the full planning pipeline: explore → evaluate → refine.
+     *
+     * @param goal - Natural language mission goal.
+     * @param context - Available tools and providers.
+     * @param onEvent - Optional callback for streaming planning progress events.
+     */
+    async plan(goal, context, onEvent) {
+        onEvent?.({ type: 'mission:planning_start', goal });
+        // Phase 1: Divergent exploration — generate N branches
+        const branches = await this.generateBranches(goal, context, onEvent);
+        // Phase 2: Evaluate and select
+        const { selectedBranch, allBranches } = await this.evaluateBranches(branches, context, onEvent);
+        // Phase 3: Refine
+        const { refinedBranch, refinements } = await this.refineBranch(selectedBranch, context, onEvent);
+        const providerAssignedBranch = this.assignProviders(refinedBranch, context);
+        // Compile to IR
+        const compiledGraph = this.compileToIR(providerAssignedBranch);
+        onEvent?.({
+            type: 'mission:graph_compiled',
+            nodeCount: compiledGraph.nodes.length,
+            edgeCount: compiledGraph.edges.length,
+            estimatedCost: providerAssignedBranch.estimatedCost,
+        });
+        return {
+            selectedBranch: providerAssignedBranch,
+            allBranches,
+            refinements,
+            compiledGraph,
+        };
+    }
+    // ---------------------------------------------------------------------------
+    // Phase 1: Divergent Exploration
+    // ---------------------------------------------------------------------------
+    async generateBranches(goal, context, onEvent) {
+        const toolList = context.tools.map((t) => `- ${t.name}: ${t.description}`).join('\n') ||
+            'No tools available';
+        const providerList = context.providers.join(', ') || 'No providers available';
+        const count = Math.min(this.config.branchCount, STRATEGIES.length);
+        const branchPromises = STRATEGIES.slice(0, count).map(async (strategy, i) => {
+            const prompt = fillPrompt(this.decompositionPrompt, {
+                STRATEGY: strategy,
+                TOOL_LIST: toolList,
+                PROVIDER_LIST: providerList,
+                GOAL: goal,
+            });
+            try {
+                const response = await this.planLlm('You are a mission planner for AgentOS. Respond with JSON only.', prompt);
+                const parsed = extractJson(response);
+                const branchId = `branch_${i}`;
+                const branch = {
+                    branchId,
+                    strategy: parsed.strategy || strategy,
+                    summary: parsed.summary || `${strategy} approach`,
+                    nodes: this.normalizeNodes(parsed.nodes),
+                    edges: this.normalizeEdges(parsed.edges),
+                    providerAssignments: [],
+                    estimatedCost: parsed.estimatedCost || 0,
+                    estimatedLatencyMs: parsed.estimatedLatencyMs || 0,
+                    scores: { feasibility: 0, costEfficiency: 0, latency: 0, robustness: 0, overall: 0 },
+                };
+                onEvent?.({
+                    type: 'mission:branch_generated',
+                    branchId,
+                    summary: branch.summary,
+                    scores: branch.scores,
+                });
+                return branch;
+            }
+            catch {
+                // Branch generation failed — return null, filtered below
+                return null;
+            }
+        });
+        const results = await Promise.allSettled(branchPromises);
+        const branches = [];
+        for (const result of results) {
+            if (result.status === 'fulfilled' && result.value) {
+                branches.push(result.value);
+            }
+        }
+        if (branches.length === 0) {
+            throw new Error('All branch generation attempts failed');
+        }
+        return branches;
+    }
+    // ---------------------------------------------------------------------------
+    // Phase 2: Evaluation
+    // ---------------------------------------------------------------------------
+    async evaluateBranches(branches, context, onEvent) {
+        const toolList = context.tools.map((t) => `- ${t.name}: ${t.description}`).join('\n') || 'None';
+        const candidatesJson = JSON.stringify(branches.map((b) => ({
+            branchId: b.branchId,
+            strategy: b.strategy,
+            summary: b.summary,
+            nodeCount: b.nodes.length,
+            nodes: b.nodes.map((n) => ({ id: n.id, type: n.type })),
+            edges: b.edges.map((e) => ({ source: e.source, target: e.target })),
+            estimatedCost: b.estimatedCost,
+            estimatedLatencyMs: b.estimatedLatencyMs,
+        })), null, 2);
+        const prompt = fillPrompt(this.evaluationPrompt, {
+            CANDIDATES_JSON: candidatesJson,
+            TOOL_LIST: toolList,
+            AVAILABLE_PROVIDERS: context.providers.join(', '),
+        });
+        const response = await this.planLlm('You are evaluating mission execution candidates. Respond with JSON only.', prompt);
+        const parsed = extractJson(response);
+        // Apply scores to branches
+        for (const evaluation of parsed.evaluations) {
+            const branch = branches.find((b) => b.branchId === evaluation.branchId);
+            if (branch)
+                branch.scores = evaluation.scores;
+        }
+        const selectedId = parsed.recommendation.selectedBranchId;
+        const selectedBranch = branches.find((b) => b.branchId === selectedId) ?? branches[0];
+        onEvent?.({
+            type: 'mission:branch_selected',
+            branchId: selectedBranch.branchId,
+            reason: parsed.recommendation.reason,
+        });
+        return { selectedBranch, allBranches: branches };
+    }
+    // ---------------------------------------------------------------------------
+    // Phase 3: Refinement (Reflexion)
+    // ---------------------------------------------------------------------------
+    async refineBranch(branch, context, onEvent) {
+        const toolList = context.tools.map((t) => `- ${t.name}: ${t.description}`).join('\n') || 'None';
+        const graphJson = JSON.stringify({ nodes: branch.nodes, edges: branch.edges, estimatedCost: branch.estimatedCost }, null, 2);
+        const prompt = fillPrompt(this.refinementPrompt, {
+            SELECTED_GRAPH_JSON: graphJson,
+            TOOL_LIST: toolList,
+        });
+        const response = await this.planLlm('You are reviewing a mission execution graph. Respond with JSON only.', prompt);
+        const parsed = extractJson(response);
+        const refinements = parsed.refinements.map((r) => r.description);
+        const refinedBranch = {
+            ...branch,
+            nodes: [...branch.nodes],
+            edges: [...branch.edges],
+            estimatedCost: parsed.finalEstimatedCost ?? branch.estimatedCost,
+            estimatedLatencyMs: parsed.finalEstimatedLatencyMs ?? branch.estimatedLatencyMs,
+        };
+        // Apply structural refinements
+        for (const refinement of parsed.refinements) {
+            if (refinement.type === 'add_node' && refinement.patch) {
+                const node = refinement.patch;
+                if (node.id)
+                    refinedBranch.nodes.push(this.normalizeNode(node));
+            }
+            if (refinement.type === 'remove_node' && refinement.nodeId) {
+                refinedBranch.nodes = refinedBranch.nodes.filter((n) => n.id !== refinement.nodeId);
+                refinedBranch.edges = refinedBranch.edges.filter((e) => e.source !== refinement.nodeId && e.target !== refinement.nodeId);
+            }
+            if (refinement.type === 'add_edge' && refinement.patch) {
+                const edge = refinement.patch;
+                if (edge.source && edge.target) {
+                    refinedBranch.edges.push(this.normalizeEdge(edge, refinedBranch.edges.length));
+                }
+            }
+        }
+        if (refinements.length > 0) {
+            onEvent?.({ type: 'mission:refinement_applied', changes: refinements });
+        }
+        return { refinedBranch, refinements };
+    }
+    // ---------------------------------------------------------------------------
+    // Compile to IR
+    // ---------------------------------------------------------------------------
+    assignProviders(branch, context) {
+        const gmiNodes = branch.nodes.filter((node) => node.type === 'gmi');
+        if (gmiNodes.length === 0) {
+            return { ...branch, providerAssignments: [] };
+        }
+        const engine = new ProviderAssignmentEngine(context.providers);
+        const assignments = engine.assign(gmiNodes.map((node) => ({
+            ...node,
+            complexity: node.complexity ?? this.estimateComplexity(node),
+        })), this.config.providerStrategy);
+        const availability = engine.checkAvailability(assignments);
+        if (!availability.available) {
+            throw new Error(`Mission provider assignment requires unavailable providers: ${availability.missing.join(', ')}`);
+        }
+        return {
+            ...branch,
+            providerAssignments: assignments,
+            nodes: branch.nodes.map((node) => this.applyProviderAssignment(node, assignments)),
+        };
+    }
+    compileToIR(branch) {
+        return {
+            id: `mission-${Date.now()}`,
+            name: branch.summary,
+            nodes: branch.nodes,
+            edges: branch.edges,
+            stateSchema: { input: {}, scratch: {}, artifacts: {} },
+            reducers: {},
+            checkpointPolicy: 'every_node',
+            memoryConsistency: 'live',
+        };
+    }
+    // ---------------------------------------------------------------------------
+    // Normalization helpers
+    // ---------------------------------------------------------------------------
+    normalizeNodes(nodes) {
+        return nodes.map((n) => this.normalizeNode(n));
+    }
+    normalizeNode(raw) {
+        const checkpoint = raw.checkpoint === true
+            ? 'after'
+            : raw.checkpoint === false
+                ? 'none'
+                : raw.checkpoint ?? 'after';
+        return {
+            id: String(raw.id ?? `node_${Date.now()}_${Math.random().toString(36).slice(2, 6)}`),
+            type: String(raw.type ?? 'gmi'),
+            executorConfig: raw.executorConfig ?? {
+                type: 'gmi',
+                instructions: String(raw.instructions ?? raw.role ?? 'Execute task'),
+            },
+            executionMode: raw.executionMode ?? 'single_turn',
+            effectClass: raw.effectClass ?? 'read',
+            checkpoint,
+            complexity: typeof raw.complexity === 'number' && Number.isFinite(raw.complexity)
+                ? Math.max(0, Math.min(1, raw.complexity))
+                : undefined,
+        };
+    }
+    normalizeEdges(edges) {
+        return edges.map((e, i) => this.normalizeEdge(e, i));
+    }
+    normalizeEdge(raw, index) {
+        return {
+            id: String(raw.id ?? `edge_${index}`),
+            source: String(raw.source ?? START),
+            target: String(raw.target ?? END),
+            type: String(raw.type ?? 'static'),
+        };
+    }
+    applyProviderAssignment(node, assignments) {
+        const assignment = assignments.find((item) => item.nodeId === node.id);
+        if (!assignment)
+            return node;
+        return {
+            ...node,
+            complexity: assignment.complexity,
+            llm: {
+                providerId: assignment.provider,
+                model: assignment.model,
+                reason: assignment.reason,
+            },
+        };
+    }
+    estimateComplexity(node) {
+        if (typeof node.complexity === 'number' && Number.isFinite(node.complexity)) {
+            return Math.max(0, Math.min(1, node.complexity));
+        }
+        if (node.type !== 'gmi')
+            return 0.1;
+        if (node.executionMode === 'planner_controlled')
+            return 0.85;
+        if (node.executionMode === 'react_bounded')
+            return 0.75;
+        const instructions = node.executorConfig.type === 'gmi'
+            ? node.executorConfig.instructions.toLowerCase()
+            : node.id.toLowerCase();
+        if (/\b(research|analy[sz]e|compare|evaluate|reason|judge|plan)\b/.test(instructions)) {
+            return 0.75;
+        }
+        if (/\b(summary|summari[sz]e|draft|write|deliver|merge|final)\b/.test(instructions)) {
+            return 0.35;
+        }
+        if (instructions.length > 180) {
+            return 0.65;
+        }
+        if (instructions.length > 80) {
+            return 0.5;
+        }
+        return 0.4;
+    }
+}
+//# sourceMappingURL=MissionPlanner.js.map

package/dist/orchestration/planning/MissionPlanner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MissionPlanner.js","sourceRoot":"","sources":["../../../src/orchestration/planning/MissionPlanner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AASzC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AAEzE,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAE1D,0DAA0D;AAC1D,SAAS,UAAU,CAAC,IAAY;IAC9B,OAAO,YAAY,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,GAAG,IAAI,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;AAC5E,CAAC;AAED,2DAA2D;AAC3D,SAAS,UAAU,CAAC,QAAgB,EAAE,IAA4B;IAChE,IAAI,MAAM,GAAG,QAAQ,CAAC;IACtB,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QAChD,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC,KAAK,GAAG,IAAI,EAAE,KAAK,CAAC,CAAC;IAClD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,8EAA8E;AAC9E,SAAS,WAAW,CAAI,IAAY;IAClC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACxC,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IACpF,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAM,CAAC;AACnC,CAAC;AAED,MAAM,UAAU,GAAG,CAAC,QAAQ,EAAE,UAAU,EAAE,cAAc,CAAU,CAAC;AAQnE;;;;;;GAMG;AACH,MAAM,OAAO,cAAc;IAQzB,YAAY,MAAqB;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,gBAAgB,IAAI,MAAM,CAAC,SAAS,CAAC;QAC3D,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC,eAAe,CAAC,CAAC;QACvD,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;QACjD,IAAI,CAAC,gBAAgB,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;IACnD,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,IAAI,CACR,IAAY,EACZ,OAAoB,EACpB,OAAiE;QAEjE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,wBAAwB,EAAE,IAAI,EAAE,CAAC,CAAC;QAEpD,uDAAuD;QACvD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAErE,+BAA+B;QAC/B,MAAM,EAAE,cAAc,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,gBAAgB,CACjE,QAAQ,EACR,OAAO,EACP,OAAO,CACR,CAAC;QAEF,kBAAkB;QAClB,MAAM,EAAE,aAAa,EAAE,WAAW,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAC5D,cAAc,EACd,OAAO,EACP,OAAO,CACR,CAAC;QAEF,MAAM,sBAAsB,GAAG,IAAI,CAAC,eAAe,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QAE5E,gBAAgB;QAChB,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,sBAAsB,CAAC,CAAC;QAE/D,OAAO,EAAE,CAAC;YACR,IAAI,EAAE,wBAAwB;YAC9B,SAAS,EAAE,aAAa,CAAC,KAAK,CAAC,MAAM;YACrC,SAAS,EAAE,aAAa,CAAC,KAAK,CAAC,MAAM;YACrC,aAAa,EAAE,sBAAsB,CAAC,aAAa;SACpD,CAAC,CAAC;QAEH,OAAO;YACL,cAAc,EAAE,sBAAsB;YACtC,WAAW;YACX,WAAW;YACX,aAAa;SACd,CAAC;IACJ,CAAC;IAED,8EAA8E;IAC9E,iCAAiC;IACjC,8EAA8E;IAEtE,KAAK,CAAC,gBAAgB,CAC5B,IAAY,EACZ,OAAoB,EACpB,OAAiE;QAEjE,MAAM,QAAQ,GACZ,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;YACpE,oBAAoB,CAAC;QACvB,MAAM,YAAY,GAAG,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,wBAAwB,CAAC;QAE9E,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,MAAM,CAAC,CAAC;QAEnE,MAAM,cAAc,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE;YAC1E,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,mBAAmB,EAAE;gBAClD,QAAQ,EAAE,QAAQ;gBAClB,SAAS,EAAE,QAAQ;gBACnB,aAAa,EAAE,YAAY;gBAC3B,IAAI,EAAE,IAAI;aACX,CAAC,CAAC;YAEH,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,gEAAgE,EAChE,MAAM,CACP,CAAC;gBAEF,MAAM,MAAM,GAAG,WAAW,CAOvB,QAAQ,CAAC,CAAC;gBAEb,MAAM,QAAQ,GAAG,UAAU,CAAC,EAAE,CAAC;gBAC/B,MAAM,MAAM,GAAoB;oBAC9B,QAAQ;oBACR,QAAQ,EAAE,MAAM,CAAC,QAAQ,IAAI,QAAQ;oBACrC,OAAO,EAAE,MAAM,CAAC,OAAO,IAAI,GAAG,QAAQ,WAAW;oBACjD,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC;oBACxC,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,KAAK,CAAC;oBACxC,mBAAmB,EAAE,EAAE;oBACvB,aAAa,EAAE,MAAM,CAAC,aAAa,IAAI,CAAC;oBACxC,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,CAAC;oBAClD,MAAM,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE;iBACrF,CAAC;gBAEF,OAAO,EAAE,CAAC;oBACR,IAAI,EAAE,0BAA0B;oBAChC,QAAQ;oBACR,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;iBACtB,CAAC,CAAC;gBAEH,OAAO,MAAM,CAAC;YAChB,CAAC;YAAC,MAAM,CAAC;gBACP,yDAAyD;gBACzD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAsB,EAAE,CAAC;QACvC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAClD,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,8EAA8E;IAC9E,sBAAsB;IACtB,8EAA8E;IAEtE,KAAK,CAAC,gBAAgB,CAC5B,QAA2B,EAC3B,OAAoB,EACpB,OAAiE;QAEjE,MAAM,QAAQ,GACZ,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC;QACjF,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CACnC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YACnB,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,QAAQ,EAAE,CAAC,CAAC,QAAQ;YACpB,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,MAAM;YACzB,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;YACvD,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC;YACnE,aAAa,EAAE,CAAC,CAAC,aAAa;YAC9B,kBAAkB,EAAE,CAAC,CAAC,kBAAkB;SACzC,CAAC,CAAC,EACH,IAAI,EACJ,CAAC,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAC/C,eAAe,EAAE,cAAc;YAC/B,SAAS,EAAE,QAAQ;YACnB,mBAAmB,EAAE,OAAO,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;SAClD,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,0EAA0E,EAC1E,MAAM,CACP,CAAC;QAEF,MAAM,MAAM,GAAG,WAAW,CAGvB,QAAQ,CAAC,CAAC;QAEb,2BAA2B;QAC3B,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5C,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,UAAU,CAAC,QAAQ,CAAC,CAAC;YACxE,IAAI,MAAM;gBAAE,MAAM,CAAC,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;QAChD,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,cAAc,CAAC,gBAAgB,CAAC;QAC1D,MAAM,cAAc,GAClB,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,UAAU,CAAC,IAAI,QAAQ,CAAC,CAAC,CAAE,CAAC;QAElE,OAAO,EAAE,CAAC;YACR,IAAI,EAAE,yBAAyB;YAC/B,QAAQ,EAAE,cAAc,CAAC,QAAQ;YACjC,MAAM,EAAE,MAAM,CAAC,cAAc,CAAC,MAAM;SACrC,CAAC,CAAC;QAEH,OAAO,EAAE,cAAc,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC;IACnD,CAAC;IAED,8EAA8E;IAC9E,kCAAkC;IAClC,8EAA8E;IAEtE,KAAK,CAAC,YAAY,CACxB,MAAuB,EACvB,OAAoB,EACpB,OAAiE;QAEjE,MAAM,QAAQ,GACZ,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC;QACjF,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAC9B,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,EACjF,IAAI,EACJ,CAAC,CACF,CAAC;QAEF,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,gBAAgB,EAAE;YAC/C,mBAAmB,EAAE,SAAS;YAC9B,SAAS,EAAE,QAAQ;SACpB,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CACjC,sEAAsE,EACtE,MAAM,CACP,CAAC;QAEF,MAAM,MAAM,GAAG,WAAW,CAUvB,QAAQ,CAAC,CAAC;QAEb,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;QACjE,MAAM,aAAa,GAAoB;YACrC,GAAG,MAAM;YACT,KAAK,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;YACxB,KAAK,EAAE,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;YACxB,aAAa,EAAE,MAAM,CAAC,kBAAkB,IAAI,MAAM,CAAC,aAAa;YAChE,kBAAkB,EAAE,MAAM,CAAC,uBAAuB,IAAI,MAAM,CAAC,kBAAkB;SAChF,CAAC;QAEF,+BAA+B;QAC/B,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,WAAW,EAAE,CAAC;YAC5C,IAAI,UAAU,CAAC,IAAI,KAAK,UAAU,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACvD,MAAM,IAAI,GAAG,UAAU,CAAC,KAAgC,CAAC;gBACzD,IAAI,IAAI,CAAC,EAAE;oBAAE,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YAClE,CAAC;YACD,IAAI,UAAU,CAAC,IAAI,KAAK,aAAa,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;gBAC3D,aAAa,CAAC,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,MAAM,CAC9C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,UAAU,CAAC,MAAM,CAClC,CAAC;gBACF,aAAa,CAAC,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,MAAM,CAC9C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,UAAU,CAAC,MAAM,CACxE,CAAC;YACJ,CAAC;YACD,IAAI,UAAU,CAAC,IAAI,KAAK,UAAU,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC;gBACvD,MAAM,IAAI,GAAG,UAAU,CAAC,KAAgC,CAAC;gBACzD,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;oBAC/B,aAAa,CAAC,KAAK,CAAC,IAAI,CACtB,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,aAAa,CAAC,KAAK,CAAC,MAAM,CAAC,CACrD,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,4BAA4B,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;QAC1E,CAAC;QAED,OAAO,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC;IACxC,CAAC;IAED,8EAA8E;IAC9E,gBAAgB;IAChB,8EAA8E;IAEtE,eAAe,CACrB,MAAuB,EACvB,OAAoB;QAEpB,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC,CAAC;QACpE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,EAAE,GAAG,MAAM,EAAE,mBAAmB,EAAE,EAAE,EAAE,CAAC;QAChD,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,wBAAwB,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC/D,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAC/B,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACtB,GAAG,IAAI;YACP,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC;SAC7D,CAAC,CAAC,EACH,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAC7B,CAAC;QACF,MAAM,YAAY,GAAG,MAAM,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAC3D,IAAI,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CACb,+DAA+D,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACjG,CAAC;QACJ,CAAC;QAED,OAAO;YACL,GAAG,MAAM;YACT,mBAAmB,EAAE,WAAW;YAChC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;SACnF,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,MAAuB;QACzC,OAAO;YACL,EAAE,EAAE,WAAW,IAAI,CAAC,GAAG,EAAE,EAAE;YAC3B,IAAI,EAAE,MAAM,CAAC,OAAO;YACpB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,WAAW,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;YACtD,QAAQ,EAAE,EAAE;YACZ,gBAAgB,EAAE,YAAY;YAC9B,iBAAiB,EAAE,MAAM;SAC1B,CAAC;IACJ,CAAC;IAED,8EAA8E;IAC9E,wBAAwB;IACxB,8EAA8E;IAEtE,cAAc,CAAC,KAAqC;QAC1D,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IACjD,CAAC;IAEO,aAAa,CAAC,GAA4B;QAChD,MAAM,UAAU,GACd,GAAG,CAAC,UAAU,KAAK,IAAI;YACrB,CAAC,CAAC,OAAO;YACT,CAAC,CAAC,GAAG,CAAC,UAAU,KAAK,KAAK;gBACxB,CAAC,CAAC,MAAM;gBACR,CAAC,CAAE,GAAG,CAAC,UAAkD,IAAI,OAAO,CAAC;QAE3E,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,IAAI,QAAQ,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACpF,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,KAAK,CAAsB;YACpD,cAAc,EAAG,GAAG,CAAC,cAA8C,IAAI;gBACrE,IAAI,EAAE,KAAc;gBACpB,YAAY,EAAE,MAAM,CAAC,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,IAAI,IAAI,cAAc,CAAC;aACrE;YACD,aAAa,EAAG,GAAG,CAAC,aAA4C,IAAI,aAAa;YACjF,WAAW,EAAG,GAAG,CAAC,WAAwC,IAAI,MAAM;YACpE,UAAU;YACV,UAAU,EACR,OAAO,GAAG,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC;gBACnE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,UAAU,CAAC,CAAC;gBAC1C,CAAC,CAAC,SAAS;SAChB,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,KAAqC;QAC1D,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC;IAEO,aAAa,CAAC,GAA4B,EAAE,KAAa;QAC/D,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,IAAI,QAAQ,KAAK,EAAE,CAAC;YACrC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC;YACnC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,CAAC;YACjC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,IAAI,IAAI,QAAQ,CAAsB;SACxD,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAC7B,IAAe,EACf,WAAqC;QAErC,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,IAAI,CAAC,EAAE,CAAC,CAAC;QACvE,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC;QAE7B,OAAO;YACL,GAAG,IAAI;YACP,UAAU,EAAE,UAAU,CAAC,UAAU;YACjC,GAAG,EAAE;gBACH,UAAU,EAAE,UAAU,CAAC,QAAQ;gBAC/B,KAAK,EAAE,UAAU,CAAC,KAAK;gBACvB,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B;SACF,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,IAAe;QACxC,IAAI,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5E,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK;YAAE,OAAO,GAAG,CAAC;QACpC,IAAI,IAAI,CAAC,aAAa,KAAK,oBAAoB;YAAE,OAAO,IAAI,CAAC;QAC7D,IAAI,IAAI,CAAC,aAAa,KAAK,eAAe;YAAE,OAAO,IAAI,CAAC;QAExD,MAAM,YAAY,GAChB,IAAI,CAAC,cAAc,CAAC,IAAI,KAAK,KAAK;YAChC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,YAAY,CAAC,WAAW,EAAE;YAChD,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QAE5B,IAAI,8DAA8D,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACtF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,4DAA4D,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACpF,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,YAAY,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;YAC9B,OAAO,IAAI,CAAC;QACd,CAAC;QACD,IAAI,YAAY,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;YAC7B,OAAO,GAAG,CAAC;QACb,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;CACF"}

package/dist/orchestration/planning/ProviderAssignmentEngine.d.ts

@@ -0,0 +1,46 @@
+/**
+ * @file ProviderAssignmentEngine.ts
+ * @description Assigns LLM providers and models to graph nodes.
+ *
+ * Five strategies:
+ *   best     — top model from each provider
+ *   cheapest — cheapest model from each provider
+ *   balanced — complexity-based tier selection
+ *   explicit — user-specified per-node assignments
+ *   mixed    — explicit overrides with a fallback strategy
+ */
+import type { GraphNode } from '../ir/types.js';
+import type { ProviderStrategyConfig, NodeProviderAssignment } from './types.js';
+/** Node with an optional complexity annotation from the planner. */
+type AnnotatedNode = GraphNode & {
+    complexity?: number;
+};
+/**
+ * Assigns LLM providers and models to graph nodes based on strategy.
+ */
+export declare class ProviderAssignmentEngine {
+    private readonly availableProviders;
+    private readonly rotation;
+    constructor(availableProviders: string[]);
+    /**
+     * Assign providers/models to all nodes in a graph.
+     *
+     * @param nodes - Graph nodes, optionally annotated with `complexity` (0-1).
+     * @param config - Strategy configuration.
+     */
+    assign(nodes: AnnotatedNode[], config: ProviderStrategyConfig): NodeProviderAssignment[];
+    /** Check whether all required providers are available (have API keys). */
+    checkAvailability(assignments: NodeProviderAssignment[]): {
+        available: boolean;
+        missing: string[];
+    };
+    private assignBest;
+    private assignCheapest;
+    private assignBalanced;
+    private assignExplicit;
+    private assignMixed;
+    /** Pick a provider from the preferred list, rotating to avoid pinning all nodes to one provider. */
+    private pickProvider;
+}
+export {};
+//# sourceMappingURL=ProviderAssignmentEngine.d.ts.map

package/dist/orchestration/planning/ProviderAssignmentEngine.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderAssignmentEngine.d.ts","sourceRoot":"","sources":["../../../src/orchestration/planning/ProviderAssignmentEngine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,KAAK,EAAE,sBAAsB,EAAE,sBAAsB,EAAsB,MAAM,YAAY,CAAC;AAqBrG,oEAAoE;AACpE,KAAK,aAAa,GAAG,SAAS,GAAG;IAAE,UAAU,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAEzD;;GAEG;AACH,qBAAa,wBAAwB;IACnC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAW;IAC9C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA6B;gBAE1C,kBAAkB,EAAE,MAAM,EAAE;IAMxC;;;;;OAKG;IACH,MAAM,CAAC,KAAK,EAAE,aAAa,EAAE,EAAE,MAAM,EAAE,sBAAsB,GAAG,sBAAsB,EAAE;IAmBxF,0EAA0E;IAC1E,iBAAiB,CAAC,WAAW,EAAE,sBAAsB,EAAE,GAAG;QACxD,SAAS,EAAE,OAAO,CAAC;QACnB,OAAO,EAAE,MAAM,EAAE,CAAC;KACnB;IAUD,OAAO,CAAC,UAAU;IAelB,OAAO,CAAC,cAAc;IAetB,OAAO,CAAC,cAAc;IAiCtB,OAAO,CAAC,cAAc;IAmBtB,OAAO,CAAC,WAAW;IAqBnB,oGAAoG;IACpG,OAAO,CAAC,YAAY;CAarB"}

package/dist/orchestration/planning/ProviderAssignmentEngine.js

@@ -0,0 +1,155 @@
+/**
+ * @file ProviderAssignmentEngine.ts
+ * @description Assigns LLM providers and models to graph nodes.
+ *
+ * Five strategies:
+ *   best     — top model from each provider
+ *   cheapest — cheapest model from each provider
+ *   balanced — complexity-based tier selection
+ *   explicit — user-specified per-node assignments
+ *   mixed    — explicit overrides with a fallback strategy
+ */
+/**
+ * Provider defaults — mirrors PROVIDER_DEFAULTS from api/provider-defaults.ts
+ * but inlined to avoid circular dependency across package boundaries.
+ */
+const DEFAULTS = {
+    openai: { text: 'gpt-4o', cheap: 'gpt-4o-mini' },
+    anthropic: { text: 'claude-sonnet-4-20250514', cheap: 'claude-haiku-4-5-20251001' },
+    gemini: { text: 'gemini-2.5-flash', cheap: 'gemini-2.0-flash' },
+    ollama: { text: 'llama3.2', cheap: 'llama3.2' },
+    openrouter: { text: 'openai/gpt-4o', cheap: 'openai/gpt-4o-mini' },
+    groq: { text: 'llama-3.3-70b-versatile', cheap: 'gemma2-9b-it' },
+    together: {
+        text: 'meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo',
+        cheap: 'meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo',
+    },
+    mistral: { text: 'mistral-large-latest', cheap: 'mistral-small-latest' },
+    xai: { text: 'grok-2', cheap: 'grok-2-mini' },
+};
+/**
+ * Assigns LLM providers and models to graph nodes based on strategy.
+ */
+export class ProviderAssignmentEngine {
+    constructor(availableProviders) {
+        this.rotation = new Map();
+        // Prefer providers we have defaults for, fall back to raw list
+        const known = availableProviders.filter((p) => p in DEFAULTS);
+        this.availableProviders = known.length > 0 ? known : availableProviders;
+    }
+    /**
+     * Assign providers/models to all nodes in a graph.
+     *
+     * @param nodes - Graph nodes, optionally annotated with `complexity` (0-1).
+     * @param config - Strategy configuration.
+     */
+    assign(nodes, config) {
+        switch (config.strategy) {
+            case 'best':
+                return nodes.map((n) => this.assignBest(n));
+            case 'cheapest':
+                return nodes.map((n) => this.assignCheapest(n));
+            case 'balanced':
+                return nodes.map((n) => this.assignBalanced(n));
+            case 'explicit':
+                return nodes.map((n) => this.assignExplicit(n, config.assignments ?? {}));
+            case 'mixed':
+                return nodes.map((n) => this.assignMixed(n, config.assignments ?? {}, config.fallback ?? 'balanced'));
+            default:
+                return nodes.map((n) => this.assignBalanced(n));
+        }
+    }
+    /** Check whether all required providers are available (have API keys). */
+    checkAvailability(assignments) {
+        const requiredProviders = new Set(assignments.map((a) => a.provider));
+        const missing = [...requiredProviders].filter((p) => !this.availableProviders.includes(p));
+        return { available: missing.length === 0, missing };
+    }
+    // ---------------------------------------------------------------------------
+    // Strategy implementations
+    // ---------------------------------------------------------------------------
+    assignBest(node) {
+        const provider = this.pickProvider('best', ['anthropic', 'openai', 'openrouter', 'gemini', 'groq', 'xai', 'mistral', 'together', 'ollama']);
+        const defaults = DEFAULTS[provider];
+        return {
+            nodeId: node.id,
+            provider,
+            model: defaults?.text ?? 'gpt-4o',
+            complexity: node.complexity ?? 0.5,
+            reason: 'best strategy: top model',
+        };
+    }
+    assignCheapest(node) {
+        const provider = this.pickProvider('cheapest', ['groq', 'gemini', 'openai', 'openrouter', 'together', 'mistral', 'xai', 'anthropic', 'ollama']);
+        const defaults = DEFAULTS[provider];
+        return {
+            nodeId: node.id,
+            provider,
+            model: defaults?.cheap ?? defaults?.text ?? 'gpt-4o-mini',
+            complexity: node.complexity ?? 0.5,
+            reason: 'cheapest strategy: cheapest model',
+        };
+    }
+    assignBalanced(node) {
+        const complexity = node.complexity ?? 0.5;
+        const provider = complexity < 0.3
+            ? this.pickProvider('balanced:cheap', ['groq', 'gemini', 'openai', 'openrouter', 'together', 'mistral', 'xai', 'anthropic', 'ollama'])
+            : complexity >= 0.7
+                ? this.pickProvider('balanced:strong', ['anthropic', 'openai', 'openrouter', 'gemini', 'groq', 'xai', 'mistral', 'together', 'ollama'])
+                : this.pickProvider('balanced:standard', ['openai', 'openrouter', 'anthropic', 'gemini', 'groq', 'mistral', 'xai', 'together', 'ollama']);
+        const defaults = DEFAULTS[provider];
+        const model = complexity < 0.3
+            ? (defaults?.cheap ?? 'gpt-4o-mini')
+            : (defaults?.text ?? 'gpt-4o');
+        return {
+            nodeId: node.id,
+            provider,
+            model,
+            complexity,
+            reason: `balanced strategy: complexity ${complexity.toFixed(2)} → ${complexity < 0.3 ? 'cheap' : 'standard'} model`,
+        };
+    }
+    assignExplicit(node, assignments) {
+        const explicit = assignments[node.id] ?? assignments._default;
+        if (explicit) {
+            const defaults = DEFAULTS[explicit.provider];
+            return {
+                nodeId: node.id,
+                provider: explicit.provider,
+                model: explicit.model ?? defaults?.text ?? 'gpt-4o',
+                complexity: node.complexity ?? 0.5,
+                reason: 'explicit assignment',
+            };
+        }
+        // No explicit assignment — fall back to balanced
+        return this.assignBalanced(node);
+    }
+    assignMixed(node, assignments, fallback) {
+        // Check for explicit assignment first
+        if (assignments[node.id]) {
+            return this.assignExplicit(node, assignments);
+        }
+        // Fall back to named strategy
+        switch (fallback) {
+            case 'best':
+                return this.assignBest(node);
+            case 'cheapest':
+                return this.assignCheapest(node);
+            default:
+                return this.assignBalanced(node);
+        }
+    }
+    /** Pick a provider from the preferred list, rotating to avoid pinning all nodes to one provider. */
+    pickProvider(bucket, preferred) {
+        const candidates = preferred?.filter((provider) => this.availableProviders.includes(provider))
+            ?? this.availableProviders;
+        const pool = candidates.length > 0 ? candidates : this.availableProviders;
+        if (pool.length === 0)
+            return 'openai';
+        const cursor = this.rotation.get(bucket) ?? 0;
+        const provider = pool[cursor % pool.length] ?? pool[0] ?? 'openai';
+        this.rotation.set(bucket, cursor + 1);
+        return provider;
+    }
+}
+//# sourceMappingURL=ProviderAssignmentEngine.js.map

package/dist/orchestration/planning/ProviderAssignmentEngine.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderAssignmentEngine.js","sourceRoot":"","sources":["../../../src/orchestration/planning/ProviderAssignmentEngine.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAKH;;;GAGG;AACH,MAAM,QAAQ,GAAoD;IAChE,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;IAChD,SAAS,EAAE,EAAE,IAAI,EAAE,0BAA0B,EAAE,KAAK,EAAE,2BAA2B,EAAE;IACnF,MAAM,EAAE,EAAE,IAAI,EAAE,kBAAkB,EAAE,KAAK,EAAE,kBAAkB,EAAE;IAC/D,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,UAAU,EAAE;IAC/C,UAAU,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,oBAAoB,EAAE;IAClE,IAAI,EAAE,EAAE,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,cAAc,EAAE;IAChE,QAAQ,EAAE;QACR,IAAI,EAAE,8CAA8C;QACpD,KAAK,EAAE,6CAA6C;KACrD;IACD,OAAO,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,KAAK,EAAE,sBAAsB,EAAE;IACxE,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;CAC9C,CAAC;AAKF;;GAEG;AACH,MAAM,OAAO,wBAAwB;IAInC,YAAY,kBAA4B;QAFvB,aAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;QAGpD,+DAA+D;QAC/D,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,QAAQ,CAAC,CAAC;QAC9D,IAAI,CAAC,kBAAkB,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC;IAC1E,CAAC;IAED;;;;;OAKG;IACH,MAAM,CAAC,KAAsB,EAAE,MAA8B;QAC3D,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;YACxB,KAAK,MAAM;gBACT,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9C,KAAK,UAAU;gBACb,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,KAAK,UAAU;gBACb,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;YAClD,KAAK,UAAU;gBACb,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC,CAAC;YAC5E,KAAK,OAAO;gBACV,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACrB,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,IAAI,EAAE,EAAE,MAAM,CAAC,QAAQ,IAAI,UAAU,CAAC,CAC7E,CAAC;YACJ;gBACE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,0EAA0E;IAC1E,iBAAiB,CAAC,WAAqC;QAIrD,MAAM,iBAAiB,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC;QACtE,MAAM,OAAO,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3F,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,OAAO,EAAE,CAAC;IACtD,CAAC;IAED,8EAA8E;IAC9E,2BAA2B;IAC3B,8EAA8E;IAEtE,UAAU,CAAC,IAAmB;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAChC,MAAM,EACN,CAAC,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,CAChG,CAAC;QACF,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACpC,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,QAAQ;YACR,KAAK,EAAE,QAAQ,EAAE,IAAI,IAAI,QAAQ;YACjC,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,GAAG;YAClC,MAAM,EAAE,0BAA0B;SACnC,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,IAAmB;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAChC,UAAU,EACV,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,QAAQ,CAAC,CAChG,CAAC;QACF,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACpC,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,QAAQ;YACR,KAAK,EAAE,QAAQ,EAAE,KAAK,IAAI,QAAQ,EAAE,IAAI,IAAI,aAAa;YACzD,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,GAAG;YAClC,MAAM,EAAE,mCAAmC;SAC5C,CAAC;IACJ,CAAC;IAEO,cAAc,CAAC,IAAmB;QACxC,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,GAAG,CAAC;QAC1C,MAAM,QAAQ,GACZ,UAAU,GAAG,GAAG;YACd,CAAC,CAAC,IAAI,CAAC,YAAY,CACf,gBAAgB,EAChB,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,WAAW,EAAE,QAAQ,CAAC,CAChG;YACH,CAAC,CAAC,UAAU,IAAI,GAAG;gBACjB,CAAC,CAAC,IAAI,CAAC,YAAY,CACf,iBAAiB,EACjB,CAAC,WAAW,EAAE,QAAQ,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,CAChG;gBACH,CAAC,CAAC,IAAI,CAAC,YAAY,CACf,mBAAmB,EACnB,CAAC,QAAQ,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,EAAE,QAAQ,CAAC,CAChG,CAAC;QACV,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEpC,MAAM,KAAK,GACT,UAAU,GAAG,GAAG;YACd,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,IAAI,aAAa,CAAC;YACpC,CAAC,CAAC,CAAC,QAAQ,EAAE,IAAI,IAAI,QAAQ,CAAC,CAAC;QAEnC,OAAO;YACL,MAAM,EAAE,IAAI,CAAC,EAAE;YACf,QAAQ;YACR,KAAK;YACL,UAAU;YACV,MAAM,EAAE,iCAAiC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,UAAU,GAAG,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,QAAQ;SACpH,CAAC;IACJ,CAAC;IAEO,cAAc,CACpB,IAAmB,EACnB,WAA+C;QAE/C,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,WAAW,CAAC,QAAQ,CAAC;QAC9D,IAAI,QAAQ,EAAE,CAAC;YACb,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC7C,OAAO;gBACL,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,QAAQ,EAAE,QAAQ,CAAC,QAAQ;gBAC3B,KAAK,EAAE,QAAQ,CAAC,KAAK,IAAI,QAAQ,EAAE,IAAI,IAAI,QAAQ;gBACnD,UAAU,EAAE,IAAI,CAAC,UAAU,IAAI,GAAG;gBAClC,MAAM,EAAE,qBAAqB;aAC9B,CAAC;QACJ,CAAC;QACD,iDAAiD;QACjD,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAEO,WAAW,CACjB,IAAmB,EACnB,WAA+C,EAC/C,QAAgB;QAEhB,sCAAsC;QACtC,IAAI,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QAChD,CAAC;QAED,8BAA8B;QAC9B,QAAQ,QAAQ,EAAE,CAAC;YACjB,KAAK,MAAM;gBACT,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YAC/B,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YACnC;gBACE,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED,oGAAoG;IAC5F,YAAY,CAAC,MAAc,EAAE,SAAoB;QACvD,MAAM,UAAU,GACd,SAAS,EAAE,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;eACtE,IAAI,CAAC,kBAAkB,CAAC;QAE/B,MAAM,IAAI,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC;QAC1E,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,QAAQ,CAAC;QAEvC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9C,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;QACnE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC;QACtC,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/orchestration/planning/buildLlmCaller.d.ts

@@ -0,0 +1,86 @@
+/**
+ * @file buildLlmCaller.ts
+ * @description Factory that creates an `llmCaller` function for any provider.
+ *
+ * Supports all 16+ AgentOS providers including CLI-based providers
+ * (claude-code-cli, gemini-cli) that don't use API keys. Uses the
+ * existing `resolveModelOption → resolveProvider → createProviderManager`
+ * chain so all provider-specific logic (auth, subprocess, fallback) is
+ * handled transparently.
+ *
+ * Usage:
+ * ```ts
+ * const caller = await buildLlmCaller({ provider: 'claude-code-cli', model: 'claude-opus-4-6' });
+ * const response = await caller('You are a planner.', 'Research AI papers');
+ * ```
+ */
+import type { PlannerConfig } from './types.js';
+/**
+ * Options for building an LLM caller function.
+ *
+ * At minimum, provide `provider` OR `model` (or both).
+ * If neither is provided, auto-detection from env vars kicks in.
+ */
+export interface BuildLlmCallerOptions {
+    /** Provider ID: 'openai', 'anthropic', 'claude-code-cli', 'gemini-cli', etc. */
+    provider?: string;
+    /** Model ID: 'gpt-4o', 'claude-opus-4-6', 'gemini-2.5-flash', etc. */
+    model?: string;
+    /** API key override (not needed for CLI providers). */
+    apiKey?: string;
+    /** Base URL override (e.g. for OpenRouter, Ollama). */
+    baseUrl?: string;
+    /** Temperature for planning calls. Default: 0.3. */
+    temperature?: number;
+    /** Max tokens for planning calls. Default: 4096. */
+    maxTokens?: number;
+}
+/** The shape of the llmCaller function used by MissionPlanner. */
+export type LlmCallerFn = PlannerConfig['llmCaller'];
+/**
+ * Build an `llmCaller` function for any AgentOS-supported provider.
+ *
+ * This uses the full provider resolution chain:
+ *   resolveModelOption → resolveProvider → createProviderManager → getProvider
+ *
+ * Works with all provider types:
+ * - API providers (openai, anthropic, groq, together, mistral, xai, openrouter)
+ * - CLI providers (claude-code-cli, gemini-cli) — no API key needed
+ * - Local providers (ollama) — requires OLLAMA_BASE_URL
+ *
+ * The returned function has the signature `(system: string, user: string) => Promise<string>`.
+ *
+ * @param options - Provider, model, and optional credential overrides.
+ * @returns A caller function compatible with `PlannerConfig.llmCaller`.
+ */
+export declare function buildLlmCaller(options?: BuildLlmCallerOptions): Promise<LlmCallerFn>;
+/**
+ * Build separate planner and execution callers with potentially different providers.
+ *
+ * @param plannerOptions - Options for the ToT planning model (strong reasoning).
+ * @param executionOptions - Options for agent node execution (can be different).
+ * @returns Object with `plannerCaller` and `executionCaller`.
+ *
+ * @example
+ * ```ts
+ * const { plannerCaller, executionCaller } = await buildSplitCallers(
+ *   { provider: 'claude-code-cli', model: 'claude-opus-4-6' },   // Strong for planning
+ *   { provider: 'openai', model: 'gpt-4o' },                      // Fast for execution
+ * );
+ *
+ * const planner = new MissionPlanner({
+ *   llmCaller: executionCaller,
+ *   plannerLlmCaller: plannerCaller,
+ *   plannerModel: 'claude-opus-4-6',
+ *   executionModel: 'gpt-4o',
+ *   ...
+ * });
+ * ```
+ */
+export declare function buildSplitCallers(plannerOptions: BuildLlmCallerOptions, executionOptions?: BuildLlmCallerOptions): Promise<{
+    plannerCaller: LlmCallerFn;
+    executionCaller: LlmCallerFn;
+    plannerModel: string;
+    executionModel: string;
+}>;
+//# sourceMappingURL=buildLlmCaller.d.ts.map