@langchain/langgraph-sdk 0.0.38 → 0.0.40

package/dist/client.cjs

@@ -2,11 +2,10 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Client = exports.StoreClient = exports.RunsClient = exports.ThreadsClient = exports.AssistantsClient = exports.CronsClient = exports.getApiKey = void 0;
 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");
 const env_js_1 = require("./utils/env.cjs");
-const fetch_js_1 = require("./singletons/fetch.cjs");
+const sse_js_1 = require("./utils/sse.cjs");
 /**
  * Get the API key from the environment.
  * Precedence:
@@ -113,7 +112,7 @@ class BaseClient {
         return [targetUrl, mutatedOptions];
     }
     async fetch(path, options) {
-        const response = await this.asyncCaller.call((0, fetch_js_1._getFetchImplementation)(), ...this.prepareFetchOptions(path, options));
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(path, options));
         if (response.status === 202 || response.status === 204) {
             return undefined;
         }
@@ -519,43 +518,15 @@ class RunsClient extends BaseClient {
             if_not_exists: payload?.ifNotExists,
         };
         const endpoint = threadId == null ? `/runs/stream` : `/threads/${threadId}/runs/stream`;
-        const response = await this.asyncCaller.call((0, fetch_js_1._getFetchImplementation)(), ...this.prepareFetchOptions(endpoint, {
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(endpoint, {
             method: "POST",
             json,
             timeoutMs: null,
             signal: payload?.signal,
         }));
-        let parser;
-        let onEndEvent;
-        const textDecoder = new TextDecoder();
-        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() })).pipeThrough(new TransformStream({
-            async start(ctrl) {
-                parser = (0, index_js_1.createParser)((event) => {
-                    if ((payload?.signal && payload.signal.aborted) ||
-                        (event.type === "event" && event.data === "[DONE]")) {
-                        ctrl.terminate();
-                        return;
-                    }
-                    if ("data" in event) {
-                        ctrl.enqueue({
-                            event: event.event ?? "message",
-                            data: JSON.parse(event.data),
-                        });
-                    }
-                });
-                onEndEvent = () => {
-                    ctrl.enqueue({ event: "end", data: undefined });
-                };
-            },
-            async transform(chunk) {
-                const payload = textDecoder.decode(chunk);
-                parser.feed(payload);
-                // eventsource-parser will ignore events
-                // that are not terminated by a newline
-                if (payload.trim() === "event: end")
-                    onEndEvent();
-            },
-        }));
+        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() }))
+            .pipeThrough(new sse_js_1.BytesLineDecoder())
+            .pipeThrough(new sse_js_1.SSEDecoder());
         yield* stream_js_1.IterableReadableStream.fromReadableStream(stream);
     }
     /**
@@ -723,43 +694,15 @@ class RunsClient extends BaseClient {
             options instanceof AbortSignal
             ? { signal: options }
             : options;
-        const response = await this.asyncCaller.call((0, fetch_js_1._getFetchImplementation)(), ...this.prepareFetchOptions(`/threads/${threadId}/runs/${runId}/stream`, {
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(`/threads/${threadId}/runs/${runId}/stream`, {
             method: "GET",
             timeoutMs: null,
             signal: opts?.signal,
             params: { cancel_on_disconnect: opts?.cancelOnDisconnect ? "1" : "0" },
         }));
-        let parser;
-        let onEndEvent;
-        const textDecoder = new TextDecoder();
-        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() })).pipeThrough(new TransformStream({
-            async start(ctrl) {
-                parser = (0, index_js_1.createParser)((event) => {
-                    if ((opts?.signal && opts.signal.aborted) ||
-                        (event.type === "event" && event.data === "[DONE]")) {
-                        ctrl.terminate();
-                        return;
-                    }
-                    if ("data" in event) {
-                        ctrl.enqueue({
-                            event: event.event ?? "message",
-                            data: JSON.parse(event.data),
-                        });
-                    }
-                });
-                onEndEvent = () => {
-                    ctrl.enqueue({ event: "end", data: undefined });
-                };
-            },
-            async transform(chunk) {
-                const payload = textDecoder.decode(chunk);
-                parser.feed(payload);
-                // eventsource-parser will ignore events
-                // that are not terminated by a newline
-                if (payload.trim() === "event: end")
-                    onEndEvent();
-            },
-        }));
+        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() }))
+            .pipeThrough(new sse_js_1.BytesLineDecoder())
+            .pipeThrough(new sse_js_1.SSEDecoder());
         yield* stream_js_1.IterableReadableStream.fromReadableStream(stream);
     }
     /**

package/dist/client.d.ts

@@ -1,6 +1,7 @@
 import { Assistant, AssistantGraph, CancelAction, Config, DefaultValues, GraphSchema, Metadata, Run, RunStatus, Thread, ThreadState, Cron, AssistantVersion, Subgraphs, Checkpoint, SearchItemsResponse, ListNamespaceResponse, Item, ThreadStatus, CronCreateResponse, CronCreateForThreadResponse } from "./schema.js";
 import { AsyncCaller, AsyncCallerParams } from "./utils/async_caller.js";
-import { RunsCreatePayload, RunsStreamPayload, RunsWaitPayload, StreamEvent, CronsCreatePayload, OnConflictBehavior } from "./types.js";
+import type { RunsCreatePayload, RunsStreamPayload, RunsWaitPayload, StreamEvent, CronsCreatePayload, OnConflictBehavior } from "./types.js";
+import type { TypedAsyncGenerator, StreamMode } from "./types.stream.js";
 /**
  * Get the API key from the environment.
  * Precedence:
@@ -13,7 +14,7 @@ import { RunsCreatePayload, RunsStreamPayload, RunsWaitPayload, StreamEvent, Cro
  * @returns The API key if found, otherwise undefined
  */
 export declare function getApiKey(apiKey?: string): string | undefined;
