Merge pull request #2066 from langchain-ai/dqbd/checkpoint-sdk-js

fix(sdk-js): pass checkpoint when creating run
langchain-ai · Oct 9, 2024 · fc276c5 · fc276c5
2 parents f9df0f4 + 9360545
commit fc276c5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/libs/sdk-js/src/client.ts b/libs/sdk-js/src/client.ts
@@ -674,6 +674,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,
@@ -756,6 +757,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,
@@ -822,6 +824,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,

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";
@@ -37,6 +37,11 @@ interface RunsInvokePayload {
    */
   checkpointId?: string;
 
+  /**
+   * Checkpoint for when creating a new run.
+   */
+  checkpoint?: Omit<Checkpoint, "thread_id">;
+
   /**
    * Interrupt execution before entering these nodes.
    */