@langchain/langgraph-api 0.0.60 → 0.0.61

package/dist/loopback.d.mts

@@ -1,3 +1,4 @@
 import { Hono } from "hono";
+import type { StorageEnv } from "./storage/types.mjs";
 export declare function getLoopbackFetch(): (url: string, init?: RequestInit) => Promise<Response>;
-export declare const bindLoopbackFetch: (app: Hono) => void;
+export declare const bindLoopbackFetch: (app: Hono<StorageEnv>) => void;

package/dist/queue.d.mts

@@ -1 +1,2 @@
-export declare const queue: () => Promise<never>;
+import type { Ops } from "./storage/types.mjs";
+export declare const queue: (ops: Ops) => Promise<never>;

package/dist/queue.mjs

@@ -1,4 +1,3 @@
-import { Runs, Threads } from "./storage/ops.mjs";
 import { streamState, } from "./stream.mjs";
 import { logError, logger } from "./logging.mjs";
 import { serializeError } from "./utils/serde.mjs";
@@ -6,16 +5,16 @@ import { callWebhook } from "./webhook.mjs";
 import { getGraph } from "./graph/load.mjs";
 const MAX_RETRY_ATTEMPTS = 3;
 const sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
-export const queue = async () => {
+export const queue = async (ops) => {
     while (true) {
-        for await (const { run, attempt, signal } of Runs.next()) {
-            await worker(run, attempt, signal);
+        for await (const { run, attempt, signal } of ops.runs.next()) {
+            await worker(ops, run, attempt, signal);
         }
         // TODO: this is very suboptimal, we should implement subscription to the run
         await sleep(1000 * Math.random());
     }
 };
-const worker = async (run, attempt, signal) => {
+const worker = async (ops, run, attempt, signal) => {
     const startedAt = new Date();
     let endedAt = undefined;
     let checkpoint = undefined;
@@ -56,11 +55,11 @@ const worker = async (run, attempt, signal) => {
                 ...(!temporary ? { onCheckpoint, onTaskResult } : undefined),
             });
             for await (const { event, data } of stream) {
-                await Runs.Stream.publish({ runId, resumable, event, data });
+                await ops.runs.stream.publish({ runId, resumable, event, data });
             }
         }
         catch (error) {
-            await Runs.Stream.publish({
+            await ops.runs.stream.publish({
                 runId,
                 resumable,
                 event: "error",
@@ -78,7 +77,7 @@ const worker = async (run, attempt, signal) => {
             run_exec_ms: endedAt.valueOf() - startedAt.valueOf(),
         });
         status = "success";
-        await Runs.setStatus(run.run_id, status);
+        await ops.runs.setStatus(run.run_id, status);
     }
     catch (error) {
         endedAt = new Date();
@@ -96,14 +95,14 @@ const worker = async (run, attempt, signal) => {
             },
         });
         status = "error";
-        await Runs.setStatus(run.run_id, "error");
+        await ops.runs.setStatus(run.run_id, "error");
     }
     finally {
         if (temporary) {
-            await Threads.delete(run.thread_id, undefined);
+            await ops.threads.delete(run.thread_id, undefined);
         }
         else {
-            await Threads.setStatus(run.thread_id, { checkpoint, exception });
+            await ops.threads.setStatus(run.thread_id, { checkpoint, exception });
         }
         if (webhook) {
             await callWebhook({

package/dist/schemas.d.mts

@@ -109,10 +109,10 @@ export declare const Assistant: z.ZodObject<{
     } & {
         [k: string]: unknown;
     };
-    graph_id: string;
-    assistant_id: string;
     created_at: string;
     updated_at: string;
+    graph_id: string;
+    assistant_id: string;
 }, {
     metadata: {} & {
         [k: string]: any;
@@ -127,10 +127,10 @@ export declare const Assistant: z.ZodObject<{
     } & {
         [k: string]: unknown;
     };
-    graph_id: string;
-    assistant_id: string;
     created_at: string;
     updated_at: string;
+    graph_id: string;
+    assistant_id: string;
 }>;
 export declare const AssistantCreate: z.ZodObject<{
     assistant_id: z.ZodOptional<z.ZodString>;
@@ -198,8 +198,8 @@ export declare const AssistantCreate: z.ZodObject<{
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
     name?: string | undefined;
-    if_exists?: "do_nothing" | "raise" | undefined;
     assistant_id?: string | undefined;
+    if_exists?: "raise" | "do_nothing" | undefined;
 }, {
     graph_id: string;
     context?: unknown;
@@ -219,8 +219,8 @@ export declare const AssistantCreate: z.ZodObject<{
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
     name?: string | undefined;
-    if_exists?: "do_nothing" | "raise" | undefined;
     assistant_id?: string | undefined;
+    if_exists?: "raise" | "do_nothing" | undefined;
 }>;
 export declare const AssistantPatch: z.ZodObject<{
     graph_id: z.ZodOptional<z.ZodString>;
@@ -353,12 +353,12 @@ export declare const CheckpointSchema: z.ZodObject<{
     checkpoint_ns: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     checkpoint_map: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
 }, "strip", z.ZodTypeAny, {
-    checkpoint_ns?: string | null | undefined;
     checkpoint_id?: string | undefined;
+    checkpoint_ns?: string | null | undefined;
     checkpoint_map?: Record<string, unknown> | null | undefined;
 }, {
-    checkpoint_ns?: string | null | undefined;
     checkpoint_id?: string | undefined;
+    checkpoint_ns?: string | null | undefined;
     checkpoint_map?: Record<string, unknown> | null | undefined;
 }>;
 export declare const CronCreate: z.ZodObject<{
@@ -418,6 +418,8 @@ export declare const CronCreate: z.ZodObject<{
     context?: unknown;
     metadata?: z.objectOutputType<{}, z.ZodAny, "strip"> | undefined;
     input?: z.objectOutputType<{}, z.ZodAny, "strip">[] | z.objectOutputType<{}, z.ZodAny, "strip"> | undefined;
+    interrupt_before?: string[] | "*" | undefined;
+    interrupt_after?: string[] | "*" | undefined;
     config?: z.objectOutputType<{
         tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         recursion_limit: z.ZodOptional<z.ZodNumber>;
@@ -432,10 +434,8 @@ export declare const CronCreate: z.ZodObject<{
             thread_ts: z.ZodOptional<z.ZodString>;
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
-    checkpoint_id?: string | undefined;
-    interrupt_before?: string[] | "*" | undefined;
-    interrupt_after?: string[] | "*" | undefined;
     webhook?: string | undefined;
+    checkpoint_id?: string | undefined;
     multitask_strategy?: "reject" | "rollback" | "interrupt" | "enqueue" | undefined;
 }, {
     thread_id: string;
@@ -443,6 +443,8 @@ export declare const CronCreate: z.ZodObject<{
     context?: unknown;
     metadata?: z.objectInputType<{}, z.ZodAny, "strip"> | undefined;
     input?: z.objectInputType<{}, z.ZodAny, "strip">[] | z.objectInputType<{}, z.ZodAny, "strip"> | undefined;
+    interrupt_before?: string[] | "*" | undefined;
+    interrupt_after?: string[] | "*" | undefined;
     config?: z.objectInputType<{
         tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         recursion_limit: z.ZodOptional<z.ZodNumber>;
@@ -457,10 +459,8 @@ export declare const CronCreate: z.ZodObject<{
             thread_ts: z.ZodOptional<z.ZodString>;
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
-    checkpoint_id?: string | undefined;
-    interrupt_before?: string[] | "*" | undefined;
-    interrupt_after?: string[] | "*" | undefined;
     webhook?: string | undefined;
+    checkpoint_id?: string | undefined;
     multitask_strategy?: "reject" | "rollback" | "interrupt" | "enqueue" | undefined;
 }>;
 export declare const CronSearch: z.ZodObject<{
@@ -470,14 +470,14 @@ export declare const CronSearch: z.ZodObject<{
     offset: z.ZodOptional<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
     thread_id?: string | undefined;
+    assistant_id?: string | undefined;
     limit?: number | undefined;
     offset?: number | undefined;
-    assistant_id?: string | undefined;
 }, {
     thread_id?: string | undefined;
+    assistant_id?: string | undefined;
     limit?: number | undefined;
     offset?: number | undefined;
-    assistant_id?: string | undefined;
 }>;
 export declare const GraphSchema: z.ZodObject<{
     graph_id: z.ZodString;
@@ -517,29 +517,29 @@ export declare const Run: z.ZodObject<{
     kwargs: z.ZodObject<{}, "strip", z.ZodAny, z.objectOutputType<{}, z.ZodAny, "strip">, z.objectInputType<{}, z.ZodAny, "strip">>;
     multitask_strategy: z.ZodEnum<["reject", "rollback", "interrupt", "enqueue"]>;
 }, "strip", z.ZodTypeAny, {
-    status: "error" | "timeout" | "interrupted" | "pending" | "running" | "success";
+    status: "error" | "interrupted" | "pending" | "running" | "success" | "timeout";
+    thread_id: string;
     metadata: {} & {
         [k: string]: any;
     };
-    run_id: string;
-    thread_id: string;
-    assistant_id: string;
     created_at: string;
     updated_at: string;
+    run_id: string;
+    assistant_id: string;
     multitask_strategy: "reject" | "rollback" | "interrupt" | "enqueue";
     kwargs: {} & {
         [k: string]: any;
     };
 }, {
-    status: "error" | "timeout" | "interrupted" | "pending" | "running" | "success";
+    status: "error" | "interrupted" | "pending" | "running" | "success" | "timeout";
+    thread_id: string;
     metadata: {} & {
         [k: string]: any;
     };
-    run_id: string;
-    thread_id: string;
-    assistant_id: string;
     created_at: string;
     updated_at: string;
+    run_id: string;
+    assistant_id: string;
     multitask_strategy: "reject" | "rollback" | "interrupt" | "enqueue";
     kwargs: {} & {
         [k: string]: any;
@@ -606,12 +606,12 @@ export declare const RunCreate: z.ZodObject<{
         checkpoint_ns: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         checkpoint_map: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
     }, "strip", z.ZodTypeAny, {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     }, {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     }>>;
     input: z.ZodOptional<z.ZodUnion<[z.ZodUnknown, z.ZodNull]>>;
@@ -728,6 +728,20 @@ export declare const RunCreate: z.ZodObject<{
     context?: unknown;
     metadata?: z.objectOutputType<{}, z.ZodAny, "strip"> | undefined;
     input?: unknown;
+    command?: {
+        update?: Record<string, unknown> | [string, unknown][] | undefined;
+        goto?: string | {
+            node: string;
+            input?: unknown;
+        } | (string | {
+            node: string;
+            input?: unknown;
+        })[] | undefined;
+        resume?: unknown;
+    } | undefined;
+    stream_mode?: "values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints" | ("values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints")[] | undefined;
+    interrupt_before?: string[] | "*" | undefined;
+    interrupt_after?: string[] | "*" | undefined;
     config?: z.objectOutputType<{
         tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         recursion_limit: z.ZodOptional<z.ZodNumber>;
@@ -742,33 +756,19 @@ export declare const RunCreate: z.ZodObject<{
             thread_ts: z.ZodOptional<z.ZodString>;
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
+    webhook?: string | undefined;
+    feedback_keys?: string[] | undefined;
     checkpoint_id?: string | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     } | undefined;
-    command?: {
-        update?: Record<string, unknown> | [string, unknown][] | undefined;
-        goto?: string | {
-            node: string;
-            input?: unknown;
-        } | (string | {
-            node: string;
-            input?: unknown;
-        })[] | undefined;
-        resume?: unknown;
-    } | undefined;
-    stream_mode?: "values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events" | ("values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events")[] | undefined;
-    interrupt_before?: string[] | "*" | undefined;
-    interrupt_after?: string[] | "*" | undefined;
-    webhook?: string | undefined;
-    feedback_keys?: string[] | undefined;
     multitask_strategy?: "reject" | "rollback" | "interrupt" | "enqueue" | undefined;
-    if_not_exists?: "reject" | "create" | undefined;
-    after_seconds?: number | undefined;
     stream_subgraphs?: boolean | undefined;
     stream_resumable?: boolean | undefined;
+    after_seconds?: number | undefined;
+    if_not_exists?: "reject" | "create" | undefined;
     on_completion?: "delete" | "keep" | undefined;
     langsmith_tracer?: {
         project_name?: string | undefined;
@@ -779,6 +779,20 @@ export declare const RunCreate: z.ZodObject<{
     context?: unknown;
     metadata?: z.objectInputType<{}, z.ZodAny, "strip"> | undefined;
     input?: unknown;
+    command?: {
+        update?: Record<string, unknown> | [string, unknown][] | undefined;
+        goto?: string | {
+            node: string;
+            input?: unknown;
+        } | (string | {
+            node: string;
+            input?: unknown;
+        })[] | undefined;
+        resume?: unknown;
+    } | undefined;
+    stream_mode?: "values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints" | ("values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints")[] | undefined;
+    interrupt_before?: string[] | "*" | undefined;
+    interrupt_after?: string[] | "*" | undefined;
     config?: z.objectInputType<{
         tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         recursion_limit: z.ZodOptional<z.ZodNumber>;
@@ -793,34 +807,20 @@ export declare const RunCreate: z.ZodObject<{
             thread_ts: z.ZodOptional<z.ZodString>;
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
+    webhook?: string | undefined;
+    feedback_keys?: string[] | undefined;
     checkpoint_id?: string | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     } | undefined;
-    command?: {
-        update?: Record<string, unknown> | [string, unknown][] | undefined;
-        goto?: string | {
-            node: string;
-            input?: unknown;
-        } | (string | {
-            node: string;
-            input?: unknown;
-        })[] | undefined;
-        resume?: unknown;
-    } | undefined;
-    stream_mode?: "values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events" | ("values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events")[] | undefined;
-    interrupt_before?: string[] | "*" | undefined;
-    interrupt_after?: string[] | "*" | undefined;
-    webhook?: string | undefined;
-    feedback_keys?: string[] | undefined;
     multitask_strategy?: "reject" | "rollback" | "interrupt" | "enqueue" | undefined;
-    if_not_exists?: "reject" | "create" | undefined;
-    after_seconds?: number | undefined;
     on_disconnect?: "cancel" | "continue" | undefined;
     stream_subgraphs?: boolean | undefined;
     stream_resumable?: boolean | undefined;
+    after_seconds?: number | undefined;
+    if_not_exists?: "reject" | "create" | undefined;
     on_completion?: "delete" | "keep" | undefined;
     langsmith_tracer?: {
         project_name?: string | undefined;
@@ -835,12 +835,12 @@ export declare const RunBatchCreate: z.ZodArray<z.ZodObject<{
         checkpoint_ns: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         checkpoint_map: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
     }, "strip", z.ZodTypeAny, {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     }, {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     }>>;
     input: z.ZodOptional<z.ZodUnion<[z.ZodUnknown, z.ZodNull]>>;
@@ -957,6 +957,20 @@ export declare const RunBatchCreate: z.ZodArray<z.ZodObject<{
     context?: unknown;
     metadata?: z.objectOutputType<{}, z.ZodAny, "strip"> | undefined;
     input?: unknown;
+    command?: {
+        update?: Record<string, unknown> | [string, unknown][] | undefined;
+        goto?: string | {
+            node: string;
+            input?: unknown;
+        } | (string | {
+            node: string;
+            input?: unknown;
+        })[] | undefined;
+        resume?: unknown;
+    } | undefined;
+    stream_mode?: "values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints" | ("values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints")[] | undefined;
+    interrupt_before?: string[] | "*" | undefined;
+    interrupt_after?: string[] | "*" | undefined;
     config?: z.objectOutputType<{
         tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         recursion_limit: z.ZodOptional<z.ZodNumber>;
@@ -971,33 +985,19 @@ export declare const RunBatchCreate: z.ZodArray<z.ZodObject<{
             thread_ts: z.ZodOptional<z.ZodString>;
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
+    webhook?: string | undefined;
+    feedback_keys?: string[] | undefined;
     checkpoint_id?: string | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     } | undefined;
-    command?: {
-        update?: Record<string, unknown> | [string, unknown][] | undefined;
-        goto?: string | {
-            node: string;
-            input?: unknown;
-        } | (string | {
-            node: string;
-            input?: unknown;
-        })[] | undefined;
-        resume?: unknown;
-    } | undefined;
-    stream_mode?: "values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events" | ("values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events")[] | undefined;
-    interrupt_before?: string[] | "*" | undefined;
-    interrupt_after?: string[] | "*" | undefined;
-    webhook?: string | undefined;
-    feedback_keys?: string[] | undefined;
     multitask_strategy?: "reject" | "rollback" | "interrupt" | "enqueue" | undefined;
-    if_not_exists?: "reject" | "create" | undefined;
-    after_seconds?: number | undefined;
     stream_subgraphs?: boolean | undefined;
     stream_resumable?: boolean | undefined;
+    after_seconds?: number | undefined;
+    if_not_exists?: "reject" | "create" | undefined;
     on_completion?: "delete" | "keep" | undefined;
     langsmith_tracer?: {
         project_name?: string | undefined;
@@ -1008,6 +1008,20 @@ export declare const RunBatchCreate: z.ZodArray<z.ZodObject<{
     context?: unknown;
     metadata?: z.objectInputType<{}, z.ZodAny, "strip"> | undefined;
     input?: unknown;
+    command?: {
+        update?: Record<string, unknown> | [string, unknown][] | undefined;
+        goto?: string | {
+            node: string;
+            input?: unknown;
+        } | (string | {
+            node: string;
+            input?: unknown;
+        })[] | undefined;
+        resume?: unknown;
+    } | undefined;
+    stream_mode?: "values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints" | ("values" | "debug" | "messages" | "messages-tuple" | "custom" | "updates" | "events" | "tasks" | "checkpoints")[] | undefined;
+    interrupt_before?: string[] | "*" | undefined;
+    interrupt_after?: string[] | "*" | undefined;
     config?: z.objectInputType<{
         tags: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         recursion_limit: z.ZodOptional<z.ZodNumber>;
@@ -1022,34 +1036,20 @@ export declare const RunBatchCreate: z.ZodArray<z.ZodObject<{
             thread_ts: z.ZodOptional<z.ZodString>;
         }, z.ZodUnknown, "strip">>>;
     }, z.ZodUnknown, "strip"> | undefined;
+    webhook?: string | undefined;
+    feedback_keys?: string[] | undefined;
     checkpoint_id?: string | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     } | undefined;
-    command?: {
-        update?: Record<string, unknown> | [string, unknown][] | undefined;
-        goto?: string | {
-            node: string;
-            input?: unknown;
-        } | (string | {
-            node: string;
-            input?: unknown;
-        })[] | undefined;
-        resume?: unknown;
-    } | undefined;
-    stream_mode?: "values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events" | ("values" | "debug" | "updates" | "messages" | "checkpoints" | "tasks" | "custom" | "messages-tuple" | "events")[] | undefined;
-    interrupt_before?: string[] | "*" | undefined;
-    interrupt_after?: string[] | "*" | undefined;
-    webhook?: string | undefined;
-    feedback_keys?: string[] | undefined;
     multitask_strategy?: "reject" | "rollback" | "interrupt" | "enqueue" | undefined;
-    if_not_exists?: "reject" | "create" | undefined;
-    after_seconds?: number | undefined;
     on_disconnect?: "cancel" | "continue" | undefined;
     stream_subgraphs?: boolean | undefined;
     stream_resumable?: boolean | undefined;
+    after_seconds?: number | undefined;
+    if_not_exists?: "reject" | "create" | undefined;
     on_completion?: "delete" | "keep" | undefined;
     langsmith_tracer?: {
         project_name?: string | undefined;
@@ -1085,6 +1085,29 @@ export declare const AssistantSearchRequest: z.ZodObject<{
     limit?: number | undefined;
     offset?: number | undefined;
 }>;
+export declare const AssistantCountRequest: z.ZodObject<{
+    metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    graph_id: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    metadata?: Record<string, unknown> | undefined;
+    graph_id?: string | undefined;
+}, {
+    metadata?: Record<string, unknown> | undefined;
+    graph_id?: string | undefined;
+}>;
+export declare const ThreadCountRequest: z.ZodObject<{
+    metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    status: z.ZodOptional<z.ZodEnum<["idle", "busy", "interrupted", "error"]>>;
+    values: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+}, "strip", z.ZodTypeAny, {
+    values?: Record<string, unknown> | undefined;
+    status?: "error" | "idle" | "busy" | "interrupted" | undefined;
+    metadata?: Record<string, unknown> | undefined;
+}, {
+    values?: Record<string, unknown> | undefined;
+    status?: "error" | "idle" | "busy" | "interrupted" | undefined;
+    metadata?: Record<string, unknown> | undefined;
+}>;
 export declare const ThreadSearchRequest: z.ZodObject<{
     metadata: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
     status: z.ZodOptional<z.ZodEnum<["idle", "busy", "interrupted", "error"]>>;
@@ -1243,9 +1266,9 @@ export declare const ThreadCreate: z.ZodObject<{
     metadata: z.ZodOptional<z.ZodObject<{}, "strip", z.ZodAny, z.objectOutputType<{}, z.ZodAny, "strip">, z.objectInputType<{}, z.ZodAny, "strip">>>;
     if_exists: z.ZodOptional<z.ZodUnion<[z.ZodLiteral<"raise">, z.ZodLiteral<"do_nothing">]>>;
 }, "strip", z.ZodTypeAny, {
-    metadata?: z.objectOutputType<{}, z.ZodAny, "strip"> | undefined;
     thread_id?: string | undefined;
-    if_exists?: "do_nothing" | "raise" | undefined;
+    metadata?: z.objectOutputType<{}, z.ZodAny, "strip"> | undefined;
+    if_exists?: "raise" | "do_nothing" | undefined;
     supersteps?: {
         updates: {
             as_node: string;
@@ -1264,9 +1287,9 @@ export declare const ThreadCreate: z.ZodObject<{
         }[];
     }[] | undefined;
 }, {
-    metadata?: z.objectInputType<{}, z.ZodAny, "strip"> | undefined;
     thread_id?: string | undefined;
-    if_exists?: "do_nothing" | "raise" | undefined;
+    metadata?: z.objectInputType<{}, z.ZodAny, "strip"> | undefined;
+    if_exists?: "raise" | "do_nothing" | undefined;
     supersteps?: {
         updates: {
             as_node: string;
@@ -1304,8 +1327,8 @@ export declare const ThreadState: z.ZodObject<{
     metadata: {} & {
         [k: string]: any;
     };
-    checkpoint_id: string;
     created_at: string;
+    checkpoint_id: string;
     next: string[];
     parent_checkpoint_id: string;
 }, {
@@ -1313,8 +1336,8 @@ export declare const ThreadState: z.ZodObject<{
     metadata: {} & {
         [k: string]: any;
     };
-    checkpoint_id: string;
     created_at: string;
+    checkpoint_id: string;
     next: string[];
     parent_checkpoint_id: string;
 }>;
@@ -1351,20 +1374,20 @@ export declare const ThreadStateUpdate: z.ZodObject<{
         checkpoint_ns: z.ZodOptional<z.ZodNullable<z.ZodString>>;
         checkpoint_map: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
     }, "strip", z.ZodTypeAny, {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     }, {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     }>>>;
 }, "strip", z.ZodTypeAny, {
     values?: Record<string, unknown> | Record<string, unknown>[] | null | undefined;
     checkpoint_id?: string | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     } | null | undefined;
     as_node?: string | undefined;
@@ -1372,8 +1395,8 @@ export declare const ThreadStateUpdate: z.ZodObject<{
     values?: Record<string, unknown> | Record<string, unknown>[] | null | undefined;
     checkpoint_id?: string | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | null | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | null | undefined;
         checkpoint_map?: Record<string, unknown> | null | undefined;
     } | null | undefined;
     as_node?: string | undefined;
@@ -1387,28 +1410,28 @@ export declare const ThreadHistoryRequest: z.ZodObject<{
         checkpoint_ns: z.ZodOptional<z.ZodString>;
         checkpoint_map: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
     }, "strip", z.ZodTypeAny, {
-        checkpoint_ns?: string | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | undefined;
         checkpoint_map?: Record<string, unknown> | undefined;
     }, {
-        checkpoint_ns?: string | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | undefined;
         checkpoint_map?: Record<string, unknown> | undefined;
     }>>;
 }, "strip", z.ZodTypeAny, {
     limit: number;
     metadata?: Record<string, unknown> | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | undefined;
         checkpoint_map?: Record<string, unknown> | undefined;
     } | undefined;
     before?: string | undefined;
 }, {
     metadata?: Record<string, unknown> | undefined;
     checkpoint?: {
-        checkpoint_ns?: string | undefined;
         checkpoint_id?: string | undefined;
+        checkpoint_ns?: string | undefined;
         checkpoint_map?: Record<string, unknown> | undefined;
     } | undefined;
     limit?: number | undefined;

package/dist/schemas.mjs

@@ -274,6 +274,31 @@ export const AssistantSearchRequest = z
         .optional(),
 })
     .describe("Payload for listing assistants.");
+export const AssistantCountRequest = z
+    .object({
+    metadata: z
+        .record(z.unknown())
+        .describe("Metadata to search for.")
+        .optional(),
+    graph_id: z.string().describe("Filter by graph ID.").optional(),
+})
+    .describe("Payload for counting assistants.");
+export const ThreadCountRequest = z
+    .object({
+    metadata: z
+        .record(z.unknown())
+        .describe("Metadata to search for.")
+        .optional(),
+    status: z
+        .enum(["idle", "busy", "interrupted", "error"])
+        .describe("Filter by thread status.")
+        .optional(),
+    values: z
+        .record(z.unknown())
+        .describe("Filter by thread values.")
+        .optional(),
+})
+    .describe("Payload for counting threads.");
 export const ThreadSearchRequest = z
     .object({
     metadata: z