@langchain/langgraph 0.2.7 → 0.2.9

package/dist/pregel/index.d.ts

@@ -1,12 +1,12 @@
 import { Runnable, RunnableConfig, RunnableFunc } from "@langchain/core/runnables";
 import { IterableReadableStream } from "@langchain/core/utils/stream";
-import { All, BaseCheckpointSaver, CheckpointListOptions } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, CheckpointListOptions, CheckpointTuple } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelNode } from "./read.js";
 import { ChannelWrite } from "./write.js";
 import { PregelInterface, PregelParams, StateSnapshot, StreamMode } from "./types.js";
 import { StrRecord } from "./algo.js";
-import { RetryPolicy } from "./utils.js";
+import { RetryPolicy } from "./utils/index.js";
 import { BaseStore } from "../store/base.js";
 import { ManagedValueMapping, type ManagedValueSpec } from "../managed/base.js";
 type WriteValue = Runnable | RunnableFunc<unknown, unknown> | unknown;
@@ -35,6 +35,8 @@ export interface PregelOptions<Nn extends StrRecord<string, PregelNode>, Cc exte
     interruptAfter?: All | Array<keyof Nn>;
     /** Enable debug mode for the graph run. */
     debug?: boolean;
+    /** Whether to stream subgraphs. */
+    subgraphs?: boolean;
 }
 export type PregelInputType = any;
 export type PregelOutputType = any;
@@ -45,7 +47,6 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
     channels: Cc;
     inputChannels: keyof Cc | Array<keyof Cc>;
     outputChannels: keyof Cc | Array<keyof Cc>;
-    configKeys?: string[];
     autoValidate: boolean;
     streamMode: StreamMode[];
     streamChannels?: keyof Cc | Array<keyof Cc>;
@@ -55,15 +56,25 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
     debug: boolean;
     checkpointer?: BaseCheckpointSaver;
     retryPolicy?: RetryPolicy;
+    config?: RunnableConfig;
     store?: BaseStore;
     constructor(fields: PregelParams<Nn, Cc>);
+    withConfig(config: RunnableConfig): typeof this;
     validate(): this;
     get streamChannelsList(): Array<keyof Cc>;
     get streamChannelsAsIs(): keyof Cc | Array<keyof Cc>;
+    getSubgraphs(namespace?: string, recurse?: boolean): Generator<[string, Pregel<any, any>]>;
+    protected _prepareStateSnapshot({ config, saved, subgraphCheckpointer, }: {
+        config: RunnableConfig;
+        saved?: CheckpointTuple;
+        subgraphCheckpointer?: BaseCheckpointSaver;
+    }): Promise<StateSnapshot>;
     /**
      * Get the current state of the graph.
      */
-    getState(config: RunnableConfig): Promise<StateSnapshot>;
+    getState(config: RunnableConfig, options?: {
+        subgraphs?: boolean;
+    }): Promise<StateSnapshot>;
     /**
      * Get the history of the state of the graph.
      */
@@ -73,7 +84,7 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
      * node `as_node`. If `as_node` is not provided, it will be set to the last node
      * that updated the state, if not ambiguous.
      */
-    updateState(config: RunnableConfig, values: Record<string, unknown> | unknown, asNode?: keyof Nn): Promise<RunnableConfig>;
+    updateState(inputConfig: RunnableConfig, values: Record<string, unknown> | unknown, asNode?: keyof Nn | string): Promise<RunnableConfig>;
     _defaults(config: PregelOptions<Nn, Cc>): [
         boolean,
         StreamMode[],