@osolmaz/pi-workflows 0.9.0 → 0.10.0

package/dist/workflows/composition.js

@@ -0,0 +1,471 @@
+import { createHash } from "node:crypto";
+import { readFileSync } from "node:fs";
+import { fileURLToPath } from "node:url";
+const COMPILED_WORKFLOW_BRAND = Symbol.for("pi-workflows.compiled-definition");
+const COMPOSITION_METADATA = Symbol.for("pi-workflows.composition-metadata");
+export function isCompiledWorkflow(workflow) {
+    return workflow[COMPILED_WORKFLOW_BRAND] === true;
+}
+export function compositionMetadata(workflow) {
+    return workflow[COMPOSITION_METADATA];
+}
+/**
+ * Compile direct includes into one qualified graph. Dynamic string references
+ * must be resolved to direct definitions before this function is called.
+ */
+export function compileWorkflowDefinition(workflow, options = {}) {
+    if (isCompiledWorkflow(workflow))
+        return workflow;
+    validateAuthoredWorkflow(workflow);
+    const nodes = {};
+    const edges = [];
+    const mounts = [];
+    const scopes = {};
+    const entries = {};
+    const exits = {};
+    const sources = [];
+    const activeDefinitions = [];
+    const mountWorkflow = (current, scopePath, parentPath, mountName) => {
+        const cycleAt = activeDefinitions.indexOf(current);
+        if (cycleAt >= 0) {
+            const chain = [...activeDefinitions.slice(cycleAt).map((item) => item.name), current.name];
+            throw new Error(`Workflow include cycle: ${chain.join(" -> ")}`);
+        }
+        activeDefinitions.push(current);
+        validateAuthoredWorkflow(current, scopePath !== "");
+        const authoredNodes = Object.fromEntries(Object.keys(current.nodes).map((nodeId) => [nodeId, qualify(scopePath, nodeId)]));
+        const childMounts = Object.fromEntries(Object.keys(current.includes ?? {}).map((name) => [name, qualify(scopePath, name)]));
+        scopes[scopePath] = {
+            path: scopePath,
+            workflowName: current.name,
+            ...(current.maxSteps !== undefined ? { maxSteps: current.maxSteps } : {}),
+            authoredNodes,
+            childMounts,
+        };
+        if (scopePath !== "" && parentPath !== null && mountName !== null) {
+            const source = options.sourceMap?.get(current) ?? sourceFromDefinition(current);
+            if (source !== undefined) {
+                sources.push({
+                    mountPath: scopePath.split("/"),
+                    workflowName: current.name,
+                    source,
+                });
+            }
+        }
+        for (const [localNodeId, node] of Object.entries(current.nodes)) {
+            const qualifiedNodeId = qualify(scopePath, localNodeId);
+            nodes[qualifiedNodeId] = wrapNode(node, scopePath, scopes, entries, exits);
+        }
+        for (const [name, include] of Object.entries(current.includes ?? {})) {
+            const child = typeof include.workflow === "string" ? include.contract : include.workflow;
+            if (child === undefined) {
+                throw new Error(`Workflow include ${qualify(scopePath, name)} is unresolved: ${include.workflow}`);
+            }
+            const mountPath = qualify(scopePath, name);
+            const childExits = child.exits ?? {};
+            if (Object.keys(childExits).length === 0) {
+                throw new Error(`Included workflow ${child.name} must declare at least one exit`);
+            }
+            const entryNode = mountPath;
+            entries[entryNode] = {
+                mountPath,
+                mountName: name,
+                parentPath: scopePath,
+                workflowName: child.name,
+                input: include.input,
+                normalizeInput: child.input,
+            };
+            nodes[entryNode] = {
+                nodeType: "compute",
+                statusDetail: `entering ${child.name}`,
+                run: async (context) => {
+                    const parentContext = projectWorkflowContext(context, scopePath, scopes, entries, exits);
+                    const mapped = include.input ? await include.input(parentContext) : parentContext.input;
+                    const normalized = child.input ? await child.input(mapped) : mapped;
+                    return {
+                        schema: "pi-workflows.include-entry.v1",
+                        invocation: countCompletedEntries(context.state, entryNode) + 1,
+                        input: normalized === undefined ? null : normalized,
+                    };
+                },
+            };
+            edges.push({ from: entryNode, to: qualify(mountPath, child.startAt) });
+            const exitNodes = {};
+            for (const [exitName, exit] of Object.entries(childExits)) {
+                const exitNodeId = qualify(mountPath, `__piw_exit_${exitName}`);
+                const fromNode = qualify(mountPath, exit.from);
+                exitNodes[exitName] = exitNodeId;
+                exits[exitNodeId] = {
+                    nodeId: exitNodeId,
+                    mountPath,
+                    mountName: name,
+                    parentPath: scopePath,
+                    exitName,
+                    fromNode,
+                    validate: exit.validate,
+                };
+                nodes[exitNodeId] = {
+                    nodeType: "compute",
+                    statusDetail: `leaving ${child.name} through ${exitName}`,
+                    run: async (context) => {
+                        const childContext = projectWorkflowContext(context, mountPath, scopes, entries, exits);
+                        const raw = childContext.outputs[exit.from];
+                        const output = exit.validate ? await exit.validate(raw) : raw;
+                        return { exit: exitName, output: output === undefined ? null : output };
+                    },
+                };
+                edges.push({ from: fromNode, to: exitNodeId });
+            }
+            mounts.push({
+                mountPath: mountPath.split("/"),
+                workflowName: child.name,
+                entryNode,
+                exits: exitNodes,
+                ...(child.maxSteps !== undefined ? { maxSteps: child.maxSteps } : {}),
+            });
+            mountWorkflow(child, mountPath, scopePath, name);
+        }
+        for (const edge of current.edges) {
+            const from = resolveAuthoredFrom(scopePath, current, edge.from, exits);
+            if ("to" in edge) {
+                edges.push({ from, to: resolveAuthoredTo(scopePath, current, edge.to) });
+            }
+            else {
+                edges.push({
+                    from,
+                    switch: {
+                        on: edge.switch.on,
+                        cases: Object.fromEntries(Object.entries(edge.switch.cases).map(([key, target]) => [
+                            key,
+                            resolveAuthoredTo(scopePath, current, target),
+                        ])),
+                    },
+                });
+            }
+        }
+        activeDefinitions.pop();
+    };
+    mountWorkflow(workflow, "", null, null);
+    const metadata = {
+        snapshot: { mounts: mounts.sort(compareMounts) },
+        scopes,
+        entries,
+        exits,
+        sources: sources.sort(compareSources),
+    };
+    const compiled = {
+        ...workflow,
+        nodes,
+        edges,
+    };
+    delete compiled.includes;
+    delete compiled.exits;
+    for (const symbol of Object.getOwnPropertySymbols(workflow)) {
+        const descriptor = Object.getOwnPropertyDescriptor(workflow, symbol);
+        if (descriptor !== undefined)
+            Object.defineProperty(compiled, symbol, descriptor);
+    }
+    Object.defineProperty(compiled, COMPILED_WORKFLOW_BRAND, {
+        value: true,
+        enumerable: false,
+    });
+    Object.defineProperty(compiled, COMPOSITION_METADATA, {
+        value: metadata,
+        enumerable: false,
+    });
+    return compiled;
+}
+function validateAuthoredWorkflow(workflow, included = false) {
+    if (!Object.hasOwn(workflow.nodes, workflow.startAt)) {
+        throw new Error(`Workflow start node is missing: ${workflow.startAt}`);
+    }
+    const includes = workflow.includes ?? {};
+    const exits = workflow.exits ?? {};
+    const outgoing = new Set();
+    for (const edge of workflow.edges) {
+        if (outgoing.has(edge.from)) {
+            throw new Error(`Workflow node must not declare multiple outgoing edges: ${edge.from}`);
+        }
+        outgoing.add(edge.from);
+        assertAuthoredFrom(workflow, edge.from);
+        const targets = "to" in edge ? [edge.to] : Object.values(edge.switch.cases);
+        for (const target of targets)
+            assertAuthoredTo(workflow, target);
+    }
+    const exitNodes = new Set();
+    for (const [exitName, exit] of Object.entries(exits)) {
+        if (!Object.hasOwn(workflow.nodes, exit.from)) {
+            throw new Error(`Workflow exit ${exitName} references unknown node: ${exit.from}`);
+        }
+        if (outgoing.has(exit.from)) {
+            throw new Error(`Workflow exit ${exitName} must reference a terminal node: ${exit.from}`);
+        }
+        if (exitNodes.has(exit.from)) {
+            throw new Error(`Workflow terminal node has more than one exit: ${exit.from}`);
+        }
+        exitNodes.add(exit.from);
+    }
+    if (included) {
+        const undeclared = Object.keys(workflow.nodes).filter((nodeId) => !outgoing.has(nodeId) && !exitNodes.has(nodeId));
+        if (undeclared.length > 0) {
+            throw new Error(`Included workflow ${workflow.name} has terminal nodes without named exits: ${undeclared.join(", ")}`);
+        }
+    }
+    const reachable = new Set([workflow.startAt]);
+    const queue = [workflow.startAt];
+    while (queue.length > 0) {
+        const from = queue.shift();
+        const edge = workflow.edges.find((candidate) => candidate.from === from);
+        if (edge !== undefined) {
+            const targets = "to" in edge ? [edge.to] : Object.values(edge.switch.cases);
+            for (const target of targets)
+                addReachable(target, reachable, queue);
+        }
+        if (Object.hasOwn(includes, from)) {
+            const include = includes[from];
+            const child = include && typeof include.workflow !== "string" ? include.workflow : include?.contract;
+            for (const exitName of Object.keys(child?.exits ?? {})) {
+                addReachable(`${from}.${exitName}`, reachable, queue);
+            }
+        }
+    }
+    const unreachable = [...Object.keys(workflow.nodes), ...Object.keys(includes)].filter((id) => !reachable.has(id));
+    if (unreachable.length > 0) {
+        throw new Error(`Workflow has unreachable nodes: ${unreachable.join(", ")}`);
+    }
+}
+function assertAuthoredFrom(workflow, from) {
+    if (Object.hasOwn(workflow.nodes, from))
+        return;
+    const parsed = parseExitReference(from);
+    const include = parsed ? workflow.includes?.[parsed.mount] : undefined;
+    const child = include && typeof include.workflow !== "string" ? include.workflow : include?.contract;
+    if (parsed && child?.exits && Object.hasOwn(child.exits, parsed.exit))
+        return;
+    throw new Error(`Workflow edge references unknown from-node or include exit: ${from}`);
+}
+function assertAuthoredTo(workflow, to) {
+    if (Object.hasOwn(workflow.nodes, to) || Object.hasOwn(workflow.includes ?? {}, to))
+        return;
+    throw new Error(`Workflow edge references unknown to-node: ${to}`);
+}
+function resolveAuthoredFrom(scopePath, workflow, from, exitMetadata) {
+    if (Object.hasOwn(workflow.nodes, from))
+        return qualify(scopePath, from);
+    const parsed = parseExitReference(from);
+    if (!parsed)
+        throw new Error(`Invalid include exit reference: ${from}`);
+    const mountPath = qualify(scopePath, parsed.mount);
+    const match = Object.values(exitMetadata).find((exit) => exit.mountPath === mountPath && exit.exitName === parsed.exit);
+    if (!match)
+        throw new Error(`Unknown include exit reference: ${from}`);
+    return match.nodeId;
+}
+function resolveAuthoredTo(scopePath, workflow, target) {
+    if (Object.hasOwn(workflow.nodes, target) || Object.hasOwn(workflow.includes ?? {}, target)) {
+        return qualify(scopePath, target);
+    }
+    throw new Error(`Unknown workflow target: ${target}`);
+}
+function parseExitReference(value) {
+    const dot = value.indexOf(".");
+    if (dot <= 0 || dot === value.length - 1 || value.indexOf(".", dot + 1) >= 0)
+        return undefined;
+    return { mount: value.slice(0, dot), exit: value.slice(dot + 1) };
+}
+function addReachable(value, reachable, queue) {
+    if (reachable.has(value))
+        return;
+    reachable.add(value);
+    queue.push(value);
+}
+function qualify(scopePath, localId) {
+    return scopePath === "" ? localId : `${scopePath}/${localId}`;
+}
+function compareMounts(a, b) {
+    return a.mountPath.join("/").localeCompare(b.mountPath.join("/"));
+}
+function compareSources(a, b) {
+    return a.mountPath.join("/").localeCompare(b.mountPath.join("/"));
+}
+function sourceFromDefinition(workflow) {
+    if (workflow.source === undefined)
+        return undefined;
+    let filePath;
+    try {
+        filePath = fileURLToPath(workflow.source);
+    }
+    catch {
+        throw new Error(`Workflow ${workflow.name} source must be a file URL: ${workflow.source}`);
+    }
+    const hash = createHash("sha256").update(readFileSync(filePath)).digest("hex");
+    return { kind: "file", path: filePath, hash };
+}
+function wrapNode(node, scopePath, scopes, entries, exits) {
+    if (scopePath === "" && Object.keys(scopes[scopePath]?.childMounts ?? {}).length === 0) {
+        return node;
+    }
+    const project = (context) => projectWorkflowContext(context, scopePath, scopes, entries, exits);
+    const timeoutResolver = node.timeoutMs;
+    const timeoutMs = typeof timeoutResolver === "function"
+        ? (context) => timeoutResolver(project(context))
+        : timeoutResolver;
+    const common = {
+        ...(timeoutMs !== undefined ? { timeoutMs } : {}),
+        ...(node.statusDetail !== undefined ? { statusDetail: node.statusDetail } : {}),
+    };
+    switch (node.nodeType) {
+        case "agent":
+            return {
+                ...common,
+                nodeType: "agent",
+                prompt: (context) => node.prompt(project(context)),
+                ...(node.expectedOutput !== undefined ? { expectedOutput: node.expectedOutput } : {}),
+                ...(node.validate !== undefined
+                    ? {
+                        validate: (output, context) => node.validate?.(output, project(context)),
+                    }
+                    : {}),
+            };
+        case "compute":
+            return {
+                ...common,
+                nodeType: "compute",
+                run: (context) => node.run(project(context)),
+            };
+        case "notify":
+            return {
+                ...common,
+                nodeType: "notify",
+                message: (context) => node.message(project(context)),
+                ...(node.kind !== undefined ? { kind: node.kind } : {}),
+            };
+        case "checkpoint":
+            return {
+                ...common,
+                nodeType: "checkpoint",
+                ...(node.summary !== undefined ? { summary: node.summary } : {}),
+                ...(node.run !== undefined
+                    ? { run: (context) => node.run?.(project(context)) }
+                    : {}),
+            };
+        case "action":
+            if ("exec" in node) {
+                const shell = node;
+                return {
+                    ...common,
+                    nodeType: "action",
+                    exec: (context) => shell.exec(project(context)),
+                    ...(shell.parse !== undefined
+                        ? {
+                            parse: (result, context) => shell.parse?.(result, project(context)),
+                        }
+                        : {}),
+                    ...(shell.updates !== undefined
+                        ? {
+                            updates: {
+                                ...(shell.updates.streams !== undefined
+                                    ? { streams: [...shell.updates.streams] }
+                                    : {}),
+                                parseLine: (line, context) => shell.updates?.parseLine(line, projectActionContext(context, scopePath, scopes, entries, exits)),
+                            },
+                        }
+                        : {}),
+                };
+            }
+            return {
+                ...common,
+                nodeType: "action",
+                run: (context) => node.run(projectActionContext(context, scopePath, scopes, entries, exits)),
+            };
+    }
+}
+function projectActionContext(context, scopePath, scopes, entries, exits) {
+    return {
+        ...projectWorkflowContext(context, scopePath, scopes, entries, exits),
+        publishUpdate: context.publishUpdate,
+    };
+}
+export function projectWorkflowContext(context, scopePath, scopes, _entries, exits) {
+    const scope = scopes[scopePath];
+    if (!scope)
+        throw new Error(`Workflow scope is missing: ${scopePath || "root"}`);
+    const entryIndex = scopePath === "" ? -1 : latestStepIndex(context.state.steps, scopePath);
+    if (scopePath !== "" && entryIndex < 0) {
+        throw new Error(`Workflow include entry is missing: ${scopePath}`);
+    }
+    const entry = scopePath === ""
+        ? undefined
+        : context.state.steps[entryIndex]?.output;
+    const outputs = {};
+    const results = {};
+    for (const [local, qualified] of Object.entries(scope.authoredNodes)) {
+        if (latestStepIndex(context.state.steps, qualified) <= entryIndex)
+            continue;
+        if (Object.hasOwn(context.state.outputs, qualified))
+            outputs[local] = context.state.outputs[qualified];
+        const result = context.state.results[qualified];
+        if (result !== undefined)
+            results[local] = result;
+    }
+    for (const [local, mountPath] of Object.entries(scope.childMounts)) {
+        const latestExit = Object.values(exits)
+            .filter((candidate) => candidate.mountPath === mountPath)
+            .map((candidate) => latestStepIndex(context.state.steps, candidate.nodeId))
+            .reduce((max, value) => Math.max(max, value), -1);
+        if (latestExit <= entryIndex)
+            continue;
+        if (Object.hasOwn(context.state.outputs, mountPath))
+            outputs[local] = context.state.outputs[mountPath];
+        const result = context.state.results[mountPath];
+        if (result !== undefined)
+            results[local] = result;
+    }
+    const localSteps = context.state.steps.slice(entryIndex + 1).flatMap((step) => {
+        if (Object.values(scope.authoredNodes).includes(step.nodeId)) {
+            return [{ ...step, nodeId: localNameFor(scope, step.nodeId) }];
+        }
+        const exit = exits[step.nodeId];
+        if (exit?.parentPath === scopePath) {
+            return [{ ...step, nodeId: exit.mountName }];
+        }
+        return [];
+    });
+    const currentNode = context.state.currentNode
+        ? localNameFor(scope, context.state.currentNode)
+        : undefined;
+    const localInput = scopePath === "" ? context.input : (entry?.input ?? null);
+    const state = {
+        ...context.state,
+        input: localInput,
+        outputs,
+        results,
+        steps: localSteps,
+        ...(currentNode !== undefined ? { currentNode } : {}),
+    };
+    return {
+        input: localInput,
+        outputs,
+        results,
+        state,
+        signal: context.signal,
+    };
+}
+function localNameFor(scope, qualified) {
+    for (const [local, global] of Object.entries(scope.authoredNodes)) {
+        if (global === qualified)
+            return local;
+    }
+    return qualified;
+}
+function latestStepIndex(steps, nodeId) {
+    for (let index = steps.length - 1; index >= 0; index -= 1) {
+        if (steps[index]?.nodeId === nodeId)
+            return index;
+    }
+    return -1;
+}
+function countCompletedEntries(state, entryNode) {
+    return state.steps.filter((step) => step.nodeId === entryNode && step.outcome === "ok").length;
+}
+//# sourceMappingURL=composition.js.map

