@langchain/langgraph 0.2.27 → 0.2.29

package/dist/constants.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._isCommand = exports.Command = exports._isSend = exports.Send = exports._isSendInterface = exports.CHECKPOINT_NAMESPACE_END = exports.CHECKPOINT_NAMESPACE_SEPARATOR = exports.RESERVED = exports.NULL_TASK_ID = exports.TASK_NAMESPACE = exports.PULL = exports.PUSH = exports.TASKS = exports.TAG_NOSTREAM = exports.TAG_HIDDEN = exports.RECURSION_LIMIT_DEFAULT = exports.RUNTIME_PLACEHOLDER = exports.RESUME = exports.INTERRUPT = exports.CONFIG_KEY_CHECKPOINT_MAP = exports.CONFIG_KEY_RESUME_VALUE = exports.CONFIG_KEY_STREAM = exports.CONFIG_KEY_TASK_ID = exports.CONFIG_KEY_RESUMING = exports.CONFIG_KEY_CHECKPOINTER = exports.CONFIG_KEY_READ = exports.CONFIG_KEY_SEND = exports.ERROR = exports.INPUT = exports.MISSING = void 0;
+exports._isCommand = exports.Command = exports._isSend = exports.Send = exports._isSendInterface = exports.CHECKPOINT_NAMESPACE_END = exports.CHECKPOINT_NAMESPACE_SEPARATOR = exports.RESERVED = exports.NULL_TASK_ID = exports.TASK_NAMESPACE = exports.PULL = exports.PUSH = exports.TASKS = exports.SELF = exports.TAG_NOSTREAM = exports.TAG_HIDDEN = exports.RECURSION_LIMIT_DEFAULT = exports.RUNTIME_PLACEHOLDER = exports.RESUME = exports.INTERRUPT = exports.CONFIG_KEY_CHECKPOINT_MAP = exports.CONFIG_KEY_RESUME_VALUE = exports.CONFIG_KEY_STREAM = exports.CONFIG_KEY_TASK_ID = exports.CONFIG_KEY_RESUMING = exports.CONFIG_KEY_CHECKPOINTER = exports.CONFIG_KEY_READ = exports.CONFIG_KEY_SEND = exports.ERROR = exports.INPUT = exports.MISSING = void 0;
 exports.MISSING = Symbol.for("__missing__");
 exports.INPUT = "__input__";
 exports.ERROR = "__error__";
@@ -19,6 +19,7 @@ exports.RUNTIME_PLACEHOLDER = "__pregel_runtime_placeholder__";
 exports.RECURSION_LIMIT_DEFAULT = 25;
 exports.TAG_HIDDEN = "langsmith:hidden";
 exports.TAG_NOSTREAM = "langsmith:nostream";
+exports.SELF = "__self__";
 exports.TASKS = "__pregel_tasks";
 exports.PUSH = "__pregel_push";
 exports.PULL = "__pregel_pull";
