Merge pull request #1993 from langchain-ai/brace/after-seconds-js

fix(js): Add afterSeconds run arg
langchain-ai · Oct 3, 2024 · f9afd3c · f9afd3c
2 parents ac7903b + 92e75aa
commit f9afd3c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/libs/sdk-js/src/client.ts b/libs/sdk-js/src/client.ts
@@ -677,6 +677,7 @@ export class RunsClient extends BaseClient {
       multitask_strategy: payload?.multitaskStrategy,
       on_completion: payload?.onCompletion,
       on_disconnect: payload?.onDisconnect,
+      after_seconds: payload?.afterSeconds,
     };
 
     const endpoint =
@@ -755,6 +756,7 @@ export class RunsClient extends BaseClient {
       webhook: payload?.webhook,
       checkpoint_id: payload?.checkpointId,
       multitask_strategy: payload?.multitaskStrategy,
+      after_seconds: payload?.afterSeconds,
     };
     return this.fetch<Run>(`/threads/${threadId}/runs`, {
       method: "POST",
@@ -823,6 +825,7 @@ export class RunsClient extends BaseClient {
       multitask_strategy: payload?.multitaskStrategy,
       on_completion: payload?.onCompletion,
       on_disconnect: payload?.onDisconnect,
+      after_seconds: payload?.afterSeconds,
     };
     const endpoint =
       threadId == null ? `/runs/wait` : `/threads/${threadId}/runs/wait`;

diff --git a/libs/sdk-js/src/types.ts b/libs/sdk-js/src/types.ts
@@ -84,6 +84,12 @@ interface RunsInvokePayload {
    * - "continue": Continue the run.
    */
   onDisconnect?: DisconnectMode;
+
+  /**
+   * The number of seconds to wait before starting the run.
+   * Use to schedule future runs.
+   */
+  afterSeconds?: number;
 }
 
 export interface RunsStreamPayload extends RunsInvokePayload {