@langchain/langgraph 0.2.10 → 0.2.11

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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "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",