@langchain/core 0.2.15 → 0.2.16

package/dist/runnables/tests/runnable_stream_events_v2.test.js

@@ -7,10 +7,12 @@ import { RunnableLambda, RunnableMap, RunnablePassthrough, RunnablePick, } from
 import { ChatPromptTemplate } from "../../prompts/chat.js";
 import { FakeChatModel, FakeLLM, FakeListChatModel, FakeRetriever, FakeStreamingLLM, } from "../../utils/testing/index.js";
 import { AIMessage, AIMessageChunk, HumanMessage, SystemMessage, } from "../../messages/index.js";
-import { DynamicStructuredTool, DynamicTool, tool } from "../../tools.js";
+import { DynamicStructuredTool, DynamicTool, tool } from "../../tools/index.js";
 import { Document } from "../../documents/document.js";
 import { PromptTemplate } from "../../prompts/prompt.js";
 import { GenerationChunk } from "../../outputs.js";
+// Import from web to avoid side-effects from AsyncLocalStorage
+import { dispatchCustomEvent } from "../../callbacks/dispatch/web.js";
 function reverse(s) {
     // Reverse a string.
     return s.split("").reverse().join("");
@@ -1748,6 +1750,109 @@ test("Runnable streamEvents method with simple tools", async () => {
         },
     ]);
 });
