@langchain/langgraph-sdk 0.0.1-rc.10 → 0.0.1-rc.11

package/dist/client.d.mts

@@ -1,6 +1,6 @@
 import { Assistant, AssistantGraph, Config, DefaultValues, GraphSchema, Metadata, Run, RunEvent, Thread, ThreadState } from "./schema.js";
 import { AsyncCaller, AsyncCallerParams } from "./utils/async_caller.mjs";
-import { RunsCreatePayload, RunsStreamPayload, RunsWaitPayload } from "./types.mjs";
+import { RunsCreatePayload, RunsStreamPayload, RunsWaitPayload, StreamEvent } from "./types.mjs";
 interface ClientConfig {
     apiUrl?: string;
     callerOptions?: AsyncCallerParams;
@@ -63,6 +63,12 @@ declare class AssistantsClient extends BaseClient {
         config?: Config;
         metadata?: Metadata;
     }): Promise<Assistant>;
+    /**
+     * Delete an assistant.
+     *
+     * @param assistantId ID of the assistant.
+     */
+    delete(assistantId: string): Promise<void>;
     /**
      * List assistants.
      * @param query Query options.
@@ -173,15 +179,12 @@ declare class ThreadsClient extends BaseClient {
     }): Promise<ThreadState<ValuesType>[]>;
 }
 declare class RunsClient extends BaseClient {
-    /**
-     * Create a run and stream the results.
-     *
-     * @param threadId The ID of the thread.
-     * @param assistantId Assistant ID to use for this run.
-     * @param payload Payload for creating a run.
-     */
+    stream(threadId: null, assistantId: string, payload?: Omit<RunsStreamPayload, "multitaskStrategy">): AsyncGenerator<{
+        event: StreamEvent;
+        data: any;
+    }>;
     stream(threadId: string, assistantId: string, payload?: RunsStreamPayload): AsyncGenerator<{
-        event: "events" | "metadata" | "debug" | "updates" | "values" | "messages/partial" | "messages/metadata" | "messages/complete" | (string & {});
+        event: StreamEvent;
         data: any;
     }>;
     /**
@@ -193,14 +196,7 @@ declare class RunsClient extends BaseClient {
      * @returns The created run.
      */
     create(threadId: string, assistantId: string, payload?: RunsCreatePayload): Promise<Run>;
-    /**
-     * Create a run and wait for it to complete.
-     *
-     * @param threadId The ID of the thread.
-     * @param assistantId Assistant ID to use for this run.
-     * @param payload Payload for creating a run.
-     * @returns The last values chunk of the thread.
-     */
+    wait(threadId: null, assistantId: string, payload?: Omit<RunsWaitPayload, "multitaskStrategy">): Promise<ThreadState["values"]>;
     wait(threadId: string, assistantId: string, payload?: RunsWaitPayload): Promise<ThreadState["values"]>;
     /**
      * List all runs for a thread.
@@ -229,6 +225,23 @@ declare class RunsClient extends BaseClient {
      * @returns The run.
      */
     get(threadId: string, runId: string): Promise<Run>;
+    /**
+     * Cancel a run.
+     *
+     * @param threadId The ID of the thread.
+     * @param runId The ID of the run.
+     * @param wait Whether to block when canceling
+     * @returns
+     */
+    cancel(threadId: string, runId: string, wait?: boolean): Promise<void>;
+    /**
+     * Block until a run is done.
+     *
+     * @param threadId The ID of the thread.
+     * @param runId The ID of the run.
+     * @returns
+     */
+    join(threadId: string, runId: string): Promise<void>;
     /**
      * List all events for a run.
      *
@@ -249,6 +262,14 @@ declare class RunsClient extends BaseClient {
          */
         offset?: number;
     }): Promise<RunEvent[]>;
