@langchain/core 0.1.4 → 0.1.6

package/dist/runnables/base.d.ts

@@ -1,4 +1,4 @@
-import { CallbackManager, CallbackManagerForChainRun, BaseCallbackConfig } from "../callbacks/manager.js";
+import { CallbackManager, CallbackManagerForChainRun } from "../callbacks/manager.js";
 import { LogStreamCallbackHandlerInput, RunLogPatch } from "../tracers/log_stream.js";
 import { Serializable } from "../load/serializable.js";
 import { IterableReadableStream, type IterableReadableStreamInterface } from "../utils/stream.js";
@@ -20,7 +20,6 @@ export interface RunnableInterface<RunInput = any, RunOutput = any, CallOptions
         returnExceptions: true;
     }): Promise<(RunOutput | Error)[]>;
     batch(inputs: RunInput[], options?: Partial<CallOptions> | Partial<CallOptions>[], batchOptions?: RunnableBatchOptions): Promise<(RunOutput | Error)[]>;
-    batch(inputs: RunInput[], options?: Partial<CallOptions> | Partial<CallOptions>[], batchOptions?: RunnableBatchOptions): Promise<(RunOutput | Error)[]>;
     stream(input: RunInput, options?: Partial<CallOptions>): Promise<IterableReadableStreamInterface<RunOutput>>;
     transform(generator: AsyncGenerator<RunInput>, options: Partial<CallOptions>): AsyncGenerator<RunOutput>;
 }