-interface ClientConfig {
+export interface ClientConfig {
     apiUrl?: string;
     apiKey?: string;
     callerOptions?: AsyncCallerParams;
@@ -167,14 +168,14 @@ export declare class AssistantsClient extends BaseClient {
      */
     setLatest(assistantId: string, version: number): Promise<Assistant>;
 }
-export declare class ThreadsClient extends BaseClient {
+export declare class ThreadsClient<TStateType = DefaultValues, TUpdateType = TStateType> extends BaseClient {
     /**
      * Get a thread by ID.
      *
      * @param threadId ID of the thread.
      * @returns The thread.
      */
-    get(threadId: string): Promise<Thread>;
+    get<ValuesType = TStateType>(threadId: string): Promise<Thread<ValuesType>>;
     /**
      * Create a new thread.
      *
@@ -188,13 +189,13 @@ export declare class ThreadsClient extends BaseClient {
         metadata?: Metadata;
         threadId?: string;
         ifExists?: OnConflictBehavior;
-    }): Promise<Thread>;
+    }): Promise<Thread<TStateType>>;
     /**
      * Copy an existing thread
      * @param threadId ID of the thread to be copied
      * @returns Newly copied thread
      */
-    copy(threadId: string): Promise<Thread>;
+    copy(threadId: string): Promise<Thread<TStateType>>;
     /**
      * Update a thread.
      *
@@ -220,7 +221,7 @@ export declare class ThreadsClient extends BaseClient {
      * @param query Query options
      * @returns List of threads
      */
-    search(query?: {
+    search<ValuesType = TStateType>(query?: {
         /**
          * Metadata to filter threads by.
          */
@@ -239,14 +240,14 @@ export declare class ThreadsClient extends BaseClient {
          * Must be one of 'idle', 'busy', 'interrupted' or 'error'.
          */
         status?: ThreadStatus;
-    }): Promise<Thread[]>;
+    }): Promise<Thread<ValuesType>[]>;
     /**
      * Get state for a thread.
      *
      * @param threadId ID of the thread.
      * @returns Thread state.
      */
-    getState<ValuesType = DefaultValues>(threadId: string, checkpoint?: Checkpoint | string, options?: {
+    getState<ValuesType = TStateType>(threadId: string, checkpoint?: Checkpoint | string, options?: {
         subgraphs?: boolean;
     }): Promise<ThreadState<ValuesType>>;
     /**
@@ -255,7 +256,7 @@ export declare class ThreadsClient extends BaseClient {
      * @param threadId The ID of the thread.
      * @returns
      */
-    updateState<ValuesType = DefaultValues>(threadId: string, options: {
+    updateState<ValuesType = TUpdateType>(threadId: string, options: {
         values: ValuesType;
         checkpoint?: Checkpoint;
         checkpointId?: string;
@@ -275,22 +276,16 @@ export declare class ThreadsClient extends BaseClient {
      * @param options Additional options.
      * @returns List of thread states.
      */
-    getHistory<ValuesType = DefaultValues>(threadId: string, options?: {
+    getHistory<ValuesType = TStateType>(threadId: string, options?: {
         limit?: number;
         before?: Config;
         checkpoint?: Partial<Omit<Checkpoint, "thread_id">>;
         metadata?: Metadata;
     }): Promise<ThreadState<ValuesType>[]>;
 }
-export declare class RunsClient extends BaseClient {
-    stream(threadId: null, assistantId: string, payload?: Omit<RunsStreamPayload, "multitaskStrategy" | "onCompletion">): AsyncGenerator<{
-        event: StreamEvent;
-        data: any;
-    }>;
-    stream(threadId: string, assistantId: string, payload?: RunsStreamPayload): AsyncGenerator<{
-        event: StreamEvent;
-        data: any;
-    }>;
+export declare class RunsClient<TStateType = DefaultValues, TUpdateType = TStateType, TCustomEventType = unknown> extends BaseClient {
+    stream<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false>(threadId: null, assistantId: string, payload?: Omit<RunsStreamPayload<TStreamMode, TSubgraphs>, "multitaskStrategy" | "onCompletion">): TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType, TCustomEventType>;
+    stream<TStreamMode extends StreamMode | StreamMode[] = StreamMode, TSubgraphs extends boolean = false>(threadId: string, assistantId: string, payload?: RunsStreamPayload<TStreamMode, TSubgraphs>): TypedAsyncGenerator<TStreamMode, TSubgraphs, TStateType, TUpdateType, TCustomEventType>;
     /**
      * Create a run.
      *
@@ -447,7 +442,7 @@ export declare class StoreClient extends BaseClient {
         offset?: number;
     }): Promise<ListNamespaceResponse>;
 }
-export declare class Client {
+export declare class Client<TStateType = DefaultValues, TUpdateType = TStateType, TCustomEventType = unknown> {
     /**
      * The client for interacting with assistants.
      */
@@ -455,11 +450,11 @@ export declare class Client {
     /**
      * The client for interacting with threads.
      */
-    threads: ThreadsClient;
+    threads: ThreadsClient<TStateType, TUpdateType>;
     /**
      * The client for interacting with runs.
      */
-    runs: RunsClient;
+    runs: RunsClient<TStateType, TUpdateType, TCustomEventType>;
     /**
      * The client for interacting with cron runs.
      */

package/dist/client.js

@@ -1,9 +1,8 @@
 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";
 import { getEnvironmentVariable } from "./utils/env.js";
-import { _getFetchImplementation } from "./singletons/fetch.js";
+import { BytesLineDecoder, SSEDecoder } from "./utils/sse.js";
 /**
  * Get the API key from the environment.
  * Precedence:
@@ -109,7 +108,7 @@ class BaseClient {
         return [targetUrl, mutatedOptions];
     }
     async fetch(path, options) {
-        const response = await this.asyncCaller.call(_getFetchImplementation(), ...this.prepareFetchOptions(path, options));
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(path, options));
         if (response.status === 202 || response.status === 204) {
             return undefined;
         }
@@ -512,43 +511,15 @@ export class RunsClient extends BaseClient {
             if_not_exists: payload?.ifNotExists,
         };
         const endpoint = threadId == null ? `/runs/stream` : `/threads/${threadId}/runs/stream`;
-        const response = await this.asyncCaller.call(_getFetchImplementation(), ...this.prepareFetchOptions(endpoint, {
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(endpoint, {
             method: "POST",
             json,
             timeoutMs: null,
             signal: payload?.signal,
         }));
-        let parser;
-        let onEndEvent;
-        const textDecoder = new TextDecoder();
-        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() })).pipeThrough(new TransformStream({
-            async start(ctrl) {
-                parser = createParser((event) => {
-                    if ((payload?.signal && payload.signal.aborted) ||
-                        (event.type === "event" && event.data === "[DONE]")) {
-                        ctrl.terminate();
-                        return;
-                    }
-                    if ("data" in event) {
-                        ctrl.enqueue({
-                            event: event.event ?? "message",
-                            data: JSON.parse(event.data),
-                        });
-                    }
-                });
-                onEndEvent = () => {
-                    ctrl.enqueue({ event: "end", data: undefined });
-                };
-            },
-            async transform(chunk) {
-                const payload = textDecoder.decode(chunk);
-                parser.feed(payload);
-                // eventsource-parser will ignore events
-                // that are not terminated by a newline
-                if (payload.trim() === "event: end")
-                    onEndEvent();
-            },
-        }));
+        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() }))
+            .pipeThrough(new BytesLineDecoder())
+            .pipeThrough(new SSEDecoder());
         yield* IterableReadableStream.fromReadableStream(stream);
     }
     /**
@@ -716,43 +687,15 @@ export class RunsClient extends BaseClient {
             options instanceof AbortSignal
             ? { signal: options }
             : options;
-        const response = await this.asyncCaller.call(_getFetchImplementation(), ...this.prepareFetchOptions(`/threads/${threadId}/runs/${runId}/stream`, {
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(`/threads/${threadId}/runs/${runId}/stream`, {
             method: "GET",
             timeoutMs: null,
             signal: opts?.signal,
             params: { cancel_on_disconnect: opts?.cancelOnDisconnect ? "1" : "0" },
         }));
-        let parser;
-        let onEndEvent;
-        const textDecoder = new TextDecoder();
-        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() })).pipeThrough(new TransformStream({
-            async start(ctrl) {
-                parser = createParser((event) => {
-                    if ((opts?.signal && opts.signal.aborted) ||
-                        (event.type === "event" && event.data === "[DONE]")) {
-                        ctrl.terminate();
-                        return;
-                    }
-                    if ("data" in event) {
-                        ctrl.enqueue({
-                            event: event.event ?? "message",
-                            data: JSON.parse(event.data),
-                        });
-                    }
-                });
-                onEndEvent = () => {
-                    ctrl.enqueue({ event: "end", data: undefined });
-                };
-            },
-            async transform(chunk) {
-                const payload = textDecoder.decode(chunk);
-                parser.feed(payload);
-                // eventsource-parser will ignore events
-                // that are not terminated by a newline
-                if (payload.trim() === "event: end")
-                    onEndEvent();
-            },
-        }));
+        const stream = (response.body || new ReadableStream({ start: (ctrl) => ctrl.close() }))
+            .pipeThrough(new BytesLineDecoder())
+            .pipeThrough(new SSEDecoder());
         yield* IterableReadableStream.fromReadableStream(stream);
     }
     /**

package/dist/index.d.ts

@@ -2,3 +2,6 @@ export { Client } from "./client.js";
 export type { Assistant, AssistantVersion, AssistantGraph, Config, DefaultValues, GraphSchema, Metadata, Run, Thread, ThreadTask, ThreadState, ThreadStatus, Cron, Checkpoint, Interrupt, ListNamespaceResponse, Item, SearchItem, SearchItemsResponse, CronCreateResponse, CronCreateForThreadResponse, } from "./schema.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export type { OnConflictBehavior, Command } from "./types.js";
+export type { StreamMode } from "./types.stream.js";
+export type { ValuesStreamEvent, MessagesTupleStreamEvent, MetadataStreamEvent, UpdatesStreamEvent, CustomStreamEvent, MessagesStreamEvent, DebugStreamEvent, EventsStreamEvent, ErrorStreamEvent, FeedbackStreamEvent, } from "./types.stream.js";
+export type { Message, HumanMessage, AIMessage, ToolMessage, SystemMessage, FunctionMessage, RemoveMessage, } from "./types.messages.js";

package/dist/react/debug.cjs

@@ -0,0 +1,32 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DebugSegmentsView = void 0;
+const jsx_runtime_1 = require("react/jsx-runtime");
+function DebugSegmentsView(props) {
+    const concatContent = (value) => {
+        let content;
+        try {
+            content = value.values?.messages?.at(-1)?.content ?? "";
+        }
+        catch {
+            content = JSON.stringify(value.values);
+        }
+        content = content.replace(/(\n|\r\n)/g, "");
+        if (content.length <= 23)
+            return content;
+        return `${content.slice(0, 10)}...${content.slice(-10)}`;
+    };
+    return ((0, jsx_runtime_1.jsx)("div", { children: props.sequence.items.map((item, index) => {
+            if (item.type === "fork") {
+                return ((0, jsx_runtime_1.jsx)("div", { children: item.items.map((fork, idx) => {
+                        const [first] = fork.items;
+                        return ((0, jsx_runtime_1.jsxs)("details", { children: [(0, jsx_runtime_1.jsxs)("summary", { children: ["Fork", " ", (0, jsx_runtime_1.jsxs)("span", { className: "font-mono", children: ["...", first.path.at(-1)?.slice(-4)] })] }), (0, jsx_runtime_1.jsx)("div", { className: "ml-4", children: (0, jsx_runtime_1.jsx)(DebugSegmentsView, { sequence: fork }) })] }, idx));
+                    }) }, index));
+            }
+            if (item.type === "node") {
+                return ((0, jsx_runtime_1.jsxs)("div", { className: "flex items-center gap-2", children: [(0, jsx_runtime_1.jsxs)("pre", { children: ["(", item.value.metadata?.step, ") ...", item.value.checkpoint.checkpoint_id?.slice(-4), " (", item.value.metadata?.source, "): ", concatContent(item.value)] }), (0, jsx_runtime_1.jsx)("button", { type: "button", className: "border rounded-sm text-sm py-0.5 px-1 text-muted-foreground", onClick: () => console.log(item.path, item.value), children: "console.log" })] }, index));
+            }
+            return null;
+        }) }));
+}
+exports.DebugSegmentsView = DebugSegmentsView;

package/dist/react/debug.d.ts

@@ -0,0 +1,23 @@
+import { ThreadState } from "../schema.js";
+interface Node<StateType = any> {
+    type: "node";
+    value: ThreadState<StateType>;
+    path: string[];
+}
+interface ValidFork<StateType = any> {
+    type: "fork";
+    items: Array<ValidSequence<StateType>>;
+}
+interface ValidSequence<StateType = any> {
+    type: "sequence";
+    items: [Node<StateType>, ...(Node<StateType> | ValidFork<StateType>)[]];
+}
+export type CheckpointBranchPath = string[];
+export type MessageBranch = {
+    current: CheckpointBranchPath;
+    options: CheckpointBranchPath[];
+};
+export declare function DebugSegmentsView(props: {
+    sequence: ValidSequence<ThreadState>;
+}): import("react/jsx-runtime").JSX.Element;
+export {};

package/dist/react/debug.js

@@ -0,0 +1,28 @@
+import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+export function DebugSegmentsView(props) {
+    const concatContent = (value) => {
+        let content;
+        try {
+            content = value.values?.messages?.at(-1)?.content ?? "";
+        }
+        catch {
+            content = JSON.stringify(value.values);
+        }
+        content = content.replace(/(\n|\r\n)/g, "");
+        if (content.length <= 23)
+            return content;
+        return `${content.slice(0, 10)}...${content.slice(-10)}`;
+    };
+    return (_jsx("div", { children: props.sequence.items.map((item, index) => {
+            if (item.type === "fork") {
+                return (_jsx("div", { children: item.items.map((fork, idx) => {
+                        const [first] = fork.items;
+                        return (_jsxs("details", { children: [_jsxs("summary", { children: ["Fork", " ", _jsxs("span", { className: "font-mono", children: ["...", first.path.at(-1)?.slice(-4)] })] }), _jsx("div", { className: "ml-4", children: _jsx(DebugSegmentsView, { sequence: fork }) })] }, idx));
+                    }) }, index));
+            }
+            if (item.type === "node") {
+                return (_jsxs("div", { className: "flex items-center gap-2", children: [_jsxs("pre", { children: ["(", item.value.metadata?.step, ") ...", item.value.checkpoint.checkpoint_id?.slice(-4), " (", item.value.metadata?.source, "): ", concatContent(item.value)] }), _jsx("button", { type: "button", className: "border rounded-sm text-sm py-0.5 px-1 text-muted-foreground", onClick: () => console.log(item.path, item.value), children: "console.log" })] }, index));
+            }
+            return null;
+        }) }));
+}

package/dist/react/index.cjs

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useStream = void 0;
+var stream_js_1 = require("./stream.cjs");
+Object.defineProperty(exports, "useStream", { enumerable: true, get: function () { return stream_js_1.useStream; } });

package/dist/react/index.d.ts

@@ -0,0 +1 @@
+export { useStream, type MessageMetadata } from "./stream.js";

package/dist/react/index.js

@@ -0,0 +1 @@
+export { useStream } from "./stream.js";