@langchain/langgraph 0.2.10 → 0.2.12

package/README.md

@@ -59,7 +59,7 @@ import { AIMessage, BaseMessage, HumanMessage } from "@langchain/core/messages";
 import { tool } from "@langchain/core/tools";
 import { z } from "zod";
 import { ChatAnthropic } from "@langchain/anthropic";
-import { StateGraph, StateGraphArgs } from "@langchain/langgraph";
+import { StateGraph } from "@langchain/langgraph";
 import { MemorySaver, Annotation } from "@langchain/langgraph";
 import { ToolNode } from "@langchain/langgraph/prebuilt";
 

package/dist/graph/graph.cjs

@@ -5,6 +5,7 @@ exports.CompiledGraph = exports.Graph = exports.Branch = exports.END = exports.S
 const runnables_1 = require("@langchain/core/runnables");
 const graph_1 = require("@langchain/core/runnables/graph");
 const zod_1 = require("zod");
+const uuid_1 = require("uuid");
 const read_js_1 = require("../pregel/read.cjs");
 const index_js_1 = require("../pregel/index.cjs");
 const ephemeral_value_js_1 = require("../channels/ephemeral_value.cjs");
@@ -358,50 +359,126 @@ class CompiledGraph extends index_js_1.Pregel {
                 schema: zod_1.z.any(),
             }, exports.START),
         };
