@langchain/core 0.3.21 → 0.3.23

package/dist/callbacks/base.cjs

@@ -23,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseCallbackHandler = void 0;
+exports.isBaseCallbackHandler = exports.BaseCallbackHandler = void 0;
 const uuid = __importStar(require("uuid"));
 const serializable_js_1 = require("../load/serializable.cjs");
 const env_js_1 = require("../utils/env.cjs");
@@ -166,3 +166,11 @@ class BaseCallbackHandler extends BaseCallbackHandlerMethodsClass {
     }
 }
 exports.BaseCallbackHandler = BaseCallbackHandler;
+const isBaseCallbackHandler = (x) => {
+    const callbackHandler = x;
+    return (callbackHandler !== undefined &&
+        typeof callbackHandler.copy === "function" &&
+        typeof callbackHandler.name === "string" &&
+        typeof callbackHandler.awaitHandlers === "boolean");
+};
+exports.isBaseCallbackHandler = isBaseCallbackHandler;

package/dist/callbacks/base.d.ts

@@ -276,4 +276,5 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
         handleCustomEvent?(eventName: string, data: any, runId: string, tags?: string[] | undefined, metadata?: Record<string, any> | undefined): any;
     };
 }
+export declare const isBaseCallbackHandler: (x: unknown) => boolean;
 export {};

package/dist/callbacks/base.js

@@ -139,3 +139,10 @@ export class BaseCallbackHandler extends BaseCallbackHandlerMethodsClass {
         return new Handler();
     }
 }
+export const isBaseCallbackHandler = (x) => {
+    const callbackHandler = x;
+    return (callbackHandler !== undefined &&
+        typeof callbackHandler.copy === "function" &&
+        typeof callbackHandler.name === "string" &&
+        typeof callbackHandler.awaitHandlers === "boolean");
+};

package/dist/callbacks/manager.cjs

@@ -10,6 +10,7 @@ const tracer_langchain_js_1 = require("../tracers/tracer_langchain.cjs");
 const promises_js_1 = require("./promises.cjs");
 const callbacks_js_1 = require("../utils/callbacks.cjs");
 const base_js_2 = require("../tracers/base.cjs");
+const context_js_1 = require("../singletons/async_local_storage/context.cjs");
 function parseCallbackConfigArg(arg) {
     if (!arg) {
         return {};
@@ -777,6 +778,23 @@ class CallbackManager extends BaseCallbackManager {
                 callbackManager.addMetadata(localMetadata ?? {}, false);
             }
         }
+        for (const { contextVar, inheritable = true, handlerClass, envVar, } of (0, context_js_1._getConfigureHooks)()) {
+            const createIfNotInContext = envVar && (0, env_js_1.getEnvironmentVariable)(envVar) === "true" && handlerClass;
+            let handler;
+            const contextVarValue = contextVar !== undefined ? (0, context_js_1.getContextVariable)(contextVar) : undefined;
+            if (contextVarValue && (0, base_js_1.isBaseCallbackHandler)(contextVarValue)) {
+                handler = contextVarValue;
+            }
+            else if (createIfNotInContext) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                handler = new handlerClass({});
+            }
+            if (handler !== undefined) {
+                if (!callbackManager?.handlers.some((h) => h.name === handler.name)) {
+                    callbackManager?.addHandler(handler, inheritable);
+                }
+            }
+        }
         return callbackManager;
     }
 }

package/dist/callbacks/manager.js

@@ -1,5 +1,5 @@
 import { v4 as uuidv4 } from "uuid";
-import { BaseCallbackHandler, } from "./base.js";
+import { BaseCallbackHandler, isBaseCallbackHandler, } from "./base.js";
 import { ConsoleCallbackHandler } from "../tracers/console.js";
 import { getBufferString } from "../messages/utils.js";
 import { getEnvironmentVariable } from "../utils/env.js";
