@langchain/core 0.2.16 → 0.2.18-rc.0

package/dist/messages/chat.js

@@ -29,6 +29,12 @@ export class ChatMessage extends BaseMessage {
     static isInstance(message) {
         return message._getType() === "generic";
     }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            role: this.role,
+        };
+    }
 }
 /**
  * Represents a chunk of a chat message, which can be concatenated with
@@ -64,4 +70,10 @@ export class ChatMessageChunk extends BaseMessageChunk {
             id: this.id ?? chunk.id,
         });
     }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            role: this.role,
+        };
+    }
 }

package/dist/messages/index.cjs

@@ -23,6 +23,7 @@ __exportStar(require("./human.cjs"), exports);
 __exportStar(require("./system.cjs"), exports);
 __exportStar(require("./utils.cjs"), exports);
 __exportStar(require("./transformers.cjs"), exports);
+__exportStar(require("./modifier.cjs"), exports);
 // TODO: Use a star export when we deprecate the
 // existing "ToolCall" type in "base.js".
 var tool_js_1 = require("./tool.cjs");

package/dist/messages/index.d.ts

@@ -6,4 +6,5 @@ export * from "./human.js";
 export * from "./system.js";
 export * from "./utils.js";
 export * from "./transformers.js";
+export * from "./modifier.js";
 export { type ToolMessageFieldsWithToolCallId, ToolMessage, ToolMessageChunk, type InvalidToolCall, } from "./tool.js";

package/dist/messages/index.js

@@ -6,6 +6,7 @@ export * from "./human.js";
 export * from "./system.js";
 export * from "./utils.js";
 export * from "./transformers.js";
+export * from "./modifier.js";
 // TODO: Use a star export when we deprecate the
 // existing "ToolCall" type in "base.js".
 export { ToolMessage, ToolMessageChunk, } from "./tool.js";

package/dist/messages/modifier.cjs

@@ -0,0 +1,35 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RemoveMessage = void 0;
+const base_js_1 = require("./base.cjs");
+/**
+ * Message responsible for deleting other messages.
+ */
+class RemoveMessage extends base_js_1.BaseMessage {
+    constructor(fields) {
+        super({
+            ...fields,
+            content: "",
+        });
+        /**
+         * The ID of the message to remove.
+         */
+        Object.defineProperty(this, "id", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.id = fields.id;
+    }
+    _getType() {
+        return "remove";
+    }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            id: this.id,
+        };
+    }
+}
+exports.RemoveMessage = RemoveMessage;

package/dist/messages/modifier.d.ts

@@ -0,0 +1,19 @@
+import { BaseMessage, BaseMessageFields, MessageType } from "./base.js";
+export interface RemoveMessageFields extends Omit<BaseMessageFields, "content"> {
+    /**
+     * The ID of the message to remove.
+     */
+    id: string;
+}
+/**
+ * Message responsible for deleting other messages.
+ */
+export declare class RemoveMessage extends BaseMessage {
+    /**
+     * The ID of the message to remove.
+     */
+    id: string;
+    constructor(fields: RemoveMessageFields);
+    _getType(): MessageType;
+    get _printableFields(): Record<string, unknown>;
+}

package/dist/messages/modifier.js

@@ -0,0 +1,31 @@
+import { BaseMessage } from "./base.js";
+/**
+ * Message responsible for deleting other messages.
+ */
+export class RemoveMessage extends BaseMessage {
+    constructor(fields) {
+        super({
+            ...fields,
+            content: "",
+        });
+        /**
+         * The ID of the message to remove.
+         */
+        Object.defineProperty(this, "id", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.id = fields.id;
+    }
+    _getType() {
+        return "remove";
+    }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            id: this.id,
+        };
+    }
+}

package/dist/messages/tool.cjs