+test("Runnable streamEvents method with a custom event", async () => {
+    const lambda = RunnableLambda.from(async (params, config) => {
+        await dispatchCustomEvent("testEvent", { someval: "test" }, config);
+        await dispatchCustomEvent("testEvent", { someval: "test2" }, config);
+        return JSON.stringify({ x: params.x, y: params.y });
+    });
+    const events = [];
+    const eventStream = await lambda.streamEvents({ x: 1, y: "2" }, { version: "v2" });
+    for await (const event of eventStream) {
+        events.push(event);
+    }
+    expect(events).toEqual([
+        {
+            event: "on_chain_start",
+            data: { input: { x: 1, y: "2" } },
+            name: "RunnableLambda",
+            tags: [],
+            run_id: expect.any(String),
+            metadata: {},
+        },
+        {
+            event: "on_custom_event",
+            run_id: expect.any(String),
+            name: "testEvent",
+            tags: [],
+            metadata: {},
+            data: { someval: "test" },
+        },
+        {
+            event: "on_custom_event",
+            run_id: expect.any(String),
+            name: "testEvent",
+            tags: [],
+            metadata: {},
+            data: { someval: "test2" },
+        },
+        {
+            event: "on_chain_stream",
+            run_id: expect.any(String),
+            name: "RunnableLambda",
+            tags: [],
+            metadata: {},
+            data: { chunk: '{"x":1,"y":"2"}' },
+        },
+        {
+            event: "on_chain_end",
+            data: { output: '{"x":1,"y":"2"}' },
+            run_id: expect.any(String),
+            name: "RunnableLambda",
+            tags: [],
+            metadata: {},
+        },
+    ]);
+});
+test("Custom event inside a custom tool", async () => {
+    const customTool = tool(async (params, config) => {
+        await dispatchCustomEvent("testEvent", { someval: "test" }, config);
+        await dispatchCustomEvent("testEvent", { someval: "test2" }, config);
+        return JSON.stringify({ x: params.x, y: params.y });
+    }, {
+        schema: z.object({ x: z.number(), y: z.string() }),
+        name: "testtool",
+    });
+    const events = [];
+    const eventStream = await customTool.streamEvents({ x: 1, y: "2" }, { version: "v2" });
+    for await (const event of eventStream) {
+        events.push(event);
+    }
+    expect(events).toEqual([
+        {
+            event: "on_tool_start",
+            data: { input: { x: 1, y: "2" } },
+            name: "testtool",
+            tags: [],
+            run_id: expect.any(String),
+            metadata: {},
+        },
+        {
+            event: "on_custom_event",
+            run_id: expect.any(String),
+            name: "testEvent",
+            tags: [],
+            metadata: {},
+            data: { someval: "test" },
+        },
+        {
+            event: "on_custom_event",
+            run_id: expect.any(String),
+            name: "testEvent",
+            tags: [],
+            metadata: {},
+            data: { someval: "test2" },
+        },
+        {
+            event: "on_tool_end",
+            data: { output: '{"x":1,"y":"2"}' },
+            run_id: expect.any(String),
+            name: "testtool",
+            tags: [],
+            metadata: {},
+        },
+    ]);
+});
 test("Runnable streamEvents method with tools that return objects", async () => {
     const adderFunc = (_params) => {
         return JSON.stringify({ sum: 3 });

package/dist/runnables/tests/runnable_tools.test.d.ts

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

package/dist/runnables/tests/runnable_tools.test.js

@@ -0,0 +1,111 @@
+import { z } from "zod";
+import { RunnableLambda, RunnableToolLike } from "../base.js";
+test("Runnable asTool works", async () => {
+    const schema = z.object({
+        foo: z.string(),
+    });
+    const runnable = RunnableLambda.from((input, config) => {
+        return `${input.foo}${config?.configurable.foo}`;
+    });
+    const tool = runnable.asTool({
+        schema,
+    });
+    expect(tool).toBeInstanceOf(RunnableToolLike);
+    expect(tool.schema).toBe(schema);
+    expect(tool.name).toBe(runnable.getName());
+});
+test("Runnable asTool works with all populated fields", async () => {
+    const schema = z.object({
+        foo: z.string(),
+    });
+    const runnable = RunnableLambda.from((input, config) => {
+        return `${input.foo}${config?.configurable.foo}`;
+    });
+    const tool = runnable.asTool({
+        schema,
+        name: "test",
+        description: "test",
+    });
+    expect(tool).toBeInstanceOf(RunnableToolLike);
+    expect(tool.schema).toBe(schema);
+    expect(tool.description).toBe("test");
+    expect(tool.name).toBe("test");
+});
+test("Runnable asTool can invoke", async () => {
+    const schema = z.object({
+        foo: z.string(),
+    });
+    const runnable = RunnableLambda.from((input, config) => {
+        return `${input.foo}${config?.configurable.foo}`;
+    });
+    const tool = runnable.asTool({
+        schema,
+    });
+    const toolResponse = await tool.invoke({
+        foo: "bar",
+    }, {
+        configurable: {
+            foo: "bar",
+        },
+    });
+    expect(toolResponse).toBe("barbar");
+});
+test("asTool should type error with mismatched schema", async () => {
+    // asTool infers the type of the Zod schema from the existing runnable's RunInput generic.
+    // If the Zod schema does not match the RunInput, it should throw a type error.
+    const schema = z.object({
+        foo: z.string(),
+    });
+    const runnable = RunnableLambda.from((input, config) => {
+        return `${input.bar}${config?.configurable.foo}`;
+    });
+    runnable.asTool({
+        // @ts-expect-error - Should error. If this does not give a type error, the generics/typing of `asTool` is broken.
+        schema,
+    });
+});
+test("Create a runnable tool directly from RunnableToolLike", async () => {
+    const schema = z.object({
+        foo: z.string(),
+    });
+    const adderFunc = (_) => {
+        return Promise.resolve(true);
+    };
+    const tool = new RunnableToolLike({
+        schema,
+        name: "test",
+        description: "test",
+        bound: RunnableLambda.from(adderFunc),
+    });
+    const result = await tool.invoke({ foo: "bar" });
+    expect(result).toBe(true);
+});
+test("asTool can take a single string input", async () => {
+    const firstRunnable = RunnableLambda.from((input) => {
+        return `${input}a`;
+    });
+    const secondRunnable = RunnableLambda.from((input) => {
+        return `${input}z`;
+    });
+    const runnable = firstRunnable.pipe(secondRunnable);
+    const asTool = runnable.asTool({
+        schema: z.string(),
+    });
+    const result = await asTool.invoke("b");
+    expect(result).toBe("baz");
+});
+test("Runnable asTool uses Zod schema description if not provided", async () => {
+    const description = "Test schema";
+    const schema = z
+        .object({
+        foo: z.string(),
+    })
+        .describe(description);
+    const runnable = RunnableLambda.from((input, config) => {
+        return `${input.foo}${config?.configurable.foo}`;
+    });
+    const tool = runnable.asTool({
+        schema,
+    });
+    expect(tool.description).toBe(description);
+});

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

@@ -2,9 +2,11 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.tool = exports.BaseToolkit = exports.DynamicStructuredTool = exports.DynamicTool = exports.Tool = exports.StructuredTool = exports.ToolInputParsingException = void 0;
 const zod_1 = require("zod");
-const manager_js_1 = require("./callbacks/manager.cjs");
-const base_js_1 = require("./language_models/base.cjs");
-const config_js_1 = require("./runnables/config.cjs");
+const manager_js_1 = require("../callbacks/manager.cjs");
+const base_js_1 = require("../language_models/base.cjs");
+const config_js_1 = require("../runnables/config.cjs");
+const tool_js_1 = require("../messages/tool.cjs");
+const index_js_1 = require("../singletons/index.cjs");
 /**
  * Custom error class used to handle exceptions related to tool input parsing.
  * It extends the built-in `Error` class and adds an optional `output`
@@ -38,6 +40,22 @@ class StructuredTool extends base_js_1.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.
@@ -46,7 +64,23 @@ class StructuredTool extends base_js_1.BaseLangChain {
      * @returns A Promise that resolves with a string.
      */
     async invoke(input, config) {
-        return this.call(input, (0, config_js_1.ensureConfig)(config));
+        let tool_call_id;
+        let toolInput;
+        if (_isToolCall(input)) {
+            tool_call_id = input.id;
+            toolInput = input.args;
+        }
+        else {
+            toolInput = input;
+        }
+        const ensuredConfig = (0, config_js_1.ensureConfig)(config);
+        return this.call(toolInput, {
+            ...ensuredConfig,
+            configurable: {
+                ...ensuredConfig.configurable,
+                tool_call_id,
+            },
+        });
     }
     /**
      * @deprecated Use .invoke() instead. Will be removed in 0.3.0.
@@ -81,8 +115,32 @@ class StructuredTool extends base_js_1.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;
     }
 }
 exports.StructuredTool = StructuredTool;
@@ -158,8 +216,8 @@ 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);
     }
 }
 exports.DynamicTool = DynamicTool;
@@ -217,8 +275,8 @@ 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);
     }
 }
 exports.DynamicStructuredTool = DynamicStructuredTool;
@@ -236,15 +294,17 @@ exports.BaseToolkit = 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.
  */
 function tool(func, fields) {
     const schema = fields.schema ??
@@ -254,7 +314,63 @@ 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 = (0, config_js_1.patchConfig)(config, {
+                    callbacks: runManager?.getChild(),
+                });
+                void index_js_1.AsyncLocalStorageProviderSingleton.getInstance().run(childConfig, async () => {
+                    try {
+                        resolve(func(input, childConfig));
+                    }
+                    catch (e) {
+                        reject(e);
+                    }
+                });
+            });
+        },
+        responseFormat: fields.responseFormat,
     });
 }
 exports.tool = tool;
+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 tool_js_1.ToolMessage({
+                content,
+                artifact,
+                tool_call_id: toolCallId,
+                name: params.name,
+            });
+        }
+        else {
+            return new tool_js_1.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.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 {};