@@ -7,6 +7,7 @@ import { LangChainTracer, } from "../tracers/tracer_langchain.js";
 import { consumeCallback } from "./promises.js";
 import { isTracingEnabled } from "../utils/callbacks.js";
 import { isBaseTracer } from "../tracers/base.js";
+import { getContextVariable, _getConfigureHooks, } from "../singletons/async_local_storage/context.js";
 export function parseCallbackConfigArg(arg) {
     if (!arg) {
         return {};
@@ -767,6 +768,23 @@ export class CallbackManager extends BaseCallbackManager {
                 callbackManager.addMetadata(localMetadata ?? {}, false);
             }
         }
+        for (const { contextVar, inheritable = true, handlerClass, envVar, } of _getConfigureHooks()) {
+            const createIfNotInContext = envVar && getEnvironmentVariable(envVar) === "true" && handlerClass;
+            let handler;
+            const contextVarValue = contextVar !== undefined ? getContextVariable(contextVar) : undefined;
+            if (contextVarValue && isBaseCallbackHandler(contextVarValue)) {
+                handler = contextVarValue;
+            }
+            else if (createIfNotInContext) {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                handler = new handlerClass({});
+            }
+            if (handler !== undefined) {
+                if (!callbackManager?.handlers.some((h) => h.name === handler.name)) {
+                    callbackManager?.addHandler(handler, inheritable);
+                }
+            }
+        }
         return callbackManager;
     }
 }

package/dist/context.cjs

