@langchain/langgraph 0.0.18 → 0.0.20

package/dist/checkpoint/base.d.ts

@@ -13,7 +13,7 @@ export interface CheckpointMetadata {
      * -1 for the first "input" checkpoint.
      * 0 for the first "loop" checkpoint.
      * ... for the nth checkpoint afterwards. */
-    writes?: Record<string, unknown>;
+    writes: Record<string, unknown> | null;
 }
 export interface Checkpoint<N extends string = string, C extends string = string> {
     /**

package/dist/graph/graph.cjs

@@ -25,12 +25,6 @@ class Branch {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "then", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         this.condition = options.path;
         this.ends = Array.isArray(options.pathMap)
             ? options.pathMap.reduce((acc, n) => {
@@ -38,7 +32,6 @@ class Branch {
                 return acc;
             }, {})
             : options.pathMap;
-        this.then = options.then;
     }
     compile(writer, reader) {
         return write_js_1.ChannelWrite.registerWriter(new utils_js_1.RunnableCallable({
@@ -57,6 +50,9 @@ class Branch {
         else {
             destinations = result;
         }
+        if (destinations.some((dest) => !dest)) {
+            throw new Error("Branch condition returned unknown or null destination");
+        }
         return writer(destinations);
     }
 }
@@ -207,26 +203,8 @@ class Graph {
     validate(interrupt) {
         // assemble sources
         const allSources = new Set([...this.allEdges].map(([src, _]) => src));
-        for (const [start, branches] of Object.entries(this.branches)) {
+        for (const [start] of Object.entries(this.branches)) {
             allSources.add(start);
-            for (const branch of Object.values(branches)) {
-                if (branch.then) {
-                    if (branch.ends) {
-                        for (const end of Object.values(branch.ends)) {
-                            if (end !== exports.END) {
-                                allSources.add(end);
-                            }
-                        }
-                    }
-                    else {
-                        for (const node of Object.keys(this.nodes)) {
-                            if (node !== start) {
-                                allSources.add(node);
-                            }
-                        }
-                    }
-                }
-            }
         }
         // validate sources
         for (const node of Object.keys(this.nodes)) {
@@ -243,9 +221,6 @@ class Graph {
         const allTargets = new Set([...this.allEdges].map(([_, target]) => target));
         for (const [start, branches] of Object.entries(this.branches)) {
             for (const branch of Object.values(branches)) {
-                if (branch.then) {
-                    allTargets.add(branch.then);
-                }
                 if (branch.ends) {
                     for (const end of Object.values(branch.ends)) {
                         allTargets.add(end);
@@ -254,7 +229,7 @@ class Graph {
                 else {
                     allTargets.add(exports.END);
                     for (const node of Object.keys(this.nodes)) {
-                        if (node !== start && node !== branch.then) {
+                        if (node !== start) {
                             allTargets.add(node);
                         }
                     }

package/dist/graph/graph.d.ts

@@ -11,12 +11,10 @@ export interface BranchOptions<IO, N extends string> {
     source: N;
     path: Branch<IO, N>["condition"];
     pathMap?: Record<string, N | typeof END> | N[];
-    then?: N | typeof END;
 }
 export declare class Branch<IO, N extends string> {
     condition: (input: IO, config?: RunnableConfig) => string | string[] | Promise<string> | Promise<string[]>;
     ends?: Record<string, N | typeof END>;
-    then?: BranchOptions<IO, N>["then"];
     constructor(options: Omit<BranchOptions<IO, N>, "source">);
     compile(writer: (dests: string[]) => Runnable | undefined, reader?: (config: RunnableConfig) => IO): RunnableCallable<unknown, unknown>;
     _route(input: IO, config: RunnableConfig, writer: (dests: string[]) => Runnable | undefined, reader?: (config: RunnableConfig) => IO): Promise<Runnable | undefined>;

package/dist/graph/graph.js

@@ -22,12 +22,6 @@ export class Branch {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "then", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         this.condition = options.path;
         this.ends = Array.isArray(options.pathMap)
             ? options.pathMap.reduce((acc, n) => {
@@ -35,7 +29,6 @@ export class Branch {
                 return acc;
             }, {})
             : options.pathMap;
-        this.then = options.then;
     }
     compile(writer, reader) {
         return ChannelWrite.registerWriter(new RunnableCallable({
@@ -54,6 +47,9 @@ export class Branch {
         else {
             destinations = result;
         }
+        if (destinations.some((dest) => !dest)) {
+            throw new Error("Branch condition returned unknown or null destination");
+        }
         return writer(destinations);
     }
 }
@@ -203,26 +199,8 @@ export class Graph {
     validate(interrupt) {
         // assemble sources
         const allSources = new Set([...this.allEdges].map(([src, _]) => src));
-        for (const [start, branches] of Object.entries(this.branches)) {
+        for (const [start] of Object.entries(this.branches)) {
             allSources.add(start);
-            for (const branch of Object.values(branches)) {
-                if (branch.then) {
-                    if (branch.ends) {
-                        for (const end of Object.values(branch.ends)) {
-                            if (end !== END) {
-                                allSources.add(end);
-                            }
-                        }
-                    }
-                    else {
-                        for (const node of Object.keys(this.nodes)) {
-                            if (node !== start) {
-                                allSources.add(node);
-                            }
-                        }
-                    }
-                }
-            }
         }
         // validate sources
         for (const node of Object.keys(this.nodes)) {
@@ -239,9 +217,6 @@ export class Graph {
         const allTargets = new Set([...this.allEdges].map(([_, target]) => target));
         for (const [start, branches] of Object.entries(this.branches)) {
             for (const branch of Object.values(branches)) {
-                if (branch.then) {
-                    allTargets.add(branch.then);
-                }
                 if (branch.ends) {
                     for (const end of Object.values(branch.ends)) {
                         allTargets.add(end);
@@ -250,7 +225,7 @@ export class Graph {
                 else {
                     allTargets.add(END);
                     for (const node of Object.keys(this.nodes)) {
-                        if (node !== start && node !== branch.then) {
+                        if (node !== start) {
                             allTargets.add(node);
                         }
                     }

package/dist/graph/state.cjs

@@ -11,7 +11,6 @@ const ephemeral_value_js_1 = require("../channels/ephemeral_value.cjs");
 const utils_js_1 = require("../utils.cjs");
 const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
-const dynamic_barrier_value_js_1 = require("../channels/dynamic_barrier_value.cjs");
 const ROOT = "__root__";
 class StateGraph extends graph_js_1.Graph {
     constructor(fields) {
@@ -241,12 +240,6 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
                 channel: `branch:${start}:${name}:${dest}`,
                 value: start,
             }));
-            if (branch.then && branch.then !== graph_js_1.END) {
-                writes.push({
-                    channel: `branch:${start}:${name}:then`,
-                    value: { __names: filteredDests },
-                });
-            }
             return new write_js_1.ChannelWrite(writes, [constants_js_1.TAG_HIDDEN]);
         }, 
         // reader
@@ -254,7 +247,7 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
         // attach branch subscribers
         const ends = branch.ends
             ? Object.values(branch.ends)
-            : Object.keys(this.builder.nodes).filter((n) => n !== branch.then);
+            : Object.keys(this.builder.nodes);
         for (const end of ends) {
             if (end === graph_js_1.END) {
                 continue;
@@ -264,18 +257,6 @@ class CompiledStateGraph extends graph_js_1.CompiledGraph {
                 new ephemeral_value_js_1.EphemeralValue();
             this.nodes[end].triggers.push(channelName);
         }
-        if (branch.then && branch.then !== graph_js_1.END) {
-            const channelName = `branch:${start}:${name}:then`;
-            this.channels[channelName] =
-                new dynamic_barrier_value_js_1.DynamicBarrierValue();
-            this.nodes[branch.then].triggers.push(channelName);
-            for (const end of ends) {
-                if (end === graph_js_1.END) {
-                    continue;
-                }
-                this.nodes[end].writers.push(new write_js_1.ChannelWrite([{ channel: channelName, value: end }], [constants_js_1.TAG_HIDDEN]));
-            }
-        }
     }
 }
 exports.CompiledStateGraph = CompiledStateGraph;

package/dist/graph/state.js

@@ -8,7 +8,6 @@ import { EphemeralValue } from "../channels/ephemeral_value.js";
 import { RunnableCallable } from "../utils.js";
 import { TAG_HIDDEN } from "../constants.js";
 import { InvalidUpdateError } from "../errors.js";
-import { DynamicBarrierValue } from "../channels/dynamic_barrier_value.js";
 const ROOT = "__root__";
 export class StateGraph extends Graph {
     constructor(fields) {
@@ -237,12 +236,6 @@ export class CompiledStateGraph extends CompiledGraph {
                 channel: `branch:${start}:${name}:${dest}`,
                 value: start,
             }));
-            if (branch.then && branch.then !== END) {
-                writes.push({
-                    channel: `branch:${start}:${name}:then`,
-                    value: { __names: filteredDests },
-                });
-            }
             return new ChannelWrite(writes, [TAG_HIDDEN]);
         }, 
         // reader
@@ -250,7 +243,7 @@ export class CompiledStateGraph extends CompiledGraph {
         // attach branch subscribers
         const ends = branch.ends
             ? Object.values(branch.ends)
-            : Object.keys(this.builder.nodes).filter((n) => n !== branch.then);
+            : Object.keys(this.builder.nodes);
         for (const end of ends) {
             if (end === END) {
                 continue;
@@ -260,17 +253,5 @@ export class CompiledStateGraph extends CompiledGraph {
                 new EphemeralValue();
             this.nodes[end].triggers.push(channelName);
         }
-        if (branch.then && branch.then !== END) {
-            const channelName = `branch:${start}:${name}:then`;
-            this.channels[channelName] =
-                new DynamicBarrierValue();
-            this.nodes[branch.then].triggers.push(channelName);
-            for (const end of ends) {
-                if (end === END) {
-                    continue;
-                }
-                this.nodes[end].writers.push(new ChannelWrite([{ channel: channelName, value: end }], [TAG_HIDDEN]));
-            }
-        }
     }
 }

package/dist/prebuilt/react_agent_executor.cjs

@@ -7,8 +7,8 @@ const prompts_1 = require("@langchain/core/prompts");
 const index_js_1 = require("../graph/index.cjs");
 const tool_node_js_1 = require("./tool_node.cjs");
 /**
- * Creates a StateGraph agent that relies on a chat model utilizing tool calling.
- * @param model The chat model that can utilize OpenAI-style function calling.
+ * Creates a StateGraph agent that relies on a chat llm utilizing tool calling.
+ * @param llm The chat llm that can utilize OpenAI-style function calling.
  * @param tools A list of tools or a ToolNode.
  * @param messageModifier An optional message modifier to apply to messages before being passed to the LLM.
  * Can be a SystemMessage, string, function that takes and returns a list of messages, or a Runnable.
@@ -17,7 +17,8 @@ const tool_node_js_1 = require("./tool_node.cjs");
  * @param interruptAfter An optional list of node names to interrupt after running.
  * @returns A compiled agent as a LangChain Runnable.
  */
-function createReactAgent(model, tools, messageModifier, checkpointSaver, interruptBefore, interruptAfter) {
+function createReactAgent(props) {
+    const { llm, tools, messageModifier, checkpointSaver, interruptBefore, interruptAfter, } = props;
     const schema = {
         messages: {
             value: (left, right) => left.concat(right),
@@ -31,10 +32,10 @@ function createReactAgent(model, tools, messageModifier, checkpointSaver, interr
     else {
         toolClasses = tools;
     }
-    if (!("bindTools" in model) || typeof model.bindTools !== "function") {
-        throw new Error(`Model ${model} must define bindTools method.`);
+    if (!("bindTools" in llm) || typeof llm.bindTools !== "function") {
+        throw new Error(`llm ${llm} must define bindTools method.`);
     }
-    const modelWithTools = model.bindTools(toolClasses);
+    const modelWithTools = llm.bindTools(toolClasses);
     const modelRunnable = _createModelWrapper(modelWithTools, messageModifier);
     const shouldContinue = (state) => {
         const { messages } = state;
@@ -60,7 +61,7 @@ function createReactAgent(model, tools, messageModifier, checkpointSaver, interr
         .addEdge(index_js_1.START, "agent")
         .addConditionalEdges("agent", shouldContinue, {
         continue: "tools",
-        end: index_js_1.END,
+        [index_js_1.END]: index_js_1.END,
     })
         .addEdge("tools", "agent");
     return workflow.compile({

package/dist/prebuilt/react_agent_executor.d.ts

@@ -12,9 +12,17 @@ export interface AgentState {
     messages: BaseMessage[];
 }
 export type N = typeof START | "agent" | "tools";
+export type CreateReactAgentParams = {
+    llm: BaseChatModel;
+    tools: ToolNode<MessagesState> | StructuredTool[];
+    messageModifier?: SystemMessage | string | ((messages: BaseMessage[]) => BaseMessage[]) | ((messages: BaseMessage[]) => Promise<BaseMessage[]>) | Runnable;
+    checkpointSaver?: BaseCheckpointSaver;
+    interruptBefore?: N[] | All;
+    interruptAfter?: N[] | All;
+};
 /**
- * Creates a StateGraph agent that relies on a chat model utilizing tool calling.
- * @param model The chat model that can utilize OpenAI-style function calling.
+ * Creates a StateGraph agent that relies on a chat llm utilizing tool calling.
+ * @param llm The chat llm that can utilize OpenAI-style function calling.
  * @param tools A list of tools or a ToolNode.
  * @param messageModifier An optional message modifier to apply to messages before being passed to the LLM.
  * Can be a SystemMessage, string, function that takes and returns a list of messages, or a Runnable.
@@ -23,4 +31,4 @@ export type N = typeof START | "agent" | "tools";
  * @param interruptAfter An optional list of node names to interrupt after running.
  * @returns A compiled agent as a LangChain Runnable.
  */
-export declare function createReactAgent(model: BaseChatModel, tools: ToolNode<MessagesState> | StructuredTool[], messageModifier?: SystemMessage | string | ((messages: BaseMessage[]) => BaseMessage[]) | Runnable, checkpointSaver?: BaseCheckpointSaver, interruptBefore?: N[] | All, interruptAfter?: N[] | All): CompiledStateGraph<AgentState, Partial<AgentState>, typeof START | "agent" | "tools">;
+export declare function createReactAgent(props: CreateReactAgentParams): CompiledStateGraph<AgentState, Partial<AgentState>, typeof START | "agent" | "tools">;

package/dist/prebuilt/react_agent_executor.js

@@ -4,8 +4,8 @@ import { ChatPromptTemplate } from "@langchain/core/prompts";
 import { END, START, StateGraph } from "../graph/index.js";
 import { ToolNode } from "./tool_node.js";
 /**
- * Creates a StateGraph agent that relies on a chat model utilizing tool calling.
- * @param model The chat model that can utilize OpenAI-style function calling.
+ * Creates a StateGraph agent that relies on a chat llm utilizing tool calling.
+ * @param llm The chat llm that can utilize OpenAI-style function calling.
  * @param tools A list of tools or a ToolNode.
  * @param messageModifier An optional message modifier to apply to messages before being passed to the LLM.
  * Can be a SystemMessage, string, function that takes and returns a list of messages, or a Runnable.
@@ -14,7 +14,8 @@ import { ToolNode } from "./tool_node.js";
  * @param interruptAfter An optional list of node names to interrupt after running.
  * @returns A compiled agent as a LangChain Runnable.
  */
-export function createReactAgent(model, tools, messageModifier, checkpointSaver, interruptBefore, interruptAfter) {
+export function createReactAgent(props) {
+    const { llm, tools, messageModifier, checkpointSaver, interruptBefore, interruptAfter, } = props;
     const schema = {
         messages: {
             value: (left, right) => left.concat(right),
@@ -28,10 +29,10 @@ export function createReactAgent(model, tools, messageModifier, checkpointSaver,
     else {
         toolClasses = tools;
     }
-    if (!("bindTools" in model) || typeof model.bindTools !== "function") {
-        throw new Error(`Model ${model} must define bindTools method.`);
+    if (!("bindTools" in llm) || typeof llm.bindTools !== "function") {
+        throw new Error(`llm ${llm} must define bindTools method.`);
     }
-    const modelWithTools = model.bindTools(toolClasses);
+    const modelWithTools = llm.bindTools(toolClasses);
     const modelRunnable = _createModelWrapper(modelWithTools, messageModifier);
     const shouldContinue = (state) => {
         const { messages } = state;
@@ -57,7 +58,7 @@ export function createReactAgent(model, tools, messageModifier, checkpointSaver,
         .addEdge(START, "agent")
         .addConditionalEdges("agent", shouldContinue, {
         continue: "tools",
-        end: END,
+        [END]: END,
     })
         .addEdge("tools", "agent");
     return workflow.compile({

package/dist/pregel/index.cjs

@@ -222,6 +222,7 @@ class Pregel extends runnables_1.Runnable {
             next: nextTasks.map((task) => task.name),
             metadata: saved?.metadata,
             config: saved ? saved.config : config,
+            createdAt: saved?.checkpoint.ts,
             parentConfig: saved?.parentConfig,
         };
     }
@@ -238,6 +239,7 @@ class Pregel extends runnables_1.Runnable {
                 next: nextTasks.map((task) => task.name),
                 metadata: saved.metadata,
                 config: saved.config,
+                createdAt: saved.checkpoint.ts,
                 parentConfig: saved.parentConfig,
             };
         }
@@ -480,7 +482,7 @@ class Pregel extends runnables_1.Runnable {
                     bg.push(this.checkpointer.put(checkpointConfig, checkpoint, {
                         source: "loop",
                         step,
-                        writes: (0, io_js_1.single)(streamMode === "values"
+                        writes: (0, io_js_1.single)(this.streamMode === "values"
                             ? (0, io_js_1.mapOutputValues)(outputKeys, pendingWrites, channels)
                             : (0, io_js_1.mapOutputUpdates)(outputKeys, nextTasks)),
                     }));

package/dist/pregel/index.js

@@ -218,6 +218,7 @@ export class Pregel extends Runnable {
             next: nextTasks.map((task) => task.name),
             metadata: saved?.metadata,
             config: saved ? saved.config : config,
+            createdAt: saved?.checkpoint.ts,
             parentConfig: saved?.parentConfig,
         };
     }
@@ -234,6 +235,7 @@ export class Pregel extends Runnable {
                 next: nextTasks.map((task) => task.name),
                 metadata: saved.metadata,
                 config: saved.config,
+                createdAt: saved.checkpoint.ts,
                 parentConfig: saved.parentConfig,
             };
         }
@@ -476,7 +478,7 @@ export class Pregel extends Runnable {
                     bg.push(this.checkpointer.put(checkpointConfig, checkpoint, {
                         source: "loop",
                         step,
-                        writes: single(streamMode === "values"
+                        writes: single(this.streamMode === "values"
                             ? mapOutputValues(outputKeys, pendingWrites, channels)
                             : mapOutputUpdates(outputKeys, nextTasks)),
                     }));

package/dist/pregel/io.cjs

@@ -138,6 +138,6 @@ function single(iter) {
     for (const value of iter) {
         return value;
     }
-    return undefined;
+    return null;
 }
 exports.single = single;

package/dist/pregel/io.d.ts

@@ -14,4 +14,4 @@ export declare function mapOutputValues<C extends PropertyKey>(outputChannels: C
  * Map pending writes (a sequence of tuples (channel, value)) to output chunk.
  */
 export declare function mapOutputUpdates<N extends PropertyKey, C extends PropertyKey>(outputChannels: C | Array<C>, tasks: readonly PregelExecutableTask<N, C>[]): Generator<Record<N, any | Record<string, any>>>;
-export declare function single<T>(iter: IterableIterator<T>): T | undefined;
+export declare function single<T>(iter: IterableIterator<T>): T | null;

package/dist/pregel/io.js

@@ -130,5 +130,5 @@ export function single(iter) {
     for (const value of iter) {
         return value;
     }
-    return undefined;
+    return null;
 }

package/dist/pregel/types.d.ts

@@ -28,6 +28,10 @@ export interface StateSnapshot {
      * Metadata about the checkpoint
      */
     readonly metadata?: CheckpointMetadata;
+    /**
+     * Time when the snapshot was created
+     */
+    readonly createdAt?: string;
     /**
      * Config used to fetch the parent snapshot, if any
      * @default undefined

package/dist/tests/chatbot.int.test.js

@@ -1,7 +1,7 @@
 import { describe, it } from "@jest/globals";
 import { ChatOpenAI } from "@langchain/openai";
 import { HumanMessage, ToolMessage, } from "@langchain/core/messages";
-import { Calculator } from "langchain/tools/calculator";
+import { Calculator } from "@langchain/community/tools/calculator";
 import { convertToOpenAITool } from "@langchain/core/utils/function_calling";
 import { END, MessageGraph, START } from "../index.js";
 describe("Chatbot", () => {

package/dist/tests/checkpoints.test.js

@@ -69,7 +69,7 @@ describe("MemorySaver", () => {
     it("should save and retrieve checkpoints correctly", async () => {
         const memorySaver = new MemorySaver();
         // save checkpoint
-        const runnableConfig = await memorySaver.put({ configurable: { thread_id: "1" } }, checkpoint1, { source: "update", step: -1 });
+        const runnableConfig = await memorySaver.put({ configurable: { thread_id: "1" } }, checkpoint1, { source: "update", step: -1, writes: null });
         expect(runnableConfig).toEqual({
             configurable: {
                 thread_id: "1",
@@ -91,6 +91,7 @@ describe("MemorySaver", () => {
         await memorySaver.put({ configurable: { thread_id: "1" } }, checkpoint2, {
             source: "update",
             step: -1,
+            writes: null,
         });
         // list checkpoints
         const checkpointTupleGenerator = await memorySaver.list({
@@ -116,7 +117,7 @@ describe("SqliteSaver", () => {
         });
         expect(undefinedCheckpoint).toBeUndefined();
         // save first checkpoint
-        const runnableConfig = await sqliteSaver.put({ configurable: { thread_id: "1" } }, checkpoint1, { source: "update", step: -1 });
+        const runnableConfig = await sqliteSaver.put({ configurable: { thread_id: "1" } }, checkpoint1, { source: "update", step: -1, writes: null });
         expect(runnableConfig).toEqual({
             configurable: {
                 thread_id: "1",
@@ -141,7 +142,7 @@ describe("SqliteSaver", () => {
                 thread_id: "1",
                 checkpoint_id: "2024-04-18T17:19:07.952Z",
             },
-        }, checkpoint2, { source: "update", step: -1 });
+        }, checkpoint2, { source: "update", step: -1, writes: null });
         // verify that parentTs is set and retrieved correctly for second checkpoint
         const secondCheckpointTuple = await sqliteSaver.getTuple({
             configurable: { thread_id: "1" },

package/dist/tests/prebuilt.int.test.js

@@ -120,7 +120,7 @@ describe("createReactAgent", () => {
             }
         }
         const tools = [new SanFranciscoWeatherTool()];
-        const reactAgent = createReactAgent(model, tools);
+        const reactAgent = createReactAgent({ llm: model, tools });
         const response = await reactAgent.invoke({
             messages: [new HumanMessage("What's the weather like in SF?")],
         });
@@ -156,7 +156,7 @@ describe("createReactAgent", () => {
             }
         }
         const tools = [new SanFranciscoWeatherTool()];
-        const reactAgent = createReactAgent(model, tools);
+        const reactAgent = createReactAgent({ llm: model, tools });
         const stream = await reactAgent.stream({
             messages: [new HumanMessage("What's the weather like in SF?")],
         }, { streamMode: "values" });

package/dist/tests/prebuilt.test.js

@@ -299,7 +299,11 @@ describe("createReactAgent", () => {
                 new AIMessage("result2"),
             ],
         });
-        const agent = createReactAgent(llm, tools, "You are a helpful assistant");
+        const agent = createReactAgent({
+            llm,
+            tools,
+            messageModifier: "You are a helpful assistant",
+        });
         const result = await agent.invoke({
             messages: [new HumanMessage("Hello Input!")],
         });
@@ -331,11 +335,14 @@ describe("createReactAgent", () => {
                 new AIMessage("result2"),
             ],
         });
-        const agent = createReactAgent(llm, tools, new SystemMessage("You are a helpful assistant"));
+        const agent = createReactAgent({
+            llm,
+            tools,
+            messageModifier: new SystemMessage("You are a helpful assistant"),
+        });
         const result = await agent.invoke({
             messages: [],
         });
-        console.log("RESULT THING", result);
         expect(result.messages).toEqual([
             new AIMessage({
                 content: "result1",
@@ -366,7 +373,37 @@ describe("createReactAgent", () => {
             new SystemMessage("You are a helpful assistant"),
             ...messages,
         ];
-        const agent = createReactAgent(llm, tools, messageModifier);
+        const agent = createReactAgent({ llm, tools, messageModifier });
+        const result = await agent.invoke({
+            messages: [new HumanMessage("Hello Input!")],
+        });
+        expect(result.messages).toEqual([
+            new HumanMessage("Hello Input!"),
+            aiM1,
+            new ToolMessage({
+                name: "search_api",
+                content: "result for foo",
+                tool_call_id: "tool_abcd123",
+            }),
+            aiM2,
+        ]);
+    });
+    it("Can use async custom function message modifier", async () => {
+        const aiM1 = new AIMessage({
+            content: "result1",
+            tool_calls: [
+                { name: "search_api", id: "tool_abcd123", args: { query: "foo" } },
+            ],
+        });
+        const aiM2 = new AIMessage("result2");
+        const llm = new FakeToolCallingChatModel({
+            responses: [aiM1, aiM2],
+        });
+        const messageModifier = async (messages) => [
+            new SystemMessage("You are a helpful assistant"),
+            ...messages,
+        ];
+        const agent = createReactAgent({ llm, tools, messageModifier });
         const result = await agent.invoke({
             messages: [new HumanMessage("Hello Input!")],
         });
@@ -398,7 +435,7 @@ describe("createReactAgent", () => {
                 ...messages,
             ],
         });
-        const agent = createReactAgent(llm, tools, messageModifier);
+        const agent = createReactAgent({ llm, tools, messageModifier });
         const result = await agent.invoke({
             messages: [
                 new HumanMessage("Hello Input!"),

package/dist/tests/pregel.test.js

@@ -1777,8 +1777,9 @@ it("StateGraph start branch then end", async () => {
         .addConditionalEdges({
         source: START,
         path: (state) => state.market === "DE" ? "tool_two_slow" : "tool_two_fast",
-        then: END,
-    });
+    })
+        .addEdge("tool_two_fast", END)
+        .addEdge("tool_two_slow", END);
     const toolTwo = toolTwoBuilder.compile();
     expect(await toolTwo.invoke({ my_key: "value", market: "DE" })).toEqual({
         my_key: "value slow",
@@ -1793,71 +1794,45 @@ it("StateGraph start branch then end", async () => {
         interruptBefore: ["tool_two_fast", "tool_two_slow"],
     });
     await expect(() => toolTwoWithCheckpointer.invoke({ my_key: "value", market: "DE" })).rejects.toThrowError("thread_id");
-    // const thread1 = { configurable: { thread_id: "1" } }
-    // expect(toolTwoWithCheckpointer.invoke({ my_key: "value", market: "DE" }, thread1)).toEqual({ my_key: "value", market: "DE" })
-    // expect(toolTwoWithCheckpointer.getState(thread1)).toEqual({
-    //   values: { my_key: "value", market: "DE" },
-    //   next: ["tool_two_slow"],
-    //   config: toolTwoWithCheckpointer.checkpointer.getTuple(thread1).config,
-    //   metadata: { source: "loop", step: 0, writes: null },
-    //   parentConfig: [...toolTwoWithCheckpointer.checkpointer.list(thread1, { limit: 2 })].pop().config
-    // })
-    // expect(toolTwoWithCheckpointer.invoke(null, thread1, { debug: 1 })).toEqual({ my_key: "value slow", market: "DE" })
-    // expect(toolTwoWithCheckpointer.getState(thread1)).toEqual({
-    //   values: { my_key
-    //     : "value slow", market: "DE" },
-    //   next: [],
-    //   config: (await toolTwoWithCheckpointer.checkpointer!.getTuple(thread1))!.config,
-    //   metadata: { source: "loop", step: 1, writes: { tool_two_slow: { my_key: " slow" } } },
-    //   parentConfig: [...toolTwoWithCheckpointer.checkpointer!.list(thread1, { limit: 2 })].pop().config
-});
-/**
- * def test_branch_then_node(snapshot: SnapshotAssertion) -> None:
-    class State(TypedDict):
-        my_key: Annotated[str, operator.add]
-        market: str
-
-    # this graph is invalid because there is no path to "finish"
-    invalid_graph = StateGraph(State)
-    invalid_graph.set_entry_point("prepare")
-    invalid_graph.set_finish_point("finish")
-    invalid_graph.add_conditional_edges(
-        source="prepare",
-        path=lambda s: "tool_two_slow" if s["market"] == "DE" else "tool_two_fast",
-        path_map=["tool_two_slow", "tool_two_fast"],
-    )
-    invalid_graph.add_node("prepare", lambda s: {"my_key": " prepared"})
-    invalid_graph.add_node("tool_two_slow", lambda s: {"my_key": " slow"})
-    invalid_graph.add_node("tool_two_fast", lambda s: {"my_key": " fast"})
-    invalid_graph.add_node("finish", lambda s: {"my_key": " finished"})
-    with pytest.raises(ValueError):
-        invalid_graph.compile()
-
-    tool_two_graph = StateGraph(State)
-    tool_two_graph.set_entry_point("prepare")
-    tool_two_graph.set_finish_point("finish")
-    tool_two_graph.add_conditional_edges(
-        source="prepare",
-        path=lambda s: "tool_two_slow" if s["market"] == "DE" else "tool_two_fast",
-        then="finish",
-    )
-    tool_two_graph.add_node("prepare", lambda s: {"my_key": " prepared"})
-    tool_two_graph.add_node("tool_two_slow", lambda s: {"my_key": " slow"})
-    tool_two_graph.add_node("tool_two_fast", lambda s: {"my_key": " fast"})
-    tool_two_graph.add_node("finish", lambda s: {"my_key": " finished"})
-    tool_two = tool_two_graph.compile()
-    assert tool_two.get_graph().draw_mermaid(with_styles=False) == snapshot
-    assert tool_two.get_graph().draw_mermaid() == snapshot
-
-    assert tool_two.invoke({"my_key": "value", "market": "DE"}, debug=1) == {
-        "my_key": "value prepared slow finished",
-        "market": "DE",
-    }
-    assert tool_two.invoke({"my_key": "value", "market": "US"}) == {
-        "my_key": "value prepared fast finished",
-        "market": "US",
+    async function last(iter) {
+        // eslint-disable-next-line no-undef-init
+        let value = undefined;
+        for await (value of iter) {
+            // do nothing
+        }
+        return value;
     }
- */
+    const thread1 = { configurable: { thread_id: "1" } };
+    expect(await toolTwoWithCheckpointer.invoke({ my_key: "value", market: "DE" }, thread1)).toEqual({ my_key: "value", market: "DE" });
+    expect(await toolTwoWithCheckpointer.getState(thread1)).toEqual({
+        values: { my_key: "value", market: "DE" },
+        next: ["tool_two_slow"],
+        config: (await toolTwoWithCheckpointer.checkpointer.getTuple(thread1))
+            .config,
+        createdAt: (await toolTwoWithCheckpointer.checkpointer.getTuple(thread1))
+            .checkpoint.ts,
+        metadata: { source: "loop", step: 0, writes: null },
+        parentConfig: (await last(toolTwoWithCheckpointer.checkpointer.list(thread1, 2))).config,
+    });
+    expect(await toolTwoWithCheckpointer.invoke(null, thread1)).toEqual({
+        my_key: "value slow",
+        market: "DE",
+    });
+    expect(await toolTwoWithCheckpointer.getState(thread1)).toEqual({
+        values: { my_key: "value slow", market: "DE" },
+        next: [],
+        config: (await toolTwoWithCheckpointer.checkpointer.getTuple(thread1))
+            .config,
+        createdAt: (await toolTwoWithCheckpointer.checkpointer.getTuple(thread1))
+            .checkpoint.ts,
+        metadata: {
+            source: "loop",
+            step: 1,
+            writes: { tool_two_slow: { my_key: " slow" } },
+        },
+        parentConfig: (await last(toolTwoWithCheckpointer.checkpointer.list(thread1, 2))).config,
+    });
+});
 it("StateGraph branch then node", async () => {
     const invalidBuilder = new StateGraph({
         channels: {
@@ -1891,8 +1866,9 @@ it("StateGraph branch then node", async () => {
         .addConditionalEdges({
         source: "prepare",
         path: (state) => state.market === "DE" ? "tool_two_slow" : "tool_two_fast",
-        then: "finish",
     })
+        .addEdge("tool_two_fast", "finish")
+        .addEdge("tool_two_slow", "finish")
         .addEdge("finish", END);
     const tool = toolBuilder.compile();
     expect(await tool.invoke({ my_key: "value", market: "DE" })).toEqual({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.0.18",
+  "version": "0.0.20",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -37,13 +37,14 @@
   "author": "LangChain",
   "license": "MIT",
   "dependencies": {
-    "@langchain/core": "^0.1.61",
+    "@langchain/core": ">0.1.61 <0.3.0",
     "uuid": "^9.0.1"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
-    "@langchain/community": "^0.0.43",
-    "@langchain/openai": "latest",
+    "@langchain/anthropic": "^0.1.21",
+    "@langchain/community": "^0.2.2",
+    "@langchain/openai": "^0.0.33",
     "@langchain/scripts": "^0.0.13",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",
@@ -63,7 +64,7 @@
     "eslint-plugin-prettier": "^4.2.1",
     "jest": "^29.5.0",
     "jest-environment-node": "^29.6.4",
-    "langchain": "^0.1.29",
+    "langchain": "^0.2.1",
     "prettier": "^2.8.3",
     "release-it": "^15.10.1",
     "rollup": "^4.5.2",