package/dist/workflows/composition.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"composition.js","sourceRoot":"","sources":["../../src/workflows/composition.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,aAAa,EAAE,MAAM,UAAU,CAAC;AAyBzC,MAAM,uBAAuB,GAAG,MAAM,CAAC,GAAG,CAAC,kCAAkC,CAAC,CAAC;AAC/E,MAAM,oBAAoB,GAAG,MAAM,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;AA4C7E,MAAM,UAAU,kBAAkB,CAAC,QAA2C;IAC5E,OAAQ,QAAyC,CAAC,uBAAuB,CAAC,KAAK,IAAI,CAAC;AACtF,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,QAA2C;IAE3C,OAAQ,QAAyC,CAAC,oBAAoB,CAEzD,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,yBAAyB,CACvC,QAAmB,EACnB,OAAO,GAA2B,EAAE;IAEpC,IAAI,kBAAkB,CAAC,QAAQ,CAAC;QAAE,OAAO,QAAQ,CAAC;IAElD,wBAAwB,CAAC,QAAQ,CAAC,CAAC;IACnC,MAAM,KAAK,GAA2C,EAAE,CAAC;IACzD,MAAM,KAAK,GAAmB,EAAE,CAAC;IACjC,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,MAAM,MAAM,GAA0C,EAAE,CAAC;IACzD,MAAM,OAAO,GAA0C,EAAE,CAAC;IAC1D,MAAM,KAAK,GAAyC,EAAE,CAAC;IACvD,MAAM,OAAO,GAA4B,EAAE,CAAC;IAC5C,MAAM,iBAAiB,GAAwC,EAAE,CAAC;IAElE,MAAM,aAAa,GAAG,CACpB,OAA0C,EAC1C,SAAiB,EACjB,UAAyB,EACzB,SAAwB,EAClB,EAAE;QACR,MAAM,OAAO,GAAG,iBAAiB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACnD,IAAI,OAAO,IAAI,CAAC,EAAE,CAAC;YACjB,MAAM,KAAK,GAAG,CAAC,GAAG,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;YAC3F,MAAM,IAAI,KAAK,CAAC,2BAA2B,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACnE,CAAC;QACD,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChC,wBAAwB,CAAC,OAAO,EAAE,SAAS,KAAK,EAAE,CAAC,CAAC;QAEpD,MAAM,aAAa,GAAG,MAAM,CAAC,WAAW,CACtC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,CACjF,CAAC;QACF,MAAM,WAAW,GAAG,MAAM,CAAC,WAAW,CACpC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,CACpF,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,GAAG;YAClB,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,OAAO,CAAC,IAAI;YAC1B,GAAG,CAAC,OAAO,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,aAAa;YACb,WAAW;SACZ,CAAC;QAEF,IAAI,SAAS,KAAK,EAAE,IAAI,UAAU,KAAK,IAAI,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,IAAI,oBAAoB,CAAC,OAAO,CAAC,CAAC;YAChF,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;gBACzB,OAAO,CAAC,IAAI,CAAC;oBACX,SAAS,EAAE,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC;oBAC/B,YAAY,EAAE,OAAO,CAAC,IAAI;oBAC1B,MAAM;iBACP,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,KAAK,MAAM,CAAC,WAAW,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAChE,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;YACxD,KAAK,CAAC,eAAe,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAC7E,CAAC;QAED,KAAK,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,IAAI,EAAE,CAGhE,EAAE,CAAC;YACJ,MAAM,KAAK,GAAG,OAAO,OAAO,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC;YACzF,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CACb,oBAAoB,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,mBAAmB,OAAO,CAAC,QAAQ,EAAE,CAClF,CAAC;YACJ,CAAC;YACD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YAC3C,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;YACrC,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,qBAAqB,KAAK,CAAC,IAAI,iCAAiC,CAAC,CAAC;YACpF,CAAC;YACD,MAAM,SAAS,GAAG,SAAS,CAAC;YAC5B,OAAO,CAAC,SAAS,CAAC,GAAG;gBACnB,SAAS;gBACT,SAAS,EAAE,IAAI;gBACf,UAAU,EAAE,SAAS;gBACrB,YAAY,EAAE,KAAK,CAAC,IAAI;gBACxB,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,cAAc,EAAE,KAAK,CAAC,KAAK;aAC5B,CAAC;YACF,KAAK,CAAC,SAAS,CAAC,GAAG;gBACjB,QAAQ,EAAE,SAAS;gBACnB,YAAY,EAAE,YAAY,KAAK,CAAC,IAAI,EAAE;gBACtC,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;oBACrB,MAAM,aAAa,GAAG,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;oBACzF,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC;oBACxF,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;oBACpE,OAAO;wBACL,MAAM,EAAE,+BAA+B;wBACvC,UAAU,EAAE,qBAAqB,CAAC,OAAO,CAAC,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC;wBAC/D,KAAK,EAAE,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU;qBACpD,CAAC;gBACJ,CAAC;aACF,CAAC;YACF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YAEvE,MAAM,SAAS,GAA2B,EAAE,CAAC;YAC7C,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAGrD,EAAE,CAAC;gBACJ,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,EAAE,cAAc,QAAQ,EAAE,CAAC,CAAC;gBAChE,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC/C,SAAS,CAAC,QAAQ,CAAC,GAAG,UAAU,CAAC;gBACjC,KAAK,CAAC,UAAU,CAAC,GAAG;oBAClB,MAAM,EAAE,UAAU;oBAClB,SAAS;oBACT,SAAS,EAAE,IAAI;oBACf,UAAU,EAAE,SAAS;oBACrB,QAAQ;oBACR,QAAQ;oBACR,QAAQ,EAAE,IAAI,CAAC,QAAQ;iBACxB,CAAC;gBACF,KAAK,CAAC,UAAU,CAAC,GAAG;oBAClB,QAAQ,EAAE,SAAS;oBACnB,YAAY,EAAE,WAAW,KAAK,CAAC,IAAI,YAAY,QAAQ,EAAE;oBACzD,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;wBACrB,MAAM,YAAY,GAAG,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;wBACxF,MAAM,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;wBAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;wBAC9D,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;oBAC1E,CAAC;iBACF,CAAC;gBACF,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,CAAC,IAAI,CAAC;gBACV,SAAS,EAAE,SAAS,CAAC,KAAK,CAAC,GAAG,CAAC;gBAC/B,YAAY,EAAE,KAAK,CAAC,IAAI;gBACxB,SAAS;gBACT,KAAK,EAAE,SAAS;gBAChB,GAAG,CAAC,KAAK,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtE,CAAC,CAAC;YACH,aAAa,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACnD,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;YACjC,MAAM,IAAI,GAAG,mBAAmB,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACvE,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;gBACjB,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,iBAAiB,CAAC,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;YAC3E,CAAC;iBAAM,CAAC;gBACN,KAAK,CAAC,IAAI,CAAC;oBACT,IAAI;oBACJ,MAAM,EAAE;wBACN,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;wBAClB,KAAK,EAAE,MAAM,CAAC,WAAW,CACvB,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC;4BACvD,GAAG;4BACH,iBAAiB,CAAC,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC;yBAC9C,CAAC,CACH;qBACF;iBACF,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,iBAAiB,CAAC,GAAG,EAAE,CAAC;IAC1B,CAAC,CAAC;IAEF,aAAa,CAAC,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;IACxC,MAAM,QAAQ,GAAgC;QAC5C,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE;QAChD,MAAM;QACN,OAAO;QACP,KAAK;QACL,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC;KACtC,CAAC;IACF,MAAM,QAAQ,GAAG;QACf,GAAG,QAAQ;QACX,KAAK;QACL,KAAK;KACO,CAAC;IACf,OAAQ,QAAmC,CAAC,QAAQ,CAAC;IACrD,OAAQ,QAAgC,CAAC,KAAK,CAAC;IAC/C,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,qBAAqB,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC5D,MAAM,UAAU,GAAG,MAAM,CAAC,wBAAwB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QACrE,IAAI,UAAU,KAAK,SAAS;YAAE,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;IACpF,CAAC;IACD,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,uBAAuB,EAAE;QACvD,KAAK,EAAE,IAAI;QACX,UAAU,EAAE,KAAK;KAClB,CAAC,CAAC;IACH,MAAM,CAAC,cAAc,CAAC,QAAQ,EAAE,oBAAoB,EAAE;QACpD,KAAK,EAAE,QAAQ;QACf,UAAU,EAAE,KAAK;KAClB,CAAC,CAAC;IACH,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,wBAAwB,CAC/B,QAA2C,EAC3C,QAAQ,GAAG,KAAK;IAEhB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACrD,MAAM,IAAI,KAAK,CAAC,mCAAmC,QAAQ,CAAC,OAAO,EAAE,CAAC,CAAC;IACzE,CAAC;IACD,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,IAAI,EAAE,CAAC;IACzC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,IAAI,EAAE,CAAC;IACnC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;IACnC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;QAClC,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,2DAA2D,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1F,CAAC;QACD,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxB,kBAAkB,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,KAAK,MAAM,MAAM,IAAI,OAAO;YAAE,gBAAgB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;IACnE,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC;IACpC,KAAK,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAuC,EAAE,CAAC;QAC3F,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,iBAAiB,QAAQ,6BAA6B,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACrF,CAAC;QACD,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,iBAAiB,QAAQ,oCAAoC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC5F,CAAC;QACD,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,kDAAkD,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACjF,CAAC;QACD,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC;IACD,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,MAAM,CACnD,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAC5D,CAAC;QACF,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CACb,qBAAqB,QAAQ,CAAC,IAAI,4CAA4C,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACtG,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,GAAG,CAAS,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACjC,OAAO,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAY,CAAC;QACrC,MAAM,IAAI,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QACzE,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,OAAO,GAAG,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,KAAK,MAAM,MAAM,IAAI,OAAO;gBAAE,YAAY,CAAC,MAAM,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,KAAK,GACT,OAAO,IAAI,OAAO,OAAO,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC;YACzF,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC;gBACvD,YAAY,CAAC,GAAG,IAAI,IAAI,QAAQ,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;YACxD,CAAC;QACH,CAAC;IACH,CAAC;IACD,MAAM,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CACnF,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAC3B,CAAC;IACF,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,mCAAmC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC/E,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,QAA2C,EAAE,IAAY;IACnF,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC;QAAE,OAAO;IAChD,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACvE,MAAM,KAAK,GACT,OAAO,IAAI,OAAO,OAAO,CAAC,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,EAAE,QAAQ,CAAC;IACzF,IAAI,MAAM,IAAI,KAAK,EAAE,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC;QAAE,OAAO;IAC9E,MAAM,IAAI,KAAK,CAAC,+DAA+D,IAAI,EAAE,CAAC,CAAC;AACzF,CAAC;AAED,SAAS,gBAAgB,CAAC,QAA2C,EAAE,EAAU;IAC/E,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,EAAE,EAAE,CAAC;QAAE,OAAO;IAC5F,MAAM,IAAI,KAAK,CAAC,6CAA6C,EAAE,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,mBAAmB,CAC1B,SAAiB,EACjB,QAA2C,EAC3C,IAAY,EACZ,YAAkD;IAElD,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,IAAI,CAAC;QAAE,OAAO,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC;IACxC,IAAI,CAAC,MAAM;QAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,IAAI,EAAE,CAAC,CAAC;IACxE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;IACnD,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAC5C,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,CAAC,IAAI,CACxE,CAAC;IACF,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,mCAAmC,IAAI,EAAE,CAAC,CAAC;IACvE,OAAO,KAAK,CAAC,MAAM,CAAC;AACtB,CAAC;AAED,SAAS,iBAAiB,CACxB,SAAiB,EACjB,QAA2C,EAC3C,MAAc;IAEd,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,IAAI,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC;QAC5F,OAAO,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,4BAA4B,MAAM,EAAE,CAAC,CAAC;AACxD,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAa;IACvC,MAAM,GAAG,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,GAAG,IAAI,CAAC,IAAI,GAAG,KAAK,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC;QAAE,OAAO,SAAS,CAAC;IAC/F,OAAO,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC;AACpE,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,SAAsB,EAAE,KAAe;IAC1E,IAAI,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC;QAAE,OAAO;IACjC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACrB,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;AACpB,CAAC;AAED,SAAS,OAAO,CAAC,SAAiB,EAAE,OAAe;IACjD,OAAO,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,SAAS,IAAI,OAAO,EAAE,CAAC;AAChE,CAAC;AAED,SAAS,aAAa,CAAC,CAAwB,EAAE,CAAwB;IACvE,OAAO,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,cAAc,CAAC,CAAwB,EAAE,CAAwB;IACxE,OAAO,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,SAAS,oBAAoB,CAC3B,QAA2C;IAE3C,IAAI,QAAQ,CAAC,MAAM,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACpD,IAAI,QAAgB,CAAC;IACrB,IAAI,CAAC;QACH,QAAQ,GAAG,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,KAAK,CAAC,YAAY,QAAQ,CAAC,IAAI,+BAA+B,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAC7F,CAAC;IACD,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAC/E,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;AAChD,CAAC;AAED,SAAS,QAAQ,CACf,IAA4B,EAC5B,SAAiB,EACjB,MAA6C,EAC7C,OAA8C,EAC9C,KAA2C;IAE3C,IAAI,SAAS,KAAK,EAAE,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,WAAW,IAAI,EAAE,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvF,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,OAAO,GAAG,CAAC,OAA4B,EAAE,EAAE,CAC/C,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;IACrE,MAAM,eAAe,GAAG,IAAI,CAAC,SAAS,CAAC;IACvC,MAAM,SAAS,GACb,OAAO,eAAe,KAAK,UAAU;QACnC,CAAC,CAAC,CAAC,OAA4B,EAAE,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACrE,CAAC,CAAC,eAAe,CAAC;IACtB,MAAM,MAAM,GAAG;QACb,GAAG,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACjD,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAChF,CAAC;IACF,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;QACtB,KAAK,OAAO;YACV,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,OAAO;gBACjB,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,CAAE,IAA4B,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC3E,GAAG,CAAC,IAAI,CAAC,cAAc,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACrF,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,SAAS;oBAC7B,CAAC,CAAC;wBACE,QAAQ,EAAE,CAAC,MAAe,EAAE,OAA4B,EAAE,EAAE,CAC1D,IAAI,CAAC,QAAQ,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;qBAC5C;oBACH,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;QACJ,KAAK,SAAS;YACZ,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,SAAS;gBACnB,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,CAAE,IAA8B,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aACxE,CAAC;QACJ,KAAK,QAAQ;YACX,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE,CAAC,OAAO,EAAE,EAAE,CAAE,IAA6B,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC9E,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACxD,CAAC;QACJ,KAAK,YAAY;YACf,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,YAAY;gBACtB,GAAG,CAAC,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChE,GAAG,CAAC,IAAI,CAAC,GAAG,KAAK,SAAS;oBACxB,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,OAA4B,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE;oBACzE,CAAC,CAAC,EAAE,CAAC;aACoB,CAAC;QAChC,KAAK,QAAQ;YACX,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;gBACnB,MAAM,KAAK,GAAG,IAAiC,CAAC;gBAChD,OAAO;oBACL,GAAG,MAAM;oBACT,QAAQ,EAAE,QAAQ;oBAClB,IAAI,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;oBAC/C,GAAG,CAAC,KAAK,CAAC,KAAK,KAAK,SAAS;wBAC3B,CAAC,CAAC;4BACE,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,CAAC;yBACpE;wBACH,CAAC,CAAC,EAAE,CAAC;oBACP,GAAG,CAAC,KAAK,CAAC,OAAO,KAAK,SAAS;wBAC7B,CAAC,CAAC;4BACE,OAAO,EAAE;gCACP,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,KAAK,SAAS;oCACrC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;oCACzC,CAAC,CAAC,EAAE,CAAC;gCACP,SAAS,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAC3B,KAAK,CAAC,OAAO,EAAE,SAAS,CACtB,IAAI,EACJ,oBAAoB,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CACjE;6BACJ;yBACF;wBACH,CAAC,CAAC,EAAE,CAAC;iBACqB,CAAC;YACjC,CAAC;YACD,OAAO;gBACL,GAAG,MAAM;gBACT,QAAQ,EAAE,QAAQ;gBAClB,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,CACd,IAAqC,CAAC,GAAG,CACxC,oBAAoB,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC,CACjE;aACoB,CAAC;IAC9B,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAC3B,OAA8B,EAC9B,SAAiB,EACjB,MAA6C,EAC7C,OAA8C,EAC9C,KAA2C;IAE3C,OAAO;QACL,GAAG,sBAAsB,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,KAAK,CAAC;QACrE,aAAa,EAAE,OAAO,CAAC,aAAa;KACrC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,sBAAsB,CACpC,OAA4B,EAC5B,SAAiB,EACjB,MAA6C,EAC7C,QAA+C,EAC/C,KAA2C;IAE3C,MAAM,KAAK,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;IAChC,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,8BAA8B,SAAS,IAAI,MAAM,EAAE,CAAC,CAAC;IACjF,MAAM,UAAU,GAAG,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IAC3F,IAAI,SAAS,KAAK,EAAE,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;QACvC,MAAM,IAAI,KAAK,CAAC,sCAAsC,SAAS,EAAE,CAAC,CAAC;IACrE,CAAC;IACD,MAAM,KAAK,GACT,SAAS,KAAK,EAAE;QACd,CAAC,CAAC,SAAS;QACX,CAAC,CAAE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,MAEpB,CAAC;IACrB,MAAM,OAAO,GAA4B,EAAE,CAAC;IAC5C,MAAM,OAAO,GAAuC,EAAE,CAAC;IACvD,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;QACrE,IAAI,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,UAAU;YAAE,SAAS;QAC5E,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,SAAS,CAAC;YACjD,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAChD,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC;IACpD,CAAC;IACD,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC,EAAE,CAAC;QACnE,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC;aACpC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,SAAS,KAAK,SAAS,CAAC;aACxD,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,eAAe,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;aAC1E,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;QACpD,IAAI,UAAU,IAAI,UAAU;YAAE,SAAS;QACvC,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,SAAS,CAAC;YACjD,OAAO,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QACpD,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAChD,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC;IACpD,CAAC;IACD,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5E,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7D,OAAO,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACjE,CAAC;QACD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,IAAI,EAAE,UAAU,KAAK,SAAS,EAAE,CAAC;YACnC,OAAO,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QACD,OAAO,EAAE,CAAC;IACZ,CAAC,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC,WAAW;QAC3C,CAAC,CAAC,YAAY,CAAC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,WAAW,CAAC;QAChD,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,UAAU,GAAG,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,KAAK,IAAI,IAAI,CAAC,CAAC;IAC7E,MAAM,KAAK,GAAqB;QAC9B,GAAG,OAAO,CAAC,KAAK;QAChB,KAAK,EAAE,UAAU;QACjB,OAAO;QACP,OAAO;QACP,KAAK,EAAE,UAAU;QACjB,GAAG,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACtD,CAAC;IACF,OAAO;QACL,KAAK,EAAE,UAAU;QACjB,OAAO;QACP,OAAO;QACP,KAAK;QACL,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,KAA4B,EAAE,SAAiB;IACnE,KAAK,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC;QAClE,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;IACzC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,eAAe,CAAC,KAA2B,EAAE,MAAc;IAClE,KAAK,IAAI,KAAK,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;QAC1D,IAAI,KAAK,CAAC,KAAK,CAAC,EAAE,MAAM,KAAK,MAAM;YAAE,OAAO,KAAK,CAAC;IACpD,CAAC;IACD,OAAO,CAAC,CAAC,CAAC;AACZ,CAAC;AAED,SAAS,qBAAqB,CAAC,KAAuB,EAAE,SAAiB;IACvE,OAAO,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,OAAO,KAAK,IAAI,CAAC,CAAC,MAAM,CAAC;AACjG,CAAC"}

