@langchain/core 0.2.14 → 0.2.16

package/dist/{tools.d.ts → tools/index.d.ts}

@@ -1,13 +1,28 @@
 import { z } from "zod";
-import { CallbackManagerForToolRun, Callbacks } from "./callbacks/manager.js";
-import { BaseLangChain, type BaseLangChainParams } from "./language_models/base.js";
-import { type RunnableConfig } from "./runnables/config.js";
-import type { RunnableFunc, RunnableInterface } from "./runnables/base.js";
-type ZodAny = z.ZodObject<any, any, any, any>;
+import { CallbackManagerForToolRun, Callbacks } from "../callbacks/manager.js";
+import { BaseLangChain, type BaseLangChainParams } from "../language_models/base.js";
+import { type RunnableConfig } from "../runnables/config.js";
+import type { RunnableFunc, RunnableInterface } from "../runnables/base.js";
+import { ToolCall } from "../messages/tool.js";
+import { ZodAny } from "../types/zod.js";
+import { MessageContent } from "../messages/base.js";
+export type ResponseFormat = "content" | "content_and_artifact" | string;
+type ToolReturnType = any;
+export type ContentAndArtifact = [MessageContent, any];
 /**
  * Parameters for the Tool classes.
  */
 export interface ToolParams extends BaseLangChainParams {
+    /**
+     * The tool response format.
+     *
+     * If "content" then the output of the tool is interpreted as the contents of a
+     * ToolMessage. If "content_and_artifact" then the output is expected to be a
+     * two-tuple corresponding to the (content, artifact) of a ToolMessage.
+     *
+     * @default "content"
+     */
+    responseFormat?: ResponseFormat;
 }
 /**
  * Custom error class used to handle exceptions related to tool input parsing.
@@ -18,7 +33,7 @@ export declare class ToolInputParsingException extends Error {
     output?: string;
     constructor(message: string, output?: string);
 }
-export interface StructuredToolInterface<T extends ZodAny = ZodAny> extends RunnableInterface<(z.output<T> extends string ? string : never) | z.input<T>, string> {
+export interface StructuredToolInterface<T extends ZodAny = ZodAny> extends RunnableInterface<(z.output<T> extends string ? string : never) | z.input<T> | ToolCall, ToolReturnType> {
     lc_namespace: string[];
     schema: T | z.ZodEffects<T>;
     /**
@@ -32,9 +47,9 @@ export interface StructuredToolInterface<T extends ZodAny = ZodAny> extends Runn
      * @param tags Optional tags for the tool.
      * @returns A Promise that resolves with a string.
      */
-    call(arg: (z.output<T> extends string ? string : never) | z.input<T>, configArg?: Callbacks | RunnableConfig, 
+    call(arg: (z.output<T> extends string ? string : never) | z.input<T> | ToolCall, configArg?: Callbacks | RunnableConfig, 
     /** @deprecated */
-    tags?: string[]): Promise<string>;
+    tags?: string[]): Promise<ToolReturnType>;
     name: string;
     description: string;
     returnDirect: boolean;
@@ -42,18 +57,31 @@ export interface StructuredToolInterface<T extends ZodAny = ZodAny> extends Runn
 /**
  * Base class for Tools that accept input of any shape defined by a Zod schema.
  */
