feat(js-sdk): Add status field in search args

langchain-ai · Nov 6, 2024 · 82c9aa8 · 82c9aa8
1 parent f7a93f9
commit 82c9aa8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/libs/sdk-js/src/client.ts b/libs/sdk-js/src/client.ts
@@ -15,6 +15,7 @@ import {
   SearchItemsResponse,
   ListNamespaceResponse,
   Item,
+  ThreadStatus,
 } from "./schema.js";
 import { AsyncCaller, AsyncCallerParams } from "./utils/async_caller.js";
 import {
@@ -515,13 +516,19 @@ export class ThreadsClient extends BaseClient {
      * Offset to start from.
      */
     offset?: number;
+    /**
+     * Thread status to filter on.
+     * Must be one of 'idle', 'busy', 'interrupted' or 'error'.
+     */
+    status?: ThreadStatus;
   }): Promise<Thread[]> {
     return this.fetch<Thread[]>("/threads/search", {
       method: "POST",
       json: {
         metadata: query?.metadata ?? undefined,
         limit: query?.limit ?? 10,
         offset: query?.offset ?? 0,
+        status: query?.status,
       },
     });
   }

diff --git a/libs/sdk-js/src/schema.ts b/libs/sdk-js/src/schema.ts
@@ -10,7 +10,7 @@ type RunStatus =
   | "timeout"
   | "interrupted";
 
-type ThreadStatus = "idle" | "busy" | "interrupted" | "error";
+export type ThreadStatus = "idle" | "busy" | "interrupted" | "error";
 
 type MultitaskStrategy = "reject" | "interrupt" | "rollback" | "enqueue";