agentweaver 0.1.16 → 0.1.18

package/dist/runtime/interactive-execution-routing.js

@@ -1,6 +1,7 @@
 import path from "node:path";
 import { flowRoutingGroups, flowRoutingKey } from "../pipeline/flow-catalog.js";
 import { loadNamedDeclarativeFlow } from "../pipeline/declarative-flows.js";
+import { createPipelineRegistryContext } from "../pipeline/plugin-loader.js";
 import { EXECUTION_ROUTING_GROUPS, } from "../pipeline/execution-routing-config.js";
 import { DEFAULT_LAUNCH_PROFILE, defaultModelForExecutor, isAllowedModelForExecutor, } from "../pipeline/launch-profile-config.js";
 import { FlowInterruptedError, TaskRunnerError } from "../errors.js";
@@ -58,15 +59,9 @@ function executionPresetSelectionForm(flowEntry, flowDefaultAvailable, lastUsedA
         ],
     };
 }
-function executorFieldOptions() {
-    return [
-        { value: "codex", label: "codex" },
-        { value: "opencode", label: "opencode" },
-    ];
-}
-function selectedExecutorFromValues(values, fieldId, fallbackExecutor) {
-    const candidate = values[fieldId];
-    return candidate === "codex" || candidate === "opencode" ? candidate : fallbackExecutor;
+function selectedExecutorFromValues(values, fieldId, fallbackExecutor, registryContext) {
+    const candidate = typeof values[fieldId] === "string" ? values[fieldId] : "";
+    return registryContext.executors.getRouting(candidate)?.kind === "llm" ? candidate : fallbackExecutor;
 }
 function routingEditorDraftFromRouting(routing) {
     return {
@@ -83,7 +78,11 @@ function routingEditorDraftFromRouting(routing) {
         ])),
     };
 }
