langchain 0.0.152 → 0.0.153

package/dist/llms/ollama.js

@@ -287,14 +287,30 @@ export class Ollama extends LLM {
     async *_streamResponseChunks(prompt, options, runManager) {
         const stream = await this.caller.call(async () => createOllamaStream(this.baseUrl, { ...this.invocationParams(options), prompt }, options));
         for await (const chunk of stream) {
-            yield new GenerationChunk({
-                text: chunk.response,
-                generationInfo: {
-                    ...chunk,
-                    response: undefined,
-                },
-            });
-            await runManager?.handleLLMNewToken(chunk.response ?? "");
+            if (!chunk.done) {
+                yield new GenerationChunk({
+                    text: chunk.response,
+                    generationInfo: {
+                        ...chunk,
+                        response: undefined,
+                    },
+                });
+                await runManager?.handleLLMNewToken(chunk.response ?? "");
+            }
+            else {
+                yield new GenerationChunk({
+                    text: "",
+                    generationInfo: {
+                        model: chunk.model,
+                        total_duration: chunk.total_duration,
+                        load_duration: chunk.load_duration,
+                        prompt_eval_count: chunk.prompt_eval_count,
+                        prompt_eval_duration: chunk.prompt_eval_duration,
+                        eval_count: chunk.eval_count,
+                        eval_duration: chunk.eval_duration,
+                    },
+                });
+            }
         }
     }
     /** @ignore */

package/dist/llms/openai-chat.cjs

@@ -35,11 +35,7 @@ class OpenAIChat extends base_js_1.LLM {
         return "OpenAIChat";
     }
     get callKeys() {
-        return [
-            ...super.callKeys,
-            "options",
-            "promptIndex",
-        ];
+        return [...super.callKeys, "options", "promptIndex"];
     }
     get lc_secrets() {
         return {

package/dist/llms/openai-chat.d.ts

@@ -36,7 +36,7 @@ export interface OpenAIChatCallOptions extends OpenAICallOptions {
  */
 export declare class OpenAIChat extends LLM<OpenAIChatCallOptions> implements OpenAIChatInput, AzureOpenAIInput {
     static lc_name(): string;
-    get callKeys(): (keyof OpenAIChatCallOptions)[];
+    get callKeys(): string[];
     lc_serializable: boolean;
     get lc_secrets(): {
         [key: string]: string;

package/dist/llms/openai-chat.js

@@ -32,11 +32,7 @@ export class OpenAIChat extends LLM {
         return "OpenAIChat";
     }
     get callKeys() {
-        return [
-            ...super.callKeys,
-            "options",
-            "promptIndex",
-        ];
+        return [...super.callKeys, "options", "promptIndex"];
     }
     get lc_secrets() {
         return {

package/dist/llms/openai.cjs

@@ -59,7 +59,7 @@ class OpenAI extends base_js_1.BaseLLM {
         if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
             fields?.modelName?.startsWith("gpt-4")) &&
             !fields?.modelName?.includes("-instruct")) {
-            // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
+            // eslint-disable-next-line no-constructor-return
             return new openai_chat_js_1.OpenAIChat(fields, configuration);
         }
         super(fields ?? {});

package/dist/llms/openai.d.ts

@@ -22,9 +22,9 @@ export { AzureOpenAIInput, OpenAICallOptions, OpenAIInput };
  * `openai.createCompletion`} can be passed through {@link modelKwargs}, even
  * if not explicitly available on this class.
  */
-export declare class OpenAI extends BaseLLM<OpenAICallOptions> implements OpenAIInput, AzureOpenAIInput {
+export declare class OpenAI<CallOptions extends OpenAICallOptions = OpenAICallOptions> extends BaseLLM<CallOptions> implements OpenAIInput, AzureOpenAIInput {
     static lc_name(): string;
-    get callKeys(): (keyof OpenAICallOptions)[];
+    get callKeys(): string[];
     lc_serializable: boolean;
     get lc_secrets(): {
         [key: string]: string;

package/dist/llms/openai.js

@@ -56,7 +56,7 @@ export class OpenAI extends BaseLLM {
         if ((fields?.modelName?.startsWith("gpt-3.5-turbo") ||
             fields?.modelName?.startsWith("gpt-4")) &&
             !fields?.modelName?.includes("-instruct")) {
-            // eslint-disable-next-line no-constructor-return, @typescript-eslint/no-explicit-any
+            // eslint-disable-next-line no-constructor-return
             return new OpenAIChat(fields, configuration);
         }
         super(fields ?? {});

package/dist/load/import_map.cjs

@@ -24,8 +24,8 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = void 0;
+exports.retrievers__databerry = exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__fireworks = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
+exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -44,6 +44,7 @@ exports.llms__openai = __importStar(require("../llms/openai.cjs"));
 exports.llms__ai21 = __importStar(require("../llms/ai21.cjs"));
 exports.llms__aleph_alpha = __importStar(require("../llms/aleph_alpha.cjs"));
 exports.llms__ollama = __importStar(require("../llms/ollama.cjs"));
+exports.llms__fireworks = __importStar(require("../llms/fireworks.cjs"));
 exports.prompts = __importStar(require("../prompts/index.cjs"));
 exports.vectorstores__base = __importStar(require("../vectorstores/base.cjs"));
 exports.vectorstores__memory = __importStar(require("../vectorstores/memory.cjs"));
@@ -60,6 +61,7 @@ exports.document_transformers__openai_functions = __importStar(require("../docum
 exports.chat_models__base = __importStar(require("../chat_models/base.cjs"));
 exports.chat_models__openai = __importStar(require("../chat_models/openai.cjs"));
 exports.chat_models__anthropic = __importStar(require("../chat_models/anthropic.cjs"));
+exports.chat_models__fireworks = __importStar(require("../chat_models/fireworks.cjs"));
 exports.chat_models__baiduwenxin = __importStar(require("../chat_models/baiduwenxin.cjs"));
 exports.chat_models__ollama = __importStar(require("../chat_models/ollama.cjs"));
 exports.chat_models__minimax = __importStar(require("../chat_models/minimax.cjs"));

package/dist/load/import_map.d.ts

@@ -16,6 +16,7 @@ export * as llms__openai from "../llms/openai.js";
 export * as llms__ai21 from "../llms/ai21.js";
 export * as llms__aleph_alpha from "../llms/aleph_alpha.js";
 export * as llms__ollama from "../llms/ollama.js";
+export * as llms__fireworks from "../llms/fireworks.js";
 export * as prompts from "../prompts/index.js";
 export * as vectorstores__base from "../vectorstores/base.js";
 export * as vectorstores__memory from "../vectorstores/memory.js";
@@ -32,6 +33,7 @@ export * as document_transformers__openai_functions from "../document_transforme
 export * as chat_models__base from "../chat_models/base.js";
 export * as chat_models__openai from "../chat_models/openai.js";
 export * as chat_models__anthropic from "../chat_models/anthropic.js";
+export * as chat_models__fireworks from "../chat_models/fireworks.js";
 export * as chat_models__baiduwenxin from "../chat_models/baiduwenxin.js";
 export * as chat_models__ollama from "../chat_models/ollama.js";
 export * as chat_models__minimax from "../chat_models/minimax.js";

package/dist/load/import_map.js

@@ -17,6 +17,7 @@ export * as llms__openai from "../llms/openai.js";
 export * as llms__ai21 from "../llms/ai21.js";
 export * as llms__aleph_alpha from "../llms/aleph_alpha.js";
 export * as llms__ollama from "../llms/ollama.js";
+export * as llms__fireworks from "../llms/fireworks.js";
 export * as prompts from "../prompts/index.js";
 export * as vectorstores__base from "../vectorstores/base.js";
 export * as vectorstores__memory from "../vectorstores/memory.js";
@@ -33,6 +34,7 @@ export * as document_transformers__openai_functions from "../document_transforme
 export * as chat_models__base from "../chat_models/base.js";
 export * as chat_models__openai from "../chat_models/openai.js";
 export * as chat_models__anthropic from "../chat_models/anthropic.js";
+export * as chat_models__fireworks from "../chat_models/fireworks.js";
 export * as chat_models__baiduwenxin from "../chat_models/baiduwenxin.js";
 export * as chat_models__ollama from "../chat_models/ollama.js";
 export * as chat_models__minimax from "../chat_models/minimax.js";

package/dist/schema/output_parser.cjs

@@ -162,20 +162,52 @@ class BytesOutputParser extends BaseTransformOutputParser {
 }
 exports.BytesOutputParser = BytesOutputParser;
 /**
- * Custom error class used to handle exceptions related to output parsing.
- * It extends the built-in `Error` class and adds an optional `output`
- * property that can hold the output that caused the exception.
+ * Exception that output parsers should raise to signify a parsing error.
+ *
+ * This exists to differentiate parsing errors from other code or execution errors
+ * that also may arise inside the output parser. OutputParserExceptions will be
+ * available to catch and handle in ways to fix the parsing error, while other
+ * errors will be raised.
+ *
+ * @param message - The error that's being re-raised or an error message.
+ * @param llmOutput - String model output which is error-ing.
+ * @param observation - String explanation of error which can be passed to a
+ *     model to try and remediate the issue.
+ * @param sendToLLM - Whether to send the observation and llm_output back to an Agent
+ *     after an OutputParserException has been raised. This gives the underlying
+ *     model driving the agent the context that the previous output was improperly
+ *     structured, in the hopes that it will update the output to the correct
+ *     format.
  */
 class OutputParserException extends Error {
-    constructor(message, output) {
+    constructor(message, llmOutput, observation, sendToLLM = false) {
         super(message);
-        Object.defineProperty(this, "output", {
+        Object.defineProperty(this, "llmOutput", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        this.output = output;
+        Object.defineProperty(this, "observation", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sendToLLM", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.llmOutput = llmOutput;
+        this.observation = observation;
+        this.sendToLLM = sendToLLM;
+        if (sendToLLM) {
+            if (observation === undefined || llmOutput === undefined) {
+                throw new Error("Arguments 'observation' & 'llmOutput' are required if 'sendToLlm' is true");
+            }
+        }
     }
 }
 exports.OutputParserException = OutputParserException;

package/dist/schema/output_parser.d.ts

@@ -118,11 +118,26 @@ export declare class BytesOutputParser extends BaseTransformOutputParser<Uint8Ar
     getFormatInstructions(): string;
 }
 /**
- * Custom error class used to handle exceptions related to output parsing.
- * It extends the built-in `Error` class and adds an optional `output`
- * property that can hold the output that caused the exception.
+ * Exception that output parsers should raise to signify a parsing error.
+ *
+ * This exists to differentiate parsing errors from other code or execution errors
+ * that also may arise inside the output parser. OutputParserExceptions will be
+ * available to catch and handle in ways to fix the parsing error, while other
+ * errors will be raised.
+ *
+ * @param message - The error that's being re-raised or an error message.
+ * @param llmOutput - String model output which is error-ing.
+ * @param observation - String explanation of error which can be passed to a
+ *     model to try and remediate the issue.
+ * @param sendToLLM - Whether to send the observation and llm_output back to an Agent
+ *     after an OutputParserException has been raised. This gives the underlying
+ *     model driving the agent the context that the previous output was improperly
+ *     structured, in the hopes that it will update the output to the correct
+ *     format.
  */
 export declare class OutputParserException extends Error {
-    output?: string;
-    constructor(message: string, output?: string);
+    llmOutput?: string;
+    observation?: string;
+    sendToLLM: boolean;
+    constructor(message: string, llmOutput?: string, observation?: string, sendToLLM?: boolean);
 }

package/dist/schema/output_parser.js

@@ -154,19 +154,51 @@ export class BytesOutputParser extends BaseTransformOutputParser {
     }
 }
 /**
- * Custom error class used to handle exceptions related to output parsing.
- * It extends the built-in `Error` class and adds an optional `output`
- * property that can hold the output that caused the exception.
+ * Exception that output parsers should raise to signify a parsing error.
+ *
+ * This exists to differentiate parsing errors from other code or execution errors
+ * that also may arise inside the output parser. OutputParserExceptions will be
+ * available to catch and handle in ways to fix the parsing error, while other
+ * errors will be raised.
+ *
+ * @param message - The error that's being re-raised or an error message.
+ * @param llmOutput - String model output which is error-ing.
+ * @param observation - String explanation of error which can be passed to a
+ *     model to try and remediate the issue.
+ * @param sendToLLM - Whether to send the observation and llm_output back to an Agent
+ *     after an OutputParserException has been raised. This gives the underlying
+ *     model driving the agent the context that the previous output was improperly
+ *     structured, in the hopes that it will update the output to the correct
+ *     format.
  */
 export class OutputParserException extends Error {
-    constructor(message, output) {
+    constructor(message, llmOutput, observation, sendToLLM = false) {
         super(message);
-        Object.defineProperty(this, "output", {
+        Object.defineProperty(this, "llmOutput", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
-        this.output = output;
+        Object.defineProperty(this, "observation", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "sendToLLM", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.llmOutput = llmOutput;
+        this.observation = observation;
+        this.sendToLLM = sendToLLM;
+        if (sendToLLM) {
+            if (observation === undefined || llmOutput === undefined) {
+                throw new Error("Arguments 'observation' & 'llmOutput' are required if 'sendToLlm' is true");
+            }
+        }
     }
 }

package/dist/schema/runnable/base.cjs

@@ -37,7 +37,7 @@ class Runnable extends serializable_js_1.Serializable {
      */
     bind(kwargs) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
-        return new RunnableBinding({ bound: this, kwargs });
+        return new RunnableBinding({ bound: this, kwargs, config: {} });
     }
     /**
      * Return a new Runnable that maps a list of inputs to a list of outputs,
@@ -48,19 +48,33 @@ class Runnable extends serializable_js_1.Serializable {
         return new RunnableEach({ bound: this });
     }
     /**
-     * Bind arguments to a Runnable, returning a new Runnable.
+     * Add retry logic to an existing runnable.
      * @param kwargs
-     * @returns A new RunnableBinding that, when invoked, will apply the bound args.
+     * @returns A new RunnableRetry that, when invoked, will retry according to the parameters.
      */
     withRetry(fields) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
         return new RunnableRetry({
             bound: this,
             kwargs: {},
+            config: {},
             maxAttemptNumber: fields?.stopAfterAttempt,
             ...fields,
         });
     }
+    /**
+     * Bind config to a Runnable, returning a new Runnable.
+     * @param config New configuration parameters to attach to the new runnable.
+     * @returns A new RunnableBinding with a config matching what's passed.
+     */
+    withConfig(config) {
+        // eslint-disable-next-line @typescript-eslint/no-use-before-define
+        return new RunnableBinding({
+            bound: this,
+            config,
+            kwargs: {},
+        });
+    }
     /**
      * Create a new runnable from the current one that will try invoking
      * other passed fallback runnables if the initial invocation fails.
@@ -312,6 +326,12 @@ class RunnableBinding extends Runnable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "config", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "kwargs", {
             enumerable: true,
             configurable: true,
@@ -320,35 +340,70 @@ class RunnableBinding extends Runnable {
         });
         this.bound = fields.bound;
         this.kwargs = fields.kwargs;
+        this.config = fields.config;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _mergeConfig(options) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const copy = { ...this.config };
+        if (options) {
+            for (const key of Object.keys(options)) {
+                if (key === "metadata") {
+                    copy[key] = { ...copy[key], ...options[key] };
+                }
+                else if (key === "tags") {
+                    copy[key] = (copy[key] ?? []).concat(options[key] ?? []);
+                }
+                else {
+                    copy[key] = options[key] ?? copy[key];
+                }
+            }
+        }
+        return copy;
     }
     bind(kwargs) {
-        return new RunnableBinding({
+        return this.constructor({
             bound: this.bound,
             kwargs: { ...this.kwargs, ...kwargs },
+            config: this.config,
+        });
+    }
+    withConfig(config) {
+        return this.constructor({
+            bound: this.bound,
+            kwargs: this.kwargs,
+            config: { ...this.config, ...config },
+        });
+    }
+    withRetry(fields) {
+        return this.constructor({
+            bound: this.bound.withRetry(fields),
+            kwargs: this.kwargs,
+            config: this.config,
         });
     }
     async invoke(input, options) {
-        return this.bound.invoke(input, { ...options, ...this.kwargs });
+        return this.bound.invoke(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async batch(inputs, options, batchOptions) {
         const mergedOptions = Array.isArray(options)
-            ? options.map((individualOption) => ({
+            ? options.map((individualOption) => this._mergeConfig({
                 ...individualOption,
                 ...this.kwargs,
             }))
-            : { ...options, ...this.kwargs };
+            : this._mergeConfig({ ...options, ...this.kwargs });
         return this.bound.batch(inputs, mergedOptions, batchOptions);
     }
     async *_streamIterator(input, options) {
-        yield* this.bound._streamIterator(input, { ...options, ...this.kwargs });
+        yield* this.bound._streamIterator(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async stream(input, options) {
-        return this.bound.stream(input, { ...options, ...this.kwargs });
+        return this.bound.stream(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async *transform(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     generator, options) {
-        yield* this.bound.transform(generator, options);
+        yield* this.bound.transform(generator, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     static isRunnableBinding(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/schema/runnable/base.d.ts

@@ -33,14 +33,20 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
      */
     map(): Runnable<RunInput[], RunOutput[], CallOptions>;
     /**
-     * Bind arguments to a Runnable, returning a new Runnable.
+     * Add retry logic to an existing runnable.
      * @param kwargs
-     * @returns A new RunnableBinding that, when invoked, will apply the bound args.
+     * @returns A new RunnableRetry that, when invoked, will retry according to the parameters.
      */
     withRetry(fields?: {
         stopAfterAttempt?: number;
         onFailedAttempt?: RunnableRetryFailedAttemptHandler;
     }): RunnableRetry<RunInput, RunOutput, CallOptions>;
+    /**
+     * Bind config to a Runnable, returning a new Runnable.
+     * @param config New configuration parameters to attach to the new runnable.
+     * @returns A new RunnableBinding with a config matching what's passed.
+     */
+    withConfig(config: RunnableConfig): RunnableBinding<RunInput, RunOutput, CallOptions>;
     /**
      * Create a new runnable from the current one that will try invoking
      * other passed fallback runnables if the initial invocation fails.
@@ -122,18 +128,26 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
 export type RunnableBindingArgs<RunInput, RunOutput, CallOptions extends RunnableConfig> = {
     bound: Runnable<RunInput, RunOutput, CallOptions>;
     kwargs: Partial<CallOptions>;
+    config: RunnableConfig;
 };
 /**
  * A runnable that delegates calls to another runnable with a set of kwargs.
  */
-export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends BaseCallbackConfig> extends Runnable<RunInput, RunOutput, CallOptions> {
+export declare class RunnableBinding<RunInput, RunOutput, CallOptions extends RunnableConfig> extends Runnable<RunInput, RunOutput, CallOptions> {
     static lc_name(): string;
     lc_namespace: string[];
     lc_serializable: boolean;
     bound: Runnable<RunInput, RunOutput, CallOptions>;
+    config: RunnableConfig;
     protected kwargs: Partial<CallOptions>;
     constructor(fields: RunnableBindingArgs<RunInput, RunOutput, CallOptions>);
+    _mergeConfig(options?: Record<string, any>): Partial<CallOptions>;
     bind(kwargs: Partial<CallOptions>): RunnableBinding<RunInput, RunOutput, CallOptions>;
+    withConfig(config: RunnableConfig): RunnableBinding<RunInput, RunOutput, CallOptions>;
+    withRetry(fields?: {
+        stopAfterAttempt?: number;
+        onFailedAttempt?: RunnableRetryFailedAttemptHandler;
+    }): RunnableRetry<RunInput, RunOutput, CallOptions>;
     invoke(input: RunInput, options?: Partial<CallOptions>): Promise<RunOutput>;
     batch(inputs: RunInput[], options?: Partial<CallOptions> | Partial<CallOptions>[], batchOptions?: RunnableBatchOptions & {
         returnExceptions?: false;

package/dist/schema/runnable/base.js

@@ -31,7 +31,7 @@ export class Runnable extends Serializable {
      */
     bind(kwargs) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
-        return new RunnableBinding({ bound: this, kwargs });
+        return new RunnableBinding({ bound: this, kwargs, config: {} });
     }
     /**
      * Return a new Runnable that maps a list of inputs to a list of outputs,
@@ -42,19 +42,33 @@ export class Runnable extends Serializable {
         return new RunnableEach({ bound: this });
     }
     /**
-     * Bind arguments to a Runnable, returning a new Runnable.
+     * Add retry logic to an existing runnable.
      * @param kwargs
-     * @returns A new RunnableBinding that, when invoked, will apply the bound args.
+     * @returns A new RunnableRetry that, when invoked, will retry according to the parameters.
      */
     withRetry(fields) {
         // eslint-disable-next-line @typescript-eslint/no-use-before-define
         return new RunnableRetry({
             bound: this,
             kwargs: {},
+            config: {},
             maxAttemptNumber: fields?.stopAfterAttempt,
             ...fields,
         });
     }
+    /**
+     * Bind config to a Runnable, returning a new Runnable.
+     * @param config New configuration parameters to attach to the new runnable.
+     * @returns A new RunnableBinding with a config matching what's passed.
+     */
+    withConfig(config) {
+        // eslint-disable-next-line @typescript-eslint/no-use-before-define
+        return new RunnableBinding({
+            bound: this,
+            config,
+            kwargs: {},
+        });
+    }
     /**
      * Create a new runnable from the current one that will try invoking
      * other passed fallback runnables if the initial invocation fails.
@@ -305,6 +319,12 @@ export class RunnableBinding extends Runnable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "config", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "kwargs", {
             enumerable: true,
             configurable: true,
@@ -313,35 +333,70 @@ export class RunnableBinding extends Runnable {
         });
         this.bound = fields.bound;
         this.kwargs = fields.kwargs;
+        this.config = fields.config;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    _mergeConfig(options) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const copy = { ...this.config };
+        if (options) {
+            for (const key of Object.keys(options)) {
+                if (key === "metadata") {
+                    copy[key] = { ...copy[key], ...options[key] };
+                }
+                else if (key === "tags") {
+                    copy[key] = (copy[key] ?? []).concat(options[key] ?? []);
+                }
+                else {
+                    copy[key] = options[key] ?? copy[key];
+                }
+            }
+        }
+        return copy;
     }
     bind(kwargs) {
-        return new RunnableBinding({
+        return this.constructor({
             bound: this.bound,
             kwargs: { ...this.kwargs, ...kwargs },
+            config: this.config,
+        });
+    }
+    withConfig(config) {
+        return this.constructor({
+            bound: this.bound,
+            kwargs: this.kwargs,
+            config: { ...this.config, ...config },
+        });
+    }
+    withRetry(fields) {
+        return this.constructor({
+            bound: this.bound.withRetry(fields),
+            kwargs: this.kwargs,
+            config: this.config,
         });
     }
     async invoke(input, options) {
-        return this.bound.invoke(input, { ...options, ...this.kwargs });
+        return this.bound.invoke(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async batch(inputs, options, batchOptions) {
         const mergedOptions = Array.isArray(options)
-            ? options.map((individualOption) => ({
+            ? options.map((individualOption) => this._mergeConfig({
                 ...individualOption,
                 ...this.kwargs,
             }))
-            : { ...options, ...this.kwargs };
+            : this._mergeConfig({ ...options, ...this.kwargs });
         return this.bound.batch(inputs, mergedOptions, batchOptions);
     }
     async *_streamIterator(input, options) {
-        yield* this.bound._streamIterator(input, { ...options, ...this.kwargs });
+        yield* this.bound._streamIterator(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async stream(input, options) {
-        return this.bound.stream(input, { ...options, ...this.kwargs });
+        return this.bound.stream(input, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     async *transform(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     generator, options) {
-        yield* this.bound.transform(generator, options);
+        yield* this.bound.transform(generator, this._mergeConfig({ ...options, ...this.kwargs }));
     }
     static isRunnableBinding(
     // eslint-disable-next-line @typescript-eslint/no-explicit-any