@langchain/langgraph-api 0.0.17 → 0.0.18

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/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/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/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: {},
@@ -487,6 +488,41 @@ 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, { 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)
@@ -513,7 +549,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 +569,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;

package/package.json

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