-export declare abstract class StructuredTool<T extends ZodAny = ZodAny> extends BaseLangChain<(z.output<T> extends string ? string : never) | z.input<T>, string> {
+export declare abstract class StructuredTool<T extends ZodAny = ZodAny> extends BaseLangChain<(z.output<T> extends string ? string : never) | z.input<T> | ToolCall, ToolReturnType> {
+    abstract name: string;
+    abstract description: string;
     abstract schema: T | z.ZodEffects<T>;
+    returnDirect: boolean;
     get lc_namespace(): string[];
+    /**
+     * The tool response format.
+     *
+     * If "content" then the output of the tool is interpreted as the contents of a
+     * ToolMessage. If "content_and_artifact" then the output is expected to be a
+     * two-tuple corresponding to the (content, artifact) of a ToolMessage.
+     *
+     * @default "content"
+     */
+    responseFormat?: ResponseFormat;
     constructor(fields?: ToolParams);
-    protected abstract _call(arg: z.output<T>, runManager?: CallbackManagerForToolRun, config?: RunnableConfig): Promise<string>;
+    protected abstract _call(arg: z.output<T>, runManager?: CallbackManagerForToolRun, parentConfig?: RunnableConfig): Promise<ToolReturnType>;
     /**
      * Invokes the tool with the provided input and configuration.
      * @param input The input for the tool.
      * @param config Optional configuration for the tool.
      * @returns A Promise that resolves with a string.
      */
-    invoke(input: (z.output<T> extends string ? string : never) | z.input<T>, config?: RunnableConfig): Promise<string>;
+    invoke(input: (z.output<T> extends string ? string : never) | z.input<T> | ToolCall, config?: RunnableConfig): Promise<ToolReturnType>;
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
      *
@@ -65,14 +93,11 @@ export declare abstract class StructuredTool<T extends ZodAny = ZodAny> extends
      * @param tags Optional tags for the tool.
      * @returns A Promise that resolves with a string.
      */
-    call(arg: (z.output<T> extends string ? string : never) | z.input<T>, configArg?: Callbacks | RunnableConfig, 
+    call(arg: (z.output<T> extends string ? string : never) | z.input<T> | ToolCall, configArg?: Callbacks | RunnableConfig, 
     /** @deprecated */
-    tags?: string[]): Promise<string>;
-    abstract name: string;
-    abstract description: string;
-    returnDirect: boolean;
+    tags?: string[]): Promise<ToolReturnType>;
 }
-export interface ToolInterface extends StructuredToolInterface {
+export interface ToolInterface<T extends ZodAny = ZodAny> extends StructuredToolInterface<T> {
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
      *
@@ -82,12 +107,12 @@ export interface ToolInterface extends StructuredToolInterface {
      * @param callbacks Optional callbacks for the tool.
      * @returns A Promise that resolves with a string.
      */
-    call(arg: string | undefined | z.input<this["schema"]>, callbacks?: Callbacks | RunnableConfig): Promise<string>;
+    call(arg: string | undefined | z.input<this["schema"]> | ToolCall, callbacks?: Callbacks | RunnableConfig): Promise<ToolReturnType>;
 }
 /**
  * Base class for Tools that accept input as a string.
  */
-export declare abstract class Tool extends StructuredTool {
+export declare abstract class Tool extends StructuredTool<ZodAny> {
     schema: z.ZodEffects<z.ZodObject<{
         input: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
@@ -107,7 +132,7 @@ export declare abstract class Tool extends StructuredTool {
      * @param callbacks Optional callbacks for the tool.
      * @returns A Promise that resolves with a string.
      */
-    call(arg: string | undefined | z.input<this["schema"]>, callbacks?: Callbacks | RunnableConfig): Promise<string>;
+    call(arg: string | undefined | z.input<this["schema"]> | ToolCall, callbacks?: Callbacks | RunnableConfig): Promise<ToolReturnType>;
 }
 export interface BaseDynamicToolInput extends ToolParams {
     name: string;
@@ -118,13 +143,13 @@ export interface BaseDynamicToolInput extends ToolParams {
  * Interface for the input parameters of the DynamicTool class.
  */
 export interface DynamicToolInput extends BaseDynamicToolInput {
-    func: (input: string, runManager?: CallbackManagerForToolRun, config?: RunnableConfig) => Promise<string>;
+    func: (input: string, runManager?: CallbackManagerForToolRun, config?: RunnableConfig) => Promise<ToolReturnType>;
 }
 /**
  * Interface for the input parameters of the DynamicStructuredTool class.
  */
 export interface DynamicStructuredToolInput<T extends ZodAny = ZodAny> extends BaseDynamicToolInput {
-    func: (input: z.infer<T>, runManager?: CallbackManagerForToolRun, config?: RunnableConfig) => Promise<string>;
+    func: (input: BaseDynamicToolInput["responseFormat"] extends "content_and_artifact" ? ToolCall : z.infer<T>, runManager?: CallbackManagerForToolRun, config?: RunnableConfig) => Promise<ToolReturnType>;
     schema: T;
 }
 /**
@@ -139,9 +164,9 @@ export declare class DynamicTool extends Tool {
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
      */
-    call(arg: string | undefined | z.input<this["schema"]>, configArg?: RunnableConfig | Callbacks): Promise<string>;
+    call(arg: string | undefined | z.input<this["schema"]> | ToolCall, configArg?: RunnableConfig | Callbacks): Promise<ToolReturnType>;
     /** @ignore */
-    _call(input: string, runManager?: CallbackManagerForToolRun, config?: RunnableConfig): Promise<string>;
+    _call(input: string, runManager?: CallbackManagerForToolRun, parentConfig?: RunnableConfig): Promise<ToolReturnType>;
 }
 /**
  * A tool that can be created dynamically from a function, name, and
@@ -159,10 +184,10 @@ export declare class DynamicStructuredTool<T extends ZodAny = ZodAny> extends St
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
      */
-    call(arg: z.output<T>, configArg?: RunnableConfig | Callbacks, 
+    call(arg: z.output<T> | ToolCall, configArg?: RunnableConfig | Callbacks, 
     /** @deprecated */
-    tags?: string[]): Promise<string>;
-    protected _call(arg: z.output<T>, runManager?: CallbackManagerForToolRun, config?: RunnableConfig): Promise<string>;
+    tags?: string[]): Promise<ToolReturnType>;
+    protected _call(arg: z.output<T> | ToolCall, runManager?: CallbackManagerForToolRun, parentConfig?: RunnableConfig): Promise<ToolReturnType>;
 }
 /**
  * Abstract base class for toolkits in LangChain. Toolkits are collections
@@ -176,6 +201,7 @@ export declare abstract class BaseToolkit {
 /**
  * Parameters for the tool function.
  * @template {ZodAny} RunInput The input schema for the tool.
+ * @template {any} RunOutput The output type for the tool.
  */
 interface ToolWrapperParams<RunInput extends ZodAny = ZodAny> extends ToolParams {
     /**
@@ -194,19 +220,31 @@ interface ToolWrapperParams<RunInput extends ZodAny = ZodAny> extends ToolParams
      * for.
      */
     schema?: RunInput;
+    /**
+     * The tool response format.
+     *
+     * If "content" then the output of the tool is interpreted as the contents of a
+     * ToolMessage. If "content_and_artifact" then the output is expected to be a
+     * two-tuple corresponding to the (content, artifact) of a ToolMessage.
+     *
+     * @default "content"
+     */
+    responseFormat?: ResponseFormat;
 }
 /**
  * Creates a new StructuredTool instance with the provided function, name, description, and schema.
  * @function
- * @template {ZodAny} RunInput The input schema for the tool.
+ * @template {RunInput extends ZodAny = ZodAny} RunInput The input schema for the tool. This corresponds to the input type when the tool is invoked.
+ * @template {RunOutput = any} RunOutput The output type for the tool. This corresponds to the output type when the tool is invoked.
+ * @template {FuncInput extends z.infer<RunInput> | ToolCall = z.infer<RunInput>} FuncInput The input type for the function.
  *
- * @param {RunnableFunc<RunInput, string>} func - The function to invoke when the tool is called.
+ * @param {RunnableFunc<z.infer<RunInput> | ToolCall, RunOutput>} func - The function to invoke when the tool is called.
  * @param fields - An object containing the following properties:
  * @param {string} fields.name The name of the tool.
  * @param {string | undefined} fields.description The description of the tool. Defaults to either the description on the Zod schema, or `${fields.name} tool`.
  * @param {z.ZodObject<any, any, any, any>} fields.schema The Zod schema defining the input for the tool.
  *
- * @returns {StructuredTool<RunInput, string>} A new StructuredTool instance.
+ * @returns {DynamicStructuredTool<RunInput, RunOutput>} A new StructuredTool instance.
  */
-export declare function tool<RunInput extends ZodAny = ZodAny>(func: RunnableFunc<z.infer<RunInput>, string>, fields: ToolWrapperParams<RunInput>): DynamicStructuredTool<RunInput>;
+export declare function tool<T extends ZodAny = ZodAny>(func: RunnableFunc<z.output<T>, ToolReturnType>, fields: ToolWrapperParams<T>): DynamicStructuredTool<T>;
 export {};

package/dist/{tools.js → tools/index.js}

@@ -1,7 +1,9 @@
 import { z } from "zod";
-import { CallbackManager, parseCallbackConfigArg, } from "./callbacks/manager.js";
-import { BaseLangChain, } from "./language_models/base.js";
-import { ensureConfig } from "./runnables/config.js";
+import { CallbackManager, parseCallbackConfigArg, } from "../callbacks/manager.js";
+import { BaseLangChain, } from "../language_models/base.js";
+import { ensureConfig, patchConfig, } from "../runnables/config.js";
+import { ToolMessage } from "../messages/tool.js";
+import { AsyncLocalStorageProviderSingleton } from "../singletons/index.js";
 /**
  * Custom error class used to handle exceptions related to tool input parsing.
  * It extends the built-in `Error` class and adds an optional `output`
@@ -34,6 +36,22 @@ export class StructuredTool extends BaseLangChain {
             writable: true,
             value: false
         });
+        /**
+         * The tool response format.
+         *
+         * If "content" then the output of the tool is interpreted as the contents of a
+         * ToolMessage. If "content_and_artifact" then the output is expected to be a
+         * two-tuple corresponding to the (content, artifact) of a ToolMessage.
+         *
+         * @default "content"
+         */
+        Object.defineProperty(this, "responseFormat", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "content"
+        });
+        this.responseFormat = fields?.responseFormat ?? this.responseFormat;
     }
     /**
      * Invokes the tool with the provided input and configuration.
@@ -42,7 +60,23 @@ export class StructuredTool extends BaseLangChain {
      * @returns A Promise that resolves with a string.
      */
     async invoke(input, config) {
-        return this.call(input, ensureConfig(config));
+        let tool_call_id;
+        let toolInput;
+        if (_isToolCall(input)) {
+            tool_call_id = input.id;
+            toolInput = input.args;
+        }
+        else {
+            toolInput = input;
+        }
+        const ensuredConfig = ensureConfig(config);
+        return this.call(toolInput, {
+            ...ensuredConfig,
+            configurable: {
+                ...ensuredConfig.configurable,
+                tool_call_id,
+            },
+        });
     }
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
@@ -77,8 +111,32 @@ export class StructuredTool extends BaseLangChain {
             await runManager?.handleToolError(e);
             throw e;
         }
-        await runManager?.handleToolEnd(result);
-        return result;
+        let content;
+        let artifact;
+        if (this.responseFormat === "content_and_artifact") {
+            if (Array.isArray(result) && result.length === 2) {
+                [content, artifact] = result;
+            }
+            else {
+                throw new Error(`Tool response format is "content_and_artifact" but the output was not a two-tuple.\nResult: ${JSON.stringify(result)}`);
+            }
+        }
+        else {
+            content = result;
+        }
+        let toolCallId;
+        if (config && "configurable" in config) {
+            toolCallId = config.configurable
+                .tool_call_id;
+        }
+        const formattedOutput = _formatToolOutput({
+            content,
+            artifact,
+            toolCallId,
+            name: this.name,
+        });
+        await runManager?.handleToolEnd(formattedOutput);
+        return formattedOutput;
     }
 }
 /**
@@ -152,8 +210,8 @@ export class DynamicTool extends Tool {
         return super.call(arg, config);
     }
     /** @ignore */
-    async _call(input, runManager, config) {
-        return this.func(input, runManager, config);
+    async _call(input, runManager, parentConfig) {
+        return this.func(input, runManager, parentConfig);
     }
 }
 /**
@@ -210,8 +268,8 @@ export class DynamicStructuredTool extends StructuredTool {
         }
         return super.call(arg, config, tags);
     }
-    _call(arg, runManager, config) {
-        return this.func(arg, runManager, config);
+    _call(arg, runManager, parentConfig) {
+        return this.func(arg, runManager, parentConfig);
     }
 }
 /**
@@ -227,15 +285,17 @@ export class BaseToolkit {
 /**
  * Creates a new StructuredTool instance with the provided function, name, description, and schema.
  * @function
- * @template {ZodAny} RunInput The input schema for the tool.
+ * @template {RunInput extends ZodAny = ZodAny} RunInput The input schema for the tool. This corresponds to the input type when the tool is invoked.
+ * @template {RunOutput = any} RunOutput The output type for the tool. This corresponds to the output type when the tool is invoked.
+ * @template {FuncInput extends z.infer<RunInput> | ToolCall = z.infer<RunInput>} FuncInput The input type for the function.
  *
- * @param {RunnableFunc<RunInput, string>} func - The function to invoke when the tool is called.
+ * @param {RunnableFunc<z.infer<RunInput> | ToolCall, RunOutput>} func - The function to invoke when the tool is called.
  * @param fields - An object containing the following properties:
  * @param {string} fields.name The name of the tool.
  * @param {string | undefined} fields.description The description of the tool. Defaults to either the description on the Zod schema, or `${fields.name} tool`.
  * @param {z.ZodObject<any, any, any, any>} fields.schema The Zod schema defining the input for the tool.
  *
- * @returns {StructuredTool<RunInput, string>} A new StructuredTool instance.
+ * @returns {DynamicStructuredTool<RunInput, RunOutput>} A new StructuredTool instance.
  */
 export function tool(func, fields) {
     const schema = fields.schema ??
@@ -245,6 +305,62 @@ export function tool(func, fields) {
         name: fields.name,
         description,
         schema: schema,
-        func: async (input, _runManager, config) => func(input, config),
+        // TODO: Consider moving into DynamicStructuredTool constructor
+        func: async (input, runManager, config) => {
+            return new Promise((resolve, reject) => {
+                const childConfig = patchConfig(config, {
+                    callbacks: runManager?.getChild(),
+                });
+                void AsyncLocalStorageProviderSingleton.getInstance().run(childConfig, async () => {
+                    try {
+                        resolve(func(input, childConfig));
+                    }
+                    catch (e) {
+                        reject(e);
+                    }
+                });
+            });
+        },
+        responseFormat: fields.responseFormat,
     });
 }
+function _isToolCall(toolCall) {
+    return !!(toolCall &&
+        typeof toolCall === "object" &&
+        "type" in toolCall &&
+        toolCall.type === "tool_call");
+}
+function _formatToolOutput(params) {
+    const { content, artifact, toolCallId } = params;
+    if (toolCallId) {
+        if (typeof content === "string" ||
+            (Array.isArray(content) &&
+                content.every((item) => typeof item === "object"))) {
+            return new ToolMessage({
+                content,
+                artifact,
+                tool_call_id: toolCallId,
+                name: params.name,
+            });
+        }
+        else {
+            return new ToolMessage({
+                content: _stringify(content),
+                artifact,
+                tool_call_id: toolCallId,
+                name: params.name,
+            });
+        }
+    }
+    else {
+        return content;
+    }
+}
+function _stringify(content) {
+    try {
+        return JSON.stringify(content, null, 2);
+    }
+    catch (_noOp) {
+        return `${content}`;
+    }
+}

package/dist/tools/tests/tools.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/tools/tests/tools.test.js

@@ -0,0 +1,74 @@
+import { test, expect } from "@jest/globals";
+import { z } from "zod";
+import { tool } from "../index.js";
+import { ToolMessage } from "../../messages/tool.js";
+test("Tool should error if responseFormat is content_and_artifact but the function doesn't return a tuple", async () => {
+    const weatherSchema = z.object({
+        location: z.string(),
+    });
+    const weatherTool = tool((_) => {
+        return "str";
+    }, {
+        name: "weather",
+        schema: weatherSchema,
+        responseFormat: "content_and_artifact",
+    });
+    await expect(async () => {
+        await weatherTool.invoke({ location: "San Francisco" });
+    }).rejects.toThrow();
+});
+test("Tool works if responseFormat is content_and_artifact and returns a tuple", async () => {
+    const weatherSchema = z.object({
+        location: z.string(),
+    });
+    const weatherTool = tool((input) => {
+        return ["msg_content", input];
+    }, {
+        name: "weather",
+        schema: weatherSchema,
+        responseFormat: "content_and_artifact",
+    });
+    const toolResult = await weatherTool.invoke({ location: "San Francisco" });
+    expect(toolResult).not.toBeInstanceOf(ToolMessage);
+    expect(toolResult).toBe("msg_content");
+});
+test("Does not return tool message if responseFormat is content_and_artifact and returns a tuple and a tool call with no id is passed in", async () => {
+    const weatherSchema = z.object({
+        location: z.string(),
+    });
+    const weatherTool = tool((input) => {
+        return ["msg_content", input];
+    }, {
+        name: "weather",
+        schema: weatherSchema,
+        responseFormat: "content_and_artifact",
+    });
+    const toolResult = await weatherTool.invoke({
+        args: { location: "San Francisco" },
+        name: "weather",
+        type: "tool_call",
+    });
+    expect(toolResult).toBe("msg_content");
+});
+test("Returns tool message if responseFormat is content_and_artifact and returns a tuple and a tool call with id is passed in", async () => {
+    const weatherSchema = z.object({
+        location: z.string(),
+    });
+    const weatherTool = tool((input) => {
+        return ["msg_content", input];
+    }, {
+        name: "weather",
+        schema: weatherSchema,
+        responseFormat: "content_and_artifact",
+    });
+    const toolResult = await weatherTool.invoke({
+        id: "testid",
+        args: { location: "San Francisco" },
+        name: "weather",
+        type: "tool_call",
+    });
+    expect(toolResult).toBeInstanceOf(ToolMessage);
+    expect(toolResult.content).toBe("msg_content");
+    expect(toolResult.artifact).toEqual({ location: "San Francisco" });
+    expect(toolResult.name).toBe("weather");
+});

package/dist/tracers/base.cjs

@@ -274,6 +274,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         await this.onToolStart?.(run);
         return run;
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleToolEnd(output, runId) {
         const run = this.runMap.get(runId);
         if (!run || run?.run_type !== "tool") {

package/dist/tracers/base.d.ts

@@ -46,7 +46,7 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
         inputs?: Record<string, unknown>;
     }): Promise<Run>;
     handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, name?: string): Promise<Run>;