@@ -48,6 +48,13 @@ class ToolMessage extends base_js_1.BaseMessage {
     static isInstance(message) {
         return message._getType() === "tool";
     }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            tool_call_id: this.tool_call_id,
+            artifact: this.artifact,
+        };
+    }
 }
 exports.ToolMessage = ToolMessage;
 /**
@@ -96,6 +103,13 @@ class ToolMessageChunk extends base_js_1.BaseMessageChunk {
             id: this.id ?? chunk.id,
         });
     }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            tool_call_id: this.tool_call_id,
+            artifact: this.artifact,
+        };
+    }
 }
 exports.ToolMessageChunk = ToolMessageChunk;
 function defaultToolCallParser(

package/dist/messages/tool.d.ts

@@ -29,6 +29,7 @@ export declare class ToolMessage extends BaseMessage {
     constructor(fields: string | BaseMessageFields, tool_call_id: string, name?: string);
     _getType(): MessageType;
     static isInstance(message: BaseMessage): message is ToolMessage;
+    get _printableFields(): Record<string, unknown>;
 }
 /**
  * Represents a chunk of a tool message, which can be concatenated
@@ -48,6 +49,7 @@ export declare class ToolMessageChunk extends BaseMessageChunk {
     static lc_name(): string;
     _getType(): MessageType;
     concat(chunk: ToolMessageChunk): ToolMessageChunk;
+    get _printableFields(): Record<string, unknown>;
 }
 /**
  * A call to a tool.

package/dist/messages/tool.js

@@ -45,6 +45,13 @@ export class ToolMessage extends BaseMessage {
     static isInstance(message) {
         return message._getType() === "tool";
     }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            tool_call_id: this.tool_call_id,
+            artifact: this.artifact,
+        };
+    }
 }
 /**
  * Represents a chunk of a tool message, which can be concatenated
@@ -92,6 +99,13 @@ export class ToolMessageChunk extends BaseMessageChunk {
             id: this.id ?? chunk.id,
         });
     }
+    get _printableFields() {
+        return {
+            ...super._printableFields,
+            tool_call_id: this.tool_call_id,
+            artifact: this.artifact,
+        };
+    }
 }
 export function defaultToolCallParser(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/messages/transformers.cjs

@@ -6,6 +6,7 @@ const ai_js_1 = require("./ai.cjs");
 const chat_js_1 = require("./chat.cjs");
 const function_js_1 = require("./function.cjs");
 const human_js_1 = require("./human.cjs");
+const modifier_js_1 = require("./modifier.cjs");
 const system_js_1 = require("./system.cjs");
 const tool_js_1 = require("./tool.cjs");
 const utils_js_1 = require("./utils.cjs");
@@ -303,6 +304,10 @@ const _MSG_CHUNK_MAP = {
         message: chat_js_1.ChatMessage,
         messageChunk: chat_js_1.ChatMessageChunk,
     },
+    remove: {
+        message: modifier_js_1.RemoveMessage,
+        messageChunk: modifier_js_1.RemoveMessage, // RemoveMessage does not have a chunk class.
+    },
 };
 function _switchTypeToMessage(messageType, fields, returnChunk) {
     let chunk;

package/dist/messages/transformers.d.ts

@@ -6,10 +6,11 @@ import { BaseMessage, MessageType } from "./base.js";
 import { ChatMessage, ChatMessageChunk } from "./chat.js";
 import { FunctionMessage, FunctionMessageChunk } from "./function.js";
 import { HumanMessage, HumanMessageChunk } from "./human.js";
+import { RemoveMessage } from "./modifier.js";
 import { SystemMessage, SystemMessageChunk } from "./system.js";
 import { ToolMessage, ToolMessageChunk } from "./tool.js";
-export type MessageUnion = typeof HumanMessage | typeof AIMessage | typeof SystemMessage | typeof ChatMessage | typeof FunctionMessage | typeof ToolMessage;
-export type MessageChunkUnion = typeof HumanMessageChunk | typeof AIMessageChunk | typeof SystemMessageChunk | typeof FunctionMessageChunk | typeof ToolMessageChunk | typeof ChatMessageChunk;
+export type MessageUnion = typeof HumanMessage | typeof AIMessage | typeof SystemMessage | typeof ChatMessage | typeof FunctionMessage | typeof ToolMessage | typeof RemoveMessage;
+export type MessageChunkUnion = typeof HumanMessageChunk | typeof AIMessageChunk | typeof SystemMessageChunk | typeof FunctionMessageChunk | typeof ToolMessageChunk | typeof ChatMessageChunk | typeof RemoveMessage;
 export type MessageTypeOrClass = MessageType | MessageUnion | MessageChunkUnion;
 export interface FilterMessagesFields {
     /**

package/dist/messages/transformers.js

@@ -3,6 +3,7 @@ import { AIMessage, AIMessageChunk } from "./ai.js";
 import { ChatMessage, ChatMessageChunk, } from "./chat.js";
 import { FunctionMessage, FunctionMessageChunk, } from "./function.js";
 import { HumanMessage, HumanMessageChunk } from "./human.js";
+import { RemoveMessage } from "./modifier.js";
 import { SystemMessage, SystemMessageChunk } from "./system.js";
 import { ToolMessage, ToolMessageChunk, } from "./tool.js";
 import { convertToChunk } from "./utils.js";
@@ -297,6 +298,10 @@ const _MSG_CHUNK_MAP = {
         message: ChatMessage,
         messageChunk: ChatMessageChunk,
     },
+    remove: {
+        message: RemoveMessage,
+        messageChunk: RemoveMessage, // RemoveMessage does not have a chunk class.
+    },
 };
 function _switchTypeToMessage(messageType, fields, returnChunk) {
     let chunk;

package/dist/runnables/base.cjs

@@ -20,6 +20,7 @@ const index_js_1 = require("../singletons/index.cjs");
 const graph_js_1 = require("./graph.cjs");
 const wrappers_js_1 = require("./wrappers.cjs");
 const iter_js_1 = require("./iter.cjs");
+const utils_js_2 = require("../tools/utils.cjs");
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function _coerceToDict(value, defaultKey) {
     return value &&
@@ -1507,7 +1508,7 @@ class RunnableLambda extends Runnable {
                 callbacks: runManager?.getChild(),
                 recursionLimit: (config?.recursionLimit ?? config_js_1.DEFAULT_RECURSION_LIMIT) - 1,
             });
-            void index_js_1.AsyncLocalStorageProviderSingleton.getInstance().run(childConfig, async () => {
+            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
                 try {
                     let output = await this.func(input, {
                         ...childConfig,
@@ -1593,7 +1594,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.getInstance().run(childConfig, async () => {
+            void index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
                 try {
                     const res = await this.func(finalChunk, {
                         ...childConfig,
@@ -1929,8 +1930,26 @@ class RunnablePick extends Runnable {
 exports.RunnablePick = RunnablePick;
 class RunnableToolLike extends RunnableBinding {
     constructor(fields) {
+        const sequence = RunnableSequence.from([
+            RunnableLambda.from(async (input) => {
+                let toolInput;
+                if ((0, utils_js_2._isToolCall)(input)) {
+                    try {
+                        toolInput = await this.schema.parseAsync(input.args);
+                    }
+                    catch (e) {
+                        throw new utils_js_2.ToolInputParsingException(`Received tool input did not match expected schema`, JSON.stringify(input.args));
+                    }
+                }
+                else {
+                    toolInput = input;
+                }
+                return toolInput;
+            }).withConfig({ runName: `${fields.name}:parse_input` }),
+            fields.bound,
+        ]).withConfig({ runName: fields.name });
         super({
-            bound: fields.bound,
+            bound: sequence,
             config: fields.config ?? {},
         });
         Object.defineProperty(this, "name", {
@@ -1975,7 +1994,19 @@ exports.RunnableToolLike = RunnableToolLike;
  */
 function convertRunnableToTool(runnable, fields) {
     const name = fields.name ?? runnable.getName();
-    const description = fields.description ?? fields.schema.description;
+    const description = fields.description ?? fields.schema?.description;
+    if (fields.schema.constructor === zod_1.z.ZodString) {
+        return new RunnableToolLike({
+            name,
+            description,
+            schema: zod_1.z
+                .object({
+                input: zod_1.z.string(),
+            })
+                .transform((input) => input.input),
+            bound: runnable,
+        });
+    }
     return new RunnableToolLike({
         name,
         description,

package/dist/runnables/base.d.ts

@@ -9,6 +9,7 @@ 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";
 export { type RunnableInterface, RunnableBatchOptions };
 export type RunnableFunc<RunInput, RunOutput> = (input: RunInput, options?: ({
     /** @deprecated Use top-level config fields instead. */
@@ -201,11 +202,11 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
      * | on_llm_end           | [model name]     |                                 | 'Hello human!'                                |                                                 |
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
-     * | on_chain_start       | format_docs      |                                 |                                               |                                                 |
+     * | on_chain_start       | some_runnable    |                                 |                                               |                                                 |
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
-     * | on_chain_stream      | format_docs      | "hello world!, goodbye world!"  |                                               |                                                 |
+     * | on_chain_stream      | some_runnable    | "hello world!, goodbye world!"  |                                               |                                                 |
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
-     * | on_chain_end         | format_docs      |                                 | [Document(...)]                               | "hello world!, goodbye world!"                  |
+     * | on_chain_end         | some_runnable    |                                 | [Document(...)]                               | "hello world!, goodbye world!"                  |
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
      * | on_tool_start        | some_tool        |                                 | {"x": 1, "y": "2"}                            |                                                 |
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
@@ -219,6 +220,52 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
      * | on_prompt_end        | [template_name]  |                                 | {"question": "hello"}                         | ChatPromptValue(messages: [SystemMessage, ...]) |
      * +----------------------+------------------+---------------------------------+-----------------------------------------------+-------------------------------------------------+
+     *
+     * The "on_chain_*" events are the default for Runnables that don't fit one of the above categories.
+     *
+     * In addition to the standard events above, users can also dispatch custom events.
+     *
+     * Custom events will be only be surfaced with in the `v2` version of the API!
+     *
+     * A custom event has following format:
+     *
+     * +-----------+------+-----------------------------------------------------------------------------------------------------------+
+     * | Attribute | Type | Description                                                                                               |
+     * +===========+======+===========================================================================================================+
+     * | name      | str  | A user defined name for the event.                                                                        |
+     * +-----------+------+-----------------------------------------------------------------------------------------------------------+
+     * | data      | Any  | The data associated with the event. This can be anything, though we suggest making it JSON serializable.  |
+     * +-----------+------+-----------------------------------------------------------------------------------------------------------+
+     *
+     * Here's an example:
+     * @example
+     * ```ts
+     * import { RunnableLambda } from "@langchain/core/runnables";
+     * import { dispatchCustomEvent } from "@langchain/core/callbacks/dispatch";
+     * // Use this import for web environments that don't support "async_hooks"
+     * // and manually pass config to child runs.
+     * // import { dispatchCustomEvent } from "@langchain/core/callbacks/dispatch/web";
+     *
+     * const slowThing = RunnableLambda.from(async (someInput: string) => {
+     *   // Placeholder for some slow operation
+     *   await new Promise((resolve) => setTimeout(resolve, 100));
+     *   await dispatchCustomEvent("progress_event", {
+     *    message: "Finished step 1 of 2",
+     *  });
+     *  await new Promise((resolve) => setTimeout(resolve, 100));
+     *  return "Done";
+     * });
+     *
+     * const eventStream = await slowThing.streamEvents("hello world", {
+     *   version: "v2",
+     * });
+     *
+     * for await (const event of eventStream) {
+     *  if (event.event === "on_custom_event") {
+     *    console.log(event);
+     *  }
+     * }
+     * ```
      */
     streamEvents(input: RunInput, options: Partial<CallOptions> & {
         version: "v1" | "v2";
@@ -262,7 +309,7 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
         name?: string;
         description?: string;
         schema: z.ZodType<T>;
-    }): RunnableToolLike<z.ZodType<T>, RunOutput>;
+    }): RunnableToolLike<z.ZodType<T | ToolCall>, RunOutput>;
 }
 export type RunnableBindingArgs<RunInput, RunOutput, CallOptions extends RunnableConfig = RunnableConfig> = {
     bound: Runnable<RunInput, RunOutput, CallOptions>;
@@ -608,4 +655,4 @@ export declare function convertRunnableToTool<RunInput, RunOutput>(runnable: Run
     name?: string;
     description?: string;
     schema: z.ZodType<RunInput>;
-}): RunnableToolLike<z.ZodType<RunInput>, RunOutput>;
+}): RunnableToolLike<z.ZodType<RunInput | ToolCall>, RunOutput>;

package/dist/runnables/base.js

@@ -14,6 +14,7 @@ import { AsyncLocalStorageProviderSingleton } from "../singletons/index.js";
 import { Graph } from "./graph.js";
 import { convertToHttpEventStream } from "./wrappers.js";
 import { consumeAsyncIterableInContext, consumeIteratorInContext, isAsyncIterable, isIterableIterator, isIterator, } from "./iter.js";
+import { _isToolCall, ToolInputParsingException } from "../tools/utils.js";
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 export function _coerceToDict(value, defaultKey) {
     return value &&
@@ -1493,7 +1494,7 @@ export class RunnableLambda extends Runnable {
                 callbacks: runManager?.getChild(),
                 recursionLimit: (config?.recursionLimit ?? DEFAULT_RECURSION_LIMIT) - 1,
             });
-            void AsyncLocalStorageProviderSingleton.getInstance().run(childConfig, async () => {
+            void AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
                 try {
                     let output = await this.func(input, {
                         ...childConfig,
@@ -1579,7 +1580,7 @@ export class RunnableLambda extends Runnable {
             recursionLimit: (config?.recursionLimit ?? DEFAULT_RECURSION_LIMIT) - 1,
         });
         const output = await new Promise((resolve, reject) => {
-            void AsyncLocalStorageProviderSingleton.getInstance().run(childConfig, async () => {
+            void AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => {
                 try {
                     const res = await this.func(finalChunk, {
                         ...childConfig,
@@ -1909,8 +1910,26 @@ export class RunnablePick extends Runnable {
 }
 export class RunnableToolLike extends RunnableBinding {
     constructor(fields) {
+        const sequence = RunnableSequence.from([
+            RunnableLambda.from(async (input) => {
+                let toolInput;
+                if (_isToolCall(input)) {
+                    try {
+                        toolInput = await this.schema.parseAsync(input.args);
+                    }
+                    catch (e) {
+                        throw new ToolInputParsingException(`Received tool input did not match expected schema`, JSON.stringify(input.args));
+                    }
+                }
+                else {
+                    toolInput = input;
+                }
+                return toolInput;
+            }).withConfig({ runName: `${fields.name}:parse_input` }),
+            fields.bound,
+        ]).withConfig({ runName: fields.name });
         super({
-            bound: fields.bound,
+            bound: sequence,
             config: fields.config ?? {},
         });
         Object.defineProperty(this, "name", {
@@ -1954,7 +1973,19 @@ export class RunnableToolLike extends RunnableBinding {
  */
 export function convertRunnableToTool(runnable, fields) {
     const name = fields.name ?? runnable.getName();
-    const description = fields.description ?? fields.schema.description;
+    const description = fields.description ?? fields.schema?.description;
+    if (fields.schema.constructor === z.ZodString) {
+        return new RunnableToolLike({
+            name,
+            description,
+            schema: z
+                .object({
+                input: z.string(),
+            })
+                .transform((input) => input.input),
+            bound: runnable,
+        });
+    }
     return new RunnableToolLike({
         name,
         description,

package/dist/runnables/config.cjs

@@ -5,7 +5,7 @@ const manager_js_1 = require("../callbacks/manager.cjs");
 const index_js_1 = require("../singletons/index.cjs");
 exports.DEFAULT_RECURSION_LIMIT = 25;
 async function getCallbackManagerForConfig(config) {
-    return manager_js_1.CallbackManager.configure(config?.callbacks, undefined, config?.tags, undefined, config?.metadata);
+    return manager_js_1.CallbackManager._configureSync(config?.callbacks, undefined, config?.tags, undefined, config?.metadata);
 }
 exports.getCallbackManagerForConfig = getCallbackManagerForConfig;
 function mergeConfigs(...configs) {
@@ -84,12 +84,9 @@ exports.mergeConfigs = mergeConfigs;
 const PRIMITIVES = new Set(["string", "number", "boolean"]);
 /**
  * Ensure that a passed config is an object with all required keys present.
- *
- * Note: To make sure async local storage loading works correctly, this
- * should not be called with a default or prepopulated config argument.
  */
 function ensureConfig(config) {
-    const implicitConfig = index_js_1.AsyncLocalStorageProviderSingleton.getInstance().getStore();
+    const implicitConfig = index_js_1.AsyncLocalStorageProviderSingleton.getRunnableConfig();
     let empty = {
         tags: [],
         metadata: {},
@@ -97,10 +94,10 @@ function ensureConfig(config) {
         runId: undefined,
     };
     if (implicitConfig) {
-        // Don't allow runId to be loaded implicitly, as this can cause
+        // Don't allow runId and runName to be loaded implicitly, as this can cause
         // child runs to improperly inherit their parents' run ids.
         // eslint-disable-next-line @typescript-eslint/no-unused-vars
-        const { runId, ...rest } = implicitConfig;
+        const { runId, runName, ...rest } = implicitConfig;
         empty = Object.entries(rest).reduce(
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         (currentConfig, [key, value]) => {

package/dist/runnables/config.d.ts

@@ -1,25 +1,11 @@
-import { type BaseCallbackConfig, CallbackManager } from "../callbacks/manager.js";
+import { CallbackManager } from "../callbacks/manager.js";
+import { RunnableConfig } from "./types.js";
 export declare const DEFAULT_RECURSION_LIMIT = 25;
-export interface RunnableConfig extends BaseCallbackConfig {
-    /**
-     * Runtime values for attributes previously made configurable on this Runnable,
-     * or sub-Runnables.
-     */
-    configurable?: Record<string, any>;
-    /**
-     * Maximum number of times a call can recurse. If not provided, defaults to 25.
-     */
-    recursionLimit?: number;
-    /** Maximum number of parallel calls to make. */
-    maxConcurrency?: number;
-}
+export { type RunnableConfig };
 export declare function getCallbackManagerForConfig(config?: RunnableConfig): Promise<CallbackManager | undefined>;
 export declare function mergeConfigs<CallOptions extends RunnableConfig>(...configs: (CallOptions | RunnableConfig | undefined | null)[]): Partial<CallOptions>;
 /**
  * Ensure that a passed config is an object with all required keys present.
- *
- * Note: To make sure async local storage loading works correctly, this
- * should not be called with a default or prepopulated config argument.
  */
 export declare function ensureConfig<CallOptions extends RunnableConfig>(config?: CallOptions): CallOptions;
 /**

package/dist/runnables/config.js

@@ -1,8 +1,8 @@
-import { CallbackManager, ensureHandler, } from "../callbacks/manager.js";
+import { CallbackManager, ensureHandler } from "../callbacks/manager.js";
 import { AsyncLocalStorageProviderSingleton } from "../singletons/index.js";
 export const DEFAULT_RECURSION_LIMIT = 25;
 export async function getCallbackManagerForConfig(config) {
-    return CallbackManager.configure(config?.callbacks, undefined, config?.tags, undefined, config?.metadata);
+    return CallbackManager._configureSync(config?.callbacks, undefined, config?.tags, undefined, config?.metadata);
 }
 export function mergeConfigs(...configs) {
     // We do not want to call ensureConfig on the empty state here as this may cause
@@ -79,12 +79,9 @@ export function mergeConfigs(...configs) {
 const PRIMITIVES = new Set(["string", "number", "boolean"]);
 /**
  * Ensure that a passed config is an object with all required keys present.
- *
- * Note: To make sure async local storage loading works correctly, this
- * should not be called with a default or prepopulated config argument.
  */
 export function ensureConfig(config) {
-    const implicitConfig = AsyncLocalStorageProviderSingleton.getInstance().getStore();
+    const implicitConfig = AsyncLocalStorageProviderSingleton.getRunnableConfig();
     let empty = {
         tags: [],
         metadata: {},
@@ -92,10 +89,10 @@ export function ensureConfig(config) {
         runId: undefined,
     };
     if (implicitConfig) {
-        // Don't allow runId to be loaded implicitly, as this can cause
+        // Don't allow runId and runName to be loaded implicitly, as this can cause
         // child runs to improperly inherit their parents' run ids.
         // eslint-disable-next-line @typescript-eslint/no-unused-vars
-        const { runId, ...rest } = implicitConfig;
+        const { runId, runName, ...rest } = implicitConfig;
         empty = Object.entries(rest).reduce(
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         (currentConfig, [key, value]) => {

package/dist/runnables/graph.d.ts

@@ -15,7 +15,7 @@ export declare class Graph {
      */
     extend(graph: Graph, prefix?: string): ({
         id: string;
-        data: RunnableIOSchema | RunnableInterface<any, any, import("./config.js").RunnableConfig>;
+        data: RunnableIOSchema | RunnableInterface<any, any, import("./types.js").RunnableConfig>;
     } | undefined)[];
     trimFirstNode(): void;
     trimLastNode(): void;

package/dist/runnables/iter.cjs

@@ -23,9 +23,8 @@ function isAsyncIterable(thing) {
 }
 exports.isAsyncIterable = isAsyncIterable;
 function* consumeIteratorInContext(context, iter) {
-    const storage = index_js_1.AsyncLocalStorageProviderSingleton.getInstance();
     while (true) {
-        const { value, done } = storage.run(context, iter.next.bind(iter));
+        const { value, done } = index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(context, iter.next.bind(iter), true);
         if (done) {
             break;
         }
@@ -36,10 +35,9 @@ function* consumeIteratorInContext(context, iter) {
 }
 exports.consumeIteratorInContext = consumeIteratorInContext;
 async function* consumeAsyncIterableInContext(context, iter) {
-    const storage = index_js_1.AsyncLocalStorageProviderSingleton.getInstance();
     const iterator = iter[Symbol.asyncIterator]();
     while (true) {
-        const { value, done } = await storage.run(context, iterator.next.bind(iter));
+        const { value, done } = await index_js_1.AsyncLocalStorageProviderSingleton.runWithConfig(context, iterator.next.bind(iter), true);
         if (done) {
             break;
         }