@@ -1,128 +1,18 @@
 "use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.getContextVariable = exports.setContextVariable = void 0;
 /* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
-const node_async_hooks_1 = require("node:async_hooks");
-const langsmith_1 = require("langsmith");
-const run_trees_1 = require("langsmith/run_trees");
-const index_js_1 = require("./singletons/index.cjs");
-index_js_1.AsyncLocalStorageProviderSingleton.initializeGlobalInstance(new node_async_hooks_1.AsyncLocalStorage());
-/**
- * Set a context variable. Context variables are scoped to any
- * child runnables called by the current runnable, or globally if set outside
- * of any runnable.
- *
- * @remarks
- * This function is only supported in environments that support AsyncLocalStorage,
- * including Node.js, Deno, and Cloudflare Workers.
- *
- * @example
- * ```ts
- * import { RunnableLambda } from "@langchain/core/runnables";
- * import {
- *   getContextVariable,
- *   setContextVariable
- * } from "@langchain/core/context";
- *
- * const nested = RunnableLambda.from(() => {
- *   // "bar" because it was set by a parent
- *   console.log(getContextVariable("foo"));
- *
- *   // Override to "baz", but only for child runnables
- *   setContextVariable("foo", "baz");
- *
- *   // Now "baz", but only for child runnables
- *   return getContextVariable("foo");
- * });
- *
- * const runnable = RunnableLambda.from(async () => {
- *   // Set a context variable named "foo"
- *   setContextVariable("foo", "bar");
- *
- *   const res = await nested.invoke({});
- *
- *   // Still "bar" since child changes do not affect parents
- *   console.log(getContextVariable("foo"));
- *
- *   return res;
- * });
- *
- * // undefined, because context variable has not been set yet
- * console.log(getContextVariable("foo"));
- *
- * // Final return value is "baz"
- * const result = await runnable.invoke({});
- * ```
- *
- * @param name The name of the context variable.
- * @param value The value to set.
- */
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function setContextVariable(name, value) {
-    const runTree = index_js_1.AsyncLocalStorageProviderSingleton.getInstance().getStore();
-    const contextVars = { ...runTree?.[index_js_1._CONTEXT_VARIABLES_KEY] };
-    contextVars[name] = value;
-    let newValue = {};
-    if ((0, run_trees_1.isRunTree)(runTree)) {
-        newValue = new langsmith_1.RunTree(runTree);
-    }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    newValue[index_js_1._CONTEXT_VARIABLES_KEY] = contextVars;
-    index_js_1.AsyncLocalStorageProviderSingleton.getInstance().enterWith(newValue);
-}
-exports.setContextVariable = setContextVariable;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.registerConfigureHook = exports.setContextVariable = exports.getContextVariable = void 0;
 /**
- * Get the value of a previously set context variable. Context variables
- * are scoped to any child runnables called by the current runnable,
- * or globally if set outside of any runnable.
- *
- * @remarks
- * This function is only supported in environments that support AsyncLocalStorage,
- * including Node.js, Deno, and Cloudflare Workers.
- *
- * @example
- * ```ts
- * import { RunnableLambda } from "@langchain/core/runnables";
- * import {
- *   getContextVariable,
- *   setContextVariable
- * } from "@langchain/core/context";
- *
- * const nested = RunnableLambda.from(() => {
- *   // "bar" because it was set by a parent
- *   console.log(getContextVariable("foo"));
- *
- *   // Override to "baz", but only for child runnables
- *   setContextVariable("foo", "baz");
- *
- *   // Now "baz", but only for child runnables
- *   return getContextVariable("foo");
- * });
- *
- * const runnable = RunnableLambda.from(async () => {
- *   // Set a context variable named "foo"
- *   setContextVariable("foo", "bar");
+ * This file exists as a convenient public entrypoint for functionality
+ * related to context variables.
  *
- *   const res = await nested.invoke({});
- *
- *   // Still "bar" since child changes do not affect parents
- *   console.log(getContextVariable("foo"));
- *
- *   return res;
- * });
- *
- * // undefined, because context variable has not been set yet
- * console.log(getContextVariable("foo"));
- *
- * // Final return value is "baz"
- * const result = await runnable.invoke({});
- * ```
- *
- * @param name The name of the context variable.
+ * Because it automatically initializes AsyncLocalStorage, internal
+ * functionality SHOULD NEVER import from this file outside of tests.
  */
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-function getContextVariable(name) {
-    const runTree = index_js_1.AsyncLocalStorageProviderSingleton.getInstance().getStore();
-    return runTree?.[index_js_1._CONTEXT_VARIABLES_KEY]?.[name];
-}
-exports.getContextVariable = getContextVariable;
+const node_async_hooks_1 = require("node:async_hooks");
+const index_js_1 = require("./singletons/index.cjs");
+const context_js_1 = require("./singletons/async_local_storage/context.cjs");
+Object.defineProperty(exports, "getContextVariable", { enumerable: true, get: function () { return context_js_1.getContextVariable; } });
+Object.defineProperty(exports, "setContextVariable", { enumerable: true, get: function () { return context_js_1.setContextVariable; } });
+Object.defineProperty(exports, "registerConfigureHook", { enumerable: true, get: function () { return context_js_1.registerConfigureHook; } });
+index_js_1.AsyncLocalStorageProviderSingleton.initializeGlobalInstance(new node_async_hooks_1.AsyncLocalStorage());

package/dist/context.d.ts