-function advancedRoutingEditorForm(activeGroups, draft, validationMessage) {
+function advancedRoutingEditorForm(activeGroups, draft, registryContext, validationMessage) {
+    const executorOptions = registryContext.executors.llmExecutors().map((entry) => ({
+        value: entry.id,
+        label: entry.id,
+    }));
     const fields = [
         {
             id: "default_route_executor",
@@ -91,7 +90,7 @@ function advancedRoutingEditorForm(activeGroups, draft, validationMessage) {
             label: "Default route executor",
             required: true,
             default: draft.defaultRoute.executor,
-            options: executorFieldOptions(),
+            options: executorOptions,
         },
         {
             id: "default_route_model",
@@ -100,8 +99,8 @@ function advancedRoutingEditorForm(activeGroups, draft, validationMessage) {
             help: "Available models follow the selected default executor.",
             required: true,
             default: draft.defaultRoute.model,
-            options: modelOptionsForExecutor(draft.defaultRoute.executor),
-            optionsFromValues: (values) => modelOptionsForExecutor(selectedExecutorFromValues(values, "default_route_executor", draft.defaultRoute.executor)),
+            options: modelOptionsForExecutor(draft.defaultRoute.executor, registryContext.executors),
+            optionsFromValues: (values) => modelOptionsForExecutor(selectedExecutorFromValues(values, "default_route_executor", draft.defaultRoute.executor, registryContext), registryContext.executors),
         },
         ...activeGroups.flatMap((group) => {
             const route = draft.groups[group];
@@ -112,7 +111,7 @@ function advancedRoutingEditorForm(activeGroups, draft, validationMessage) {
                     label: `${routingGroupLabel(group)} executor`,
                     required: true,
                     default: route.executor,
-                    options: executorFieldOptions(),
+                    options: executorOptions,
                 },
                 {
                     id: `${group}_model`,
@@ -121,8 +120,8 @@ function advancedRoutingEditorForm(activeGroups, draft, validationMessage) {
                     help: `Available models follow the selected ${routingGroupLabel(group)} executor.`,
                     required: true,
                     default: route.model,
-                    options: modelOptionsForExecutor(route.executor),
-                    optionsFromValues: (values) => modelOptionsForExecutor(selectedExecutorFromValues(values, `${group}_executor`, route.executor)),
+                    options: modelOptionsForExecutor(route.executor, registryContext.executors),
+                    optionsFromValues: (values) => modelOptionsForExecutor(selectedExecutorFromValues(values, `${group}_executor`, route.executor, registryContext), registryContext.executors),
                 },
             ];
         }),
@@ -202,8 +201,8 @@ function presetNameForm() {
         ],
     };
 }
-function normalizeEditableRoute(label, route) {
-    if (isAllowedModelForExecutor(route.executor, route.model)) {
+function normalizeEditableRoute(label, route, executors) {
+    if (isAllowedModelForExecutor(route.executor, route.model, executors)) {
         return {
             route: { ...route },
             validationErrors: [],
@@ -212,20 +211,20 @@ function normalizeEditableRoute(label, route) {
     return {
         route: {
             executor: route.executor,
-            model: defaultModelForExecutor(route.executor),
+            model: defaultModelForExecutor(route.executor, executors),
         },
         validationErrors: [
             `${label} model '${route.model}' is not allowed for executor '${route.executor}'. Select a ${route.executor} model.`,
         ],
     };
 }
-function normalizeEditableRoutingDraft(draft) {
-    const defaultRoute = normalizeEditableRoute("Default route", draft.defaultRoute);
+function normalizeEditableRoutingDraft(draft, executors) {
+    const defaultRoute = normalizeEditableRoute("Default route", draft.defaultRoute, executors);
     const groups = {};
     const currentRunOverrides = {};
     const validationErrors = [...defaultRoute.validationErrors];
     for (const group of EXECUTION_ROUTING_GROUPS) {
-        const normalizedRoute = normalizeEditableRoute(routingGroupLabel(group), draft.groups[group]);
+        const normalizedRoute = normalizeEditableRoute(routingGroupLabel(group), draft.groups[group], executors);
         groups[group] = normalizedRoute.route;
         if (normalizedRoute.route.executor !== defaultRoute.route.executor
             || normalizedRoute.route.model !== defaultRoute.route.model) {
@@ -274,40 +273,47 @@ function formatAsciiTable(headers, rows, maxWidths) {
     const separator = `|-${widths.map((width) => "-".repeat(width)).join("-|-")}-|`;
     return [formatRow(headers), separator, ...rows.map((row) => formatRow(row))];
 }
-function collectEffectiveRoutedStepRows(flow, cwd, routing, prefixSegments = [], ancestry = []) {
+function collectEffectiveRoutedStepRows(flow, cwd, routing, registryContext, prefixSegments = [], ancestry = []) {
     if (ancestry.includes(flow.absolutePath)) {
-        return [];
+        return Promise.resolve([]);
     }
     const nextAncestry = [...ancestry, flow.absolutePath];
     const rows = [];
-    for (const phase of flow.phases) {
-        for (const step of phase.steps) {
-            const stepRef = `${phase.id}.${step.id}`;
-            if (step.routingGroup) {
-                const route = routing.groups[step.routingGroup];
-                rows.push({
-                    step: [...prefixSegments, stepRef].join(" > "),
-                    group: routingGroupLabel(step.routingGroup),
-                    executor: route.executor,
-                    model: route.model,
+    const run = async () => {
+        for (const phase of flow.phases) {
+            for (const step of phase.steps) {
+                const stepRef = `${phase.id}.${step.id}`;
+                if (step.routingGroup) {
+                    const route = routing.groups[step.routingGroup];
+                    rows.push({
+                        step: [...prefixSegments, stepRef].join(" > "),
+                        group: routingGroupLabel(step.routingGroup),
+                        executor: route.executor,
+                        model: route.model,
+                    });
+                }
+                if (step.node !== "flow-run") {
+                    continue;
+                }
+                const nestedFlowName = step.params?.fileName;
+                if (!nestedFlowName || !("const" in nestedFlowName) || typeof nestedFlowName.const !== "string") {
+                    continue;
+                }
+                const nestedFlow = await loadNamedDeclarativeFlow(nestedFlowName.const, cwd, {
+                    ...(registryContext ? { registryContext } : {}),
                 });
+                const nestedFlowLabel = path.basename(nestedFlow.fileName, path.extname(nestedFlow.fileName));
+                rows.push(...await collectEffectiveRoutedStepRows(nestedFlow, cwd, routing, registryContext, [...prefixSegments, stepRef, nestedFlowLabel], nextAncestry));
             }
-            if (step.node !== "flow-run") {
-                continue;
-            }
-            const nestedFlowName = step.params?.fileName;
-            if (!nestedFlowName || !("const" in nestedFlowName) || typeof nestedFlowName.const !== "string") {
-                continue;
-            }
-            const nestedFlow = loadNamedDeclarativeFlow(nestedFlowName.const, cwd);
-            const nestedFlowLabel = path.basename(nestedFlow.fileName, path.extname(nestedFlow.fileName));
-            rows.push(...collectEffectiveRoutedStepRows(nestedFlow, cwd, routing, [...prefixSegments, stepRef, nestedFlowLabel], nextAncestry));
         }
-    }
-    return rows;
+        return rows;
+    };
+    return run();
 }
-export function describeEffectiveRoutingPreview(flowEntry, routing, cwd) {
-    const previewGroups = flowRoutingGroups(flowEntry, cwd);
+export async function describeEffectiveRoutingPreview(flowEntry, routing, cwd, registryContext) {
+    const previewGroups = await flowRoutingGroups(flowEntry, cwd, {
+        ...(registryContext ? { registryContext } : {}),
+    });
     const summaryRows = [
         ["Default", routing.defaultRoute.executor, routing.defaultRoute.model],
         ...previewGroups.map((group) => [
@@ -320,7 +326,7 @@ export function describeEffectiveRoutingPreview(flowEntry, routing, cwd) {
         "Effective routes:",
         ...formatAsciiTable(["Scope", "Executor", "Model"], summaryRows, [18, 12, 36]),
     ];
-    const routedSteps = collapseRepeatedEffectiveRoutedStepRows(collectEffectiveRoutedStepRows(flowEntry.flow, cwd, routing));
+    const routedSteps = collapseRepeatedEffectiveRoutedStepRows(await collectEffectiveRoutedStepRows(flowEntry.flow, cwd, routing, registryContext));
     if (routedSteps.length === 0) {
         return lines.join("\n");
     }
@@ -332,7 +338,8 @@ export function describeEffectiveRoutingPreview(flowEntry, routing, cwd) {
     ].join("\n");
 }
 export async function requestInteractiveExecutionRouting(flowEntry, requestUserInput) {
-    const previewGroups = flowRoutingGroups(flowEntry, process.cwd());
+    const registryContext = await createPipelineRegistryContext(process.cwd());
+    const previewGroups = await flowRoutingGroups(flowEntry, process.cwd(), { registryContext });
     const flowKey = flowRoutingKey(flowEntry);
     const namedPresets = getNamedExecutionPresets();
     const flowDefault = getFlowDefaultExecutionRouting(flowKey);
@@ -346,14 +353,14 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
             throw new TaskRunnerError("Flow default routing is unavailable.");
         }
         selectedPreset = { kind: "flow-default", label: "Flow default" };
-        routing = resolveStoredExecutionRoutingSnapshot(flowDefault.routing);
+        routing = resolveStoredExecutionRoutingSnapshot(flowDefault.routing, registryContext.executors);
     }
     else if (selectedPresetValue === "last-used") {
         if (!lastUsed) {
             throw new TaskRunnerError("Last-used routing is unavailable.");
         }
         selectedPreset = { kind: "last-used", label: "Last used" };
-        routing = resolveStoredExecutionRoutingSnapshot(lastUsed.routing);
+        routing = resolveStoredExecutionRoutingSnapshot(lastUsed.routing, registryContext.executors);
     }
     else if (selectedPresetValue.startsWith("named:")) {
         const presetName = selectedPresetValue.slice("named:".length);
@@ -362,7 +369,7 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
             throw new TaskRunnerError(`Named preset '${presetName}' is unavailable.`);
         }
         selectedPreset = { kind: "named", presetId: presetName, label: `Named preset: ${presetName}` };
-        routing = resolveStoredExecutionRoutingSnapshot(namedPreset.routing);
+        routing = resolveStoredExecutionRoutingSnapshot(namedPreset.routing, registryContext.executors);
     }
     else if (selectedPresetValue.startsWith("built-in:")) {
         const presetId = selectedPresetValue.slice("built-in:".length);
@@ -371,11 +378,12 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
             throw new TaskRunnerError(`Unknown execution preset '${presetId}'.`);
         }
         selectedPreset = { kind: "built-in", presetId, label: preset.label };
-        routing = resolveExecutionRouting({ presetId });
+        routing = resolveExecutionRouting({ presetId, executors: registryContext.executors });
     }
     else {
         selectedPreset = { kind: "custom", label: "Custom" };
         routing = resolveExecutionRouting({
+            executors: registryContext.executors,
             defaultRoute: {
                 executor: DEFAULT_LAUNCH_PROFILE.executor,
                 model: DEFAULT_LAUNCH_PROFILE.model,
@@ -385,7 +393,7 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
     let editorDraft = routingEditorDraftFromRouting(routing);
     let editorValidationMessage;
     for (;;) {
-        const previewText = `Preset: ${selectedPreset.label}\n${describeEffectiveRoutingPreview(flowEntry, routing, process.cwd())}`;
+        const previewText = `Preset: ${selectedPreset.label}\n${await describeEffectiveRoutingPreview(flowEntry, routing, process.cwd(), registryContext)}`;
         const actionResult = await requestUserInput(routingActionForm(previewText));
         const action = String(actionResult.values.action ?? "start");
         if (action === "cancel") {
@@ -395,7 +403,7 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
             saveLastUsedExecutionRouting(flowKey, routing, selectedPreset);
             return { routing, selectedPreset };
         }
-        const routingFormResult = await requestUserInput(advancedRoutingEditorForm(previewGroups, editorDraft, editorValidationMessage));
+        const routingFormResult = await requestUserInput(advancedRoutingEditorForm(previewGroups, editorDraft, registryContext, editorValidationMessage));
         const requestedDefaultRoute = {
             executor: String(routingFormResult.values.default_route_executor ?? editorDraft.defaultRoute.executor),
             model: String(routingFormResult.values.default_route_model ?? editorDraft.defaultRoute.model),
@@ -418,7 +426,7 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
                 ];
             })),
         };
-        const normalizedDraft = normalizeEditableRoutingDraft(requestedDraft);
+        const normalizedDraft = normalizeEditableRoutingDraft(requestedDraft, registryContext.executors);
         editorDraft = normalizedDraft.draft;
         if (normalizedDraft.validationErrors.length > 0) {
             editorValidationMessage = normalizedDraft.validationErrors.join("\n");
@@ -426,6 +434,7 @@ export async function requestInteractiveExecutionRouting(flowEntry, requestUserI
         }
         try {
             routing = resolveExecutionRouting({
+                executors: registryContext.executors,
                 defaultRoute: {
                     executor: normalizedDraft.draft.defaultRoute.executor,
                     model: normalizedDraft.draft.defaultRoute.model,