package/dist/workflows/decision.d.ts

@@ -1,4 +1,4 @@
-import type { AgentNodeDefinition, WorkflowEdge, WorkflowNodeContext } from "./types.js";
+import type { AgentNodeDefinition, WorkflowNodeContext } from "./types.js";
 type DecisionAgentOptions = Omit<AgentNodeDefinition, "nodeType" | "prompt" | "expectedOutput" | "validate">;
 export type DecisionDefinition<TChoice extends string> = DecisionAgentOptions & {
     question: string | ((context: WorkflowNodeContext) => string | Promise<string>);
@@ -16,10 +16,16 @@ export declare function decision<TChoice extends string>(definition: DecisionDef
  * Build the matching `switch` edge for a `decision` node. Typing `cases` as
  * `Record<TChoice, string>` makes a missing case a compile error.
  */
-export declare function decisionEdge<TChoice extends string>(args: {
-    from: string;
+export declare function decisionEdge<TChoice extends string, const TFrom extends string, const TCases extends Record<TChoice, string>>(args: {
+    from: TFrom;
     choices: readonly TChoice[];
     field?: string;
-    cases: Record<TChoice, string>;
-}): WorkflowEdge;
+    cases: TCases;
+}): {
+    from: TFrom;
+    switch: {
+        on: string;
+        cases: TCases;
+    };
+};
 export {};

package/dist/workflows/decision.js.map

@@ -1 +1 @@
-{"version":3,"file":"decision.js","sourceRoot":"","sources":["../../src/workflows/decision.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAG7C,MAAM,aAAa,GAAG,OAAO,CAAC;AAC9B,MAAM,oBAAoB,GAAG,0BAA0B,CAAC;AAcxD;;;;;GAKG;AACH,MAAM,UAAU,QAAQ,CACtB,UAAuC;IAEvC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,YAAY,EAAE,GAAG,UAAU,CAAC;IAChF,MAAM,KAAK,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;IAC5C,kBAAkB,CAAC,OAAO,CAAC,CAAC;IAC5B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAS,OAAO,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAElF,OAAO,KAAK,CAAC;QACX,GAAG,YAAY;QACf,KAAK,CAAC,MAAM,CAAC,OAAO;YAClB,MAAM,IAAI,GAAG,OAAO,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;YACjF,OAAO;gBACL,IAAI;gBACJ,EAAE;gBACF,qCAAqC,aAAa,GAAG;gBACrD,iDAAiD;aAClD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC;QACD,cAAc,EAAE,KAAK,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,aAAa,qCAAqC;QACjG,QAAQ,CAAC,MAAM;YACb,MAAM,GAAG,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClE,MAAM,IAAI,KAAK,CAAC,8CAA8C,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtF,CAAC;YACD,MAAM,KAAK,GAAI,GAA+B,CAAC,KAAK,CAAC,CAAC;YACtD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACrD,MAAM,IAAI,KAAK,CACb,6BAA6B,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,qBAAqB,aAAa,EAAE,CAChG,CAAC;YACJ,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAyB,IAKpD;IACC,MAAM,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjC,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,4CAA4C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IACD,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,MAAM,EAAE;YACN,EAAE,EAAE,KAAK,KAAK,EAAE;YAChB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB;KACF,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,MAAe;IAC9C,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC;AACvD,CAAC;AAED,SAAS,kBAAkB,CAAC,OAA0B;IACpD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;IACtE,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,8CAA8C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,aAAiC;IACvD,MAAM,KAAK,GAAG,aAAa,IAAI,aAAa,CAAC;IAC7C,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CACb,qDAAqD,oBAAoB,CAAC,MAAM,EAAE,CACnF,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}
+{"version":3,"file":"decision.js","sourceRoot":"","sources":["../../src/workflows/decision.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AACxC,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAG7C,MAAM,aAAa,GAAG,OAAO,CAAC;AAC9B,MAAM,oBAAoB,GAAG,0BAA0B,CAAC;AAcxD;;;;;GAKG;AACH,MAAM,UAAU,QAAQ,CACtB,UAAuC;IAEvC,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,YAAY,EAAE,GAAG,UAAU,CAAC;IAChF,MAAM,KAAK,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;IAC5C,kBAAkB,CAAC,OAAO,CAAC,CAAC;IAC5B,MAAM,OAAO,GAAG,IAAI,GAAG,CAAS,OAAO,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAElF,OAAO,KAAK,CAAC;QACX,GAAG,YAAY;QACf,KAAK,CAAC,MAAM,CAAC,OAAO;YAClB,MAAM,IAAI,GAAG,OAAO,QAAQ,KAAK,UAAU,CAAC,CAAC,CAAC,MAAM,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;YACjF,OAAO;gBACL,IAAI;gBACJ,EAAE;gBACF,qCAAqC,aAAa,GAAG;gBACrD,iDAAiD;aAClD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC;QACD,cAAc,EAAE,KAAK,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,aAAa,qCAAqC;QACjG,QAAQ,CAAC,MAAM;YACb,MAAM,GAAG,GAAG,uBAAuB,CAAC,MAAM,CAAC,CAAC;YAC5C,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;gBAClE,MAAM,IAAI,KAAK,CAAC,8CAA8C,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACtF,CAAC;YACD,MAAM,KAAK,GAAI,GAA+B,CAAC,KAAK,CAAC,CAAC;YACtD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC;gBACrD,MAAM,IAAI,KAAK,CACb,6BAA6B,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,qBAAqB,aAAa,EAAE,CAChG,CAAC;YACJ,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CAI1B,IAKD;IAIC,MAAM,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACzC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACjC,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC;YACvC,MAAM,IAAI,KAAK,CAAC,4CAA4C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IACD,OAAO;QACL,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,MAAM,EAAE;YACN,EAAE,EAAE,KAAK,KAAK,EAAE;YAChB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB;KACF,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,MAAe;IAC9C,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,OAAO,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC;AACvD,CAAC;AAED,SAAS,kBAAkB,CAAC,OAA0B;IACpD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;IACtE,CAAC;IACD,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtD,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAChE,CAAC;QACD,IAAI,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,8CAA8C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAC1F,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,aAAiC;IACvD,MAAM,KAAK,GAAG,aAAa,IAAI,aAAa,CAAC;IAC7C,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CACb,qDAAqD,oBAAoB,CAAC,MAAM,EAAE,CACnF,CAAC;IACJ,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/workflows/definition.d.ts

@@ -1,6 +1,24 @@
-import type { AgentNodeDefinition, CheckpointNodeDefinition, ComputeNodeDefinition, FunctionActionNodeDefinition, NotifyNodeDefinition, ShellActionNodeDefinition, WorkflowDefinition } from "./types.js";
-export declare function defineWorkflow<TWorkflow extends WorkflowDefinition>(definition: TWorkflow): TWorkflow;
-export declare function isWorkflowDefinition(value: unknown): value is WorkflowDefinition;
+import type { AgentNodeDefinition, CheckpointNodeDefinition, ComputeNodeDefinition, FunctionActionNodeDefinition, NotifyNodeDefinition, ShellActionNodeDefinition, WorkflowDefinition, WorkflowExitMap, WorkflowIncludeDefinition, WorkflowIncludedResult, WorkflowIncludeMap, WorkflowInputOf, WorkflowNodeContext, WorkflowNodeDefinition, WorkflowTypedEdge, WorkflowValueParser } from "./types.js";
+type WorkflowDefinitionInput<TInput, TNodes extends Record<string, WorkflowNodeDefinition>, TIncludes extends WorkflowIncludeMap, TExits extends WorkflowExitMap> = Omit<WorkflowDefinition<TInput, TExits, TIncludes>, "input" | "nodes" | "includes" | "exits" | "edges"> & {
+    input?: WorkflowValueParser<TInput>;
+    nodes: TNodes;
+    includes?: TIncludes;
+    exits?: TExits;
+    edges: WorkflowTypedEdge<TNodes, TIncludes>[];
+};
+export declare function defineWorkflow<TInput = unknown, const TNodes extends Record<string, WorkflowNodeDefinition> = Record<string, WorkflowNodeDefinition>, const TIncludes extends WorkflowIncludeMap = Record<never, never>, const TExits extends WorkflowExitMap = Record<never, never>>(definition: WorkflowDefinitionInput<TInput, TNodes, TIncludes, TExits>): WorkflowDefinition<TInput, TExits, TIncludes> & {
+    nodes: TNodes;
+    includes?: TIncludes;
+    exits?: TExits;
+};
+export declare function includeWorkflow<TWorkflow extends WorkflowDefinition<any, any, any>>(workflow: TWorkflow, options?: {
+    input?: (context: WorkflowNodeContext) => Promise<WorkflowInputOf<TWorkflow>> | WorkflowInputOf<TWorkflow>;
+}): WorkflowIncludeDefinition<TWorkflow>;
+export declare function includeWorkflow<TWorkflow extends WorkflowDefinition<any, any, any>>(definition: WorkflowIncludeDefinition<TWorkflow>): WorkflowIncludeDefinition<TWorkflow>;
+export declare function includedResult<TWorkflow extends WorkflowDefinition<any, any, any>>(workflow: TWorkflow, value: unknown): WorkflowIncludedResult<TWorkflow>;
+/** Preserve exact workflow types while checking duplicate registry names. */
+export declare function defineWorkflowRegistry<const TRegistry extends Record<string, WorkflowDefinition<any, any, any>>>(registry: TRegistry): Readonly<TRegistry>;
+export declare function isWorkflowDefinition(value: unknown): value is WorkflowDefinition<any, any, any>;
 export declare function agent(definition: Omit<AgentNodeDefinition, "nodeType">): AgentNodeDefinition;
 export declare function compute(definition: Omit<ComputeNodeDefinition, "nodeType">): ComputeNodeDefinition;
 export declare function notify(definition: Omit<NotifyNodeDefinition, "nodeType">): NotifyNodeDefinition;
@@ -8,3 +26,4 @@ export declare function action(definition: Omit<FunctionActionNodeDefinition, "n
 export declare function action(definition: Omit<ShellActionNodeDefinition, "nodeType">): ShellActionNodeDefinition;
 export declare function shell(definition: Omit<ShellActionNodeDefinition, "nodeType">): ShellActionNodeDefinition;
 export declare function checkpoint(definition?: Omit<CheckpointNodeDefinition, "nodeType">): CheckpointNodeDefinition;
+export {};

package/dist/workflows/definition.js

@@ -2,17 +2,60 @@ import { assertValidAgentNode, assertValidActionNode, assertValidCheckpointNode,
 const WORKFLOW_DEFINITION_BRAND = Symbol.for("pi-workflows.definition");
 export function defineWorkflow(definition) {
     assertValidWorkflowDefinitionShape(definition);
-    if (isWorkflowDefinition(definition)) {
-        return definition;
+    const typed = definition;
+    if (isWorkflowDefinition(typed)) {
+        return typed;
     }
-    Object.defineProperty(definition, WORKFLOW_DEFINITION_BRAND, {
+    Object.defineProperty(typed, WORKFLOW_DEFINITION_BRAND, {
         value: true,
         enumerable: false,
         configurable: false,
         writable: false,
     });
+    return typed;
+}
+export function includeWorkflow(workflowOrDefinition, options = {}) {
+    const definition = isWorkflowDefinition(workflowOrDefinition)
+        ? { workflow: workflowOrDefinition, ...options }
+        : workflowOrDefinition;
+    if (typeof definition.workflow !== "string" && !isWorkflowDefinition(definition.workflow)) {
+        throw new Error("Included workflow must be a defined workflow or a workflow reference");
+    }
+    if (definition.input !== undefined && typeof definition.input !== "function") {
+        throw new Error("Included workflow input must be a function");
+    }
+    if (definition.contract !== undefined && !isWorkflowDefinition(definition.contract)) {
+        throw new Error("Included workflow contract must be defined with defineWorkflow");
+    }
     return definition;
 }
+export function includedResult(workflow, value) {
+    if (value === null || typeof value !== "object" || Array.isArray(value)) {
+        throw new Error(`Included ${workflow.name} result must be an object`);
+    }
+    const exit = value.exit;
+    if (typeof exit !== "string" || !Object.hasOwn(workflow.exits ?? {}, exit)) {
+        throw new Error(`Included ${workflow.name} result has unknown exit ${JSON.stringify(exit)}`);
+    }
+    if (!Object.hasOwn(value, "output")) {
+        throw new Error(`Included ${workflow.name} result requires output`);
+    }
+    return value;
+}
+/** Preserve exact workflow types while checking duplicate registry names. */
+export function defineWorkflowRegistry(registry) {
+    const names = new Set();
+    for (const [key, workflow] of Object.entries(registry)) {
+        if (!isWorkflowDefinition(workflow)) {
+            throw new Error(`Workflow registry entry ${key} is not defined with defineWorkflow`);
+        }
+        if (names.has(workflow.name)) {
+            throw new Error(`Workflow registry has duplicate workflow name: ${workflow.name}`);
+        }
+        names.add(workflow.name);
+    }
+    return Object.freeze({ ...registry });
+}
 export function isWorkflowDefinition(value) {
     return (value != null &&
         typeof value === "object" &&