@langchain/langgraph 0.2.27 → 0.2.28

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,7 +118,7 @@ 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;
 class Command {
@@ -134,10 +135,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,19 @@ export type Interrupt = {
     resumable?: boolean;
     ns?: string[];
 };
+export type CommandParams<R> = {
+    resume?: R;
+    graph?: string;
+    update?: Record<string, any>;
+    goto?: string | Send | (string | Send)[];
+};
 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,7 +113,7 @@ export class Send {
 }
 export function _isSend(x) {
     const operation = x;
-    return operation.lg_name === "Send";
+    return operation !== undefined && operation.lg_name === "Send";
 }
 export class Command {
     constructor(args) {
@@ -128,9 +129,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;
@@ -168,10 +178,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}\``);
@@ -328,7 +354,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;

package/dist/graph/graph.d.ts

@@ -15,15 +15,16 @@ 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>;
@@ -46,7 +47,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 {
@@ -164,10 +174,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}\``);
@@ -323,7 +349,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;

package/dist/graph/state.cjs

@@ -303,6 +303,14 @@ class StateGraph extends graph_js_1.Graph {
         for (const [key, node] of Object.entries(this.nodes)) {
             compiled.attachNode(key, node);
         }
+        compiled.attachBranch(graph_js_1.START, constants_js_1.SELF, _getControlBranch(), {
+            withReader: false,
+        });
+        for (const [key] of Object.entries(this.nodes)) {
+            compiled.attachBranch(key, constants_js_1.SELF, _getControlBranch(), {
+                withReader: false,
+            });
+        }
         for (const [start, end] of this.edges) {
             compiled.attachEdge(start, end);
         }
@@ -339,13 +347,30 @@ function _getChannels(schema) {
 class CompiledStateGraph extends graph_js_1.CompiledGraph {
     attachNode(key, node) {
         const stateKeys = Object.keys(this.builder.channels);
+        function _getRoot(input) {
+            if ((0, constants_js_1._isCommand)(input)) {
+                if (input.graph === constants_js_1.Command.PARENT) {
+                    return write_js_1.SKIP_WRITE;
+                }
+                return input.update;
+            }
+            return input;
+        }
+        // to avoid name collision below
+        const nodeKey = key;
         function getStateKey(key, input) {
             if (!input) {
                 return write_js_1.SKIP_WRITE;
             }
+            else if ((0, constants_js_1._isCommand)(input)) {
+                if (input.graph === constants_js_1.Command.PARENT) {
+                    return write_js_1.SKIP_WRITE;
+                }
+                return getStateKey(key, input.update);
+            }
             else if (typeof input !== "object" || Array.isArray(input)) {
                 const typeofInput = Array.isArray(input) ? "array" : typeof input;
-                throw new errors_js_1.InvalidUpdateError(`Expected node "${key.toString()}" to return an object, received ${typeofInput}`, {
+                throw new errors_js_1.InvalidUpdateError(`Expected node "${nodeKey.toString()}" to return an object, received ${typeofInput}`, {
                     lc_error_code: "INVALID_GRAPH_NODE_RETURN_VALUE",
                 });
             }
@@ -355,7 +380,16 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
         }
         // state updaters
         const stateWriteEntries = stateKeys.map((key) => key === ROOT
-            ? { channel: key, value: write_js_1.PASSTHROUGH, skipNone: true }
+            ? {
+                channel: key,
+                value: write_js_1.PASSTHROUGH,
+                skipNone: true,
+                mapper: new utils_js_1.RunnableCallable({
+                    func: _getRoot,
+                    trace: false,
+                    recurse: false,
+                }),
+            }
             : {
                 channel: key,
                 value: write_js_1.PASSTHROUGH,
@@ -430,28 +464,29 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
             this.nodes[end].triggers.push(start);
         }
     }
-    attachBranch(start, name, branch) {
-        // attach branch publisher
-        this.nodes[start].writers.push(branch.compile(
-        // writer
-        (dests) => {
-            const filteredDests = dests.filter((dest) => dest !== graph_js_1.END);
-            if (!filteredDests.length) {
+    attachBranch(start, name, branch, options = { withReader: true }) {
+        const branchWriter = async (packets, config) => {
+            const filteredPackets = packets.filter((p) => p !== graph_js_1.END);
+            if (!filteredPackets.length) {
                 return;
             }
-            const writes = filteredDests.map((dest) => {
-                if ((0, constants_js_1._isSend)(dest)) {
-                    return dest;
+            const writes = filteredPackets.map((p) => {
+                if ((0, constants_js_1._isSend)(p)) {
+                    return p;
                 }
                 return {
-                    channel: `branch:${start}:${name}:${dest}`,
+                    channel: `branch:${start}:${name}:${p}`,
                     value: start,
                 };
             });
-            return new write_js_1.ChannelWrite(writes, [constants_js_1.TAG_HIDDEN]);
-        }, 
+            await write_js_1.ChannelWrite.doWrite({ ...config, tags: (config.tags ?? []).concat([constants_js_1.TAG_HIDDEN]) }, writes);
+        };
+        // attach branch publisher
+        this.nodes[start].writers.push(branch.run(branchWriter, 
         // reader
-        (config) => read_js_1.ChannelRead.doRead(config, this.streamChannels ?? this.outputChannels, true)));
+        options.withReader
+            ? (config) => read_js_1.ChannelRead.doRead(config, this.streamChannels ?? this.outputChannels, true)
+            : undefined));
         // attach branch subscribers
         const ends = branch.ends
             ? Object.values(branch.ends)
@@ -499,3 +534,28 @@ function isStateGraphArgsWithInputOutputSchemas(obj) {
         obj.input !== undefined &&
         obj.output !== undefined);
 }
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function _controlBranch(value) {
+    if ((0, constants_js_1._isSend)(value)) {
+        return [value];
+    }
+    if (!(0, constants_js_1._isCommand)(value)) {
+        return [];
+    }
+    if (value.graph === constants_js_1.Command.PARENT) {
+        throw new errors_js_1.ParentCommand(value);
+    }
+    return Array.isArray(value.goto) ? value.goto : [value.goto];
+}
+function _getControlBranch() {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const CONTROL_BRANCH_PATH = new utils_js_1.RunnableCallable({
+        func: _controlBranch,
+        tags: [constants_js_1.TAG_HIDDEN],
+        trace: false,
+        recurse: false,
+    });
+    return new graph_js_1.Branch({
+        path: CONTROL_BRANCH_PATH,
+    });
+}

package/dist/graph/state.d.ts

@@ -133,5 +133,7 @@ export declare class CompiledStateGraph<S, U, N extends string = typeof START, I
     attachNode(key: typeof START, node?: never): void;
     attachNode(key: N, node: StateGraphNodeSpec<S, U>): void;
     attachEdge(start: N | N[] | "__start__", end: N | "__end__"): void;
-    attachBranch(start: N | typeof START, name: string, branch: Branch<S, N>): void;
+    attachBranch(start: N | typeof START, name: string, branch: Branch<S, N>, options?: {
+        withReader?: boolean;
+    }): void;
 }