+    /**
+     * Delete a run.
+     *
+     * @param threadId The ID of the thread.
+     * @param runId The ID of the run.
+     * @returns
+     */
+    delete(threadId: string, runId: string): Promise<void>;
 }
 export declare class Client {
     /**

package/dist/client.mjs

@@ -65,6 +65,9 @@ class BaseClient {
     }
     async fetch(path, options) {
         const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(path, options));
+        if (response.status == 202) {
+            return undefined;
+        }
         return response.json();
     }
 }
@@ -125,6 +128,16 @@ class AssistantsClient extends BaseClient {
             },
         });
     }
+    /**
+     * Delete an assistant.
+     *
+     * @param assistantId ID of the assistant.
+     */
+    async delete(assistantId) {
+        return this.fetch(`/assistants/${assistantId}`, {
+            method: "DELETE",
+        });
+    }
     /**
      * List assistants.
      * @param query Query options.
@@ -278,18 +291,23 @@ class RunsClient extends BaseClient {
      * @param payload Payload for creating a run.
      */
     async *stream(threadId, assistantId, payload) {
-        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(`/threads/${threadId}/runs/stream`, {
+        const json = {
+            input: payload?.input,
+            config: payload?.config,
+            metadata: payload?.metadata,
+            stream_mode: payload?.streamMode,
+            feedback_keys: payload?.feedbackKeys,
+            assistant_id: assistantId,
+            interrupt_before: payload?.interruptBefore,
+            interrupt_after: payload?.interruptAfter,
+        };
+        if (payload?.multitaskStrategy != null) {
+            json["multitask_strategy"] = payload?.multitaskStrategy;
+        }
+        const endpoint = threadId == null ? `/runs/stream` : `/threads/${threadId}/runs/stream`;
+        const response = await this.asyncCaller.fetch(...this.prepareFetchOptions(endpoint, {
             method: "POST",
-            json: {
-                input: payload?.input,
-                config: payload?.config,
-                metadata: payload?.metadata,
-                stream_mode: payload?.streamMode,
-                feedback_keys: payload?.feedbackKeys,
-                assistant_id: assistantId,
-                interrupt_before: payload?.interruptBefore,
-                interrupt_after: payload?.interruptAfter,
-            },
+            json,
             signal: payload?.signal,
         }));
         let parser;
@@ -325,17 +343,21 @@ class RunsClient extends BaseClient {
      * @returns The created run.
      */
     async create(threadId, assistantId, payload) {
+        const json = {
+            input: payload?.input,
+            config: payload?.config,
+            metadata: payload?.metadata,
+            assistant_id: assistantId,
+            interrupt_before: payload?.interruptBefore,
+            interrupt_after: payload?.interruptAfter,
+            webhook: payload?.webhook,
+        };
+        if (payload?.multitaskStrategy != null) {
+            json["multitask_strategy"] = payload?.multitaskStrategy;
+        }
         return this.fetch(`/threads/${threadId}/runs`, {
             method: "POST",
-            json: {
-                input: payload?.input,
-                config: payload?.config,
-                metadata: payload?.metadata,
-                assistant_id: assistantId,
-                interrupt_before: payload?.interruptBefore,
-                interrupt_after: payload?.interruptAfter,
-                webhook: payload?.webhook,
-            },
+            json,
             signal: payload?.signal,
         });
     }
@@ -348,16 +370,21 @@ class RunsClient extends BaseClient {
      * @returns The last values chunk of the thread.
      */
     async wait(threadId, assistantId, payload) {
-        return this.fetch(`/threads/${threadId}/runs/wait`, {
+        const json = {
+            input: payload?.input,
+            config: payload?.config,
+            metadata: payload?.metadata,
+            assistant_id: assistantId,
+            interrupt_before: payload?.interruptBefore,
+            interrupt_after: payload?.interruptAfter,
+        };
+        if (payload?.multitaskStrategy != null) {
+            json["multitask_strategy"] = payload?.multitaskStrategy;
+        }
+        const endpoint = threadId == null ? `/runs/wait` : `/threads/${threadId}/runs/wait`;
+        return this.fetch(endpoint, {
             method: "POST",
-            json: {
-                input: payload?.input,
-                config: payload?.config,
-                metadata: payload?.metadata,
-                assistant_id: assistantId,
-                interrupt_before: payload?.interruptBefore,
-                interrupt_after: payload?.interruptAfter,
-            },
+            json,
             signal: payload?.signal,
         });
     }
@@ -386,6 +413,32 @@ class RunsClient extends BaseClient {
     async get(threadId, runId) {
         return this.fetch(`/threads/${threadId}/runs/${runId}`);
     }
+    /**
+     * Cancel a run.
+     *
+     * @param threadId The ID of the thread.
+     * @param runId The ID of the run.
+     * @param wait Whether to block when canceling
+     * @returns
+     */
+    async cancel(threadId, runId, wait = false) {
+        return this.fetch(`/threads/${threadId}/runs/${runId}/cancel`, {
+            method: "POST",
+            params: {
+                wait: wait ? "1" : "0",
+            },
+        });
+    }
+    /**
+     * Block until a run is done.
+     *
+     * @param threadId The ID of the thread.
+     * @param runId The ID of the run.
+     * @returns
+     */
+    async join(threadId, runId) {
+        return this.fetch(`/threads/${threadId}/runs/${runId}/join`);
+    }
     /**
      * List all events for a run.
      *
@@ -402,6 +455,18 @@ class RunsClient extends BaseClient {
             },
         });
     }
+    /**
+     * Delete a run.
+     *
+     * @param threadId The ID of the thread.
+     * @param runId The ID of the run.
+     * @returns
+     */
+    async delete(threadId, runId) {
+        return this.fetch(`/threads/${threadId}/runs/${runId}`, {
+            method: "DELETE",
+        });
+    }
 }
 export class Client {
     constructor(config) {

package/dist/types.d.mts

@@ -1,5 +1,7 @@
 import { Config, Metadata } from "./schema.js";
 export type StreamMode = "values" | "messages" | "updates" | "events" | "debug";
+export type MultitaskStrategy = "reject" | "interrupt" | "rollback" | "enqueue";
+export type StreamEvent = "events" | "metadata" | "debug" | "updates" | "values" | "messages/partial" | "messages/metadata" | "messages/complete" | (string & {});
 interface RunsInvokePayload {
     /**
      * Input to the run. Pass `null` to resume from the current state of the thread.
@@ -21,6 +23,17 @@ interface RunsInvokePayload {
      * Interrupt execution after leaving these nodes.
      */
     interruptAfter?: string[];
+    /**
+     * Strategy to handle concurrent runs on the same thread. Only relevant if
+     * there is a pending/inflight run on the same thread. One of:
+     * - "reject": Reject the new run.
+     * - "interrupt": Interrupt the current run, keeping steps completed until now,
+         and start a new one.
+     * - "rollback": Cancel and delete the existing run, rolling back the thread to
+        the state before it had started, then start the new run.
+     * - "enqueue": Queue up the new run to start after the current run finishes.
+     */
+    multitaskStrategy?: MultitaskStrategy;
     /**
      * Abort controller signal to cancel the run.
      */

package/package.json

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