@@ -1,101 +1,2 @@
-/**
- * Set a context variable. Context variables are scoped to any
- * child runnables called by the current runnable, or globally if set outside
- * of any runnable.
- *
- * @remarks
- * This function is only supported in environments that support AsyncLocalStorage,
- * including Node.js, Deno, and Cloudflare Workers.
- *
- * @example
- * ```ts
- * import { RunnableLambda } from "@langchain/core/runnables";
- * import {
- *   getContextVariable,
- *   setContextVariable
- * } from "@langchain/core/context";
- *
- * const nested = RunnableLambda.from(() => {
- *   // "bar" because it was set by a parent
- *   console.log(getContextVariable("foo"));
- *
- *   // Override to "baz", but only for child runnables
- *   setContextVariable("foo", "baz");
- *
- *   // Now "baz", but only for child runnables
- *   return getContextVariable("foo");
- * });
- *
- * const runnable = RunnableLambda.from(async () => {
- *   // Set a context variable named "foo"
- *   setContextVariable("foo", "bar");
- *
- *   const res = await nested.invoke({});
- *
- *   // Still "bar" since child changes do not affect parents
- *   console.log(getContextVariable("foo"));
- *
- *   return res;
- * });
- *
- * // undefined, because context variable has not been set yet
- * console.log(getContextVariable("foo"));
- *
- * // Final return value is "baz"
- * const result = await runnable.invoke({});
- * ```
- *
- * @param name The name of the context variable.
- * @param value The value to set.
- */
-export declare function setContextVariable<T>(name: PropertyKey, value: T): void;
-/**
- * Get the value of a previously set context variable. Context variables
- * are scoped to any child runnables called by the current runnable,
- * or globally if set outside of any runnable.
- *
- * @remarks
- * This function is only supported in environments that support AsyncLocalStorage,
- * including Node.js, Deno, and Cloudflare Workers.
- *
- * @example
- * ```ts
- * import { RunnableLambda } from "@langchain/core/runnables";
- * import {
- *   getContextVariable,
- *   setContextVariable
- * } from "@langchain/core/context";
- *
- * const nested = RunnableLambda.from(() => {
- *   // "bar" because it was set by a parent
- *   console.log(getContextVariable("foo"));
- *
- *   // Override to "baz", but only for child runnables
- *   setContextVariable("foo", "baz");
- *
- *   // Now "baz", but only for child runnables
- *   return getContextVariable("foo");
- * });
- *
- * const runnable = RunnableLambda.from(async () => {
- *   // Set a context variable named "foo"
- *   setContextVariable("foo", "bar");
- *
- *   const res = await nested.invoke({});
- *
- *   // Still "bar" since child changes do not affect parents
- *   console.log(getContextVariable("foo"));
- *
- *   return res;
- * });
- *
- * // undefined, because context variable has not been set yet
- * console.log(getContextVariable("foo"));
- *
- * // Final return value is "baz"
- * const result = await runnable.invoke({});
- * ```
- *
- * @param name The name of the context variable.
- */
-export declare function getContextVariable<T = any>(name: PropertyKey): T | undefined;
+import { getContextVariable, setContextVariable, type ConfigureHook, registerConfigureHook } from "./singletons/async_local_storage/context.js";
+export { getContextVariable, setContextVariable, registerConfigureHook, type ConfigureHook, };

package/dist/context.js

@@ -1,123 +1,13 @@
 /* __LC_ALLOW_ENTRYPOINT_SIDE_EFFECTS__ */
