@langchain/core 0.2.30 → 0.2.32

package/dist/language_models/chat_models.cjs

@@ -90,6 +90,11 @@ class BaseChatModel extends base_js_1.BaseLanguageModel {
             let generationChunk;
             try {
                 for await (const chunk of this._streamResponseChunks(messages, callOptions, runManagers?.[0])) {
+                    if (chunk.message.id == null) {
+                        const runId = runManagers?.at(0)?.runId;
+                        if (runId != null)
+                            chunk.message._updateId(`run-${runId}`);
+                    }
                     chunk.message.response_metadata = {
                         ...chunk.generationInfo,
                         ...chunk.message.response_metadata,
@@ -150,6 +155,11 @@ class BaseChatModel extends base_js_1.BaseLanguageModel {
                 const stream = await this._streamResponseChunks(baseMessages[0], parsedOptions, runManagers?.[0]);
                 let aggregated;
                 for await (const chunk of stream) {
+                    if (chunk.message.id == null) {
+                        const runId = runManagers?.at(0)?.runId;
+                        if (runId != null)
+                            chunk.message._updateId(`run-${runId}`);
+                    }
                     if (aggregated === undefined) {
                         aggregated = chunk;
                     }
@@ -179,6 +189,11 @@ class BaseChatModel extends base_js_1.BaseLanguageModel {
                 if (pResult.status === "fulfilled") {
                     const result = pResult.value;
                     for (const generation of result.generations) {
+                        if (generation.message.id == null) {
+                            const runId = runManagers?.at(0)?.runId;
+                            if (runId != null)
+                                generation.message._updateId(`run-${runId}`);
+                        }
                         generation.message.response_metadata = {
                             ...generation.generationInfo,
                             ...generation.message.response_metadata,

package/dist/language_models/chat_models.js

@@ -86,6 +86,11 @@ export class BaseChatModel extends BaseLanguageModel {
             let generationChunk;
             try {
                 for await (const chunk of this._streamResponseChunks(messages, callOptions, runManagers?.[0])) {
+                    if (chunk.message.id == null) {
+                        const runId = runManagers?.at(0)?.runId;
+                        if (runId != null)
+                            chunk.message._updateId(`run-${runId}`);
+                    }
                     chunk.message.response_metadata = {
                         ...chunk.generationInfo,
                         ...chunk.message.response_metadata,
@@ -146,6 +151,11 @@ export class BaseChatModel extends BaseLanguageModel {
                 const stream = await this._streamResponseChunks(baseMessages[0], parsedOptions, runManagers?.[0]);
                 let aggregated;
                 for await (const chunk of stream) {
+                    if (chunk.message.id == null) {
+                        const runId = runManagers?.at(0)?.runId;
+                        if (runId != null)
+                            chunk.message._updateId(`run-${runId}`);
+                    }
                     if (aggregated === undefined) {
                         aggregated = chunk;
                     }
@@ -175,6 +185,11 @@ export class BaseChatModel extends BaseLanguageModel {
                 if (pResult.status === "fulfilled") {
                     const result = pResult.value;
                     for (const generation of result.generations) {
+                        if (generation.message.id == null) {
+                            const runId = runManagers?.at(0)?.runId;
+                            if (runId != null)
+                                generation.message._updateId(`run-${runId}`);
+                        }
                         generation.message.response_metadata = {
                             ...generation.generationInfo,
                             ...generation.message.response_metadata,

package/dist/messages/ai.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AIMessageChunk = exports.isAIMessage = exports.AIMessage = void 0;
+exports.AIMessageChunk = exports.isAIMessageChunk = exports.isAIMessage = exports.AIMessage = void 0;
 const json_js_1 = require("../utils/json.cjs");
 const base_js_1 = require("./base.cjs");
 const tool_js_1 = require("./tool.cjs");
@@ -112,6 +112,10 @@ function isAIMessage(x) {
     return x._getType() === "ai";
 }
 exports.isAIMessage = isAIMessage;
+function isAIMessageChunk(x) {
+    return x._getType() === "ai";
+}
+exports.isAIMessageChunk = isAIMessageChunk;
 /**
  * Represents a chunk of an AI message, which can be concatenated with
  * other AI message chunks.

package/dist/messages/ai.d.ts

@@ -41,6 +41,7 @@ export declare class AIMessage extends BaseMessage {
     get _printableFields(): Record<string, unknown>;
 }
 export declare function isAIMessage(x: BaseMessage): x is AIMessage;
+export declare function isAIMessageChunk(x: BaseMessageChunk): x is AIMessageChunk;
 export type AIMessageChunkFields = AIMessageFields & {
     tool_call_chunks?: ToolCallChunk[];
 };

package/dist/messages/ai.js

@@ -107,6 +107,9 @@ export class AIMessage extends BaseMessage {
 export function isAIMessage(x) {
     return x._getType() === "ai";
 }
+export function isAIMessageChunk(x) {
+    return x._getType() === "ai";
+}
 /**
  * Represents a chunk of an AI message, which can be concatenated with
  * other AI message chunks.

package/dist/messages/base.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isBaseMessageChunk = exports.isBaseMessage = exports.BaseMessageChunk = exports._mergeObj = exports._mergeLists = exports._mergeDicts = exports.isOpenAIToolCallArray = exports.BaseMessage = exports._mergeStatus = exports.mergeContent = void 0;
+exports.isBaseMessageChunk = exports.isBaseMessage = exports._isMessageFieldWithRole = exports.BaseMessageChunk = exports._mergeObj = exports._mergeLists = exports._mergeDicts = exports.isOpenAIToolCallArray = exports.BaseMessage = exports._mergeStatus = exports.mergeContent = void 0;
 const serializable_js_1 = require("../load/serializable.cjs");
 function mergeContent(firstContent, secondContent) {
     // If first content is a string
@@ -181,6 +181,14 @@ class BaseMessage extends serializable_js_1.Serializable {
             response_metadata: this.response_metadata,
         };
     }
+    // this private method is used to update the ID for the runtime
+    // value as well as in lc_kwargs for serialisation
+    _updateId(value) {
+        this.id = value;
+        // lc_attributes wouldn't work here, because jest compares the
+        // whole object
+        this.lc_kwargs.id = value;
+    }
     get [Symbol.toStringTag]() {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         return this.constructor.lc_name();
@@ -317,6 +325,10 @@ exports._mergeObj = _mergeObj;
 class BaseMessageChunk extends BaseMessage {
 }
 exports.BaseMessageChunk = BaseMessageChunk;
+function _isMessageFieldWithRole(x) {
+    return typeof x.role === "string";
+}
+exports._isMessageFieldWithRole = _isMessageFieldWithRole;
 function isBaseMessage(messageLike) {
     return typeof messageLike?._getType === "function";
 }

package/dist/messages/base.d.ts

@@ -134,6 +134,7 @@ export declare abstract class BaseMessage extends Serializable implements BaseMe
     toDict(): StoredMessage;
     static lc_name(): string;
     get _printableFields(): Record<string, unknown>;
+    _updateId(value: string | undefined): void;
     get [Symbol.toStringTag](): any;
 }
 export type OpenAIToolCall = ToolCall & {
@@ -153,9 +154,15 @@ export declare function _mergeObj<T = any>(left: T | undefined, right: T | undef
 export declare abstract class BaseMessageChunk extends BaseMessage {
     abstract concat(chunk: BaseMessageChunk): BaseMessageChunk;
 }
+export type MessageFieldWithRole = {
+    role: StringWithAutocomplete<"user" | "assistant" | MessageType>;
+    content: MessageContent;
+    name?: string;
+} & Record<string, unknown>;
+export declare function _isMessageFieldWithRole(x: BaseMessageLike): x is MessageFieldWithRole;
 export type BaseMessageLike = BaseMessage | ({
     type: MessageType | "user" | "assistant" | "placeholder";
-} & BaseMessageFields & Record<string, unknown>) | [
+} & BaseMessageFields & Record<string, unknown>) | MessageFieldWithRole | [
     StringWithAutocomplete<MessageType | "user" | "assistant" | "placeholder">,
     MessageContent
 ] | string;

package/dist/messages/base.js

@@ -176,6 +176,14 @@ export class BaseMessage extends Serializable {
             response_metadata: this.response_metadata,
         };
     }
+    // this private method is used to update the ID for the runtime
+    // value as well as in lc_kwargs for serialisation
+    _updateId(value) {
+        this.id = value;
+        // lc_attributes wouldn't work here, because jest compares the
+        // whole object
+        this.lc_kwargs.id = value;
+    }
     get [Symbol.toStringTag]() {
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         return this.constructor.lc_name();
@@ -306,6 +314,9 @@ export function _mergeObj(left, right) {
  */
 export class BaseMessageChunk extends BaseMessage {
 }
+export function _isMessageFieldWithRole(x) {
+    return typeof x.role === "string";
+}
 export function isBaseMessage(messageLike) {
     return typeof messageLike?._getType === "function";
 }

package/dist/messages/utils.cjs

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.convertToChunk = exports.mapChatMessagesToStoredMessages = exports.mapStoredMessagesToChatMessages = exports.mapStoredMessageToChatMessage = exports.getBufferString = exports.coerceMessageLikeToMessage = void 0;
+const utils_js_1 = require("../tools/utils.cjs");
 const ai_js_1 = require("./ai.cjs");
 const base_js_1 = require("./base.cjs");
 const chat_js_1 = require("./chat.cjs");
@@ -8,13 +9,43 @@ const function_js_1 = require("./function.cjs");
 const human_js_1 = require("./human.cjs");
 const system_js_1 = require("./system.cjs");
 const tool_js_1 = require("./tool.cjs");
+function _coerceToolCall(toolCall) {
+    if ((0, utils_js_1._isToolCall)(toolCall)) {
+        return toolCall;
+    }
+    else if (typeof toolCall.id === "string" &&
+        toolCall.type === "function" &&
+        typeof toolCall.function === "object" &&
+        toolCall.function !== null &&
+        "arguments" in toolCall.function &&
+        typeof toolCall.function.arguments === "string" &&
+        "name" in toolCall.function &&
+        typeof toolCall.function.name === "string") {
+        // Handle OpenAI tool call format
+        return {
+            id: toolCall.id,
+            args: JSON.parse(toolCall.function.arguments),
+            name: toolCall.function.name,
+            type: "tool_call",
+        };
+    }
+    else {
+        // TODO: Throw an error?
+        return toolCall;
+    }
+}
 function _constructMessageFromParams(params) {
     const { type, ...rest } = params;
     if (type === "human" || type === "user") {
         return new human_js_1.HumanMessage(rest);
     }
     else if (type === "ai" || type === "assistant") {
-        return new ai_js_1.AIMessage(rest);
+        const { tool_calls: rawToolCalls, ...other } = rest;
+        if (!Array.isArray(rawToolCalls)) {
+            return new ai_js_1.AIMessage(rest);
+        }
+        const tool_calls = rawToolCalls.map(_coerceToolCall);
+        return new ai_js_1.AIMessage({ ...other, tool_calls });
     }
     else if (type === "system") {
         return new system_js_1.SystemMessage(rest);
@@ -42,6 +73,10 @@ function coerceMessageLikeToMessage(messageLike) {
         const [type, content] = messageLike;
         return _constructMessageFromParams({ type, content });
     }
+    else if ((0, base_js_1._isMessageFieldWithRole)(messageLike)) {
+        const { role: type, ...rest } = messageLike;
+        return _constructMessageFromParams({ ...rest, type });
+    }
     else {
         return _constructMessageFromParams(messageLike);
     }

package/dist/messages/utils.js

@@ -1,17 +1,48 @@
+import { _isToolCall } from "../tools/utils.js";
 import { AIMessage, AIMessageChunk } from "./ai.js";
-import { isBaseMessage, } from "./base.js";
+import { isBaseMessage, _isMessageFieldWithRole, } from "./base.js";
 import { ChatMessage, ChatMessageChunk, } from "./chat.js";
 import { FunctionMessage, FunctionMessageChunk, } from "./function.js";
 import { HumanMessage, HumanMessageChunk } from "./human.js";
 import { SystemMessage, SystemMessageChunk } from "./system.js";
-import { ToolMessage } from "./tool.js";
+import { ToolMessage, } from "./tool.js";
+function _coerceToolCall(toolCall) {
+    if (_isToolCall(toolCall)) {
+        return toolCall;
+    }
+    else if (typeof toolCall.id === "string" &&
+        toolCall.type === "function" &&
+        typeof toolCall.function === "object" &&
+        toolCall.function !== null &&
+        "arguments" in toolCall.function &&
+        typeof toolCall.function.arguments === "string" &&
+        "name" in toolCall.function &&
+        typeof toolCall.function.name === "string") {
+        // Handle OpenAI tool call format
+        return {
+            id: toolCall.id,
+            args: JSON.parse(toolCall.function.arguments),
+            name: toolCall.function.name,
+            type: "tool_call",
+        };
+    }
+    else {
+        // TODO: Throw an error?
+        return toolCall;
+    }
+}
 function _constructMessageFromParams(params) {
     const { type, ...rest } = params;
     if (type === "human" || type === "user") {
         return new HumanMessage(rest);
     }
     else if (type === "ai" || type === "assistant") {
-        return new AIMessage(rest);
+        const { tool_calls: rawToolCalls, ...other } = rest;
+        if (!Array.isArray(rawToolCalls)) {
+            return new AIMessage(rest);
+        }
+        const tool_calls = rawToolCalls.map(_coerceToolCall);
+        return new AIMessage({ ...other, tool_calls });
     }
     else if (type === "system") {
         return new SystemMessage(rest);
@@ -39,6 +70,10 @@ export function coerceMessageLikeToMessage(messageLike) {
         const [type, content] = messageLike;
         return _constructMessageFromParams({ type, content });
     }
+    else if (_isMessageFieldWithRole(messageLike)) {
+        const { role: type, ...rest } = messageLike;
+        return _constructMessageFromParams({ ...rest, type });
+    }
     else {
         return _constructMessageFromParams(messageLike);
     }

package/dist/output_parsers/openai_tools/json_output_tools_parsers.cjs

@@ -3,6 +3,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.JsonOutputKeyToolsParser = exports.JsonOutputToolsParser = exports.makeInvalidToolCall = exports.convertLangChainToolCallToOpenAI = exports.parseToolCall = void 0;
 const base_js_1 = require("../base.cjs");
 const json_js_1 = require("../json.cjs");
+const transform_js_1 = require("../transform.cjs");
+const ai_js_1 = require("../../messages/ai.cjs");
 function parseToolCall(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 rawToolCall, options) {
@@ -74,7 +76,7 @@ exports.makeInvalidToolCall = makeInvalidToolCall;
 /**
  * Class for parsing the output of a tool-calling LLM into a JSON object.
  */
-class JsonOutputToolsParser extends base_js_1.BaseLLMOutputParser {
+class JsonOutputToolsParser extends transform_js_1.BaseCumulativeTransformOutputParser {
     static lc_name() {
         return "JsonOutputToolsParser";
     }
@@ -100,29 +102,58 @@ class JsonOutputToolsParser extends base_js_1.BaseLLMOutputParser {
         });
         this.returnId = fields?.returnId ?? this.returnId;
     }
+    _diff() {
+        throw new Error("Not supported.");
+    }
+    async parse() {
+        throw new Error("Not implemented.");
+    }
+    async parseResult(generations) {
+        const result = await this.parsePartialResult(generations, false);
+        return result;
+    }
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,
      * only the arguments of the function call are returned.
      * @param generations The output of the LLM to parse.
      * @returns A JSON object representation of the function call or its arguments.
      */
-    async parseResult(generations) {
-        const toolCalls = generations[0].message.additional_kwargs.tool_calls;
+    async parsePartialResult(generations, partial = true
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ) {
+        const message = generations[0].message;
+        let toolCalls;
+        if ((0, ai_js_1.isAIMessage)(message) && message.tool_calls?.length) {
+            toolCalls = message.tool_calls.map((toolCall) => {
+                const { id, ...rest } = toolCall;
+                if (!this.returnId) {
+                    return rest;
+                }
+                return {
+                    id,
+                    ...rest,
+                };
+            });
+        }
+        else if (message.additional_kwargs.tool_calls !== undefined) {
+            const rawToolCalls = JSON.parse(JSON.stringify(message.additional_kwargs.tool_calls));
+            toolCalls = rawToolCalls.map((rawToolCall) => {
+                return parseToolCall(rawToolCall, { returnId: this.returnId, partial });
+            });
+        }
         if (!toolCalls) {
-            throw new Error(`No tools_call in message ${JSON.stringify(generations)}`);
+            return [];
         }
-        const clonedToolCalls = JSON.parse(JSON.stringify(toolCalls));
         const parsedToolCalls = [];
-        for (const toolCall of clonedToolCalls) {
-            const parsedToolCall = parseToolCall(toolCall, { partial: true });
-            if (parsedToolCall !== undefined) {
+        for (const toolCall of toolCalls) {
+            if (toolCall !== undefined) {
                 // backward-compatibility with previous
                 // versions of Langchain JS, which uses `name` and `arguments`
                 // @ts-expect-error name and arguemnts are defined by Object.defineProperty
                 const backwardsCompatibleToolCall = {
-                    type: parsedToolCall.name,
-                    args: parsedToolCall.args,
-                    id: parsedToolCall.id,
+                    type: toolCall.name,
+                    args: toolCall.args,
+                    id: toolCall.id,
                 };
                 Object.defineProperty(backwardsCompatibleToolCall, "name", {
                     get() {
@@ -145,7 +176,7 @@ exports.JsonOutputToolsParser = JsonOutputToolsParser;
  * Class for parsing the output of a tool-calling LLM into a JSON object if you are
  * expecting only a single tool to be called.
  */
-class JsonOutputKeyToolsParser extends base_js_1.BaseLLMOutputParser {
+class JsonOutputKeyToolsParser extends JsonOutputToolsParser {
     static lc_name() {
         return "JsonOutputKeyToolsParser";
     }
@@ -183,12 +214,6 @@ class JsonOutputKeyToolsParser extends base_js_1.BaseLLMOutputParser {
             writable: true,
             value: false
         });
-        Object.defineProperty(this, "initialParser", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "zodSchema", {
             enumerable: true,
             configurable: true,
@@ -197,7 +222,6 @@ class JsonOutputKeyToolsParser extends base_js_1.BaseLLMOutputParser {
         });
         this.keyName = params.keyName;
         this.returnSingle = params.returnSingle ?? this.returnSingle;
-        this.initialParser = new JsonOutputToolsParser(params);
         this.zodSchema = params.zodSchema;
     }
     async _validateResult(result) {
@@ -213,11 +237,31 @@ class JsonOutputKeyToolsParser extends base_js_1.BaseLLMOutputParser {
         }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async parsePartialResult(generations) {
+        const results = await super.parsePartialResult(generations);
+        const matchingResults = results.filter((result) => result.type === this.keyName);
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        let returnedValues = matchingResults;
+        if (!matchingResults.length) {
+            return undefined;
+        }
+        if (!this.returnId) {
+            returnedValues = matchingResults.map((result) => result.args);
+        }
+        if (this.returnSingle) {
+            return returnedValues[0];
+        }
+        return returnedValues;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async parseResult(generations) {
-        const results = await this.initialParser.parseResult(generations);
+        const results = await super.parsePartialResult(generations, false);
         const matchingResults = results.filter((result) => result.type === this.keyName);
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let returnedValues = matchingResults;
+        if (!matchingResults.length) {
+            return undefined;
+        }
         if (!this.returnId) {
             returnedValues = matchingResults.map((result) => result.args);
         }

package/dist/output_parsers/openai_tools/json_output_tools_parsers.d.ts

@@ -1,7 +1,7 @@
 import { z } from "zod";
-import { ChatGeneration } from "../../outputs.js";
-import { BaseLLMOutputParser } from "../base.js";
+import { ChatGeneration, ChatGenerationChunk } from "../../outputs.js";
 import { InvalidToolCall, ToolCall } from "../../messages/tool.js";
+import { BaseCumulativeTransformOutputParser, BaseCumulativeTransformOutputParserInput } from "../transform.js";
 export type ParsedToolCall = {
     id?: string;
     type: string;
@@ -14,7 +14,7 @@ export type ParsedToolCall = {
 export type JsonOutputToolsParserParams = {
     /** Whether to return the tool call id. */
     returnId?: boolean;
-};
+} & BaseCumulativeTransformOutputParserInput;
 export declare function parseToolCall(rawToolCall: Record<string, any>, options: {
     returnId?: boolean;
     partial: true;
@@ -23,6 +23,10 @@ export declare function parseToolCall(rawToolCall: Record<string, any>, options?
     returnId?: boolean;
     partial?: false;
 }): ToolCall;
+export declare function parseToolCall(rawToolCall: Record<string, any>, options?: {
+    returnId?: boolean;
+    partial?: boolean;
+}): ToolCall | undefined;
 export declare function convertLangChainToolCallToOpenAI(toolCall: ToolCall): {
     id: string;
     type: string;
@@ -35,32 +39,33 @@ export declare function makeInvalidToolCall(rawToolCall: Record<string, any>, er
 /**
  * Class for parsing the output of a tool-calling LLM into a JSON object.
  */
-export declare class JsonOutputToolsParser extends BaseLLMOutputParser<ParsedToolCall[]> {
+export declare class JsonOutputToolsParser<T> extends BaseCumulativeTransformOutputParser<T> {
     static lc_name(): string;
     returnId: boolean;
     lc_namespace: string[];
     lc_serializable: boolean;
     constructor(fields?: JsonOutputToolsParserParams);
+    protected _diff(): void;
+    parse(): Promise<T>;
+    parseResult(generations: ChatGeneration[]): Promise<T>;
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,
      * only the arguments of the function call are returned.
      * @param generations The output of the LLM to parse.
      * @returns A JSON object representation of the function call or its arguments.
      */
-    parseResult(generations: ChatGeneration[]): Promise<ParsedToolCall[]>;
+    parsePartialResult(generations: ChatGenerationChunk[] | ChatGeneration[], partial?: boolean): Promise<any>;
 }
 export type JsonOutputKeyToolsParserParams<T extends Record<string, any> = Record<string, any>> = {
     keyName: string;
     returnSingle?: boolean;
-    /** Whether to return the tool call id. */
-    returnId?: boolean;
     zodSchema?: z.ZodType<T>;
-};
+} & JsonOutputToolsParserParams;
 /**
  * Class for parsing the output of a tool-calling LLM into a JSON object if you are
  * expecting only a single tool to be called.
  */
-export declare class JsonOutputKeyToolsParser<T extends Record<string, any> = Record<string, any>> extends BaseLLMOutputParser<T> {
+export declare class JsonOutputKeyToolsParser<T extends Record<string, any> = Record<string, any>> extends JsonOutputToolsParser<T> {
     static lc_name(): string;
     lc_namespace: string[];
     lc_serializable: boolean;
@@ -69,9 +74,9 @@ export declare class JsonOutputKeyToolsParser<T extends Record<string, any> = Re
     keyName: string;
     /** Whether to return only the first tool call. */
     returnSingle: boolean;
-    initialParser: JsonOutputToolsParser;
     zodSchema?: z.ZodType<T>;
     constructor(params: JsonOutputKeyToolsParserParams<T>);
     protected _validateResult(result: unknown): Promise<T>;
+    parsePartialResult(generations: ChatGeneration[]): Promise<any>;
     parseResult(generations: ChatGeneration[]): Promise<any>;
 }

package/dist/output_parsers/openai_tools/json_output_tools_parsers.js

@@ -1,5 +1,7 @@
-import { BaseLLMOutputParser, OutputParserException } from "../base.js";
+import { OutputParserException } from "../base.js";
 import { parsePartialJson } from "../json.js";
+import { BaseCumulativeTransformOutputParser, } from "../transform.js";
+import { isAIMessage } from "../../messages/ai.js";
 export function parseToolCall(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 rawToolCall, options) {
@@ -68,7 +70,7 @@ rawToolCall, errorMsg) {
 /**
  * Class for parsing the output of a tool-calling LLM into a JSON object.
  */
-export class JsonOutputToolsParser extends BaseLLMOutputParser {
+export class JsonOutputToolsParser extends BaseCumulativeTransformOutputParser {
     static lc_name() {
         return "JsonOutputToolsParser";
     }
@@ -94,29 +96,58 @@ export class JsonOutputToolsParser extends BaseLLMOutputParser {
         });
         this.returnId = fields?.returnId ?? this.returnId;
     }
+    _diff() {
+        throw new Error("Not supported.");
+    }
+    async parse() {
+        throw new Error("Not implemented.");
+    }
+    async parseResult(generations) {
+        const result = await this.parsePartialResult(generations, false);
+        return result;
+    }
     /**
      * Parses the output and returns a JSON object. If `argsOnly` is true,
      * only the arguments of the function call are returned.
      * @param generations The output of the LLM to parse.
      * @returns A JSON object representation of the function call or its arguments.
      */
-    async parseResult(generations) {
-        const toolCalls = generations[0].message.additional_kwargs.tool_calls;
+    async parsePartialResult(generations, partial = true
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    ) {
+        const message = generations[0].message;
+        let toolCalls;
+        if (isAIMessage(message) && message.tool_calls?.length) {
+            toolCalls = message.tool_calls.map((toolCall) => {
+                const { id, ...rest } = toolCall;
+                if (!this.returnId) {
+                    return rest;
+                }
+                return {
+                    id,
+                    ...rest,
+                };
+            });
+        }
+        else if (message.additional_kwargs.tool_calls !== undefined) {
+            const rawToolCalls = JSON.parse(JSON.stringify(message.additional_kwargs.tool_calls));
+            toolCalls = rawToolCalls.map((rawToolCall) => {
+                return parseToolCall(rawToolCall, { returnId: this.returnId, partial });
+            });
+        }
         if (!toolCalls) {
-            throw new Error(`No tools_call in message ${JSON.stringify(generations)}`);
+            return [];
         }
-        const clonedToolCalls = JSON.parse(JSON.stringify(toolCalls));
         const parsedToolCalls = [];
-        for (const toolCall of clonedToolCalls) {
-            const parsedToolCall = parseToolCall(toolCall, { partial: true });
-            if (parsedToolCall !== undefined) {
+        for (const toolCall of toolCalls) {
+            if (toolCall !== undefined) {
                 // backward-compatibility with previous
                 // versions of Langchain JS, which uses `name` and `arguments`
                 // @ts-expect-error name and arguemnts are defined by Object.defineProperty
                 const backwardsCompatibleToolCall = {
-                    type: parsedToolCall.name,
-                    args: parsedToolCall.args,
-                    id: parsedToolCall.id,
+                    type: toolCall.name,
+                    args: toolCall.args,
+                    id: toolCall.id,
                 };
                 Object.defineProperty(backwardsCompatibleToolCall, "name", {
                     get() {
@@ -138,7 +169,7 @@ export class JsonOutputToolsParser extends BaseLLMOutputParser {
  * Class for parsing the output of a tool-calling LLM into a JSON object if you are
  * expecting only a single tool to be called.
  */
-export class JsonOutputKeyToolsParser extends BaseLLMOutputParser {
+export class JsonOutputKeyToolsParser extends JsonOutputToolsParser {
     static lc_name() {
         return "JsonOutputKeyToolsParser";
     }
@@ -176,12 +207,6 @@ export class JsonOutputKeyToolsParser extends BaseLLMOutputParser {
             writable: true,
             value: false
         });
-        Object.defineProperty(this, "initialParser", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         Object.defineProperty(this, "zodSchema", {
             enumerable: true,
             configurable: true,
@@ -190,7 +215,6 @@ export class JsonOutputKeyToolsParser extends BaseLLMOutputParser {
         });
         this.keyName = params.keyName;
         this.returnSingle = params.returnSingle ?? this.returnSingle;
-        this.initialParser = new JsonOutputToolsParser(params);
         this.zodSchema = params.zodSchema;
     }
     async _validateResult(result) {
@@ -206,11 +230,31 @@ export class JsonOutputKeyToolsParser extends BaseLLMOutputParser {
         }
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async parsePartialResult(generations) {
+        const results = await super.parsePartialResult(generations);
+        const matchingResults = results.filter((result) => result.type === this.keyName);
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        let returnedValues = matchingResults;
+        if (!matchingResults.length) {
+            return undefined;
+        }
+        if (!this.returnId) {
+            returnedValues = matchingResults.map((result) => result.args);
+        }
+        if (this.returnSingle) {
+            return returnedValues[0];
+        }
+        return returnedValues;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async parseResult(generations) {
-        const results = await this.initialParser.parseResult(generations);
+        const results = await super.parsePartialResult(generations, false);
         const matchingResults = results.filter((result) => result.type === this.keyName);
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let returnedValues = matchingResults;
+        if (!matchingResults.length) {
+            return undefined;
+        }
         if (!this.returnId) {
             returnedValues = matchingResults.map((result) => result.args);
         }

package/dist/output_parsers/transform.cjs

@@ -105,5 +105,8 @@ class BaseCumulativeTransformOutputParser extends BaseTransformOutputParser {
             }
         }
     }
+    getFormatInstructions() {
+        return "";
+    }
 }
 exports.BaseCumulativeTransformOutputParser = BaseCumulativeTransformOutputParser;

package/dist/output_parsers/transform.d.ts

@@ -30,4 +30,5 @@ export declare abstract class BaseCumulativeTransformOutputParser<T = unknown> e
     protected abstract _diff(prev: any | undefined, next: any): any;
     abstract parsePartialResult(generations: Generation[] | ChatGeneration[]): Promise<T | undefined>;
     _transform(inputGenerator: AsyncGenerator<string | BaseMessage>): AsyncGenerator<T>;
+    getFormatInstructions(): string;
 }

package/dist/output_parsers/transform.js

@@ -101,4 +101,7 @@ export class BaseCumulativeTransformOutputParser extends BaseTransformOutputPars
             }
         }
     }
+    getFormatInstructions() {
+        return "";
+    }
 }

package/dist/tracers/event_stream.cjs

@@ -279,7 +279,7 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
         if (runInfo.runType === "chat_model") {
             eventName = "on_chat_model_stream";
             if (kwargs?.chunk === undefined) {
-                chunk = new ai_js_1.AIMessageChunk({ content: token });
+                chunk = new ai_js_1.AIMessageChunk({ content: token, id: `run-${run.id}` });
             }
             else {
                 chunk = kwargs.chunk.message;

package/dist/tracers/event_stream.js

@@ -275,7 +275,7 @@ export class EventStreamCallbackHandler extends BaseTracer {
         if (runInfo.runType === "chat_model") {
             eventName = "on_chat_model_stream";
             if (kwargs?.chunk === undefined) {
-                chunk = new AIMessageChunk({ content: token });
+                chunk = new AIMessageChunk({ content: token, id: `run-${run.id}` });
             }
             else {
                 chunk = kwargs.chunk.message;

package/dist/tracers/log_stream.cjs

@@ -401,7 +401,10 @@ class LogStreamCallbackHandler extends base_js_1.BaseTracer {
                 streamedOutputValue = kwargs?.chunk;
             }
             else {
-                streamedOutputValue = new ai_js_1.AIMessageChunk(token);
+                streamedOutputValue = new ai_js_1.AIMessageChunk({
+                    id: `run-${run.id}`,
+                    content: token,
+                });
             }
         }
         else {

package/dist/tracers/log_stream.js

@@ -395,7 +395,10 @@ export class LogStreamCallbackHandler extends BaseTracer {
                 streamedOutputValue = kwargs?.chunk;
             }
             else {
-                streamedOutputValue = new AIMessageChunk(token);
+                streamedOutputValue = new AIMessageChunk({
+                    id: `run-${run.id}`,
+                    content: token,
+                });
             }
         }
         else {

package/dist/utils/stream.cjs

@@ -68,6 +68,11 @@ class IterableReadableStream extends ReadableStream {
     [Symbol.asyncIterator]() {
         return this;
     }
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore Not present in Node 18 types, required in latest Node 22
+    async [Symbol.asyncDispose]() {
+        await this.return();
+    }
     static fromReadableStream(stream) {
         // From https://developer.mozilla.org/en-US/docs/Web/API/Streams_API/Using_readable_streams#reading_the_stream
         const reader = stream.getReader();
@@ -249,6 +254,11 @@ class AsyncGeneratorWithSetup {
     [Symbol.asyncIterator]() {
         return this;
     }
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore Not present in Node 18 types, required in latest Node 22
+    async [Symbol.asyncDispose]() {
+        await this.return();
+    }
 }
 exports.AsyncGeneratorWithSetup = AsyncGeneratorWithSetup;
 async function pipeGeneratorWithSetup(to, generator, startSetup, signal, ...args) {

package/dist/utils/stream.d.ts

@@ -6,6 +6,7 @@ export declare class IterableReadableStream<T> extends ReadableStream<T> impleme
     return(): Promise<IteratorResult<T>>;
     throw(e: any): Promise<IteratorResult<T>>;
     [Symbol.asyncIterator](): this;
+    [Symbol.asyncDispose](): Promise<void>;
     static fromReadableStream<T>(stream: ReadableStream<T>): IterableReadableStream<T>;
     static fromAsyncGenerator<T>(generator: AsyncGenerator<T>): IterableReadableStream<T>;
 }
@@ -25,9 +26,10 @@ export declare class AsyncGeneratorWithSetup<S = unknown, T = unknown, TReturn =
         signal?: AbortSignal;
     });
     next(...args: [] | [TNext]): Promise<IteratorResult<T>>;
-    return(value: TReturn | PromiseLike<TReturn>): Promise<IteratorResult<T>>;
+    return(value?: TReturn | PromiseLike<TReturn>): Promise<IteratorResult<T>>;
     throw(e: Error): Promise<IteratorResult<T>>;
     [Symbol.asyncIterator](): this;
+    [Symbol.asyncDispose](): Promise<void>;
 }
 export declare function pipeGeneratorWithSetup<S, A extends unknown[], T, TReturn, TNext, U, UReturn, UNext>(to: (g: AsyncGenerator<T, TReturn, TNext>, s: S, ...args: A) => AsyncGenerator<U, UReturn, UNext>, generator: AsyncGenerator<T, TReturn, TNext>, startSetup: () => Promise<S>, signal: AbortSignal | undefined, ...args: A): Promise<{
     output: AsyncGenerator<U, UReturn, UNext>;

package/dist/utils/stream.js

@@ -65,6 +65,11 @@ export class IterableReadableStream extends ReadableStream {
     [Symbol.asyncIterator]() {
         return this;
     }
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore Not present in Node 18 types, required in latest Node 22
+    async [Symbol.asyncDispose]() {
+        await this.return();
+    }
     static fromReadableStream(stream) {
         // From https://developer.mozilla.org/en-US/docs/Web/API/Streams_API/Using_readable_streams#reading_the_stream
         const reader = stream.getReader();
@@ -243,6 +248,11 @@ export class AsyncGeneratorWithSetup {
     [Symbol.asyncIterator]() {
         return this;
     }
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore Not present in Node 18 types, required in latest Node 22
+    async [Symbol.asyncDispose]() {
+        await this.return();
+    }
 }
 export async function pipeGeneratorWithSetup(to, generator, startSetup, signal, ...args) {
     const gen = new AsyncGeneratorWithSetup({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/core",
-  "version": "0.2.30",
+  "version": "0.2.32",
   "description": "Core LangChain.js abstractions and schemas",
   "type": "module",
   "engines": {