@langchain/langgraph-sdk 0.0.107 → 0.0.111

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @langchain/langgraph-sdk
 
+## 0.0.111
+
+### Patch Changes
+
+- b5f14d0: Add methods to connect with the /count endpoints
+
+## 0.0.109
+
+### Patch Changes
+
+- e8b4540: Add support for select statements in the search endpoints
+- 9c57526: fix(sdk): expose subgraph events in useStream callbacks
+
 ## 0.0.107
 
 ### Patch Changes

package/dist/client.cjs

@@ -300,6 +300,23 @@ class CronsClient extends BaseClient {
                 offset: query?.offset ?? 0,
                 sort_by: query?.sortBy ?? undefined,
                 sort_order: query?.sortOrder ?? undefined,
+                select: query?.select ?? undefined,
+            },
+        });
+    }
+    /**
+     * Count cron jobs matching filters.
+     *
+     * @param query.assistantId Assistant ID to filter by.
+     * @param query.threadId Thread ID to filter by.
+     * @returns Number of cron jobs matching the criteria.
+     */
+    async count(query) {
+        return this.fetch(`/runs/crons/count`, {
+            method: "POST",
+            json: {
+                assistant_id: query?.assistantId ?? undefined,
+                thread_id: query?.threadId ?? undefined,
             },
         });
     }
@@ -413,6 +430,23 @@ class AssistantsClient extends BaseClient {
                 offset: query?.offset ?? 0,
                 sort_by: query?.sortBy ?? undefined,
                 sort_order: query?.sortOrder ?? undefined,
+                select: query?.select ?? undefined,
+            },
+        });
+    }
+    /**
+     * Count assistants matching filters.
+     *
+     * @param query.metadata Metadata to filter by. Exact match for each key/value.
+     * @param query.graphId Optional graph id to filter by.
+     * @returns Number of assistants matching the criteria.
+     */
+    async count(query) {
+        return this.fetch(`/assistants/count`, {
+            method: "POST",
+            json: {
+                metadata: query?.metadata ?? undefined,
+                graph_id: query?.graphId ?? undefined,
             },
         });
     }
@@ -532,6 +566,25 @@ class ThreadsClient extends BaseClient {
                 status: query?.status,
                 sort_by: query?.sortBy,
                 sort_order: query?.sortOrder,
+                select: query?.select ?? undefined,
+            },
+        });
+    }
+    /**
+     * Count threads matching filters.
+     *
+     * @param query.metadata Thread metadata to filter on.
+     * @param query.values State values to filter on.
+     * @param query.status Thread status to filter on.
+     * @returns Number of threads matching the criteria.
+     */
+    async count(query) {
+        return this.fetch(`/threads/count`, {
+            method: "POST",
+            json: {
+                metadata: query?.metadata ?? undefined,
+                values: query?.values ?? undefined,
+                status: query?.status ?? undefined,
             },
         });
     }
@@ -795,6 +848,7 @@ class RunsClient extends BaseClient {
                 limit: options?.limit ?? 10,
                 offset: options?.offset ?? 0,
                 status: options?.status ?? undefined,
+                select: options?.select ?? undefined,
             },
         });
     }

package/dist/client.d.ts

@@ -1,4 +1,4 @@
-import { Assistant, AssistantGraph, AssistantSortBy, AssistantVersion, CancelAction, Checkpoint, Config, Cron, CronCreateForThreadResponse, CronCreateResponse, CronSortBy, DefaultValues, GraphSchema, Item, ListNamespaceResponse, Metadata, Run, RunStatus, SearchItemsResponse, SortOrder, Subgraphs, Thread, ThreadSortBy, ThreadState, ThreadStatus } from "./schema.js";
+import { Assistant, AssistantGraph, AssistantSortBy, AssistantSelectField, AssistantVersion, CancelAction, Checkpoint, Config, Cron, CronSelectField, CronCreateForThreadResponse, CronCreateResponse, CronSortBy, DefaultValues, GraphSchema, Item, ListNamespaceResponse, Metadata, Run, RunSelectField, RunStatus, SearchItemsResponse, SortOrder, Subgraphs, Thread, ThreadSelectField, ThreadSortBy, ThreadState, ThreadStatus } from "./schema.js";
 import type { Command, CronsCreatePayload, OnConflictBehavior, RunsCreatePayload, RunsStreamPayload, RunsWaitPayload, StreamEvent } from "./types.js";
 import type { StreamMode, TypedAsyncGenerator } from "./types.stream.js";
 import { AsyncCaller, AsyncCallerParams } from "./utils/async_caller.js";
