@langchain/langgraph-api 0.0.17 → 0.0.19

package/dist/api/assistants.mjs

@@ -76,9 +76,10 @@ api.get("/assistants/:assistant_id/graph", zValidator("query", z.object({ xray:
     const assistantId = getAssistantId(c.req.param("assistant_id"));
     const assistant = await Assistants.get(assistantId);
     const { xray } = c.req.valid("query");
-    const graph = getGraph(assistant.graph_id);
+    const config = getRunnableConfig(assistant.config);
+    const graph = await getGraph(assistant.graph_id, config);
     const drawable = await graph.getGraphAsync({
-        ...getRunnableConfig(assistant.config),
+        ...config,
         xray: xray ?? undefined,
     });
     return c.json(drawable.toJSON());
@@ -106,7 +107,8 @@ api.get("/assistants/:assistant_id/subgraphs/:namespace?", zValidator("param", z
     const { recurse } = c.req.valid("query");
     const assistantId = getAssistantId(assistant_id);
     const assistant = await Assistants.get(assistantId);
-    const graph = getGraph(assistant.graph_id);
+    const config = getRunnableConfig(assistant.config);
+    const graph = await getGraph(assistant.graph_id, config);
     const graphSchema = await getGraphSchema(assistant.graph_id);
     const rootGraphId = Object.keys(graphSchema).find((i) => !i.includes("|"));
     if (!rootGraphId) {

package/dist/api/threads.mjs

@@ -15,6 +15,9 @@ api.post("/threads", zValidator("json", schemas.ThreadCreate), async (c) => {
         metadata: payload.metadata,
         if_exists: payload.if_exists ?? "raise",
     });
+    if (payload.supersteps?.length) {
+        await Threads.State.bulk({ configurable: { thread_id: thread.thread_id } }, payload.supersteps);
+    }
     return jsonExtra(c, thread);
 });
 api.post("/threads/search", zValidator("json", schemas.ThreadSearchRequest), async (c) => {

package/dist/cli/entrypoint.mjs

@@ -37,5 +37,5 @@ logger.info(`Server running at ${host}`);
 let queryParams = `?baseUrl=http://${options.host}:${options.port}`;
 if (organizationId)
     queryParams += `&organizationId=${organizationId}`;
-asyncExitHook(cleanup, { wait: 1000 });
+asyncExitHook(cleanup, { wait: 3_000 });
 sendToParent?.({ queryParams });

package/dist/cli/spawn.d.mts

@@ -6,6 +6,9 @@ export declare function spawnServer(args: {
     config: {
         graphs: Record<string, string>;
         ui?: Record<string, string>;
+        ui_config?: {
+            shared?: string[];
+        };
     };
     env: NodeJS.ProcessEnv;
     hostUrl: string;

package/dist/cli/spawn.mjs

@@ -31,6 +31,7 @@ For production use, please use LangGraph Cloud.
             host: args.host,
             graphs: context.config.graphs,
             ui: context.config.ui,
+            ui_config: context.config.ui_config,
             cwd: options.projectCwd,
         }),
     ], {

package/dist/command.mjs

@@ -0,0 +1,15 @@
+import { Command, Send } from "@langchain/langgraph";
+export const getLangGraphCommand = (command) => {
+    let goto = command.goto != null && !Array.isArray(command.goto)
+        ? [command.goto]
+        : command.goto;
+    return new Command({
+        goto: goto?.map((item) => {
+            if (typeof item !== "string")
+                return new Send(item.node, item.input);
+            return item;
+        }),
+        update: command.update,
+        resume: command.resume,
+    });
+};

package/dist/graph/load.mjs

@@ -41,11 +41,12 @@ export async function registerFromEnv(specs, options) {
         return resolved;
     }));
 }
-export function getGraph(graphId, options) {
+export async function getGraph(graphId, config, options) {
     if (!GRAPHS[graphId])
         throw new HTTPException(404, { message: `Graph "${graphId}" not found` });
-    // TODO: have a check for the type of graph
-    const compiled = GRAPHS[graphId];
+    const compiled = typeof GRAPHS[graphId] === "function"
+        ? await GRAPHS[graphId](config ?? { configurable: {} })
+        : GRAPHS[graphId];
     if (typeof options?.checkpointer !== "undefined") {
         compiled.checkpointer = options?.checkpointer ?? undefined;
     }

package/dist/graph/load.utils.mjs

@@ -23,10 +23,17 @@ export async function resolveGraph(spec, options) {
     const resolved = await (async () => {
         if (!graph)
             throw new Error("Failed to load graph: graph is nullush");
-        const graphLike = typeof graph === "function" ? await graph() : await graph;
-        if (isGraph(graphLike))
-            return graphLike.compile();
-        return graphLike;
+        const afterResolve = (graphLike) => {
+            const graph = isGraph(graphLike) ? graphLike.compile() : graphLike;
+            return graph;
+        };
+        if (typeof graph === "function") {
+            return async (config) => {
+                const graphLike = await graph(config);
+                return afterResolve(graphLike);
+            };
+        }
+        return afterResolve(await graph);
     })();
     return { sourceFile, exportSymbol, resolved };
 }

package/dist/schemas.mjs

@@ -140,35 +140,31 @@ export const Run = z.object({
     kwargs: z.object({}).catchall(z.any()),
     multitask_strategy: z.enum(["reject", "rollback", "interrupt", "enqueue"]),
 });
+export const CommandSchema = z.object({
+    goto: z
+        .union([
+        z.union([
+            z.string(),
+            z.object({ node: z.string(), input: z.unknown().optional() }),
+        ]),
+        z.array(z.union([
+            z.string(),
+            z.object({ node: z.string(), input: z.unknown().optional() }),
+        ])),
+    ])
+        .optional(),
+    update: z
+        .union([z.record(z.unknown()), z.array(z.tuple([z.string(), z.unknown()]))])
+        .optional(),
+    resume: z.unknown().optional(),
+});
 export const RunCreate = z
     .object({
     assistant_id: z.union([z.string().uuid(), z.string()]),
     checkpoint_id: z.string().optional(),
     checkpoint: CheckpointSchema.optional(),
     input: z.union([z.unknown(), z.null()]).optional(),
-    command: z
-        .object({
-        goto: z
-            .union([
-            z.union([
-                z.string(),
-                z.object({ node: z.string(), input: z.unknown().optional() }),
-            ]),
-            z.array(z.union([
-                z.string(),
-                z.object({ node: z.string(), input: z.unknown().optional() }),
-            ])),
-        ])
-            .optional(),
-        update: z
-            .union([
-            z.record(z.unknown()),
-            z.array(z.tuple([z.string(), z.unknown()])),
-        ])
-            .optional(),
-        resume: z.unknown().optional(),
-    })
-        .optional(),
+    command: CommandSchema.optional(),
     metadata: z
         .object({})
         .catchall(z.any())
@@ -301,6 +297,16 @@ export const Thread = z.object({
 });
 export const ThreadCreate = z
     .object({
+    supersteps: z
+        .array(z.object({
+        updates: z.array(z.object({
+            values: z.unknown().nullish(),
+            command: CommandSchema.nullish(),
+            as_node: z.string(),
+        })),
+    }))
+        .describe("The supersteps to apply to the thread.")
+        .optional(),
     thread_id: z
         .string()
         .uuid()

package/dist/server.mjs

@@ -50,6 +50,7 @@ export const StartServerSchema = z.object({
     cwd: z.string(),
     graphs: z.record(z.string()),
     ui: z.record(z.string()).optional(),
+    ui_config: z.object({ shared: z.array(z.string()).optional() }).optional(),
 });
 export async function startServer(options) {
     logger.info(`Initializing storage...`);
@@ -69,7 +70,10 @@ export async function startServer(options) {
         const { api, registerGraphUi } = await import("./ui/load.mjs");
         app.route("/", api);
         logger.info(`Registering UI from ${options.cwd}`);
-        await registerGraphUi(options.ui, { cwd: options.cwd });
+        await registerGraphUi(options.ui, {
+            cwd: options.cwd,
+            config: options.ui_config,
+        });
     }
     logger.info(`Starting ${options.nWorkers} workers`);
     for (let i = 0; i < options.nWorkers; i++)

package/dist/storage/ops.mjs

@@ -6,6 +6,7 @@ import { store } from "./store.mjs";
 import { logger } from "../logging.mjs";
 import { serializeError } from "../utils/serde.mjs";
 import { FileSystemPersistence } from "./persist.mjs";
+import { getLangGraphCommand } from "../command.mjs";
 export const conn = new FileSystemPersistence(".langgraphjs_ops.json", () => ({
     runs: {},
     threads: {},
@@ -446,7 +447,10 @@ export class Threads {
                     tasks: [],
                 };
             }
-            const graph = await getGraph(graphId, { checkpointer, store });
+            const graph = await getGraph(graphId, thread.config, {
+                checkpointer,
+                store,
+            });
             const result = await graph.getState(config, { subgraphs });
             if (result.metadata != null &&
                 "checkpoint_ns" in result.metadata &&
@@ -470,7 +474,10 @@ export class Threads {
             }
             config.configurable ??= {};
             config.configurable.graph_id ??= graphId;
-            const graph = await getGraph(graphId, { checkpointer, store });
+            const graph = await getGraph(graphId, thread.config, {
+                checkpointer,
+                store,
+            });
             const updateConfig = structuredClone(config);
             updateConfig.configurable ??= {};
             updateConfig.configurable.checkpoint_ns ??= "";
@@ -487,6 +494,44 @@ export class Threads {
             });
             return { checkpoint: nextConfig.configurable };
         }
+        static async bulk(config, supersteps) {
+            const threadId = config.configurable?.thread_id;
+            if (!threadId)
+                return [];
+            const thread = await Threads.get(threadId);
+            const graphId = thread.metadata?.graph_id;
+            if (graphId == null) {
+                throw new HTTPException(400, {
+                    message: `Thread ${threadId} has no graph ID`,
+                });
+            }
+            config.configurable ??= {};
+            config.configurable.graph_id ??= graphId;
+            const graph = await getGraph(graphId, thread.config, {
+                checkpointer,
+                store,
+            });
+            const updateConfig = structuredClone(config);
+            updateConfig.configurable ??= {};
+            updateConfig.configurable.checkpoint_ns ??= "";
+            const nextConfig = await graph.bulkUpdateState(updateConfig, supersteps.map((i) => ({
+                updates: i.updates.map((j) => ({
+                    values: j.command != null ? getLangGraphCommand(j.command) : j.values,
+                    asNode: j.as_node,
+                })),
+            })));
+            const state = await Threads.State.get(config, { subgraphs: false });
+            // update thread values
+            await conn.with(async (STORE) => {
+                for (const thread of Object.values(STORE.threads)) {
+                    if (thread.thread_id === threadId) {
+                        thread.values = state.values;
+                        break;
+                    }
+                }
+            });
+            return { checkpoint: nextConfig.configurable };
+        }
         static async list(config, options) {
             const threadId = config.configurable?.thread_id;
             if (!threadId)
@@ -495,7 +540,10 @@ export class Threads {
             const graphId = thread.metadata?.graph_id;
             if (graphId == null)
                 return [];
-            const graph = await getGraph(graphId, { checkpointer, store });
+            const graph = await getGraph(graphId, thread.config, {
+                checkpointer,
+                store,
+            });
             const before = typeof options?.before === "string"
                 ? { configurable: { checkpoint_id: options.before } }
                 : options?.before;
@@ -513,7 +561,7 @@ export class Threads {
 }
 export class Runs {
     static async *next() {
-        yield* conn.withGenerator(async function* (STORE) {
+        yield* conn.withGenerator(async function* (STORE, options) {
             const now = new Date();
             const pendingRuns = Object.values(STORE.runs)
                 .filter((run) => run.status === "pending" && run.created_at < now)
@@ -533,6 +581,7 @@ export class Runs {
                     continue;
                 try {
                     const signal = StreamManager.lock(runId);
+                    options.schedulePersist();
                     STORE.retry_counter[runId] ??= 0;
                     STORE.retry_counter[runId] += 1;
                     yield { run, attempt: STORE.retry_counter[runId], signal };

package/dist/storage/persist.mjs

@@ -67,12 +67,15 @@ export class FileSystemPersistence {
         if (this.filepath == null || this.data == null) {
             throw new Error(`${this.name} not initialized`);
         }
+        let shouldPersist = false;
+        let schedulePersist = () => void (shouldPersist = true);
         try {
-            const gen = typeof fn === "function" ? fn(this.data) : fn;
+            const gen = typeof fn === "function" ? fn(this.data, { schedulePersist }) : fn;
             yield* gen;
         }
         finally {
-            this.schedulePersist();
+            if (shouldPersist)
+                this.schedulePersist();
         }
     }
 }

package/dist/stream.mjs

@@ -1,22 +1,8 @@
 import { getGraph } from "./graph/load.mjs";
 import { Client as LangSmithClient } from "langsmith";
-import { Command, Send, } from "@langchain/langgraph";
 import { runnableConfigToCheckpoint, taskRunnableConfigToCheckpoint, } from "./utils/runnableConfig.mjs";
 import { isBaseMessage } from "@langchain/core/messages";
-const getLangGraphCommand = (command) => {
-    let goto = command.goto != null && !Array.isArray(command.goto)
-        ? [command.goto]
-        : command.goto;
-    return new Command({
-        goto: goto?.map((item) => {
-            if (typeof item !== "string")
-                return new Send(item.node, item.input);
-            return item;
-        }),
-        update: command.update,
-        resume: command.resume,
-    });
-};
+import { getLangGraphCommand } from "./command.mjs";
 const isRunnableConfig = (config) => {
     if (typeof config !== "object" || config == null)
         return false;
@@ -71,7 +57,7 @@ export async function* streamState(run, attempt = 1, options) {
     if (!graphId || typeof graphId !== "string") {
         throw new Error("Invalid or missing graph_id");
     }
-    const graph = getGraph(graphId, {
+    const graph = await getGraph(graphId, kwargs.config, {
         checkpointer: kwargs.temporary ? null : undefined,
     });
     const userStreamMode = kwargs.stream_mode ?? [];

package/dist/ui/bundler.d.mts

@@ -2,6 +2,9 @@ import { type BuildOptions } from "esbuild";
 export declare function build(agentName: string, args: {
     cwd: string;
     userPath: string;
+    config?: {
+        shared?: string[];
+    };
 }): Promise<{
     basename: string;
     contents: Uint8Array;
@@ -9,6 +12,9 @@ export declare function build(agentName: string, args: {
 export declare function watch(agentName: string, args: {
     cwd: string;
     userPath: string;
+    config?: {
+        shared?: string[];
+    };
 }, onResult: (result: {
     basename: string;
     contents: Uint8Array;

package/dist/ui/bundler.mjs

@@ -68,6 +68,7 @@ function setup(agentName, args, onResult) {
             "react-dom",
             "@langchain/langgraph-sdk",
             "@langchain/langgraph-sdk/react-ui",
+            ...(args.config?.shared ?? []),
         ],
         plugins: [tailwind(), entrypointPlugin(args), registerPlugin(onResult)],
         globalName: `__LGUI_${agentName}`,

package/dist/ui/load.mjs

@@ -7,7 +7,7 @@ import { watch } from "./bundler.mjs";
 const GRAPH_UI = {};
 export async function registerGraphUi(defs, options) {
     const result = await Promise.all(Object.entries(defs).map(async ([agentName, userPath]) => {
-        const ctx = await watch(agentName, { cwd: options.cwd, userPath }, (result) => {
+        const ctx = await watch(agentName, { cwd: options.cwd, userPath, config: options.config }, (result) => {
             GRAPH_UI[agentName] = result;
         });
         return [agentName, ctx];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-api",
-  "version": "0.0.17",
+  "version": "0.0.19",
   "type": "module",
   "engines": {
     "node": "^18.19.0 || >=20.16.0"
@@ -47,13 +47,13 @@
     "zod": "^3.23.8"
   },
   "peerDependencies": {
-    "@langchain/core": "^0.3.40",
-    "@langchain/langgraph": "^0.2.49",
-    "@langchain/langgraph-checkpoint": "^0.0.15",
+    "@langchain/core": "^0.3.42",
+    "@langchain/langgraph": "^0.2.57",
+    "@langchain/langgraph-checkpoint": "^0.0.16",
     "typescript": "^5.5.4"
   },
   "devDependencies": {
-    "@langchain/langgraph-sdk": "^0.0.33",
+    "@langchain/langgraph-sdk": "^0.0.60",
     "@types/babel__code-frame": "^7.0.6",
     "@types/react": "^19.0.8",
     "@types/react-dom": "^19.0.3",