-import { AsyncLocalStorage } from "node:async_hooks";
-import { RunTree } from "langsmith";
-import { isRunTree } from "langsmith/run_trees";
-import { _CONTEXT_VARIABLES_KEY, AsyncLocalStorageProviderSingleton, } from "./singletons/index.js";
-AsyncLocalStorageProviderSingleton.initializeGlobalInstance(new AsyncLocalStorage());
 /**
- * Set a context variable. Context variables are scoped to any
- * child runnables called by the current runnable, or globally if set outside
- * of any runnable.
- *
- * @remarks
- * This function is only supported in environments that support AsyncLocalStorage,
- * including Node.js, Deno, and Cloudflare Workers.
- *
- * @example
- * ```ts
- * import { RunnableLambda } from "@langchain/core/runnables";
- * import {
- *   getContextVariable,
- *   setContextVariable
- * } from "@langchain/core/context";
- *
- * const nested = RunnableLambda.from(() => {
- *   // "bar" because it was set by a parent
- *   console.log(getContextVariable("foo"));
- *
- *   // Override to "baz", but only for child runnables
- *   setContextVariable("foo", "baz");
- *
- *   // Now "baz", but only for child runnables
- *   return getContextVariable("foo");
- * });
- *
- * const runnable = RunnableLambda.from(async () => {
- *   // Set a context variable named "foo"
- *   setContextVariable("foo", "bar");
+ * This file exists as a convenient public entrypoint for functionality
+ * related to context variables.
  *
- *   const res = await nested.invoke({});
- *
- *   // Still "bar" since child changes do not affect parents
- *   console.log(getContextVariable("foo"));
- *
- *   return res;
- * });
- *
- * // undefined, because context variable has not been set yet
- * console.log(getContextVariable("foo"));
- *
- * // Final return value is "baz"
- * const result = await runnable.invoke({});
- * ```
- *
- * @param name The name of the context variable.
- * @param value The value to set.
+ * Because it automatically initializes AsyncLocalStorage, internal
+ * functionality SHOULD NEVER import from this file outside of tests.
  */
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-export function setContextVariable(name, value) {
-    const runTree = AsyncLocalStorageProviderSingleton.getInstance().getStore();
-    const contextVars = { ...runTree?.[_CONTEXT_VARIABLES_KEY] };
-    contextVars[name] = value;
-    let newValue = {};
-    if (isRunTree(runTree)) {
-        newValue = new RunTree(runTree);
-    }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    newValue[_CONTEXT_VARIABLES_KEY] = contextVars;
-    AsyncLocalStorageProviderSingleton.getInstance().enterWith(newValue);
-}
-/**
- * Get the value of a previously set context variable. Context variables
- * are scoped to any child runnables called by the current runnable,
- * or globally if set outside of any runnable.
- *
- * @remarks
- * This function is only supported in environments that support AsyncLocalStorage,
- * including Node.js, Deno, and Cloudflare Workers.
- *
- * @example
- * ```ts
- * import { RunnableLambda } from "@langchain/core/runnables";
- * import {
- *   getContextVariable,
- *   setContextVariable
- * } from "@langchain/core/context";
- *
- * const nested = RunnableLambda.from(() => {
- *   // "bar" because it was set by a parent
- *   console.log(getContextVariable("foo"));
- *
- *   // Override to "baz", but only for child runnables
- *   setContextVariable("foo", "baz");
- *
- *   // Now "baz", but only for child runnables
- *   return getContextVariable("foo");
- * });
- *
- * const runnable = RunnableLambda.from(async () => {
- *   // Set a context variable named "foo"
- *   setContextVariable("foo", "bar");
- *
- *   const res = await nested.invoke({});
- *
- *   // Still "bar" since child changes do not affect parents
- *   console.log(getContextVariable("foo"));
- *
- *   return res;
- * });
- *
- * // undefined, because context variable has not been set yet
- * console.log(getContextVariable("foo"));
- *
- * // Final return value is "baz"
- * const result = await runnable.invoke({});
- * ```
- *
- * @param name The name of the context variable.
- */
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-export function getContextVariable(name) {
-    const runTree = AsyncLocalStorageProviderSingleton.getInstance().getStore();
-    return runTree?.[_CONTEXT_VARIABLES_KEY]?.[name];
-}
+import { AsyncLocalStorage } from "node:async_hooks";
+import { AsyncLocalStorageProviderSingleton } from "./singletons/index.js";
+import { getContextVariable, setContextVariable, registerConfigureHook, } from "./singletons/async_local_storage/context.js";
+AsyncLocalStorageProviderSingleton.initializeGlobalInstance(new AsyncLocalStorage());
+export { getContextVariable, setContextVariable, registerConfigureHook, };

package/dist/messages/tool.cjs

@@ -1,7 +1,14 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isToolMessageChunk = exports.isToolMessage = exports.defaultToolCallParser = exports.ToolMessageChunk = exports.ToolMessage = void 0;
+exports.isToolMessageChunk = exports.isToolMessage = exports.defaultToolCallParser = exports.ToolMessageChunk = exports.ToolMessage = exports.isDirectToolOutput = void 0;
 const base_js_1 = require("./base.cjs");
+function isDirectToolOutput(x) {
+    return (x != null &&
+        typeof x === "object" &&
+        "lc_direct_tool_output" in x &&
+        x.lc_direct_tool_output === true);
+}
+exports.isDirectToolOutput = isDirectToolOutput;
 /**
  * Represents a tool message in a conversation.
  */
