poly-weaver 0.15.0 → 0.15.2

package/dist/preview-panel.js

@@ -3,7 +3,8 @@ import { breakKillRun } from "./text-editing/kill-ring.js";
 import { availableDriversForRole } from "./agents/registry.js";
 import { diagramHeight, renderFlowDiagram, computeMinWidth, detailedDiagramHeight, renderDetailedFlowDiagram, computeDetailedMinWidth } from "./flow/diagram.js";
 import { configurationGroupOf, getConfigurationGroupModelEffortKeys, getProviderForConfigurationGroup, } from "./flow/params.js";
-import { defaultEffort, defaultModel, defaultSubagentModel, isKnownEffort, isKnownModel, isKnownSubagentModel, providerEfforts, providerModels, subagentModels, } from "./providers/option-catalog.js";
+import { defaultEffort, defaultFast, defaultModel, defaultSubagentModel, isFastAvailable, defaultContext, isKnownContext, isKnownEffort, isKnownModel, isKnownSubagentModel, providerContexts, providerEfforts, providerModels, subagentModels, } from "./providers/option-catalog.js";
+import { compactProviderConfigurationTokens, } from "./providers/configuration-presentation.js";
 import { visibleLength, padToWidth, dim, bold, reset, green, red } from "./ansi.js";
 import { renderWrappedLine } from "./text-editing/soft-wrap.js";
 // ─── Constants ───────────────────────────────────────────────────────────
