@langchain/core 0.2.15 → 0.2.17

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

@@ -1,24 +1,32 @@
 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 { ZodObjectAny } from "../types/zod.js";
+import { MessageContent } from "../messages/base.js";
+import { ToolInputParsingException } from "./utils.js";
+export { ToolInputParsingException };
+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.
- * It extends the built-in `Error` class and adds an optional `output`
- * property that can hold the output that caused the exception.
- */
-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 ZodObjectAny = ZodObjectAny> extends RunnableInterface<(z.output<T> extends string ? string : never) | z.input<T> | ToolCall, ToolReturnType> {
     lc_namespace: string[];
     schema: T | z.ZodEffects<T>;
     /**
@@ -32,9 +40,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 +50,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 ZodObjectAny = ZodObjectAny> 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 +86,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 ZodObjectAny = ZodObjectAny> extends StructuredToolInterface<T> {
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
      *
@@ -82,12 +100,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<ZodObjectAny> {
     schema: z.ZodEffects<z.ZodObject<{
         input: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
@@ -107,7 +125,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 +136,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>;
+export interface DynamicStructuredToolInput<T extends ZodObjectAny = ZodObjectAny> extends BaseDynamicToolInput {
+    func: (input: BaseDynamicToolInput["responseFormat"] extends "content_and_artifact" ? ToolCall : z.infer<T>, runManager?: CallbackManagerForToolRun, config?: RunnableConfig) => Promise<ToolReturnType>;
     schema: T;
 }
 /**
@@ -139,9 +157,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
@@ -149,7 +167,7 @@ export declare class DynamicTool extends Tool {
  * StructuredTool class and overrides the _call method to execute the
  * provided function when the tool is called.
  */
-export declare class DynamicStructuredTool<T extends ZodAny = ZodAny> extends StructuredTool<T> {
+export declare class DynamicStructuredTool<T extends ZodObjectAny = ZodObjectAny> extends StructuredTool<T> {
     static lc_name(): string;
     name: string;
     description: string;
@@ -159,10 +177,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
@@ -175,9 +193,9 @@ export declare abstract class BaseToolkit {
 }
 /**
  * Parameters for the tool function.
- * @template {ZodAny} RunInput The input schema for the tool.
+ * @template {ZodObjectAny | z.ZodString = ZodObjectAny} RunInput The input schema for the tool. Either any Zod object, or a Zod string.
  */
-interface ToolWrapperParams<RunInput extends ZodAny = ZodAny> extends ToolParams {
+interface ToolWrapperParams<RunInput extends ZodObjectAny | z.ZodString = ZodObjectAny> extends ToolParams {
     /**
      * The name of the tool. If using with an LLM, this
      * will be passed as the tool name.
@@ -194,19 +212,30 @@ 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 {ZodObjectAny | z.ZodString = ZodObjectAny} T The input schema for the tool. Either any Zod object, or a Zod string.
  *
- * @param {RunnableFunc<RunInput, string>} func - The function to invoke when the tool is called.
- * @param fields - An object containing the following properties:
+ * @param {RunnableFunc<z.output<T>, ToolReturnType>} func - The function to invoke when the tool is called.
+ * @param {ToolWrapperParams<T>} 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.
+ * @param {ZodObjectAny | z.ZodString | undefined} fields.schema The Zod schema defining the input for the tool. If undefined, it will default to a Zod string schema.
  *
- * @returns {StructuredTool<RunInput, string>} A new StructuredTool instance.
+ * @returns {DynamicStructuredTool<T>} A new StructuredTool instance.
  */
-export declare function tool<RunInput extends ZodAny = ZodAny>(func: RunnableFunc<z.infer<RunInput>, string>, fields: ToolWrapperParams<RunInput>): DynamicStructuredTool<RunInput>;
-export {};
+export declare function tool<T extends z.ZodString = z.ZodString>(func: RunnableFunc<z.output<T>, ToolReturnType>, fields: ToolWrapperParams<T>): DynamicTool;
+export declare function tool<T extends ZodObjectAny = ZodObjectAny>(func: RunnableFunc<z.output<T>, ToolReturnType>, fields: ToolWrapperParams<T>): DynamicStructuredTool<T>;

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

@@ -1,24 +1,11 @@
 import { z } from "zod";
-import { CallbackManager, parseCallbackConfigArg, } from "./callbacks/manager.js";
-import { BaseLangChain, } from "./language_models/base.js";
-import { ensureConfig } from "./runnables/config.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`
- * property that can hold the output that caused the exception.
- */
-export class ToolInputParsingException extends Error {
-    constructor(message, output) {
-        super(message);
-        Object.defineProperty(this, "output", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.output = output;
-    }
-}
+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";
+import { _isToolCall, ToolInputParsingException } from "./utils.js";
+export { ToolInputParsingException };
 /**
  * Base class for Tools that accept input of any shape defined by a Zod schema.
  */
@@ -34,6 +21,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 +45,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 +96,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 +195,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 +253,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);
     }
 }
 /**
@@ -224,27 +267,73 @@ export class BaseToolkit {
         return this.tools;
     }
 }
-/**
- * Creates a new StructuredTool instance with the provided function, name, description, and schema.
- * @function
- * @template {ZodAny} RunInput The input schema for the tool.
- *
- * @param {RunnableFunc<RunInput, string>} 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.
- */
 export function tool(func, fields) {
-    const schema = fields.schema ??
-        z.object({ input: z.string().optional() }).transform((obj) => obj.input);
-    const description = fields.description ?? schema.description ?? `${fields.name} tool`;
+    // If the schema is not provided, or it's a string schema, create a DynamicTool
+    if (!fields.schema || !("shape" in fields.schema) || !fields.schema.shape) {
+        return new DynamicTool({
+            name: fields.name,
+            description: fields.description ??
+                fields.schema?.description ??
+                `${fields.name} tool`,
+            responseFormat: fields.responseFormat,
+            func,
+        });
+    }
+    const description = fields.description ?? fields.schema.description ?? `${fields.name} tool`;
     return new DynamicStructuredTool({
         name: fields.name,
         description,
-        schema: schema,
-        func: async (input, _runManager, config) => func(input, config),
+        schema: fields.schema,
+        // 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 _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,85 @@
+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");
+});
+test("Tool can accept single string input", async () => {
+    const stringTool = tool((input) => {
+        return `${input}a`;
+    }, {
+        name: "string_tool",
+        description: "A tool that appends 'a' to the input string",
+        schema: z.string(),
+    });
+    const result = await stringTool.invoke("b");
+    expect(result).toBe("ba");
+});

package/dist/tools/utils.cjs

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolInputParsingException = exports._isToolCall = void 0;
+function _isToolCall(toolCall) {
+    return !!(toolCall &&
+        typeof toolCall === "object" &&
+        "type" in toolCall &&
+        toolCall.type === "tool_call");
+}
+exports._isToolCall = _isToolCall;
+/**
+ * Custom error class used to handle exceptions related to tool input parsing.
+ * It extends the built-in `Error` class and adds an optional `output`
+ * property that can hold the output that caused the exception.
+ */
+class ToolInputParsingException extends Error {
+    constructor(message, output) {
+        super(message);
+        Object.defineProperty(this, "output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.output = output;
+    }
+}
+exports.ToolInputParsingException = ToolInputParsingException;

package/dist/tools/utils.d.ts

@@ -0,0 +1,11 @@
+import { ToolCall } from "../messages/tool.js";
+export declare function _isToolCall(toolCall?: unknown): toolCall is ToolCall;
+/**
+ * Custom error class used to handle exceptions related to tool input parsing.
+ * It extends the built-in `Error` class and adds an optional `output`
+ * property that can hold the output that caused the exception.
+ */
+export declare class ToolInputParsingException extends Error {
+    output?: string;
+    constructor(message: string, output?: string);
+}

package/dist/tools/utils.js

@@ -0,0 +1,23 @@
+export function _isToolCall(toolCall) {
+    return !!(toolCall &&
+        typeof toolCall === "object" &&
+        "type" in toolCall &&
+        toolCall.type === "tool_call");
+}
+/**
+ * Custom error class used to handle exceptions related to tool input parsing.
+ * It extends the built-in `Error` class and adds an optional `output`
+ * property that can hold the output that caused the exception.
+ */
+export class ToolInputParsingException extends Error {
+    constructor(message, output) {
+        super(message);
+        Object.defineProperty(this, "output", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.output = output;
+    }
+}

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(() => {