-        const endNodes = {
-            [exports.END]: graph.addNode({
-                schema: zod_1.z.any(),
-            }, exports.END),
-        };
-        for (const [key, node] of Object.entries(this.builder.nodes)) {
-            if (config?.xray) {
-                const subgraph = isCompiledGraph(node)
-                    ? node.getGraph({
+        const endNodes = {};
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        let subgraphs = {};
+        if (xray) {
+            subgraphs = Object.fromEntries((0, utils_js_1.gatherIteratorSync)(this.getSubgraphs()).filter(
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            (x) => isCompiledGraph(x[1])));
+        }
+        function addEdge(start, end, label, conditional = false) {
+            if (end === exports.END && endNodes[exports.END] === undefined) {
+                endNodes[exports.END] = graph.addNode({ schema: zod_1.z.any() }, exports.END);
+            }
+            return graph.addEdge(startNodes[start], endNodes[end], label !== end ? label : undefined, conditional);
+        }
+        for (const [key, nodeSpec] of Object.entries(this.builder.nodes)) {
+            const displayKey = _escapeMermaidKeywords(key);
+            const node = nodeSpec.runnable;
+            const metadata = nodeSpec.metadata ?? {};
+            if (this.interruptBefore?.includes(key) &&
+                this.interruptAfter?.includes(key)) {
+                metadata.__interrupt = "before,after";
+            }
+            else if (this.interruptBefore?.includes(key)) {
+                metadata.__interrupt = "before";
+            }
+            else if (this.interruptAfter?.includes(key)) {
+                metadata.__interrupt = "after";
+            }
+            if (xray) {
+                const newXrayValue = typeof xray === "number" ? xray - 1 : xray;
+                const drawableSubgraph = subgraphs[key] !== undefined
+                    ? subgraphs[key].getGraph({
                         ...config,
-                        xray: typeof xray === "number" && xray > 0 ? xray - 1 : xray,
+                        xray: newXrayValue,
                     })
-                    : node.runnable.getGraph(config);
-                subgraph.trimFirstNode();
-                subgraph.trimLastNode();
-                if (Object.keys(subgraph.nodes).length > 1) {
-                    const [newEndNode, newStartNode] = graph.extend(subgraph, key);
-                    if (newEndNode !== undefined) {
-                        endNodes[key] = newEndNode;
+                    : node.getGraph(config);
+                drawableSubgraph.trimFirstNode();
+                drawableSubgraph.trimLastNode();
+                if (Object.keys(drawableSubgraph.nodes).length > 1) {
+                    const [e, s] = graph.extend(drawableSubgraph, displayKey);
+                    if (e === undefined) {
+                        throw new Error(`Could not extend subgraph "${key}" due to missing entrypoint.`);
+                    }
+                    // TODO: Remove default name once we stop supporting core 0.2.0
+                    // eslint-disable-next-line no-inner-declarations
+                    function _isRunnableInterface(
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    thing) {
+                        return thing ? thing.lc_runnable : false;
                     }
-                    if (newStartNode !== undefined) {
-                        startNodes[key] = newStartNode;
+                    // eslint-disable-next-line no-inner-declarations
+                    function _nodeDataStr(id, data) {
+                        if (id !== undefined && !(0, uuid_1.validate)(id)) {
+                            return id;
+                        }
+                        else if (_isRunnableInterface(data)) {
+                            try {
+                                let dataStr = data.getName();
+                                dataStr = dataStr.startsWith("Runnable")
+                                    ? dataStr.slice("Runnable".length)
+                                    : dataStr;
+                                return dataStr;
+                            }
+                            catch (error) {
+                                return data.getName();
+                            }
+                        }
+                        else {
+                            return data.name ?? "UnknownSchema";
+                        }
                     }
+                    // TODO: Remove casts when we stop supporting core 0.2.0
+                    if (s !== undefined) {
+                        startNodes[displayKey] = {
+                            name: _nodeDataStr(s.id, s.data),
+                            ...s,
+                        };
+                    }
+                    endNodes[displayKey] = {
+                        name: _nodeDataStr(e.id, e.data),
+                        ...e,
+                    };
                 }
                 else {
-                    const newNode = graph.addNode(node.runnable, key);
-                    startNodes[key] = newNode;
-                    endNodes[key] = newNode;
+                    // TODO: Remove when we stop supporting core 0.2.0
+                    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+                    // @ts-ignore
+                    const newNode = graph.addNode(node, displayKey, metadata);
+                    startNodes[displayKey] = newNode;
+                    endNodes[displayKey] = newNode;
                 }
             }
             else {
-                const newNode = graph.addNode(node.runnable, key);
-                startNodes[key] = newNode;
-                endNodes[key] = newNode;
+                // TODO: Remove when we stop supporting core 0.2.0
+                // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+                // @ts-ignore
+                const newNode = graph.addNode(node, displayKey, metadata);
+                startNodes[displayKey] = newNode;
+                endNodes[displayKey] = newNode;
             }
         }
-        for (const [start, end] of this.builder.allEdges) {
-            graph.addEdge(startNodes[start], endNodes[end]);
+        const sortedEdges = [...this.builder.allEdges].sort(([a], [b]) => {
+            if (a < b) {
+                return -1;
+            }
+            else if (b > a) {
+                return 1;
+            }
+            else {
+                return 0;
+            }
+        });
+        for (const [start, end] of sortedEdges) {
+            addEdge(_escapeMermaidKeywords(start), _escapeMermaidKeywords(end));
         }
         for (const [start, branches] of Object.entries(this.builder.branches)) {
             const defaultEnds = {
                 ...Object.fromEntries(Object.keys(this.builder.nodes)
                     .filter((k) => k !== start)
-                    .map((k) => [k, k])),
+                    .map((k) => [_escapeMermaidKeywords(k), _escapeMermaidKeywords(k)])),
                 [exports.END]: exports.END,
             };
             for (const branch of Object.values(branches)) {
@@ -413,7 +490,7 @@ class CompiledGraph extends index_js_1.Pregel {
                     ends = defaultEnds;
                 }
                 for (const [label, end] of Object.entries(ends)) {
-                    graph.addEdge(startNodes[start], endNodes[end], label !== end ? label : undefined, true);
+                    addEdge(_escapeMermaidKeywords(start), _escapeMermaidKeywords(end), label, true);
                 }
             }
         }
@@ -429,3 +506,9 @@ function isCompiledGraph(x) {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         typeof x.attachEdge === "function");
 }
+function _escapeMermaidKeywords(key) {
+    if (key === "subgraph") {
+        return `"${key}"`;
+    }
+    return key;
+}

package/dist/graph/graph.d.ts

@@ -1,5 +1,5 @@
 import { Runnable, RunnableConfig, RunnableLike } from "@langchain/core/runnables";
-import { Graph as RunnableGraph } from "@langchain/core/runnables/graph";
+import { Graph as DrawableGraph } from "@langchain/core/runnables/graph";
 import { All, BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
 import { PregelNode } from "../pregel/read.js";
 import { Pregel } from "../pregel/index.js";
@@ -74,5 +74,5 @@ export declare class CompiledGraph<N extends string, RunInput = any, RunOutput =
      */
     getGraph(config?: RunnableConfig & {
         xray?: boolean | number;
-    }): RunnableGraph;
+    }): DrawableGraph;
 }

package/dist/graph/graph.js

@@ -1,13 +1,14 @@
 /* eslint-disable @typescript-eslint/no-use-before-define */
 import { _coerceToRunnable, } from "@langchain/core/runnables";
-import { Graph as RunnableGraph, } from "@langchain/core/runnables/graph";
+import { Graph as DrawableGraph, } from "@langchain/core/runnables/graph";
 import { z } from "zod";
+import { validate as isUuid } from "uuid";
 import { PregelNode } from "../pregel/read.js";
 import { Channel, Pregel } from "../pregel/index.js";
 import { EphemeralValue } from "../channels/ephemeral_value.js";
 import { ChannelWrite, PASSTHROUGH } from "../pregel/write.js";
 import { _isSend, CHECKPOINT_NAMESPACE_END, CHECKPOINT_NAMESPACE_SEPARATOR, TAG_HIDDEN, } from "../constants.js";
-import { RunnableCallable } from "../utils.js";
+import { gatherIteratorSync, RunnableCallable } from "../utils.js";
 import { InvalidUpdateError, NodeInterrupt } from "../errors.js";
 /** Special reserved node name denoting the start of a graph. */
 export const START = "__start__";
@@ -347,56 +348,132 @@ export class CompiledGraph extends Pregel {
      */
     getGraph(config) {
         const xray = config?.xray;
-        const graph = new RunnableGraph();
+        const graph = new DrawableGraph();
         const startNodes = {
             [START]: graph.addNode({
                 schema: z.any(),
             }, START),
         };
-        const endNodes = {
-            [END]: graph.addNode({
-                schema: z.any(),
-            }, END),
-        };
-        for (const [key, node] of Object.entries(this.builder.nodes)) {
-            if (config?.xray) {
-                const subgraph = isCompiledGraph(node)
-                    ? node.getGraph({
+        const endNodes = {};
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        let subgraphs = {};
+        if (xray) {
+            subgraphs = Object.fromEntries(gatherIteratorSync(this.getSubgraphs()).filter(
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            (x) => isCompiledGraph(x[1])));
+        }
+        function addEdge(start, end, label, conditional = false) {
+            if (end === END && endNodes[END] === undefined) {
+                endNodes[END] = graph.addNode({ schema: z.any() }, END);
+            }
+            return graph.addEdge(startNodes[start], endNodes[end], label !== end ? label : undefined, conditional);
+        }
+        for (const [key, nodeSpec] of Object.entries(this.builder.nodes)) {
+            const displayKey = _escapeMermaidKeywords(key);
+            const node = nodeSpec.runnable;
+            const metadata = nodeSpec.metadata ?? {};
+            if (this.interruptBefore?.includes(key) &&
+                this.interruptAfter?.includes(key)) {
+                metadata.__interrupt = "before,after";
+            }
+            else if (this.interruptBefore?.includes(key)) {
+                metadata.__interrupt = "before";
+            }
+            else if (this.interruptAfter?.includes(key)) {
+                metadata.__interrupt = "after";
+            }
+            if (xray) {
+                const newXrayValue = typeof xray === "number" ? xray - 1 : xray;
+                const drawableSubgraph = subgraphs[key] !== undefined
+                    ? subgraphs[key].getGraph({
                         ...config,
-                        xray: typeof xray === "number" && xray > 0 ? xray - 1 : xray,
+                        xray: newXrayValue,
                     })
-                    : node.runnable.getGraph(config);
-                subgraph.trimFirstNode();
-                subgraph.trimLastNode();
-                if (Object.keys(subgraph.nodes).length > 1) {
-                    const [newEndNode, newStartNode] = graph.extend(subgraph, key);
-                    if (newEndNode !== undefined) {
-                        endNodes[key] = newEndNode;
+                    : node.getGraph(config);
+                drawableSubgraph.trimFirstNode();
+                drawableSubgraph.trimLastNode();
+                if (Object.keys(drawableSubgraph.nodes).length > 1) {
+                    const [e, s] = graph.extend(drawableSubgraph, displayKey);
+                    if (e === undefined) {
+                        throw new Error(`Could not extend subgraph "${key}" due to missing entrypoint.`);
+                    }
+                    // TODO: Remove default name once we stop supporting core 0.2.0
+                    // eslint-disable-next-line no-inner-declarations
+                    function _isRunnableInterface(
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    thing) {
+                        return thing ? thing.lc_runnable : false;
                     }
-                    if (newStartNode !== undefined) {
-                        startNodes[key] = newStartNode;
+                    // eslint-disable-next-line no-inner-declarations
+                    function _nodeDataStr(id, data) {
+                        if (id !== undefined && !isUuid(id)) {
+                            return id;
+                        }
+                        else if (_isRunnableInterface(data)) {
+                            try {
+                                let dataStr = data.getName();
+                                dataStr = dataStr.startsWith("Runnable")
+                                    ? dataStr.slice("Runnable".length)
+                                    : dataStr;
+                                return dataStr;
+                            }
+                            catch (error) {
+                                return data.getName();
+                            }
+                        }
+                        else {
+                            return data.name ?? "UnknownSchema";
+                        }
                     }
+                    // TODO: Remove casts when we stop supporting core 0.2.0
+                    if (s !== undefined) {
+                        startNodes[displayKey] = {
+                            name: _nodeDataStr(s.id, s.data),
+                            ...s,
+                        };
+                    }
+                    endNodes[displayKey] = {
+                        name: _nodeDataStr(e.id, e.data),
+                        ...e,
+                    };
                 }
                 else {
-                    const newNode = graph.addNode(node.runnable, key);
-                    startNodes[key] = newNode;
-                    endNodes[key] = newNode;
+                    // TODO: Remove when we stop supporting core 0.2.0
+                    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+                    // @ts-ignore
+                    const newNode = graph.addNode(node, displayKey, metadata);
+                    startNodes[displayKey] = newNode;
+                    endNodes[displayKey] = newNode;
                 }
             }
             else {
-                const newNode = graph.addNode(node.runnable, key);
-                startNodes[key] = newNode;
-                endNodes[key] = newNode;
+                // TODO: Remove when we stop supporting core 0.2.0
+                // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+                // @ts-ignore
+                const newNode = graph.addNode(node, displayKey, metadata);
+                startNodes[displayKey] = newNode;
+                endNodes[displayKey] = newNode;
             }
         }
-        for (const [start, end] of this.builder.allEdges) {
-            graph.addEdge(startNodes[start], endNodes[end]);
+        const sortedEdges = [...this.builder.allEdges].sort(([a], [b]) => {
+            if (a < b) {
+                return -1;
+            }
+            else if (b > a) {
+                return 1;
+            }
+            else {
+                return 0;
+            }
+        });
+        for (const [start, end] of sortedEdges) {
+            addEdge(_escapeMermaidKeywords(start), _escapeMermaidKeywords(end));
         }
         for (const [start, branches] of Object.entries(this.builder.branches)) {
             const defaultEnds = {
                 ...Object.fromEntries(Object.keys(this.builder.nodes)
                     .filter((k) => k !== start)
-                    .map((k) => [k, k])),
+                    .map((k) => [_escapeMermaidKeywords(k), _escapeMermaidKeywords(k)])),
                 [END]: END,
             };
             for (const branch of Object.values(branches)) {
@@ -408,7 +485,7 @@ export class CompiledGraph extends Pregel {
                     ends = defaultEnds;
                 }
                 for (const [label, end] of Object.entries(ends)) {
-                    graph.addEdge(startNodes[start], endNodes[end], label !== end ? label : undefined, true);
+                    addEdge(_escapeMermaidKeywords(start), _escapeMermaidKeywords(end), label, true);
                 }
             }
         }
@@ -423,3 +500,9 @@ function isCompiledGraph(x) {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         typeof x.attachEdge === "function");
 }
+function _escapeMermaidKeywords(key) {
+    if (key === "subgraph") {
+        return `"${key}"`;
+    }
+    return key;
+}

package/dist/pregel/index.cjs

@@ -825,7 +825,7 @@ class Pregel extends runnables_1.Runnable {
                 await loop.store?.stop();
             }
             await Promise.all([
-                loop?.checkpointerPromises ?? [],
+                ...(loop?.checkpointerPromises ?? []),
                 ...Array.from(managed.values()).map((mv) => mv.promises()),
             ]);
         }

package/dist/pregel/index.js

@@ -818,7 +818,7 @@ export class Pregel extends Runnable {
                 await loop.store?.stop();
             }
             await Promise.all([
-                loop?.checkpointerPromises ?? [],
+                ...(loop?.checkpointerPromises ?? []),
                 ...Array.from(managed.values()).map((mv) => mv.promises()),
             ]);
         }

package/dist/web.cjs

@@ -14,7 +14,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MessagesAnnotation = exports.MemoryStore = exports.AsyncBatchedStore = exports.BaseStore = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.BinaryOperatorAggregate = exports.BaseChannel = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
+exports.MessagesAnnotation = exports.InMemoryStore = exports.AsyncBatchedStore = exports.BaseStore = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.BinaryOperatorAggregate = exports.BaseChannel = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
 var index_js_1 = require("./graph/index.cjs");
 Object.defineProperty(exports, "END", { enumerable: true, get: function () { return index_js_1.END; } });
 Object.defineProperty(exports, "Graph", { enumerable: true, get: function () { return index_js_1.Graph; } });
@@ -36,7 +36,7 @@ Object.defineProperty(exports, "emptyCheckpoint", { enumerable: true, get: funct
 Object.defineProperty(exports, "BaseCheckpointSaver", { enumerable: true, get: function () { return langgraph_checkpoint_1.BaseCheckpointSaver; } });
 Object.defineProperty(exports, "BaseStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.BaseStore; } });
 Object.defineProperty(exports, "AsyncBatchedStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.AsyncBatchedStore; } });
-Object.defineProperty(exports, "MemoryStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.MemoryStore; } });
+Object.defineProperty(exports, "InMemoryStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.InMemoryStore; } });
 __exportStar(require("./managed/index.cjs"), exports);
 var messages_annotation_js_1 = require("./graph/messages_annotation.cjs");
 Object.defineProperty(exports, "MessagesAnnotation", { enumerable: true, get: function () { return messages_annotation_js_1.MessagesAnnotation; } });

package/dist/web.d.ts

@@ -5,7 +5,7 @@ export { BaseChannel, type BinaryOperator, BinaryOperatorAggregate, type AnyValu
 export { type AnnotationRoot as _INTERNAL_ANNOTATION_ROOT } from "./graph/index.js";
 export { type RetryPolicy } from "./pregel/utils/index.js";
 export { Send } from "./constants.js";
-export { MemorySaver, type Checkpoint, type CheckpointMetadata, type CheckpointTuple, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, type Item, type GetOperation, type SearchOperation, type PutOperation, type Operation, type OperationResults, BaseStore, AsyncBatchedStore, MemoryStore, type NameSpacePath, type NamespaceMatchType, type MatchCondition, type ListNamespacesOperation, } from "@langchain/langgraph-checkpoint";
+export { MemorySaver, type Checkpoint, type CheckpointMetadata, type CheckpointTuple, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, type Item, type GetOperation, type SearchOperation, type PutOperation, type Operation, type OperationResults, BaseStore, AsyncBatchedStore, InMemoryStore, type NameSpacePath, type NamespaceMatchType, type MatchCondition, type ListNamespacesOperation, } from "@langchain/langgraph-checkpoint";
 export * from "./managed/index.js";
 export { MessagesAnnotation } from "./graph/messages_annotation.js";
 export { type LangGraphRunnableConfig } from "./pregel/runnable_types.js";

package/dist/web.js

@@ -2,6 +2,6 @@ export { END, Graph, START, StateGraph, MessageGraph, messagesStateReducer, Anno
 export * from "./errors.js";
 export { BaseChannel, BinaryOperatorAggregate, } from "./channels/index.js";
 export { Send } from "./constants.js";
-export { MemorySaver, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, BaseStore, AsyncBatchedStore, MemoryStore, } from "@langchain/langgraph-checkpoint";
+export { MemorySaver, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, BaseStore, AsyncBatchedStore, InMemoryStore, } from "@langchain/langgraph-checkpoint";
 export * from "./managed/index.js";
 export { MessagesAnnotation } from "./graph/messages_annotation.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.2.10",
+  "version": "0.2.12",
   "description": "LangGraph",
   "type": "module",
   "engines": {
@@ -43,7 +43,7 @@
     "@jest/globals": "^29.5.0",
     "@langchain/anthropic": "^0.3.0",
     "@langchain/community": "^0.3.0",
-    "@langchain/core": "^0.3.0",
+    "@langchain/core": "^0.3.6",
     "@langchain/langgraph-checkpoint-sqlite": "workspace:*",
     "@langchain/openai": "^0.3.0",
     "@langchain/scripts": ">=0.1.3 <0.2.0",
@@ -70,7 +70,7 @@
     "langchain": "^0.3.0",
     "prettier": "^2.8.3",
     "release-it": "^17.6.0",
-    "rollup": "^4.22.4",
+    "rollup": "^4.23.0",
     "ts-jest": "^29.1.0",
     "tsx": "^4.7.0",
     "typescript": "^4.9.5 || ^5.4.5",