@@ -85,7 +85,19 @@ export declare class CronsClient extends BaseClient {
         offset?: number;
         sortBy?: CronSortBy;
         sortOrder?: SortOrder;
+        select?: CronSelectField[];
     }): Promise<Cron[]>;
+    /**
+     * Count cron jobs matching filters.
+     *
+     * @param query.assistantId Assistant ID to filter by.
+     * @param query.threadId Thread ID to filter by.
+     * @returns Number of cron jobs matching the criteria.
+     */
+    count(query?: {
+        assistantId?: string;
+        threadId?: string;
+    }): Promise<number>;
 }
 export declare class AssistantsClient extends BaseClient {
     /**
@@ -168,7 +180,19 @@ export declare class AssistantsClient extends BaseClient {
         offset?: number;
         sortBy?: AssistantSortBy;
         sortOrder?: SortOrder;
+        select?: AssistantSelectField[];
     }): Promise<Assistant[]>;
+    /**
+     * Count assistants matching filters.
+     *
+     * @param query.metadata Metadata to filter by. Exact match for each key/value.
+     * @param query.graphId Optional graph id to filter by.
+     * @returns Number of assistants matching the criteria.
+     */
+    count(query?: {
+        metadata?: Metadata;
+        graphId?: string;
+    }): Promise<number>;
     /**
      * List all versions of an assistant.
      *
@@ -295,7 +319,21 @@ export declare class ThreadsClient<TStateType = DefaultValues, TUpdateType = TSt
          * Must be one of 'asc' or 'desc'.
          */
         sortOrder?: SortOrder;
+        select?: ThreadSelectField[];
     }): Promise<Thread<ValuesType>[]>;
+    /**
+     * Count threads matching filters.
+     *
+     * @param query.metadata Thread metadata to filter on.
+     * @param query.values State values to filter on.
+     * @param query.status Thread status to filter on.
+     * @returns Number of threads matching the criteria.
+     */
+    count<ValuesType = TStateType>(query?: {
+        metadata?: Metadata;
+        values?: ValuesType;
+        status?: ThreadStatus;
+    }): Promise<number>;
     /**
      * Get state for a thread.
      *
@@ -383,6 +421,7 @@ export declare class RunsClient<TStateType = DefaultValues, TUpdateType = TState
          * Status of the run to filter by.
          */
         status?: RunStatus;
+        select?: RunSelectField[];
     }): Promise<Run[]>;
     /**
      * Get a run by ID.

package/dist/client.js

@@ -295,6 +295,23 @@ export class CronsClient extends BaseClient {
                 offset: query?.offset ?? 0,
                 sort_by: query?.sortBy ?? undefined,
                 sort_order: query?.sortOrder ?? undefined,
+                select: query?.select ?? undefined,
+            },
+        });
+    }
+    /**
+     * Count cron jobs matching filters.
+     *
+     * @param query.assistantId Assistant ID to filter by.
+     * @param query.threadId Thread ID to filter by.
+     * @returns Number of cron jobs matching the criteria.
+     */
+    async count(query) {
+        return this.fetch(`/runs/crons/count`, {
+            method: "POST",
+            json: {
+                assistant_id: query?.assistantId ?? undefined,
+                thread_id: query?.threadId ?? undefined,
             },
         });
     }
@@ -407,6 +424,23 @@ export class AssistantsClient extends BaseClient {
                 offset: query?.offset ?? 0,
                 sort_by: query?.sortBy ?? undefined,
                 sort_order: query?.sortOrder ?? undefined,
+                select: query?.select ?? undefined,
+            },
+        });
+    }
+    /**
+     * Count assistants matching filters.
+     *
+     * @param query.metadata Metadata to filter by. Exact match for each key/value.
+     * @param query.graphId Optional graph id to filter by.
+     * @returns Number of assistants matching the criteria.
+     */
+    async count(query) {
+        return this.fetch(`/assistants/count`, {
+            method: "POST",
+            json: {
+                metadata: query?.metadata ?? undefined,
+                graph_id: query?.graphId ?? undefined,
             },
         });
     }
@@ -525,6 +559,25 @@ export class ThreadsClient extends BaseClient {
                 status: query?.status,
                 sort_by: query?.sortBy,
                 sort_order: query?.sortOrder,
+                select: query?.select ?? undefined,
+            },
+        });
+    }
+    /**
+     * Count threads matching filters.
+     *
+     * @param query.metadata Thread metadata to filter on.
+     * @param query.values State values to filter on.
+     * @param query.status Thread status to filter on.
+     * @returns Number of threads matching the criteria.
+     */
+    async count(query) {
+        return this.fetch(`/threads/count`, {
+            method: "POST",
+            json: {
+                metadata: query?.metadata ?? undefined,
+                values: query?.values ?? undefined,
+                status: query?.status ?? undefined,
             },
         });
     }