@@ -44,6 +43,8 @@ export type RunnableRetryFailedAttemptHandler = (error: any) => any;
  */
 export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOptions extends RunnableConfig = RunnableConfig> extends Serializable implements RunnableInterface<RunInput, RunOutput, CallOptions> {
     protected lc_runnable: boolean;
+    name?: string;
+    getName(suffix?: string): string;
     abstract invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
     /**
      * Bind arguments to a Runnable, returning a new Runnable.
@@ -136,10 +137,10 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
      * Output values, with callbacks.
      * Use this to implement `stream()` or `transform()` in Runnable subclasses.
      */
-    protected _transformStreamWithConfig<I extends RunInput, O extends RunOutput>(inputGenerator: AsyncGenerator<I>, transformer: (generator: AsyncGenerator<I>, runManager?: CallbackManagerForChainRun, options?: Partial<CallOptions>) => AsyncGenerator<O>, options?: CallOptions & {
+    protected _transformStreamWithConfig<I extends RunInput, O extends RunOutput>(inputGenerator: AsyncGenerator<I>, transformer: (generator: AsyncGenerator<I>, runManager?: Promise<CallbackManagerForChainRun | undefined>, options?: Partial<CallOptions>) => AsyncGenerator<O>, options?: CallOptions & {
         runType?: string;
     }): AsyncGenerator<O>;
-    _patchConfig(config?: Partial<CallOptions>, callbackManager?: CallbackManager | undefined): Partial<CallOptions>;
+    _patchConfig(config?: Partial<CallOptions>, callbackManager?: CallbackManager | undefined, recursionLimit?: number | undefined): Partial<CallOptions>;
     /**
      * Create a new runnable sequence that runs each individual runnable in series,
      * piping the output of one runnable into another runnable or runnable-like.
@@ -147,6 +148,14 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
      * @returns A new runnable sequence.
      */
     pipe<NewRunOutput>(coerceable: RunnableLike<RunOutput, NewRunOutput>): RunnableSequence<RunInput, Exclude<NewRunOutput, Error>>;
+    /**
+     * Pick keys from the dict output of this runnable. Returns a new runnable.
+     */
+    pick(keys: string | string[]): RunnableSequence;
+    /**
+     * Assigns new fields to the dict output of this runnable. Returns a new runnable.
+     */
+    assign(mapping: RunnableMapLike<Record<string, unknown>, Record<string, unknown>>): RunnableSequence;
     /**
      * Default implementation of transform, which buffers input and then calls stream.
      * Subclasses should override this method if they can start producing output while
@@ -185,7 +194,7 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
         onError?: (run: Run, config?: RunnableConfig) => void | Promise<void>;
     }): Runnable<RunInput, RunOutput, CallOptions>;
 }
-export type RunnableBindingArgs<RunInput, RunOutput, CallOptions extends RunnableConfig> = {
+export type RunnableBindingArgs<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = {
     bound: Runnable<RunInput, RunOutput, CallOptions>;
     kwargs?: Partial<CallOptions>;
     config: RunnableConfig;
@@ -194,7 +203,7 @@ export type RunnableBindingArgs<RunInput, RunOutput, CallOptions extends Runnabl
 /**
  * A runnable that delegates calls to another runnable with a set of kwargs.
  */
-export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends RunnableConfig> extends Runnable<RunInput, RunOutput, CallOptions> {
+export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> extends Runnable<RunInput, RunOutput, CallOptions> {
     static lc_name(): string;
     lc_namespace: string[];
     lc_serializable: boolean;
@@ -203,6 +212,7 @@ export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends Ru
     protected kwargs?: Partial<CallOptions>;
     configFactories?: Array<(config: RunnableConfig) => RunnableConfig | Promise<RunnableConfig>>;
     constructor(fields: RunnableBindingArgs<RunInput, RunOutput, CallOptions>);
+    getName(suffix?: string | undefined): string;
     _mergeConfig(options?: Record<string, any>): Promise<Partial<CallOptions>>;
     bind(kwargs: Partial<CallOptions>): RunnableBinding<RunInput, RunOutput, CallOptions>;
     withConfig(config: RunnableConfig): RunnableBinding<RunInput, RunOutput, CallOptions>;
@@ -243,7 +253,7 @@ export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends Ru
  * A runnable that delegates calls to another runnable
  * with each element of the input sequence.
  */
-export declare class RunnableEach<RunInputItem, RunOutputItem, CallOptions extends BaseCallbackConfig> extends Runnable<RunInputItem[], RunOutputItem[], CallOptions> {
+export declare class RunnableEach<RunInputItem, RunOutputItem, CallOptions extends RunnableConfig> extends Runnable<RunInputItem[], RunOutputItem[], CallOptions> {
     static lc_name(): string;
     lc_serializable: boolean;
     lc_namespace: string[];
@@ -345,8 +355,9 @@ export declare class RunnableSequence<RunInput = any, RunOutput = any> extends R
         first: Runnable<RunInput>;
         middle?: Runnable[];
         last: Runnable<any, RunOutput>;
+        name?: string;
     });
-    get steps(): Runnable<RunInput, any, BaseCallbackConfig>[];
+    get steps(): Runnable<RunInput, any, RunnableConfig>[];
     invoke(input: RunInput, options?: RunnableConfig): Promise<RunOutput>;
     batch(inputs: RunInput[], options?: Partial<RunnableConfig> | Partial<RunnableConfig>[], batchOptions?: RunnableBatchOptions & {
         returnExceptions?: false;
@@ -362,7 +373,7 @@ export declare class RunnableSequence<RunInput = any, RunOutput = any> extends R
         RunnableLike<RunInput>,
         ...RunnableLike[],
         RunnableLike<any, RunOutput>
-    ]): RunnableSequence<RunInput, Exclude<RunOutput, Error>>;
+    ], name?: string): RunnableSequence<RunInput, Exclude<RunOutput, Error>>;
 }
 /**
  * A runnable that runs a mapping of runnables in parallel,
@@ -390,7 +401,10 @@ export declare class RunnableMap<RunInput = any, RunOutput extends Record<string
         steps: RunnableMapLike<RunInput, RunOutput>;
     });
     static from<RunInput, RunOutput extends Record<string, any> = Record<string, any>>(steps: RunnableMapLike<RunInput, RunOutput>): RunnableMap<RunInput, RunOutput>;
-    invoke(input: RunInput, options?: Partial<BaseCallbackConfig>): Promise<RunOutput>;
+    invoke(input: RunInput, options?: Partial<RunnableConfig>): Promise<RunOutput>;
+    _transform(generator: AsyncGenerator<RunInput>, runManagerPromise?: Promise<CallbackManagerForChainRun | undefined>, options?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    transform(generator: AsyncGenerator<RunInput>, options?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    stream(input: RunInput, options?: Partial<RunnableConfig>): Promise<IterableReadableStream<RunOutput>>;
 }
 /**
  * A runnable that runs a callable.
@@ -398,13 +412,16 @@ export declare class RunnableMap<RunInput = any, RunOutput extends Record<string
 export declare class RunnableLambda<RunInput, RunOutput> extends Runnable<RunInput, RunOutput> {
     static lc_name(): string;
     lc_namespace: string[];
-    protected func: RunnableFunc<RunInput, RunOutput>;
+    protected func: RunnableFunc<RunInput, RunOutput | Runnable<RunInput, RunOutput>>;
     constructor(fields: {
-        func: RunnableFunc<RunInput, RunOutput>;
+        func: RunnableFunc<RunInput, RunOutput | Runnable<RunInput, RunOutput>>;
     });
-    static from<RunInput, RunOutput>(func: RunnableFunc<RunInput, RunOutput>): RunnableLambda<RunInput, RunOutput>;
-    _invoke(input: RunInput, config?: Partial<BaseCallbackConfig>, runManager?: CallbackManagerForChainRun): Promise<RunOutput>;
-    invoke(input: RunInput, options?: Partial<BaseCallbackConfig>): Promise<RunOutput>;
+    static from<RunInput, RunOutput>(func: RunnableFunc<RunInput, RunOutput | Runnable<RunInput, RunOutput>>): RunnableLambda<RunInput, RunOutput>;
+    _invoke(input: RunInput, config?: Partial<RunnableConfig>, runManager?: CallbackManagerForChainRun): Promise<RunOutput>;
+    invoke(input: RunInput, options?: Partial<RunnableConfig>): Promise<RunOutput>;
+    _transform(generator: AsyncGenerator<RunInput>, runManagerPromise?: Promise<CallbackManagerForChainRun | undefined>, config?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    transform(generator: AsyncGenerator<RunInput>, options?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    stream(input: RunInput, options?: Partial<RunnableConfig>): Promise<IterableReadableStream<RunOutput>>;
 }
 export declare class RunnableParallel<RunInput> extends RunnableMap<RunInput> {
 }
@@ -421,14 +438,49 @@ export declare class RunnableWithFallbacks<RunInput, RunOutput> extends Runnable
         runnable: Runnable<RunInput, RunOutput>;
         fallbacks: Runnable<RunInput, RunOutput>[];
     });
-    runnables(): Generator<Runnable<RunInput, RunOutput, BaseCallbackConfig>, void, unknown>;
-    invoke(input: RunInput, options?: Partial<BaseCallbackConfig>): Promise<RunOutput>;
-    batch(inputs: RunInput[], options?: Partial<BaseCallbackConfig> | Partial<BaseCallbackConfig>[], batchOptions?: RunnableBatchOptions & {
+    runnables(): Generator<Runnable<RunInput, RunOutput, RunnableConfig>, void, unknown>;
+    invoke(input: RunInput, options?: Partial<RunnableConfig>): Promise<RunOutput>;
+    batch(inputs: RunInput[], options?: Partial<RunnableConfig> | Partial<RunnableConfig>[], batchOptions?: RunnableBatchOptions & {
         returnExceptions?: false;
     }): Promise<RunOutput[]>;
-    batch(inputs: RunInput[], options?: Partial<BaseCallbackConfig> | Partial<BaseCallbackConfig>[], batchOptions?: RunnableBatchOptions & {
+    batch(inputs: RunInput[], options?: Partial<RunnableConfig> | Partial<RunnableConfig>[], batchOptions?: RunnableBatchOptions & {
         returnExceptions: true;
     }): Promise<(RunOutput | Error)[]>;
-    batch(inputs: RunInput[], options?: Partial<BaseCallbackConfig> | Partial<BaseCallbackConfig>[], batchOptions?: RunnableBatchOptions): Promise<(RunOutput | Error)[]>;
+    batch(inputs: RunInput[], options?: Partial<RunnableConfig> | Partial<RunnableConfig>[], batchOptions?: RunnableBatchOptions): Promise<(RunOutput | Error)[]>;
 }
 export declare function _coerceToRunnable<RunInput, RunOutput>(coerceable: RunnableLike<RunInput, RunOutput>): Runnable<RunInput, Exclude<RunOutput, Error>>;
+export interface RunnableAssignFields<RunInput> {
+    mapper: RunnableMap<RunInput>;
+}
+/**
+ * A runnable that assigns key-value pairs to inputs of type `Record<string, unknown>`.
+ */
+export declare class RunnableAssign<RunInput extends Record<string, any> = Record<string, any>, RunOutput extends Record<string, any> = Record<string, any>, CallOptions extends RunnableConfig = RunnableConfig> extends Runnable<RunInput, RunOutput> implements RunnableAssignFields<RunInput> {
+    static lc_name(): string;
+    lc_namespace: string[];
+    lc_serializable: boolean;
+    mapper: RunnableMap<RunInput>;
+    constructor(fields: RunnableMap<RunInput> | RunnableAssignFields<RunInput>);
+    invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
+    _transform(generator: AsyncGenerator<RunInput>, runManagerPromise?: Promise<CallbackManagerForChainRun | undefined>, options?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    transform(generator: AsyncGenerator<RunInput>, options?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    stream(input: RunInput, options?: Partial<RunnableConfig>): Promise<IterableReadableStream<RunOutput>>;
+}
+export interface RunnablePickFields {
+    keys: string | string[];
+}
+/**
+ * A runnable that assigns key-value pairs to inputs of type `Record<string, unknown>`.
+ */
+export declare class RunnablePick<RunInput extends Record<string, any> = Record<string, any>, RunOutput extends Record<string, any> | any = Record<string, any> | any, CallOptions extends RunnableConfig = RunnableConfig> extends Runnable<RunInput, RunOutput> implements RunnablePickFields {
+    static lc_name(): string;
+    lc_namespace: string[];
+    lc_serializable: boolean;
+    keys: string | string[];
+    constructor(fields: string | string[] | RunnablePickFields);
+    _pick(input: RunInput): Promise<RunOutput>;
+    invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
+    _transform(generator: AsyncGenerator<RunInput>): AsyncGenerator<RunOutput>;
+    transform(generator: AsyncGenerator<RunInput>, options?: Partial<RunnableConfig>): AsyncGenerator<RunOutput>;
+    stream(input: RunInput, options?: Partial<RunnableConfig>): Promise<IterableReadableStream<RunOutput>>;
+}