@@ -117,9 +118,12 @@ class Send {
 exports.Send = Send;
 function _isSend(x) {
     const operation = x;
-    return operation.lg_name === "Send";
+    return operation !== undefined && operation.lg_name === "Send";
 }
 exports._isSend = _isSend;
+/**
+ * One or more commands to update the graph's state and send messages to nodes.
+ */
 class Command {
     constructor(args) {
         Object.defineProperty(this, "lg_name", {
@@ -134,10 +138,40 @@ class Command {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "graph", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        Object.defineProperty(this, "update", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "goto", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
         this.resume = args.resume;
+        this.graph = args.graph;
+        this.update = args.update;
+        if (args.goto) {
+            this.goto = Array.isArray(args.goto) ? args.goto : [args.goto];
+        }
     }
 }
 exports.Command = Command;
+Object.defineProperty(Command, "PARENT", {
+    enumerable: true,
+    configurable: true,
+    writable: true,
+    value: "__parent__"
+});
 function _isCommand(x) {
     return typeof x === "object" && !!x && x.lg_name === "Command";
 }

package/dist/constants.d.ts

@@ -15,6 +15,7 @@ export declare const RUNTIME_PLACEHOLDER = "__pregel_runtime_placeholder__";
 export declare const RECURSION_LIMIT_DEFAULT = 25;
 export declare const TAG_HIDDEN = "langsmith:hidden";
 export declare const TAG_NOSTREAM = "langsmith:nostream";
+export declare const SELF = "__self__";
 export declare const TASKS = "__pregel_tasks";
 export declare const PUSH = "__pregel_push";
 export declare const PULL = "__pregel_pull";
@@ -87,11 +88,40 @@ export type Interrupt = {
     resumable?: boolean;
     ns?: string[];
 };
+export type CommandParams<R> = {
+    /**
+     * Value to resume execution with. To be used together with {@link interrupt}.
+     */
+    resume?: R;
+    /**
+     * Graph to send the command to. Supported values are:
+     *   - None: the current graph (default)
+     *   - GraphCommand.PARENT: closest parent graph
+     */
+    graph?: string;
+    /**
+     * Update to apply to the graph's state.
+     */
+    update?: Record<string, any>;
+    /**
+     * Can be one of the following:
+     *   - name of the node to navigate to next (any node that belongs to the specified `graph`)
+     *   - sequence of node names to navigate to next
+     *   - `Send` object (to execute a node with the input provided)
+     *   - sequence of `Send` objects
+     */
+    goto?: string | Send | (string | Send)[];
+};
+/**
+ * One or more commands to update the graph's state and send messages to nodes.
+ */
 export declare class Command<R = unknown> {
     lg_name: string;
-    resume: R;
-    constructor(args: {
-        resume: R;
-    });
+    resume?: R;
+    graph?: string;
+    update?: Record<string, any>;
+    goto: string | Send | (string | Send)[];
+    static PARENT: string;
+    constructor(args: CommandParams<R>);
 }
 export declare function _isCommand(x: unknown): x is Command;

package/dist/constants.js

@@ -16,6 +16,7 @@ export const RUNTIME_PLACEHOLDER = "__pregel_runtime_placeholder__";
 export const RECURSION_LIMIT_DEFAULT = 25;
 export const TAG_HIDDEN = "langsmith:hidden";
 export const TAG_NOSTREAM = "langsmith:nostream";
+export const SELF = "__self__";
 export const TASKS = "__pregel_tasks";
 export const PUSH = "__pregel_push";
 export const PULL = "__pregel_pull";
@@ -112,8 +113,11 @@ export class Send {
 }
 export function _isSend(x) {
     const operation = x;
-    return operation.lg_name === "Send";
+    return operation !== undefined && operation.lg_name === "Send";
 }
+/**
+ * One or more commands to update the graph's state and send messages to nodes.
+ */
 export class Command {
     constructor(args) {
         Object.defineProperty(this, "lg_name", {
@@ -128,9 +132,39 @@ export class Command {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "graph", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        Object.defineProperty(this, "update", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "goto", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: []
+        });
         this.resume = args.resume;
+        this.graph = args.graph;
+        this.update = args.update;
+        if (args.goto) {
+            this.goto = Array.isArray(args.goto) ? args.goto : [args.goto];
+        }
     }
 }
+Object.defineProperty(Command, "PARENT", {
+    enumerable: true,
+    configurable: true,
+    writable: true,
+    value: "__parent__"
+});
 export function _isCommand(x) {
     return typeof x === "object" && !!x && x.lg_name === "Command";
 }

package/dist/errors.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getSubgraphsSeenSet = exports.RemoteException = exports.MultipleSubgraphsError = exports.InvalidUpdateError = exports.EmptyChannelError = exports.EmptyInputError = exports.isGraphInterrupt = exports.isGraphBubbleUp = exports.NodeInterrupt = exports.GraphInterrupt = exports.GraphValueError = exports.GraphRecursionError = exports.GraphBubbleUp = exports.BaseLangGraphError = void 0;
+exports.getSubgraphsSeenSet = exports.RemoteException = exports.MultipleSubgraphsError = exports.InvalidUpdateError = exports.EmptyChannelError = exports.EmptyInputError = exports.isGraphInterrupt = exports.isGraphBubbleUp = exports.isParentCommand = exports.ParentCommand = exports.NodeInterrupt = exports.GraphInterrupt = exports.GraphValueError = exports.GraphRecursionError = exports.GraphBubbleUp = exports.BaseLangGraphError = void 0;
 // TODO: Merge with base LangChain error class when we drop support for core@0.2.0
 class BaseLangGraphError extends Error {
     constructor(message, fields) {
@@ -79,6 +79,28 @@ class NodeInterrupt extends GraphInterrupt {
     }
 }
 exports.NodeInterrupt = NodeInterrupt;
+class ParentCommand extends GraphBubbleUp {
+    constructor(command) {
+        super();
+        Object.defineProperty(this, "command", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.name = "ParentCommand";
+        this.command = command;
+    }
+    static get unminifiable_name() {
+        return "ParentCommand";
+    }
+}
+exports.ParentCommand = ParentCommand;
+function isParentCommand(e) {
+    return (e !== undefined &&
+        e.name === ParentCommand.unminifiable_name);
+}
+exports.isParentCommand = isParentCommand;
 function isGraphBubbleUp(e) {
     return e !== undefined && e.is_bubble_up === true;
 }

package/dist/errors.d.ts

@@ -1,4 +1,4 @@
-import { Interrupt } from "./constants.js";
+import { Command, Interrupt } from "./constants.js";
 export type BaseLangGraphErrorFields = {
     lc_error_code?: string;
 };
@@ -27,6 +27,12 @@ export declare class NodeInterrupt extends GraphInterrupt {
     constructor(message: any, fields?: BaseLangGraphErrorFields);
     static get unminifiable_name(): string;
 }
+export declare class ParentCommand extends GraphBubbleUp {
+    command: Command;
+    constructor(command: Command);
+    static get unminifiable_name(): string;
+}
+export declare function isParentCommand(e?: Error): e is ParentCommand;
 export declare function isGraphBubbleUp(e?: Error): e is GraphBubbleUp;
 export declare function isGraphInterrupt(e?: GraphInterrupt | Error): e is GraphInterrupt;
 export declare class EmptyInputError extends BaseLangGraphError {

package/dist/errors.js

@@ -70,6 +70,26 @@ export class NodeInterrupt extends GraphInterrupt {
         return "NodeInterrupt";
     }
 }
+export class ParentCommand extends GraphBubbleUp {
+    constructor(command) {
+        super();
+        Object.defineProperty(this, "command", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.name = "ParentCommand";
+        this.command = command;
+    }
+    static get unminifiable_name() {
+        return "ParentCommand";
+    }
+}
+export function isParentCommand(e) {
+    return (e !== undefined &&
+        e.name === ParentCommand.unminifiable_name);
+}
 export function isGraphBubbleUp(e) {
     return e !== undefined && e.is_bubble_up === true;
 }

package/dist/graph/graph.cjs

@@ -32,7 +32,14 @@ class Branch {
             writable: true,
             value: void 0
         });
-        this.condition = options.path;
+        if (runnables_1.Runnable.isRunnable(options.path)) {
+            this.condition = options.path;
+        }
+        else {
+            this.condition = (0, runnables_1._coerceToRunnable)(options.path).withConfig({
+                runName: `Branch`,
+            });
+        }
         this.ends = Array.isArray(options.pathMap)
             ? options.pathMap.reduce((acc, n) => {
                 acc[n] = n;
@@ -40,7 +47,7 @@ class Branch {
             }, {})
             : options.pathMap;
     }
-    compile(writer, reader) {
+    run(writer, reader) {
         return write_js_1.ChannelWrite.registerWriter(new utils_js_1.RunnableCallable({
             trace: false,
             func: async (input, config) => {
@@ -59,8 +66,10 @@ class Branch {
             },
         }));
     }
-    async _route(input, config, writer, reader) {
-        let result = await this.condition(reader ? reader(config) : input, config);
+    async _route(input, config, writer, reader
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ) {
+        let result = await this.condition.invoke(reader ? reader(config) : input, config);
         if (!Array.isArray(result)) {
             result = [result];
         }
@@ -77,7 +86,8 @@ class Branch {
         if (destinations.filter(constants_js_1._isSend).some((packet) => packet.node === exports.END)) {
             throw new errors_js_1.InvalidUpdateError("Cannot send a packet to the END node");
         }
-        return writer(destinations);
+        const writeResult = await writer(destinations, config);
+        return writeResult ?? input;
     }
 }
 exports.Branch = Branch;
@@ -149,6 +159,7 @@ class Graph {
             runnable,
             metadata: options?.metadata,
             subgraphs: (0, subgraph_js_1.isPregelLike)(runnable) ? [runnable] : options?.subgraphs,
+            ends: options?.ends,
         };
         return this;
     }
@@ -168,10 +179,26 @@ class Graph {
         return this;
     }
     addConditionalEdges(source, path, pathMap) {
-        const options = typeof source === "object" ? source : { source, path: path, pathMap };
+        const options = typeof source === "object"
+            ? source
+            : {
+                source,
+                path: path,
+                pathMap,
+            };
         this.warnIfCompiled("Adding an edge to a graph that has already been compiled. This will not be reflected in the compiled graph.");
+        if (!runnables_1.Runnable.isRunnable(options.path)) {
+            const pathDisplayValues = Array.isArray(options.pathMap)
+                ? options.pathMap.join(",")
+                : Object.keys(options.pathMap ?? {}).join(",");
+            options.path = (0, runnables_1._coerceToRunnable)(options.path).withConfig({
+                runName: `Branch<${options.source}${pathDisplayValues !== "" ? `,${pathDisplayValues}` : ""}>`.slice(0, 63),
+            });
+        }
         // find a name for condition
-        const name = options.path.name || "condition";
+        const name = options.path.getName() === "RunnableLambda"
+            ? "condition"
+            : options.path.getName();
         // validate condition
         if (this.branches[options.source] && this.branches[options.source][name]) {
             throw new Error(`Condition \`${name}\` already present for node \`${source}\``);
@@ -264,6 +291,11 @@ class Graph {
                 }
             }
         }
+        for (const node of Object.values(this.nodes)) {
+            for (const target of node.ends ?? []) {
+                allTargets.add(target);
+            }
+        }
         // validate targets
         for (const node of Object.keys(this.nodes)) {
             if (!allTargets.has(node)) {
@@ -305,6 +337,7 @@ class CompiledGraph extends index_js_1.Pregel {
             triggers: [],
             metadata: node.metadata,
             subgraphs: node.subgraphs,
+            ends: node.ends,
         })
             .pipe(node.runnable)
             .pipe(new write_js_1.ChannelWrite([{ channel: key, value: write_js_1.PASSTHROUGH }], [constants_js_1.TAG_HIDDEN]));
@@ -328,7 +361,7 @@ class CompiledGraph extends index_js_1.Pregel {
             this.nodes[exports.START] = index_js_1.Channel.subscribeTo(exports.START, { tags: [constants_js_1.TAG_HIDDEN] });
         }
         // attach branch writer
-        this.nodes[start].pipe(branch.compile((dests) => {
+        this.nodes[start].pipe(branch.run((dests) => {
             const writes = dests.map((dest) => {
                 if ((0, constants_js_1._isSend)(dest)) {
                     return dest;
@@ -500,6 +533,13 @@ class CompiledGraph extends index_js_1.Pregel {
                 }
             }
         }
+        for (const [key, node] of Object.entries(this.builder.nodes)) {
+            if (node.ends !== undefined) {
+                for (const end of node.ends) {
+                    addEdge(_escapeMermaidKeywords(key), _escapeMermaidKeywords(end), undefined, true);
+                }
+            }
+        }
         return graph;
     }
     /**

package/dist/graph/graph.d.ts

@@ -15,24 +15,27 @@ export declare const START = "__start__";
 export declare const END = "__end__";
 export interface BranchOptions<IO, N extends string, CallOptions extends LangGraphRunnableConfig = LangGraphRunnableConfig> {
     source: N;
-    path: Branch<IO, N, CallOptions>["condition"];
+    path: RunnableLike<IO, BranchPathReturnValue, CallOptions>;
     pathMap?: Record<string, N | typeof END> | (N | typeof END)[];
 }
+export type BranchPathReturnValue = string | Send | (string | Send)[] | Promise<string | Send | (string | Send)[]>;
 export declare class Branch<IO, N extends string, CallOptions extends LangGraphRunnableConfig = LangGraphRunnableConfig> {
-    condition: (input: IO, config: CallOptions) => string | Send | (string | Send)[] | Promise<string | Send | (string | Send)[]>;
+    condition: Runnable<IO, BranchPathReturnValue, CallOptions>;
     ends?: Record<string, N | typeof END>;
     constructor(options: Omit<BranchOptions<IO, N, CallOptions>, "source">);
-    compile(writer: (dests: (string | Send)[]) => Runnable | undefined, reader?: (config: CallOptions) => IO): RunnableCallable<unknown, unknown>;
-    _route(input: IO, config: CallOptions, writer: (dests: (string | Send)[]) => Runnable | undefined, reader?: (config: CallOptions) => IO): Promise<Runnable | undefined>;
+    run(writer: (dests: (string | Send)[], config: LangGraphRunnableConfig) => Runnable | void | Promise<void>, reader?: (config: CallOptions) => IO): RunnableCallable<unknown, unknown>;
+    _route(input: IO, config: CallOptions, writer: (dests: (string | Send)[], config: LangGraphRunnableConfig) => Runnable | void | Promise<void>, reader?: (config: CallOptions) => IO): Promise<Runnable | any>;
 }
 export type NodeSpec<RunInput, RunOutput> = {
     runnable: Runnable<RunInput, RunOutput>;
     metadata?: Record<string, unknown>;
     subgraphs?: Pregel<any, any>[];
+    ends?: string[];
 };
 export type AddNodeOptions = {
     metadata?: Record<string, unknown>;
     subgraphs?: Pregel<any, any>[];
+    ends?: string[];
 };
 export declare class Graph<N extends string = typeof END, RunInput = any, RunOutput = any, NodeSpecType extends NodeSpec<RunInput, RunOutput> = NodeSpec<RunInput, RunOutput>, C extends StateDefinition = StateDefinition> {
     nodes: Record<N, NodeSpecType>;
@@ -46,7 +49,7 @@ export declare class Graph<N extends string = typeof END, RunInput = any, RunOut
     addNode<K extends string, NodeInput = RunInput>(key: K, action: RunnableLike<NodeInput, RunOutput extends object ? RunOutput & Record<string, any> : RunOutput, LangGraphRunnableConfig<StateType<C>>>, options?: AddNodeOptions): Graph<N | K, RunInput, RunOutput>;
     addEdge(startKey: N | typeof START, endKey: N | typeof END): this;
     addConditionalEdges(source: BranchOptions<RunInput, N, LangGraphRunnableConfig<StateType<C>>>): this;
-    addConditionalEdges(source: N, path: Branch<RunInput, N, LangGraphRunnableConfig<StateType<C>>>["condition"], pathMap?: BranchOptions<RunInput, N, LangGraphRunnableConfig<StateType<C>>>["pathMap"]): this;
+    addConditionalEdges(source: N, path: RunnableLike<RunInput, BranchPathReturnValue, LangGraphRunnableConfig<StateType<C>>>, pathMap?: BranchOptions<RunInput, N, LangGraphRunnableConfig<StateType<C>>>["pathMap"]): this;
     /**
      * @deprecated use `addEdge(START, key)` instead
      */

package/dist/graph/graph.js

@@ -1,5 +1,5 @@
 /* eslint-disable @typescript-eslint/no-use-before-define */
-import { _coerceToRunnable, } from "@langchain/core/runnables";
+import { _coerceToRunnable, Runnable, } from "@langchain/core/runnables";
 import { Graph as DrawableGraph, } from "@langchain/core/runnables/graph";
 import { z } from "zod";
 import { validate as isUuid } from "uuid";
@@ -29,7 +29,14 @@ export class Branch {
             writable: true,
             value: void 0
         });
-        this.condition = options.path;
+        if (Runnable.isRunnable(options.path)) {
+            this.condition = options.path;
+        }
+        else {
+            this.condition = _coerceToRunnable(options.path).withConfig({
+                runName: `Branch`,
+            });
+        }
         this.ends = Array.isArray(options.pathMap)
             ? options.pathMap.reduce((acc, n) => {
                 acc[n] = n;
@@ -37,7 +44,7 @@ export class Branch {
             }, {})
             : options.pathMap;
     }
-    compile(writer, reader) {
+    run(writer, reader) {
         return ChannelWrite.registerWriter(new RunnableCallable({
             trace: false,
             func: async (input, config) => {
@@ -56,8 +63,10 @@ export class Branch {
             },
         }));
     }
-    async _route(input, config, writer, reader) {
-        let result = await this.condition(reader ? reader(config) : input, config);
+    async _route(input, config, writer, reader
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ) {
+        let result = await this.condition.invoke(reader ? reader(config) : input, config);
         if (!Array.isArray(result)) {
             result = [result];
         }
@@ -74,7 +83,8 @@ export class Branch {
         if (destinations.filter(_isSend).some((packet) => packet.node === END)) {
             throw new InvalidUpdateError("Cannot send a packet to the END node");
         }
-        return writer(destinations);
+        const writeResult = await writer(destinations, config);
+        return writeResult ?? input;
     }
 }
 export class Graph {
@@ -145,6 +155,7 @@ export class Graph {
             runnable,
             metadata: options?.metadata,
             subgraphs: isPregelLike(runnable) ? [runnable] : options?.subgraphs,
+            ends: options?.ends,
         };
         return this;
     }
@@ -164,10 +175,26 @@ export class Graph {
         return this;
     }
     addConditionalEdges(source, path, pathMap) {
-        const options = typeof source === "object" ? source : { source, path: path, pathMap };
+        const options = typeof source === "object"
+            ? source
+            : {
+                source,
+                path: path,
+                pathMap,
+            };
         this.warnIfCompiled("Adding an edge to a graph that has already been compiled. This will not be reflected in the compiled graph.");
+        if (!Runnable.isRunnable(options.path)) {
+            const pathDisplayValues = Array.isArray(options.pathMap)
+                ? options.pathMap.join(",")
+                : Object.keys(options.pathMap ?? {}).join(",");
+            options.path = _coerceToRunnable(options.path).withConfig({
+                runName: `Branch<${options.source}${pathDisplayValues !== "" ? `,${pathDisplayValues}` : ""}>`.slice(0, 63),
+            });
+        }
         // find a name for condition
-        const name = options.path.name || "condition";
+        const name = options.path.getName() === "RunnableLambda"
+            ? "condition"
+            : options.path.getName();
         // validate condition
         if (this.branches[options.source] && this.branches[options.source][name]) {
             throw new Error(`Condition \`${name}\` already present for node \`${source}\``);
@@ -260,6 +287,11 @@ export class Graph {
                 }
             }
         }
+        for (const node of Object.values(this.nodes)) {
+            for (const target of node.ends ?? []) {
+                allTargets.add(target);
+            }
+        }
         // validate targets
         for (const node of Object.keys(this.nodes)) {
             if (!allTargets.has(node)) {
@@ -300,6 +332,7 @@ export class CompiledGraph extends Pregel {
             triggers: [],
             metadata: node.metadata,
             subgraphs: node.subgraphs,
+            ends: node.ends,
         })
             .pipe(node.runnable)
             .pipe(new ChannelWrite([{ channel: key, value: PASSTHROUGH }], [TAG_HIDDEN]));
@@ -323,7 +356,7 @@ export class CompiledGraph extends Pregel {
             this.nodes[START] = Channel.subscribeTo(START, { tags: [TAG_HIDDEN] });
         }
         // attach branch writer
-        this.nodes[start].pipe(branch.compile((dests) => {
+        this.nodes[start].pipe(branch.run((dests) => {
             const writes = dests.map((dest) => {
                 if (_isSend(dest)) {
                     return dest;
@@ -495,6 +528,13 @@ export class CompiledGraph extends Pregel {
                 }
             }
         }
+        for (const [key, node] of Object.entries(this.builder.nodes)) {
+            if (node.ends !== undefined) {
+                for (const end of node.ends) {
+                    addEdge(_escapeMermaidKeywords(key), _escapeMermaidKeywords(end), undefined, true);
+                }
+            }
+        }
         return graph;
     }
     /**