@@ -787,6 +840,7 @@ export class RunsClient extends BaseClient {
                 limit: options?.limit ?? 10,
                 offset: options?.offset ?? 0,
                 status: options?.status ?? undefined,
+                select: options?.select ?? undefined,
             },
         });
     }

package/dist/react/stream.cjs

@@ -284,6 +284,9 @@ function useStreamValuesState() {
     return [values?.[0] ?? null, setStreamValues, mutate];
 }
 function useStream(options) {
+    const matchEventType = (expected, actual, _data) => {
+        return expected === actual || actual.startsWith(`${expected}|`);
+    };
     let { messagesKey } = options;
     const { assistantId, fetchStateHistory } = options;
     const { onCreated, onError, onFinish } = options;
@@ -454,36 +457,37 @@ function useStream(options) {
                     streamError = new StreamError(data);
                     break;
                 }
-                if (event === "updates")
-                    options.onUpdateEvent?.(data);
-                if (event === "custom" ||
-                    // if `streamSubgraphs: true`, then we also want
-                    // to also receive custom events from subgraphs
-                    event.startsWith("custom|"))
-                    options.onCustomEvent?.(data, {
-                        mutate: getMutateFn("stream", historyValues),
-                    });
+                const namespace = event.includes("|")
+                    ? event.split("|").slice(1)
+                    : undefined;
+                const mutate = getMutateFn("stream", historyValues);
                 if (event === "metadata")
                     options.onMetadataEvent?.(data);
                 if (event === "events")
                     options.onLangChainEvent?.(data);
-                if (event === "debug")
-                    options.onDebugEvent?.(data);
-                if (event === "checkpoints")
-                    options.onCheckpointEvent?.(data);
-                if (event === "tasks")
-                    options.onTaskEvent?.(data);
+                if (matchEventType("updates", event, data)) {
+                    options.onUpdateEvent?.(data, { namespace, mutate });
+                }
+                if (matchEventType("custom", event, data)) {
+                    options.onCustomEvent?.(data, { namespace, mutate });
+                }
+                if (matchEventType("checkpoints", event, data)) {
+                    options.onCheckpointEvent?.(data, { namespace });
+                }
+                if (matchEventType("tasks", event, data)) {
+                    options.onTaskEvent?.(data, { namespace });
+                }
+                if (matchEventType("debug", event, data)) {
+                    options.onDebugEvent?.(data, { namespace });
+                }
                 if (event === "values") {
-                    if ("__interrupt__" in data) {
-                        // don't update values on interrupt values event
+                    // don't update values on interrupt values event
+                    if ("__interrupt__" in data)
                         continue;
-                    }
                     setStreamValues(data);
                 }
-                if (event === "messages" ||
-                    // if `streamSubgraphs: true`, then we also want
-                    // to also receive messages from subgraphs
-                    event.startsWith("messages|")) {
+                // Consume subgraph messages as well
+                if (matchEventType("messages", event, data)) {
                     const [serialized, metadata] = data;
                     const messageId = messageManagerRef.current.add(serialized, metadata);
                     if (!messageId) {

package/dist/react/stream.d.ts

@@ -109,11 +109,15 @@ export interface UseStreamOptions<StateType extends Record<string, unknown> = Re
     /**
      * Callback that is called when an update event is received.
      */
-    onUpdateEvent?: (data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>["data"]) => void;
+    onUpdateEvent?: (data: UpdatesStreamEvent<GetUpdateType<Bag, StateType>>["data"], options: {
+        namespace: string[] | undefined;
+        mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
+    }) => void;
     /**
      * Callback that is called when a custom event is received.
      */
     onCustomEvent?: (data: CustomStreamEvent<GetCustomEventType<Bag>>["data"], options: {
+        namespace: string[] | undefined;
         mutate: (update: Partial<StateType> | ((prev: StateType) => Partial<StateType>)) => void;
     }) => void;
     /**
@@ -129,15 +133,21 @@ export interface UseStreamOptions<StateType extends Record<string, unknown> = Re
      * Callback that is called when a debug event is received.
      * @internal This API is experimental and subject to change.
      */
-    onDebugEvent?: (data: DebugStreamEvent["data"]) => void;
+    onDebugEvent?: (data: DebugStreamEvent["data"], options: {
+        namespace: string[] | undefined;
+    }) => void;
     /**
      * Callback that is called when a checkpoints event is received.
      */
-    onCheckpointEvent?: (data: CheckpointsStreamEvent<StateType>["data"]) => void;
+    onCheckpointEvent?: (data: CheckpointsStreamEvent<StateType>["data"], options: {
+        namespace: string[] | undefined;
+    }) => void;
     /**
      * Callback that is called when a tasks event is received.
      */
-    onTaskEvent?: (data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>["data"]) => void;
+    onTaskEvent?: (data: TasksStreamEvent<StateType, GetUpdateType<Bag, StateType>>["data"], options: {
+        namespace: string[] | undefined;
+    }) => void;
     /**
      * Callback that is called when the stream is stopped by the user.
      * Provides a mutate function to update the stream state immediately

package/dist/react/stream.js

@@ -280,6 +280,9 @@ function useStreamValuesState() {
     return [values?.[0] ?? null, setStreamValues, mutate];
 }
 export function useStream(options) {
+    const matchEventType = (expected, actual, _data) => {
+        return expected === actual || actual.startsWith(`${expected}|`);
+    };
     let { messagesKey } = options;
     const { assistantId, fetchStateHistory } = options;
     const { onCreated, onError, onFinish } = options;
@@ -450,36 +453,37 @@ export function useStream(options) {
                     streamError = new StreamError(data);
                     break;
                 }
-                if (event === "updates")
-                    options.onUpdateEvent?.(data);
-                if (event === "custom" ||
-                    // if `streamSubgraphs: true`, then we also want
-                    // to also receive custom events from subgraphs
-                    event.startsWith("custom|"))
-                    options.onCustomEvent?.(data, {
-                        mutate: getMutateFn("stream", historyValues),
-                    });
+                const namespace = event.includes("|")
+                    ? event.split("|").slice(1)
+                    : undefined;
+                const mutate = getMutateFn("stream", historyValues);
                 if (event === "metadata")
                     options.onMetadataEvent?.(data);
                 if (event === "events")
                     options.onLangChainEvent?.(data);
-                if (event === "debug")
-                    options.onDebugEvent?.(data);
-                if (event === "checkpoints")
-                    options.onCheckpointEvent?.(data);
-                if (event === "tasks")
-                    options.onTaskEvent?.(data);
+                if (matchEventType("updates", event, data)) {
+                    options.onUpdateEvent?.(data, { namespace, mutate });
+                }
+                if (matchEventType("custom", event, data)) {
+                    options.onCustomEvent?.(data, { namespace, mutate });
+                }
+                if (matchEventType("checkpoints", event, data)) {
+                    options.onCheckpointEvent?.(data, { namespace });
+                }
+                if (matchEventType("tasks", event, data)) {
+                    options.onTaskEvent?.(data, { namespace });
+                }
+                if (matchEventType("debug", event, data)) {
+                    options.onDebugEvent?.(data, { namespace });
+                }
                 if (event === "values") {
-                    if ("__interrupt__" in data) {
-                        // don't update values on interrupt values event
+                    // don't update values on interrupt values event
+                    if ("__interrupt__" in data)
                         continue;
-                    }
                     setStreamValues(data);
                 }
-                if (event === "messages" ||
-                    // if `streamSubgraphs: true`, then we also want
-                    // to also receive messages from subgraphs
-                    event.startsWith("messages|")) {
+                // Consume subgraph messages as well
+                if (matchEventType("messages", event, data)) {
                     const [serialized, metadata] = data;
                     const messageId = messageManagerRef.current.add(serialized, metadata);
                     if (!messageId) {

package/dist/schema.d.ts

@@ -263,4 +263,8 @@ export type AssistantSortBy = "assistant_id" | "graph_id" | "name" | "created_at
 export type ThreadSortBy = "thread_id" | "status" | "created_at" | "updated_at";
 export type CronSortBy = "cron_id" | "assistant_id" | "thread_id" | "created_at" | "updated_at" | "next_run_date";
 export type SortOrder = "asc" | "desc";
+export type AssistantSelectField = "assistant_id" | "graph_id" | "name" | "description" | "config" | "context" | "created_at" | "updated_at" | "metadata" | "version";
+export type ThreadSelectField = "thread_id" | "created_at" | "updated_at" | "metadata" | "config" | "context" | "status" | "values" | "interrupts";
+export type RunSelectField = "run_id" | "thread_id" | "assistant_id" | "created_at" | "updated_at" | "status" | "metadata" | "kwargs" | "multitask_strategy";
+export type CronSelectField = "cron_id" | "assistant_id" | "thread_id" | "end_time" | "schedule" | "created_at" | "updated_at" | "user_id" | "payload" | "next_run_date" | "metadata" | "now";
 export {};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-sdk",
-  "version": "0.0.107",
+  "version": "0.0.111",
   "description": "Client library for interacting with the LangGraph API",
   "type": "module",
   "scripts": {