@@ -23,7 +24,7 @@ export function buildFieldsForTemplate(template, defaults, existingFields) {
     const fields = [];
     const groupBuckets = new Map();
     for (const param of template.params) {
-        if (param.type !== "driver" && param.type !== "model" && param.type !== "effort") {
+        if (!isProviderConfigurationParam(param)) {
             continue;
         }
         const group = configurationGroupOf(param);
@@ -34,6 +35,10 @@ export function buildFieldsForTemplate(template, defaults, existingFields) {
             bucket.model = param;
         if (param.type === "effort")
             bucket.effort = param;
+        if (param.type === "fast")
+            bucket.fast = param;
+        if (param.type === "context")
+            bucket.context = param;
         groupBuckets.set(group, bucket);
     }
     const groupedConfigurations = new Set([...groupBuckets.entries()]
@@ -41,16 +46,18 @@ export function buildFieldsForTemplate(template, defaults, existingFields) {
         .map(([group]) => group));
     const emittedGroups = new Set();
     for (const param of template.params) {
-        if (param.type === "driver" || param.type === "model" || param.type === "effort") {
+        if (isProviderConfigurationParam(param)) {
             const configurationGroup = configurationGroupOf(param);
             const bucket = groupBuckets.get(configurationGroup);
             const isGroupedParam = groupedConfigurations.has(configurationGroup) && bucket && (param.key === bucket.driver?.key ||
                 param.key === bucket.model?.key ||
-                param.key === bucket.effort?.key);
+                param.key === bucket.effort?.key ||
+                param.key === bucket.fast?.key ||
+                param.key === bucket.context?.key);
             if (isGroupedParam) {
                 if (emittedGroups.has(configurationGroup))
                     continue;
-                fields.push(buildRoleGroupField(param.role, configurationGroup, bucket.driver, bucket.model, bucket.effort, template, defaults, existingFields));
+                fields.push(buildRoleGroupField(param.role, configurationGroup, bucket.driver, bucket.model, bucket.effort, bucket.fast, bucket.context, template, defaults, existingFields));
                 emittedGroups.add(configurationGroup);
                 continue;
             }
@@ -113,6 +120,38 @@ export function buildFieldsForTemplate(template, defaults, existingFields) {
                 });
                 break;
             }
+            case "fast": {
+                const fp = param;
+                const provider = getAvailableProviderForGroup(configurationGroupOf(fp), fp.role, template, defaults, existingFields);
+                fields.push({
+                    kind: "toggle",
+                    label: fp.label,
+                    key: fp.key,
+                    value: defaultVal ?? defaultFast(provider),
+                    hidden: true,
+                });
+                break;
+            }
+            case "context": {
+                const cp = param;
+                const provider = getAvailableProviderForGroup(configurationGroupOf(cp), cp.role, template, defaults, existingFields);
+                const options = providerContexts(provider);
+                const requested = defaultVal;
+                const value = requested && isKnownContext(provider, requested)
+                    ? requested
+                    : defaultContext(provider) ?? "default";
+                fields.push({
+                    kind: "select",
+                    label: cp.label,
+                    key: cp.key,
+                    options: [...options],
+                    value,
+                    expanded: false,
+                    expandedIndex: 0,
+                    hidden: options.length === 0,
+                });
+                break;
+            }
             case "number": {
                 const np = param;
                 fields.push({
@@ -188,7 +227,7 @@ function getAvailableModelForGroup(configurationGroup, template, defaults, exist
         return value;
     return defaultModel(provider);
 }
-function buildRoleGroupField(role, configurationGroup, driverParam, modelParam, effortParam, template, defaults, existingFields) {
+function buildRoleGroupField(role, configurationGroup, driverParam, modelParam, effortParam, fastParam, contextParam, template, defaults, existingFields) {
     const existingProvider = existingFields ? findFieldByKey(existingFields, driverParam.key) : undefined;
     const previous = existingFields?.find((f) => f.kind === "group" && f.providerKey === driverParam.key);
     const availableProviders = availableDriverOptions(role);
@@ -198,8 +237,22 @@ function buildRoleGroupField(role, configurationGroup, driverParam, modelParam,
         driverParam.default));
     const existingModel = existingFields ? findFieldByKey(existingFields, modelParam.key) : undefined;
     const existingEffort = existingFields ? findFieldByKey(existingFields, effortParam.key) : undefined;
+    const existingFast = fastParam && existingFields ? findFieldByKey(existingFields, fastParam.key) : undefined;
+    const existingContext = contextParam && existingFields ? findFieldByKey(existingFields, contextParam.key) : undefined;
     const model = (existingModel?.value ?? defaults[modelParam.key] ?? defaultModel(provider));
     const effort = (existingEffort?.value ?? defaults[effortParam.key] ?? defaultEffort(provider));
+    const rawFast = fastParam
+        ? existingFast?.value
+            ?? defaults[fastParam.key]
+            ?? defaultFast(provider)
+        : undefined;
+    const context = contextParam
+        ? existingContext?.value
+            ?? defaults[contextParam.key]
+            ?? defaultContext(provider)
+            ?? "default"
+        : undefined;
+    const focusKey = previous?.children[previous.focusedChild]?.key;
     const group = {
         kind: "group",
         key: driverParam.key,
@@ -210,6 +263,8 @@ function buildRoleGroupField(role, configurationGroup, driverParam, modelParam,
         providerKey: driverParam.key,
         modelKey: modelParam.key,
         effortKey: effortParam.key,
+        fastKey: fastParam?.key,
+        contextKey: contextParam?.key,
         children: [
             {
                 kind: "select",
@@ -238,11 +293,27 @@ function buildRoleGroupField(role, configurationGroup, driverParam, modelParam,
                 expanded: false,
                 expandedIndex: 0,
             },
+            ...(contextParam ? [{
+                    kind: "select",
+                    label: "Context",
+                    key: contextParam.key,
+                    options: [],
+                    value: context,
+                    expanded: false,
+                    expandedIndex: 0,
+                }] : []),
+            ...(fastParam ? [{
+                    kind: "toggle",
+                    label: "Fast",
+                    key: fastParam.key,
+                    value: rawFast,
+                }] : []),
         ],
         groupExpanded: previous?.groupExpanded ?? false,
         focusedChild: previous?.focusedChild ?? 0,
     };
-    reconcileModelEffortForGroup(group);
+    reconcileProviderSettingsForGroup(group);
+    restoreGroupFocus(group, focusKey);
     return group;
 }
 // ─── Field Utilities ─────────────────────────────────────────────────────
@@ -279,6 +350,93 @@ function selectedOptionLabel(field) {
 function formatEffortValue(value, label = value) {
     return value === "ultra" ? `${red}${label}${reset}` : label;
 }
+function renderCompactProviderConfigurationToken(token) {
+    const text = token.kind === "effort"
+        ? formatEffortValue(token.text)
+        : token.text;
+    return token.dimmed ? `${dim}${text}${reset}` : text;
+}
+function selectChild(group, key) {
+    const field = group.children.find((child) => child.kind === "select" && child.key === key);
+    if (!field)
+        throw new Error(`Missing select child "${key}" in ${group.key}`);
+    return field;
+}
+function optionalSelectChild(group, key) {
+    if (!key)
+        return undefined;
+    return group.children.find((child) => child.kind === "select" && child.key === key);
+}
+function toggleChild(group, key) {
+    const field = group.children.find((child) => child.kind === "toggle" && child.key === key);
+    if (!field)
+        throw new Error(`Missing toggle child "${key}" in ${group.key}`);
+    return field;
+}
+function optionalToggleChild(group, key) {
+    if (!key)
+        return undefined;
+    return group.children.find((child) => child.kind === "toggle" && child.key === key);
+}
+function providerChild(group) {
+    return selectChild(group, group.providerKey);
+}
+function modelChild(group) {
+    return selectChild(group, group.modelKey);
+}
+function effortChild(group) {
+    return selectChild(group, group.effortKey);
+}
+function fastChild(group) {
+    return optionalToggleChild(group, group.fastKey);
+}
+function contextChild(group) {
+    return optionalSelectChild(group, group.contextKey);
+}
+function visibleGroupChildIndexes(group) {
+    return group.children
+        .map((child, index) => child.hidden ? -1 : index)
+        .filter((index) => index >= 0);
+}
+function firstVisibleGroupChildIndex(group) {
+    return visibleGroupChildIndexes(group)[0] ?? 0;
+}
+function nextVisibleGroupChildIndex(group, direction) {
+    const visible = visibleGroupChildIndexes(group);
+    if (visible.length === 0)
+        return 0;
+    const currentIndex = visible.indexOf(group.focusedChild);
+    const current = currentIndex >= 0 ? currentIndex : 0;
+    return visible[(current + direction + visible.length) % visible.length];
+}
+function ensureGroupFocusVisible(group) {
+    if (!group.children[group.focusedChild]?.hidden)
+        return;
+    group.focusedChild = firstVisibleGroupChildIndex(group);
+}
+function restoreGroupFocus(group, focusKey) {
+    if (focusKey) {
+        const next = group.children.findIndex((child) => child.key === focusKey);
+        if (next >= 0)
+            group.focusedChild = next;
+    }
+    ensureGroupFocusVisible(group);
+}
+export function snapshotGroupToDriverMemory(memory, group, providerOverride) {
+    const provider = providerOverride ?? providerChild(group).value;
+    const fast = fastChild(group);
+    const entry = {
+        model: modelChild(group).value,
+        effort: effortChild(group).value,
+        ...(fast ? { fast: fast.value } : {}),
+    };
+    const context = contextChild(group);
+    if (context && isKnownContext(provider, context.value)) {
+        Object.assign(entry, { context: context.value });
+    }
+    memory[group.configurationGroup] ??= {};
+    memory[group.configurationGroup][provider] = entry;
+}
 function findFieldOwnerByKey(fields, key) {
     for (const owner of fields) {
         if (owner.kind === "group") {
@@ -311,41 +469,80 @@ function defaultParamValue(param) {
             return param.default;
         case "model":
         case "effort":
+        case "fast":
+        case "context":
             return undefined;
     }
 }
-export function reconcileModelEffortForGroup(group, driverMemory) {
-    const provider = group.children[0].value;
+export function reconcileProviderSettingsForGroup(group, driverMemory, opts = {}) {
+    const provider = providerChild(group).value;
     group.value = provider;
-    const model = group.children[1];
-    const effort = group.children[2];
+    const model = modelChild(group);
+    const effort = effortChild(group);
+    const fast = fastChild(group);
+    const context = contextChild(group);
     model.options = [...providerModels(provider)];
     const memory = driverMemory?.[group.configurationGroup]?.[provider];
     if (memory?.model && isKnownModel(provider, memory.model)) {
         model.value = memory.model;
     }
-    else if (!isKnownModel(provider, model.value)) {
+    else if (opts.providerChanged || !isKnownModel(provider, model.value)) {
         model.value = defaultModel(provider);
     }
     effort.options = [...providerEfforts(provider, model.value)];
     if (memory?.effort && isKnownEffort(provider, memory.effort, model.value)) {
         effort.value = memory.effort;
     }
-    else if (!isKnownEffort(provider, effort.value, model.value)) {
+    else if (opts.providerChanged || !isKnownEffort(provider, effort.value, model.value)) {
         effort.value = defaultEffort(provider);
     }
+    if (fast) {
+        if (typeof memory?.fast === "boolean") {
+            fast.value = memory.fast;
+        }
+        else if (opts.providerChanged) {
+            fast.value = defaultFast(provider);
+        }
+        fast.hidden = !isFastAvailable(provider, model.value);
+    }
+    if (context) {
+        const options = providerContexts(provider);
+        context.options = [...options];
+        if (options.length === 0) {
+            context.hidden = true;
+        }
+        else {
+            context.hidden = false;
+            if (memory?.context && isKnownContext(provider, memory.context)) {
+                context.value = memory.context;
+            }
+            else if (opts.providerChanged ||
+                !isKnownContext(provider, context.value)) {
+                context.value = defaultContext(provider) ?? "default";
+            }
+            context.expandedIndex = Math.max(0, selectOptionIndex(context, context.value));
+        }
+    }
     model.expandedIndex = Math.max(0, selectOptionIndex(model, model.value));
     effort.expandedIndex = Math.max(0, selectOptionIndex(effort, effort.value));
+    ensureGroupFocusVisible(group);
 }
-function reconcileEffortForCurrentModel(group) {
-    const provider = group.children[0].value;
-    const model = group.children[1];
-    const effort = group.children[2];
+export function reconcileModelEffortForGroup(group, driverMemory) {
+    reconcileProviderSettingsForGroup(group, driverMemory);
+}
+function reconcileSettingsForCurrentModel(group) {
+    const provider = providerChild(group).value;
+    const model = modelChild(group);
+    const effort = effortChild(group);
     effort.options = [...providerEfforts(provider, model.value)];
     if (!isKnownEffort(provider, effort.value, model.value)) {
         effort.value = defaultEffort(provider);
     }
     effort.expandedIndex = Math.max(0, selectOptionIndex(effort, effort.value));
+    const fast = fastChild(group);
+    if (fast)
+        fast.hidden = !isFastAvailable(provider, model.value);
+    ensureGroupFocusVisible(group);
 }
 /**
  * Mutate `field.hidden` in place based on each template param's
@@ -390,8 +587,8 @@ export function configRowCount(fields) {
             rows += f.options.length;
         }
         else if (f.kind === "group" && f.groupExpanded) {
-            rows += f.children.length;
-            const expandedChild = f.children.find((c) => c.expanded);
+            rows += f.children.filter((child) => !child.hidden).length;
+            const expandedChild = f.children.find((c) => c.kind === "select" && !c.hidden && c.expanded);
             if (expandedChild)
                 rows += expandedChild.options.length;
         }
@@ -402,12 +599,33 @@ export function extractFlowValues(fields, template) {
     const values = {};
     for (const param of template.params) {
         const field = findFieldByKey(fields, param.key);
+        if (param.type === "context" && !isApplicableContextField(fields, param)) {
+            continue;
+        }
         const value = field?.value ?? defaultParamValue(param);
         if (value !== undefined)
             values[param.key] = value;
     }
     return values;
 }
+function isApplicableContextField(fields, param) {
+    const group = fields.find((field) => field.kind === "group" &&
+        field.contextKey === param.key);
+    if (!group)
+        return false;
+    const provider = providerChild(group).value;
+    return defaultContext(provider) !== undefined;
+}
+function isProviderConfigurationParam(param) {
+    return param.type === "driver" ||
+        param.type === "model" ||
+        param.type === "effort" ||
+        param.type === "fast" ||
+        param.type === "context";
+}
+function isDriverName(value) {
+    return value === "claude" || value === "codex" || value === "copilot";
+}
 export function diagramWidth(flow, termCols, detailed) {
     const minW = detailed ? computeDetailedMinWidth(flow.steps) : computeMinWidth(flow.steps);
     return Math.min(minW + DIAGRAM_PADDING, termCols - 4);
@@ -518,62 +736,95 @@ function handleExpandedSelectEvent(field, event) {
     }
 }
 function handleGroupFieldEvent(group, event, driverMemory) {
-    const beforeProvider = group.children[0].value;
-    const beforeModel = group.children[1].value;
+    ensureGroupFocusVisible(group);
+    const provider = providerChild(group);
+    const model = modelChild(group);
+    const effort = effortChild(group);
+    const beforeProvider = provider.value;
+    const beforeModel = model.value;
     let consumed = false;
     if (!group.groupExpanded) {
+        const isArrow = event.type === "arrow";
+        const ctrl = isArrow && event.ctrl === true;
+        const alt = isArrow && event.alt === true;
         if (event.type === "enter") {
             group.groupExpanded = true;
-            group.focusedChild = 0;
+            group.focusedChild = firstVisibleGroupChildIndex(group);
+            consumed = true;
+        }
+        else if (isArrow
+            && ctrl
+            && (event.direction === "left" || event.direction === "right")) {
+            clampSelectValue(effort, event.direction === "left" ? -1 : 1);
+            consumed = true;
+        }
+        else if (isArrow
+            && ctrl
+            && (event.direction === "up" || event.direction === "down")) {
+            cycleSelectValue(model, event.direction === "up" ? -1 : 1);
             consumed = true;
         }
-        else if (event.type === "arrow"
-            && event.ctrl
+        else if (isArrow
+            && alt
             && (event.direction === "left" || event.direction === "right")) {
-            clampSelectValue(group.children[2], event.direction === "left" ? -1 : 1);
+            const context = contextChild(group);
+            if (context && !context.hidden) {
+                cycleSelectValue(context, event.direction === "left" ? -1 : 1);
+            }
             consumed = true;
         }
-        else if (event.type === "arrow"
-            && event.ctrl
+        else if (isArrow
+            && alt
             && (event.direction === "up" || event.direction === "down")) {
-            cycleSelectValue(group.children[1], event.direction === "up" ? -1 : 1);
+            const fast = fastChild(group);
+            if (fast && !fast.hidden) {
+                fast.value = !fast.value;
+            }
             consumed = true;
         }
-        else if (event.type === "arrow"
+        else if (isArrow
+            && !ctrl
+            && !alt
             && (event.direction === "left" || event.direction === "right")) {
-            cycleSelectValue(group.children[0], event.direction === "left" ? -1 : 1);
+            cycleSelectValue(provider, event.direction === "left" ? -1 : 1);
             consumed = true;
         }
     }
     else {
         const child = group.children[group.focusedChild];
-        if (child.expanded) {
+        if (child?.kind === "select" && child.expanded) {
             consumed = handleExpandedSelectEvent(child, event);
         }
         else {
             switch (event.type) {
                 case "arrow":
                     if (event.direction === "up") {
-                        group.focusedChild = (group.focusedChild + group.children.length - 1) %
-                            group.children.length;
+                        group.focusedChild = nextVisibleGroupChildIndex(group, -1);
                     }
                     else if (event.direction === "down") {
-                        group.focusedChild = (group.focusedChild + 1) % group.children.length;
+                        group.focusedChild = nextVisibleGroupChildIndex(group, 1);
                     }
-                    else {
-                        cycleSelectValue(child, event.direction === "left" ? -1 : 1);
+                    else if (child) {
+                        handleFieldAdjust(child, event.direction === "left" ? -1 : 1);
                     }
                     consumed = true;
                     break;
                 case "enter":
-                    child.expanded = true;
-                    child.expandedIndex = Math.max(0, selectOptionIndex(child, child.value));
+                    if (child?.kind === "select") {
+                        child.expanded = true;
+                        child.expandedIndex = Math.max(0, selectOptionIndex(child, child.value));
+                    }
+                    else if (child?.kind === "toggle") {
+                        child.value = !child.value;
+                    }
                     consumed = true;
                     break;
                 case "escape":
                     group.groupExpanded = false;
-                    for (const c of group.children)
-                        c.expanded = false;
+                    for (const c of group.children) {
+                        if (c.kind === "select")
+                            c.expanded = false;
+                    }
                     consumed = true;
                     break;
                 case "tab":
@@ -582,13 +833,16 @@ function handleGroupFieldEvent(group, event, driverMemory) {
             }
         }
     }
-    if (group.children[0].value !== beforeProvider) {
-        reconcileModelEffortForGroup(group, driverMemory);
+    if (provider.value !== beforeProvider) {
+        if (driverMemory && isDriverName(beforeProvider)) {
+            snapshotGroupToDriverMemory(driverMemory, group, beforeProvider);
+        }
+        reconcileProviderSettingsForGroup(group, driverMemory, { providerChanged: true });
     }
-    else if (group.children[1].value !== beforeModel) {
-        reconcileEffortForCurrentModel(group);
+    else if (model.value !== beforeModel) {
+        reconcileSettingsForCurrentModel(group);
     }
-    group.value = group.children[0].value;
+    group.value = provider.value;
     return consumed;
 }
 function handleNumberEditingEvent(field, event) {
@@ -706,9 +960,20 @@ export function renderConfigFields(fields, focusedIndex, isConfigFocused, cols)
                 : `${dim}off${reset}`;
         }
         else {
-            const [provider, model, effort] = field.children;
+            const provider = providerChild(field);
+            const model = modelChild(field);
+            const effort = effortChild(field);
+            const fast = fastChild(field);
+            const context = contextChild(field);
+            const summary = compactProviderConfigurationTokens({
+                driver: provider.value,
+                model: model.value,
+                effort: effort.value,
+                fast: fast?.value,
+                context: context && !context.hidden ? context.value : undefined,
+            }).map(renderCompactProviderConfigurationToken).join(` ${dim}/${reset} `);
             const marker = field.groupExpanded ? `${bold}▴${reset}` : `${dim}▾${reset}`;
-            valueStr = `${provider.value} ${dim}/${reset} ${model.value} ${dim}/${reset} ${formatEffortValue(effort.value)} ${marker}`;
+            valueStr = `${summary} ${marker}`;
         }
         lines.push(padToWidth(`  ${indicator} ${dimWrap}${labelStr}${dimEnd}${valueStr}`, cols));
         // Expanded dropdown options
@@ -723,17 +988,27 @@ export function renderConfigFields(fields, focusedIndex, isConfigFocused, cols)
         else if (field.kind === "group" && field.groupExpanded) {
             for (let childIdx = 0; childIdx < field.children.length; childIdx++) {
                 const child = field.children[childIdx];
+                if (child.hidden)
+                    continue;
                 const childFocused = focused && field.focusedChild === childIdx;
                 const childIndicator = childFocused ? `${bold}▸${reset}` : " ";
                 const childLabel = child.label.padEnd(LABEL_WIDTH - 2);
-                const childDisplayValue = child.key === field.effortKey
-                    ? formatEffortValue(child.value)
-                    : child.value;
-                const childValue = child.expanded
-                    ? `${childDisplayValue} ${bold}▴${reset}`
-                    : `${childDisplayValue} ${dim}▾${reset}`;
+                let childValue;
+                if (child.kind === "toggle") {
+                    childValue = child.value
+                        ? `${green}on${reset}`
+                        : `${dim}off${reset}`;
+                }
+                else {
+                    const childDisplayValue = child.key === field.effortKey
+                        ? formatEffortValue(child.value)
+                        : child.value;
+                    childValue = child.expanded
+                        ? `${childDisplayValue} ${bold}▴${reset}`
+                        : `${childDisplayValue} ${dim}▾${reset}`;
+                }
                 lines.push(padToWidth(`    ${childIndicator} ${dimWrap}${childLabel}${dimEnd}${childValue}`, cols));
-                if (child.expanded) {
+                if (child.kind === "select" && child.expanded) {
                     for (let j = 0; j < child.options.length; j++) {
                         const opt = child.options[j];
                         const optIndicator = j === child.expandedIndex ? `${bold}▸${reset}` : " ";