WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-28 10:49:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Tat Dat Duong
2024-10-09 20:34:19 +02:00
parent f9df0f4700
commit 8df533b489
2 changed files with 28 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/sdk-js/src/client.ts
+21
View File
@@ -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,6 +829,12 @@ 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,
@@ -822,6 +842,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,
libs/sdk-js/src/types.ts
+7 -1
View File
@@ -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.
*/