fix(sdk-js): allow passing checkpoint when creating a run

langchain-ai · Oct 9, 2024 · 8df533b · 8df533b
1 parent f9df0f4
commit 8df533b
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/libs/sdk-js/src/client.ts b/libs/sdk-js/src/client.ts
@@ -664,6 +664,12 @@ export class RunsClient extends BaseClient {
     event: StreamEvent;
     data: any;
   }> {
+    if (!!payload?.checkpointId && !!payload?.checkpoint) {
+      throw new Error(
+        "Cannot specify both a checkpoint and a checkpoint ID in the payload.",
+      );
+    }
+
     const json: Record<string, any> = {
       input: payload?.input,
       config: payload?.config,
@@ -674,6 +680,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,
@@ -748,6 +755,12 @@ export class RunsClient extends BaseClient {
     assistantId: string,
     payload?: RunsCreatePayload,
   ): Promise<Run> {
+    if (!!payload?.checkpointId && !!payload?.checkpoint) {
+      throw new Error(
+        "Cannot specify both a checkpoint and a checkpoint ID in the payload.",
+      );
+    }
+
     const json: Record<string, any> = {
       input: payload?.input,
       config: payload?.config,
@@ -756,6 +769,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,
@@ -815,13 +829,20 @@ export class RunsClient extends BaseClient {
     assistantId: string,
     payload?: RunsWaitPayload,
   ): Promise<ThreadState["values"]> {
+    if (!!payload?.checkpointId && !!payload?.checkpoint) {
+      throw new Error(
+        "Cannot specify both a checkpoint and a checkpoint ID in the payload.",
+      );
+    }
+
     const json: Record<string, any> = {
       input: payload?.input,
       config: payload?.config,
       metadata: payload?.metadata,
       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,

diff --git a/libs/sdk-js/src/types.ts b/libs/sdk-js/src/types.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";
@@ -34,9 +34,15 @@ interface RunsInvokePayload {
 
   /**
    * Checkpoint ID for when creating a new run.
+   * @deprecated Use `checkpoint` instead.
    */
   checkpointId?: string;
 
+  /**
+   * Checkpoint for when creating a new run.
+   */
+  checkpoint?: Omit<Checkpoint, "thread_id">;
+
   /**
    * Interrupt execution before entering these nodes.
    */