@langchain/langgraph-sdk 0.0.16 → 0.0.18

package/dist/client.cjs

@@ -4,6 +4,7 @@ exports.Client = exports.StoreClient = exports.RunsClient = exports.ThreadsClien
 const async_caller_js_1 = require("./utils/async_caller.cjs");
 const index_js_1 = require("./utils/eventsource-parser/index.cjs");
 const stream_js_1 = require("./utils/stream.cjs");
+const signals_js_1 = require("./utils/signals.cjs");
 class BaseClient {
     constructor(config) {
         Object.defineProperty(this, "asyncCaller", {
@@ -36,6 +37,8 @@ class BaseClient {
             ...config?.callerOptions,
         });
         this.timeoutMs = config?.timeoutMs || 12_000;
+        // default limit being capped by Chrome
+        // https://github.com/nodejs/undici/issues/1373
         this.apiUrl = config?.apiUrl || "http://localhost:8123";
         this.defaultHeaders = config?.defaultHeaders || {};
         if (config?.apiKey != null) {
@@ -55,6 +58,16 @@ class BaseClient {
             };
             delete mutatedOptions.json;
         }
+        let timeoutSignal = null;
+        if (typeof options?.timeoutMs !== "undefined") {
+            if (options.timeoutMs != null) {
+                timeoutSignal = AbortSignal.timeout(options.timeoutMs);
+            }
+        }
+        else {
+            timeoutSignal = AbortSignal.timeout(this.timeoutMs);
+        }
+        mutatedOptions.signal = (0, signals_js_1.mergeSignals)(timeoutSignal, mutatedOptions.signal);
         const targetUrl = new URL(`${this.apiUrl}${path}`);
         if (mutatedOptions.params) {
             for (const [key, value] of Object.entries(mutatedOptions.params)) {
@@ -462,6 +475,7 @@ class RunsClient extends BaseClient {
             assistant_id: assistantId,
             interrupt_before: payload?.interruptBefore,
             interrupt_after: payload?.interruptAfter,
+            checkpoint: payload?.checkpoint,
             checkpoint_id: payload?.checkpointId,
             webhook: payload?.webhook,
             multitask_strategy: payload?.multitaskStrategy,
@@ -473,6 +487,7 @@ class RunsClient extends BaseClient {
         const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(endpoint, {
             method: "POST",
             json,
+            timeoutMs: null,
             signal: payload?.signal,
         }));
         let parser;
@@ -525,6 +540,7 @@ class RunsClient extends BaseClient {
             interrupt_before: payload?.interruptBefore,
             interrupt_after: payload?.interruptAfter,
             webhook: payload?.webhook,
+            checkpoint: payload?.checkpoint,
             checkpoint_id: payload?.checkpointId,
             multitask_strategy: payload?.multitaskStrategy,
             after_seconds: payload?.afterSeconds,
@@ -568,6 +584,7 @@ class RunsClient extends BaseClient {
             assistant_id: assistantId,
             interrupt_before: payload?.interruptBefore,
             interrupt_after: payload?.interruptAfter,
+            checkpoint: payload?.checkpoint,
             checkpoint_id: payload?.checkpointId,
             webhook: payload?.webhook,
             multitask_strategy: payload?.multitaskStrategy,
@@ -579,6 +596,7 @@ class RunsClient extends BaseClient {
         return this.fetch(endpoint, {
             method: "POST",
             json,
+            timeoutMs: null,
             signal: payload?.signal,
         });
     }
@@ -630,8 +648,11 @@ class RunsClient extends BaseClient {
      * @param runId The ID of the run.
      * @returns
      */
-    async join(threadId, runId) {
-        return this.fetch(`/threads/${threadId}/runs/${runId}/join`);
+    async join(threadId, runId, options) {
+        return this.fetch(`/threads/${threadId}/runs/${runId}/join`, {
+            timeoutMs: null,
+            signal: options?.signal,
+        });
     }
     /**
      * Stream output from a run in real-time, until the run is done.
@@ -646,6 +667,7 @@ class RunsClient extends BaseClient {
     async *joinStream(threadId, runId, signal) {
         const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(`/threads/${threadId}/runs/${runId}/stream`, {
             method: "GET",
+            timeoutMs: null,
             signal,
         }));
         let parser;

package/dist/client.d.ts

@@ -17,10 +17,13 @@ declare class BaseClient {
     protected prepareFetchOptions(path: string, options?: RequestInit & {
         json?: unknown;
         params?: Record<string, unknown>;
+        timeoutMs?: number | null;
     }): [url: URL, init: RequestInit];
     protected fetch<T>(path: string, options?: RequestInit & {
         json?: unknown;
         params?: Record<string, unknown>;
+        timeoutMs?: number | null;
+        signal?: AbortSignal;
     }): Promise<T>;
 }
 export declare class CronsClient extends BaseClient {
@@ -334,7 +337,9 @@ export declare class RunsClient extends BaseClient {
      * @param runId The ID of the run.
      * @returns
      */
-    join(threadId: string, runId: string): Promise<void>;
+    join(threadId: string, runId: string, options?: {
+        signal?: AbortSignal;
+    }): Promise<void>;
     /**
      * Stream output from a run in real-time, until the run is done.
      * Output is not buffered, so any output produced before this call will

package/dist/client.js

@@ -1,6 +1,7 @@
 import { AsyncCaller } from "./utils/async_caller.js";
 import { createParser, } from "./utils/eventsource-parser/index.js";
 import { IterableReadableStream } from "./utils/stream.js";
+import { mergeSignals } from "./utils/signals.js";
 class BaseClient {
     constructor(config) {
         Object.defineProperty(this, "asyncCaller", {
@@ -33,6 +34,8 @@ class BaseClient {
             ...config?.callerOptions,
         });
         this.timeoutMs = config?.timeoutMs || 12_000;
+        // default limit being capped by Chrome
+        // https://github.com/nodejs/undici/issues/1373
         this.apiUrl = config?.apiUrl || "http://localhost:8123";
         this.defaultHeaders = config?.defaultHeaders || {};
         if (config?.apiKey != null) {
@@ -52,6 +55,16 @@ class BaseClient {
             };
             delete mutatedOptions.json;
         }
+        let timeoutSignal = null;
+        if (typeof options?.timeoutMs !== "undefined") {
+            if (options.timeoutMs != null) {
+                timeoutSignal = AbortSignal.timeout(options.timeoutMs);
+            }
+        }
+        else {
+            timeoutSignal = AbortSignal.timeout(this.timeoutMs);
+        }
+        mutatedOptions.signal = mergeSignals(timeoutSignal, mutatedOptions.signal);
         const targetUrl = new URL(`${this.apiUrl}${path}`);
         if (mutatedOptions.params) {
             for (const [key, value] of Object.entries(mutatedOptions.params)) {
@@ -456,6 +469,7 @@ export class RunsClient extends BaseClient {
             assistant_id: assistantId,
             interrupt_before: payload?.interruptBefore,
             interrupt_after: payload?.interruptAfter,
+            checkpoint: payload?.checkpoint,
             checkpoint_id: payload?.checkpointId,
             webhook: payload?.webhook,
             multitask_strategy: payload?.multitaskStrategy,
@@ -467,6 +481,7 @@ export class RunsClient extends BaseClient {
         const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(endpoint, {
             method: "POST",
             json,
+            timeoutMs: null,
             signal: payload?.signal,
         }));
         let parser;
@@ -519,6 +534,7 @@ export class RunsClient extends BaseClient {
             interrupt_before: payload?.interruptBefore,
             interrupt_after: payload?.interruptAfter,
             webhook: payload?.webhook,
+            checkpoint: payload?.checkpoint,
             checkpoint_id: payload?.checkpointId,
             multitask_strategy: payload?.multitaskStrategy,
             after_seconds: payload?.afterSeconds,
@@ -562,6 +578,7 @@ export class RunsClient extends BaseClient {
             assistant_id: assistantId,
             interrupt_before: payload?.interruptBefore,
             interrupt_after: payload?.interruptAfter,
+            checkpoint: payload?.checkpoint,
             checkpoint_id: payload?.checkpointId,
             webhook: payload?.webhook,
             multitask_strategy: payload?.multitaskStrategy,
@@ -573,6 +590,7 @@ export class RunsClient extends BaseClient {
         return this.fetch(endpoint, {
             method: "POST",
             json,
+            timeoutMs: null,
             signal: payload?.signal,
         });
     }
@@ -624,8 +642,11 @@ export class RunsClient extends BaseClient {
      * @param runId The ID of the run.
      * @returns
      */
-    async join(threadId, runId) {
-        return this.fetch(`/threads/${threadId}/runs/${runId}/join`);
+    async join(threadId, runId, options) {
+        return this.fetch(`/threads/${threadId}/runs/${runId}/join`, {
+            timeoutMs: null,
+            signal: options?.signal,
+        });
     }
     /**
      * Stream output from a run in real-time, until the run is done.
@@ -640,6 +661,7 @@ export class RunsClient extends BaseClient {
     async *joinStream(threadId, runId, signal) {
         const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(`/threads/${threadId}/runs/${runId}/stream`, {
             method: "GET",
+            timeoutMs: null,
             signal,
         }));
         let parser;

package/dist/schema.d.ts

@@ -1,7 +1,7 @@
 import type { JSONSchema7 } from "json-schema";
 type Optional<T> = T | null | undefined;
 type RunStatus = "pending" | "running" | "error" | "success" | "timeout" | "interrupted";
-type ThreadStatus = "idle" | "busy" | "interrupted";
+type ThreadStatus = "idle" | "busy" | "interrupted" | "error";
 type MultitaskStrategy = "reject" | "interrupt" | "rollback" | "enqueue";
 export interface Config {
     /**

package/dist/types.d.ts

@@ -1,4 +1,4 @@
-import { Config, Metadata } from "./schema.js";
+import { Checkpoint, Config, Metadata } from "./schema.js";
 export type StreamMode = "values" | "messages" | "updates" | "events" | "debug";
 export type MultitaskStrategy = "reject" | "interrupt" | "rollback" | "enqueue";
 export type OnConflictBehavior = "raise" | "do_nothing";
@@ -22,6 +22,10 @@ interface RunsInvokePayload {
      * Checkpoint ID for when creating a new run.
      */
     checkpointId?: string;
+    /**
+     * Checkpoint for when creating a new run.
+     */
+    checkpoint?: Omit<Checkpoint, "thread_id">;
     /**
      * Interrupt execution before entering these nodes.
      */

package/dist/utils/signals.cjs

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.mergeSignals = void 0;
+function mergeSignals(...signals) {
+    const nonZeroSignals = signals.filter((signal) => signal != null);
+    if (nonZeroSignals.length === 0)
+        return undefined;
+    if (nonZeroSignals.length === 1)
+        return nonZeroSignals[0];
+    const controller = new AbortController();
+    for (const signal of signals) {
+        if (signal?.aborted) {
+            controller.abort(signal.reason);
+            return controller.signal;
+        }
+        signal?.addEventListener("abort", () => controller.abort(signal.reason), {
+            once: true,
+        });
+    }
+    return controller.signal;
+}
+exports.mergeSignals = mergeSignals;

package/dist/utils/signals.d.ts

@@ -0,0 +1 @@
+export declare function mergeSignals(...signals: (AbortSignal | null | undefined)[]): AbortSignal | undefined;

package/dist/utils/signals.js

@@ -0,0 +1,18 @@
+export function mergeSignals(...signals) {
+    const nonZeroSignals = signals.filter((signal) => signal != null);
+    if (nonZeroSignals.length === 0)
+        return undefined;
+    if (nonZeroSignals.length === 1)
+        return nonZeroSignals[0];
+    const controller = new AbortController();
+    for (const signal of signals) {
+        if (signal?.aborted) {
+            controller.abort(signal.reason);
+            return controller.signal;
+        }
+        signal?.addEventListener("abort", () => controller.abort(signal.reason), {
+            once: true,
+        });
+    }
+    return controller.signal;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-sdk",
-  "version": "0.0.16",
+  "version": "0.0.18",
   "description": "Client library for interacting with the LangGraph API",
   "type": "module",
   "packageManager": "yarn@1.22.19",