@@ -19,6 +26,12 @@ class ToolMessage extends base_js_1.BaseMessage {
             fields = { content: fields, name, tool_call_id: tool_call_id };
         }
         super(fields);
+        Object.defineProperty(this, "lc_direct_tool_output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         /**
          * Status of the tool invocation.
          * @version 0.2.19

package/dist/messages/tool.d.ts

@@ -15,12 +15,24 @@ export interface ToolMessageFieldsWithToolCallId extends BaseMessageFields {
      */
     status?: "success" | "error";
 }
+/**
+ * Marker parameter for objects that tools can return directly.
+ *
+ * If a custom BaseTool is invoked with a ToolCall and the output of custom code is
+ * not an instance of DirectToolOutput, the output will automatically be coerced to
+ * a string and wrapped in a ToolMessage.
+ */
+export interface DirectToolOutput {
+    readonly lc_direct_tool_output: boolean;
+}
+export declare function isDirectToolOutput(x: unknown): x is DirectToolOutput;
 /**
  * Represents a tool message in a conversation.
  */
-export declare class ToolMessage extends BaseMessage {
+export declare class ToolMessage extends BaseMessage implements DirectToolOutput {
     static lc_name(): string;
     get lc_aliases(): Record<string, string>;
+    lc_direct_tool_output: boolean;
     /**
      * Status of the tool invocation.
      * @version 0.2.19

package/dist/messages/tool.js

@@ -1,4 +1,10 @@
 import { BaseMessage, BaseMessageChunk, mergeContent, _mergeDicts, _mergeObj, _mergeStatus, } from "./base.js";
+export function isDirectToolOutput(x) {
+    return (x != null &&
+        typeof x === "object" &&
+        "lc_direct_tool_output" in x &&
+        x.lc_direct_tool_output === true);
+}
 /**
  * Represents a tool message in a conversation.
  */
@@ -16,6 +22,12 @@ export class ToolMessage extends BaseMessage {
             fields = { content: fields, name, tool_call_id: tool_call_id };
         }
         super(fields);
+        Object.defineProperty(this, "lc_direct_tool_output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
         /**
          * Status of the tool invocation.
          * @version 0.2.19

package/dist/output_parsers/transform.cjs

@@ -1,11 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseCumulativeTransformOutputParser = exports.BaseTransformOutputParser = void 0;
+const json_schema_1 = require("@cfworker/json-schema");
 const base_js_1 = require("./base.cjs");
 const base_js_2 = require("../messages/base.cjs");
 const utils_js_1 = require("../messages/utils.cjs");
 const outputs_js_1 = require("../outputs.cjs");
-const index_js_1 = require("../utils/@cfworker/json-schema/index.cjs");
 /**
  * Class to parse the output of an LLM call that also allows streaming inputs.
  */
@@ -94,7 +94,7 @@ class BaseCumulativeTransformOutputParser extends BaseTransformOutputParser {
             const parsed = await this.parsePartialResult([accGen]);
             if (parsed !== undefined &&
                 parsed !== null &&
-                !(0, index_js_1.deepCompareStrict)(parsed, prevParsed)) {
+                !(0, json_schema_1.deepCompareStrict)(parsed, prevParsed)) {
                 if (this.diff) {
                     yield this._diff(prevParsed, parsed);
                 }

package/dist/output_parsers/transform.js

@@ -1,8 +1,8 @@
+import { deepCompareStrict } from "@cfworker/json-schema";
 import { BaseOutputParser } from "./base.js";
 import { isBaseMessage, isBaseMessageChunk, } from "../messages/base.js";
 import { convertToChunk } from "../messages/utils.js";
 import { GenerationChunk, ChatGenerationChunk, } from "../outputs.js";
-import { deepCompareStrict } from "../utils/@cfworker/json-schema/index.js";
 /**
  * Class to parse the output of an LLM call that also allows streaming inputs.
  */

package/dist/runnables/base.cjs

@@ -1657,7 +1657,7 @@ class RunnableLambda extends Runnable {
                 callbacks: runManager?.getChild(),
                 recursionLimit: (config?.recursionLimit ?? config_js_1.DEFAULT_RECURSION_LIMIT) - 1,
             });
-            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
+            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig((0, config_js_1.pickRunnableConfigKeys)(childConfig), async () => {
                 try {
                     let output = await this.func(input, {
                         ...childConfig,
@@ -1744,7 +1744,7 @@ class RunnableLambda extends Runnable {
             recursionLimit: (config?.recursionLimit ?? config_js_1.DEFAULT_RECURSION_LIMIT) - 1,
         });
         const output = await new Promise((resolve, reject) => {
-            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
+            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig((0, config_js_1.pickRunnableConfigKeys)(childConfig), async () => {
                 try {
                     const res = await this.func(finalChunk, {
                         ...childConfig,

package/dist/runnables/base.d.ts

@@ -1,12 +1,11 @@
 import { z } from "zod";
 import { type TraceableFunction } from "langsmith/singletons/traceable";
-import type { RunnableInterface, RunnableBatchOptions } from "./types.js";
+import type { RunnableInterface, RunnableBatchOptions, RunnableConfig } from "./types.js";
 import { CallbackManagerForChainRun } from "../callbacks/manager.js";
 import { LogStreamCallbackHandler, LogStreamCallbackHandlerInput, RunLogPatch } from "../tracers/log_stream.js";
 import { EventStreamCallbackHandlerInput, StreamEvent } from "../tracers/event_stream.js";
 import { Serializable } from "../load/serializable.js";
 import { IterableReadableStream } from "../utils/stream.js";
-import { RunnableConfig } from "./config.js";
 import { Run } from "../tracers/base.js";
 import { Graph } from "./graph.js";
 import { ToolCall } from "../messages/tool.js";

package/dist/runnables/base.js

@@ -7,7 +7,7 @@ import { EventStreamCallbackHandler, isStreamEventsHandler, } from "../tracers/e
 import { Serializable } from "../load/serializable.js";
 import { IterableReadableStream, concat, atee, pipeGeneratorWithSetup, AsyncGeneratorWithSetup, } from "../utils/stream.js";
 import { raceWithSignal } from "../utils/signal.js";
-import { DEFAULT_RECURSION_LIMIT, ensureConfig, getCallbackManagerForConfig, mergeConfigs, patchConfig, } from "./config.js";
+import { DEFAULT_RECURSION_LIMIT, ensureConfig, getCallbackManagerForConfig, mergeConfigs, patchConfig, pickRunnableConfigKeys, } from "./config.js";
 import { AsyncCaller } from "../utils/async_caller.js";
 import { RootListenersTracer } from "../tracers/root_listener.js";
 import { _RootEventFilter, isRunnableInterface } from "./utils.js";
@@ -1643,7 +1643,7 @@ export class RunnableLambda extends Runnable {
                 callbacks: runManager?.getChild(),
                 recursionLimit: (config?.recursionLimit ?? DEFAULT_RECURSION_LIMIT) - 1,
             });
-            void AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
+            void AsyncLocalStorageProviderSingleton.runWithConfig(pickRunnableConfigKeys(childConfig), async () => {
                 try {
                     let output = await this.func(input, {
                         ...childConfig,
@@ -1730,7 +1730,7 @@ export class RunnableLambda extends Runnable {
             recursionLimit: (config?.recursionLimit ?? DEFAULT_RECURSION_LIMIT) - 1,
         });
         const output = await new Promise((resolve, reject) => {
-            void AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
+            void AsyncLocalStorageProviderSingleton.runWithConfig(pickRunnableConfigKeys(childConfig), async () => {
                 try {
                     const res = await this.func(finalChunk, {
                         ...childConfig,