-    handleToolEnd(output: string, runId: string): Promise<Run>;
+    handleToolEnd(output: any, runId: string): Promise<Run>;
     handleToolError(error: unknown, runId: string): Promise<Run>;
     handleAgentAction(action: AgentAction, runId: string): Promise<void>;
     handleAgentEnd(action: AgentFinish, runId: string): Promise<void>;

package/dist/tracers/base.js

@@ -271,6 +271,7 @@ export class BaseTracer extends BaseCallbackHandler {
         await this.onToolStart?.(run);
         return run;
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleToolEnd(output, runId) {
         const run = this.runMap.get(runId);
         if (!run || run?.run_type !== "tool") {

package/dist/tracers/event_stream.cjs

@@ -489,6 +489,21 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
             metadata: runInfo.metadata,
         }, runInfo);
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async handleCustomEvent(eventName, data, runId) {
+        const runInfo = this.runInfoMap.get(runId);
+        if (runInfo === undefined) {
+            throw new Error(`handleCustomEvent: Run ID ${runId} not found in run map.`);
+        }
+        await this.send({
+            event: "on_custom_event",
+            run_id: runId,
+            name: eventName,
+            tags: runInfo.tags,
+            metadata: runInfo.metadata,
+            data,
+        }, runInfo);
+    }
     async finish() {
         const pendingPromises = [...this.tappedPromises.values()];
         void Promise.all(pendingPromises).finally(() => {

package/dist/tracers/event_stream.d.ts

@@ -133,6 +133,7 @@ export declare class EventStreamCallbackHandler extends BaseTracer {
     onToolEnd(run: Run): Promise<void>;
     onRetrieverStart(run: Run): Promise<void>;
     onRetrieverEnd(run: Run): Promise<void>;
+    handleCustomEvent(eventName: string, data: any, runId: string): Promise<void>;
     finish(): Promise<void>;
 }
 export {};

package/dist/tracers/event_stream.js

@@ -485,6 +485,21 @@ export class EventStreamCallbackHandler extends BaseTracer {
             metadata: runInfo.metadata,
         }, runInfo);
     }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async handleCustomEvent(eventName, data, runId) {
+        const runInfo = this.runInfoMap.get(runId);
+        if (runInfo === undefined) {
+            throw new Error(`handleCustomEvent: Run ID ${runId} not found in run map.`);
+        }
+        await this.send({
+            event: "on_custom_event",
+            run_id: runId,
+            name: eventName,
+            tags: runInfo.tags,
+            metadata: runInfo.metadata,
+            data,
+        }, runInfo);
+    }
     async finish() {
         const pendingPromises = [...this.tappedPromises.values()];
         void Promise.all(pendingPromises).finally(() => {

package/dist/tracers/initialize.cjs

@@ -24,6 +24,8 @@ async function getTracingCallbackHandler(session) {
 }
 exports.getTracingCallbackHandler = getTracingCallbackHandler;
 /**
+ * @deprecated Instantiate directly using the LangChainTracer constructor.
+ *
  * Function that returns an instance of `LangChainTracer`. It does not
  * load any session data.
  * @returns An instance of `LangChainTracer`.

package/dist/tracers/initialize.d.ts

@@ -11,6 +11,8 @@ import { LangChainTracerV1 } from "./tracer_langchain_v1.js";
  */
 export declare function getTracingCallbackHandler(session?: string): Promise<LangChainTracerV1>;
 /**
+ * @deprecated Instantiate directly using the LangChainTracer constructor.
+ *
  * Function that returns an instance of `LangChainTracer`. It does not
  * load any session data.
  * @returns An instance of `LangChainTracer`.

package/dist/tracers/initialize.js

@@ -20,6 +20,8 @@ export async function getTracingCallbackHandler(session) {
     return tracer;
 }
 /**
+ * @deprecated Instantiate directly using the LangChainTracer constructor.
+ *
  * Function that returns an instance of `LangChainTracer`. It does not
  * load any session data.
  * @returns An instance of `LangChainTracer`.

package/dist/types/zod.cjs

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/types/zod.d.ts

@@ -0,0 +1,2 @@
+import type { z } from "zod";
+export type ZodAny = z.ZodObject<any, any, any, any>;

package/dist/types/zod.js

@@ -0,0 +1 @@
+export {};

package/dist/utils/callbacks.cjs

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.isTracingEnabled = void 0;
+const env_js_1 = require("./env.cjs");
+const isTracingEnabled = (tracingEnabled) => {
+    if (tracingEnabled !== undefined) {
+        return tracingEnabled;
+    }
+    const envVars = [
+        "LANGSMITH_TRACING_V2",
+        "LANGCHAIN_TRACING_V2",
+        "LANGSMITH_TRACING",
+        "LANGCHAIN_TRACING",
+    ];
+    return !!envVars.find((envVar) => (0, env_js_1.getEnvironmentVariable)(envVar) === "true");
+};
+exports.isTracingEnabled = isTracingEnabled;

package/dist/utils/callbacks.d.ts

@@ -0,0 +1 @@
+export declare const isTracingEnabled: (tracingEnabled?: boolean) => boolean;

package/dist/utils/callbacks.js

@@ -0,0 +1,13 @@
+import { getEnvironmentVariable } from "./env.js";
+export const isTracingEnabled = (tracingEnabled) => {
+    if (tracingEnabled !== undefined) {
+        return tracingEnabled;
+    }
+    const envVars = [
+        "LANGSMITH_TRACING_V2",
+        "LANGCHAIN_TRACING_V2",
+        "LANGSMITH_TRACING",
+        "LANGCHAIN_TRACING",
+    ];
+    return !!envVars.find((envVar) => getEnvironmentVariable(envVar) === "true");
+};