@langchain/langgraph 0.4.8 → 1.0.0-alpha.0

package/dist/pregel/messages.js

@@ -1,196 +1,107 @@
-import { BaseCallbackHandler, } from "@langchain/core/callbacks/base";
-import { AIMessageChunk, isBaseMessage, isBaseMessageChunk, isToolMessage, } from "@langchain/core/messages";
 import { TAG_HIDDEN, TAG_NOSTREAM } from "../constants.js";
+import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
+import { AIMessageChunk, isBaseMessage, isBaseMessageChunk, isToolMessage } from "@langchain/core/messages";
+
+//#region src/pregel/messages.ts
 function isChatGenerationChunk(x) {
-    return isBaseMessage(x?.message);
+	return isBaseMessage(x?.message);
 }
 /**
- * A callback handler that implements stream_mode=messages.
- * Collects messages from (1) chat model stream events and (2) node outputs.
- */
-// TODO: Make this import and explicitly implement the
-// CallbackHandlerPrefersStreaming interface once we drop support for core 0.2
-export class StreamMessagesHandler extends BaseCallbackHandler {
-    constructor(streamFn) {
-        super();
-        Object.defineProperty(this, "name", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "StreamMessagesHandler"
-        });
-        Object.defineProperty(this, "streamFn", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "metadatas", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {}
-        });
-        Object.defineProperty(this, "seen", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {}
-        });
-        Object.defineProperty(this, "emittedChatModelRunIds", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {}
-        });
-        Object.defineProperty(this, "stableMessageIdMap", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {}
-        });
-        Object.defineProperty(this, "lc_prefer_streaming", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: true
-        });
-        this.streamFn = streamFn;
-    }
-    _emit(meta, message, runId, dedupe = false) {
-        if (dedupe &&
-            message.id !== undefined &&
-            this.seen[message.id] !== undefined) {
-            return;
-        }
-        let messageId = message.id;
-        if (runId != null) {
-            if (isToolMessage(message)) {
-                // Distinguish tool messages by tool call ID.
-                messageId ??= `run-${runId}-tool-${message.tool_call_id}`;
-            }
-            else {
-                // For instance in ChatAnthropic, the first chunk has an message ID
-                // but the subsequent chunks do not. To avoid clients seeing two messages
-                // we rename the message ID if it's being auto-set to `run-${runId}`
-                // (see https://github.com/langchain-ai/langchainjs/pull/6646).
-                if (messageId == null || messageId === `run-${runId}`) {
-                    messageId =
-                        this.stableMessageIdMap[runId] ?? messageId ?? `run-${runId}`;
-                }
-                this.stableMessageIdMap[runId] ??= messageId;
-            }
-        }
-        if (messageId !== message.id) {
-            // eslint-disable-next-line no-param-reassign
-            message.id = messageId;
-            // eslint-disable-next-line no-param-reassign
-            message.lc_kwargs.id = messageId;
-        }
-        if (message.id != null)
-            this.seen[message.id] = message;
-        this.streamFn([meta[0], "messages", [message, meta[1]]]);
-    }
-    handleChatModelStart(_llm, _messages, runId, _parentRunId, _extraParams, tags, metadata, name) {
-        if (metadata &&
-            // Include legacy LangGraph SDK tag
-            (!tags || (!tags.includes(TAG_NOSTREAM) && !tags.includes("nostream")))) {
-            this.metadatas[runId] = [
-                metadata.langgraph_checkpoint_ns.split("|"),
-                { tags, name, ...metadata },
-            ];
-        }
-    }
-    handleLLMNewToken(token, _idx, runId, _parentRunId, _tags, fields) {
-        const chunk = fields?.chunk;
-        this.emittedChatModelRunIds[runId] = true;
-        if (this.metadatas[runId] !== undefined) {
-            if (isChatGenerationChunk(chunk)) {
-                this._emit(this.metadatas[runId], chunk.message, runId);
-            }
-            else {
-                this._emit(this.metadatas[runId], new AIMessageChunk({ content: token }), runId);
-            }
-        }
-    }
-    handleLLMEnd(output, runId) {
-        // Filter out runs that we do not have metadata for
-        if (this.metadatas[runId] === undefined)
-            return;
-        // In JS, non-streaming runs do not call handleLLMNewToken at the model level
-        if (!this.emittedChatModelRunIds[runId]) {
-            const chatGeneration = output.generations?.[0]?.[0];
-            if (isBaseMessage(chatGeneration?.message)) {
-                this._emit(this.metadatas[runId], chatGeneration?.message, runId, true);
-            }
-            delete this.emittedChatModelRunIds[runId];
-        }
-        delete this.metadatas[runId];
-        delete this.stableMessageIdMap[runId];
-    }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    handleLLMError(_err, runId) {
-        delete this.metadatas[runId];
-    }
-    handleChainStart(_chain, inputs, runId, _parentRunId, tags, metadata, _runType, name) {
-        if (metadata !== undefined &&
-            name === metadata.langgraph_node &&
-            (tags === undefined || !tags.includes(TAG_HIDDEN))) {
-            this.metadatas[runId] = [
-                metadata.langgraph_checkpoint_ns.split("|"),
-                { tags, name, ...metadata },
-            ];
-            if (typeof inputs === "object") {
-                for (const value of Object.values(inputs)) {
-                    if ((isBaseMessage(value) || isBaseMessageChunk(value)) &&
-                        value.id !== undefined) {
-                        this.seen[value.id] = value;
-                    }
-                    else if (Array.isArray(value)) {
-                        for (const item of value) {
-                            if ((isBaseMessage(item) || isBaseMessageChunk(item)) &&
-                                item.id !== undefined) {
-                                this.seen[item.id] = item;
-                            }
-                        }
-                    }
-                }
-            }
-        }
-    }
-    handleChainEnd(outputs, runId) {
-        const metadata = this.metadatas[runId];
-        delete this.metadatas[runId];
-        if (metadata !== undefined) {
-            if (isBaseMessage(outputs)) {
-                this._emit(metadata, outputs, runId, true);
-            }
-            else if (Array.isArray(outputs)) {
-                for (const value of outputs) {
-                    if (isBaseMessage(value)) {
-                        this._emit(metadata, value, runId, true);
-                    }
-                }
-            }
-            else if (outputs != null && typeof outputs === "object") {
-                for (const value of Object.values(outputs)) {
-                    if (isBaseMessage(value)) {
-                        this._emit(metadata, value, runId, true);
-                    }
-                    else if (Array.isArray(value)) {
-                        for (const item of value) {
-                            if (isBaseMessage(item)) {
-                                this._emit(metadata, item, runId, true);
-                            }
-                        }
-                    }
-                }
-            }
-        }
-    }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    handleChainError(_err, runId) {
-        delete this.metadatas[runId];
-    }
-}
+* A callback handler that implements stream_mode=messages.
+* Collects messages from (1) chat model stream events and (2) node outputs.
+*/
+var StreamMessagesHandler = class extends BaseCallbackHandler {
+	name = "StreamMessagesHandler";
+	streamFn;
+	metadatas = {};
+	seen = {};
+	emittedChatModelRunIds = {};
+	stableMessageIdMap = {};
+	lc_prefer_streaming = true;
+	constructor(streamFn) {
+		super();
+		this.streamFn = streamFn;
+	}
+	_emit(meta, message, runId, dedupe = false) {
+		if (dedupe && message.id !== void 0 && this.seen[message.id] !== void 0) return;
+		let messageId = message.id;
+		if (runId != null) if (isToolMessage(message)) messageId ??= `run-${runId}-tool-${message.tool_call_id}`;
+		else {
+			if (messageId == null || messageId === `run-${runId}`) messageId = this.stableMessageIdMap[runId] ?? messageId ?? `run-${runId}`;
+			this.stableMessageIdMap[runId] ??= messageId;
+		}
+		if (messageId !== message.id) {
+			message.id = messageId;
+			message.lc_kwargs.id = messageId;
+		}
+		if (message.id != null) this.seen[message.id] = message;
+		this.streamFn([
+			meta[0],
+			"messages",
+			[message, meta[1]]
+		]);
+	}
+	handleChatModelStart(_llm, _messages, runId, _parentRunId, _extraParams, tags, metadata, name) {
+		if (metadata && (!tags || !tags.includes(TAG_NOSTREAM) && !tags.includes("nostream"))) this.metadatas[runId] = [metadata.langgraph_checkpoint_ns.split("|"), {
+			tags,
+			name,
+			...metadata
+		}];
+	}
+	handleLLMNewToken(token, _idx, runId, _parentRunId, _tags, fields) {
+		const chunk = fields?.chunk;
+		this.emittedChatModelRunIds[runId] = true;
+		if (this.metadatas[runId] !== void 0) if (isChatGenerationChunk(chunk)) this._emit(this.metadatas[runId], chunk.message, runId);
+		else this._emit(this.metadatas[runId], new AIMessageChunk({ content: token }), runId);
+	}
+	handleLLMEnd(output, runId) {
+		if (this.metadatas[runId] === void 0) return;
+		if (!this.emittedChatModelRunIds[runId]) {
+			const chatGeneration = output.generations?.[0]?.[0];
+			if (isBaseMessage(chatGeneration?.message)) this._emit(this.metadatas[runId], chatGeneration?.message, runId, true);
+			delete this.emittedChatModelRunIds[runId];
+		}
+		delete this.metadatas[runId];
+		delete this.stableMessageIdMap[runId];
+	}
+	handleLLMError(_err, runId) {
+		delete this.metadatas[runId];
+	}
+	handleChainStart(_chain, inputs, runId, _parentRunId, tags, metadata, _runType, name) {
+		if (metadata !== void 0 && name === metadata.langgraph_node && (tags === void 0 || !tags.includes(TAG_HIDDEN))) {
+			this.metadatas[runId] = [metadata.langgraph_checkpoint_ns.split("|"), {
+				tags,
+				name,
+				...metadata
+			}];
+			if (typeof inputs === "object") {
+				for (const value of Object.values(inputs)) if ((isBaseMessage(value) || isBaseMessageChunk(value)) && value.id !== void 0) this.seen[value.id] = value;
+				else if (Array.isArray(value)) {
+					for (const item of value) if ((isBaseMessage(item) || isBaseMessageChunk(item)) && item.id !== void 0) this.seen[item.id] = item;
+				}
+			}
+		}
+	}
+	handleChainEnd(outputs, runId) {
+		const metadata = this.metadatas[runId];
+		delete this.metadatas[runId];
+		if (metadata !== void 0) {
+			if (isBaseMessage(outputs)) this._emit(metadata, outputs, runId, true);
+			else if (Array.isArray(outputs)) {
+				for (const value of outputs) if (isBaseMessage(value)) this._emit(metadata, value, runId, true);
+			} else if (outputs != null && typeof outputs === "object") {
+				for (const value of Object.values(outputs)) if (isBaseMessage(value)) this._emit(metadata, value, runId, true);
+				else if (Array.isArray(value)) {
+					for (const item of value) if (isBaseMessage(item)) this._emit(metadata, item, runId, true);
+				}
+			}
+		}
+	}
+	handleChainError(_err, runId) {
+		delete this.metadatas[runId];
+	}
+};
+
+//#endregion
+export { StreamMessagesHandler };
 //# sourceMappingURL=messages.js.map

package/dist/pregel/messages.js.map

@@ -1 +1 @@
-{"version":3,"file":"messages.js","sourceRoot":"","sources":["../../src/pregel/messages.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,GAGpB,MAAM,gCAAgC,CAAC;AACxC,OAAO,EACL,cAAc,EAEd,aAAa,EACb,kBAAkB,EAClB,aAAa,GACd,MAAM,0BAA0B,CAAC;AASlC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAM3D,SAAS,qBAAqB,CAAC,CAAU;IACvC,OAAO,aAAa,CAAE,CAAyB,EAAE,OAAO,CAAC,CAAC;AAC5D,CAAC;AAED;;;GAGG;AACH,sDAAsD;AACtD,8EAA8E;AAC9E,MAAM,OAAO,qBAAsB,SAAQ,mBAAmB;IAe5D,YAAY,QAA4C;QACtD,KAAK,EAAE,CAAC;QAfV;;;;mBAAO,uBAAuB;WAAC;QAE/B;;;;;WAA6C;QAE7C;;;;mBAA8C,EAAE;WAAC;QAEjD;;;;mBAAoC,EAAE;WAAC;QAEvC;;;;mBAAkD,EAAE;WAAC;QAErD;;;;mBAA6C,EAAE;WAAC;QAEhD;;;;mBAAsB,IAAI;WAAC;QAIzB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CACH,IAAU,EACV,OAAoB,EACpB,KAAyB,EACzB,MAAM,GAAG,KAAK;QAEd,IACE,MAAM;YACN,OAAO,CAAC,EAAE,KAAK,SAAS;YACxB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,KAAK,SAAS,EACnC,CAAC;YACD,OAAO;QACT,CAAC;QAED,IAAI,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;QAE3B,IAAI,KAAK,IAAI,IAAI,EAAE,CAAC;YAClB,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3B,6CAA6C;gBAC7C,SAAS,KAAK,OAAO,KAAK,SAAS,OAAO,CAAC,YAAY,EAAE,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACN,mEAAmE;gBACnE,yEAAyE;gBACzE,oEAAoE;gBACpE,+DAA+D;gBAC/D,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,KAAK,OAAO,KAAK,EAAE,EAAE,CAAC;oBACtD,SAAS;wBACP,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,SAAS,IAAI,OAAO,KAAK,EAAE,CAAC;gBAClE,CAAC;gBAED,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,KAAK,SAAS,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,IAAI,SAAS,KAAK,OAAO,CAAC,EAAE,EAAE,CAAC;YAC7B,6CAA6C;YAC7C,OAAO,CAAC,EAAE,GAAG,SAAS,CAAC;YAEvB,6CAA6C;YAC7C,OAAO,CAAC,SAAS,CAAC,EAAE,GAAG,SAAS,CAAC;QACnC,CAAC;QAED,IAAI,OAAO,CAAC,EAAE,IAAI,IAAI;YAAE,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC;QACxD,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC;IAED,oBAAoB,CAClB,IAAgB,EAChB,SAA0B,EAC1B,KAAa,EACb,YAAqB,EACrB,YAAsC,EACtC,IAAe,EACf,QAAkC,EAClC,IAAa;QAEb,IACE,QAAQ;YACR,mCAAmC;YACnC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EACvE,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG;gBACrB,QAAQ,CAAC,uBAAkC,CAAC,KAAK,CAAC,GAAG,CAAC;gBACvD,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE;aAC5B,CAAC;QACJ,CAAC;IACH,CAAC;IAED,iBAAiB,CACf,KAAa,EACb,IAAqB,EACrB,KAAa,EACb,YAAqB,EACrB,KAAgB,EAChB,MAAwC;QAExC,MAAM,KAAK,GAAG,MAAM,EAAE,KAAK,CAAC;QAC5B,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;QAC1C,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,SAAS,EAAE,CAAC;YACxC,IAAI,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC,KAAK,CACR,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EACrB,IAAI,cAAc,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,EACtC,KAAK,CACN,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,YAAY,CAAC,MAAiB,EAAE,KAAa;QAC3C,mDAAmD;QACnD,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,SAAS;YAAE,OAAO;QAEhD,6EAA6E;QAC7E,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE,CAAC;YACxC,MAAM,cAAc,GAAG,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAmB,CAAC;YACtE,IAAI,aAAa,CAAC,cAAc,EAAE,OAAO,CAAC,EAAE,CAAC;gBAC3C,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC1E,CAAC;YACD,OAAO,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,OAAO,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAED,8DAA8D;IAC9D,cAAc,CAAC,IAAS,EAAE,KAAa;QACrC,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;IAED,gBAAgB,CACd,MAAkB,EAClB,MAAmB,EACnB,KAAa,EACb,YAAqB,EACrB,IAAe,EACf,QAAkC,EAClC,QAAiB,EACjB,IAAa;QAEb,IACE,QAAQ,KAAK,SAAS;YACtB,IAAI,KAAK,QAAQ,CAAC,cAAc;YAChC,CAAC,IAAI,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,EAClD,CAAC;YACD,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG;gBACrB,QAAQ,CAAC,uBAAkC,CAAC,KAAK,CAAC,GAAG,CAAC;gBACvD,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE;aAC5B,CAAC;YAEF,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC/B,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;oBAC1C,IACE,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,kBAAkB,CAAC,KAAK,CAAC,CAAC;wBACnD,KAAK,CAAC,EAAE,KAAK,SAAS,EACtB,CAAC;wBACD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC;oBAC9B,CAAC;yBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;wBAChC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;4BACzB,IACE,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAC;gCACjD,IAAI,CAAC,EAAE,KAAK,SAAS,EACrB,CAAC;gCACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC;4BAC5B,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,cAAc,CAAC,OAAoB,EAAE,KAAa;QAChD,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACvC,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC7B,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC3B,IAAI,aAAa,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC7C,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;oBAC5B,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;wBACzB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;oBAC3C,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;gBAC1D,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC3C,IAAI,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;wBACzB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;oBAC3C,CAAC;yBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;wBAChC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;4BACzB,IAAI,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;gCACxB,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;4BAC1C,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,8DAA8D;IAC9D,gBAAgB,CAAC,IAAS,EAAE,KAAa;QACvC,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IAC/B,CAAC;CACF"}
+{"version":3,"file":"messages.js","names":[],"sources":["../../src/pregel/messages.ts"],"sourcesContent":["import {\n  BaseCallbackHandler,\n  HandleLLMNewTokenCallbackFields,\n  NewTokenIndices,\n} from \"@langchain/core/callbacks/base\";\nimport {\n  AIMessageChunk,\n  BaseMessage,\n  isBaseMessage,\n  isBaseMessageChunk,\n  isToolMessage,\n} from \"@langchain/core/messages\";\nimport { Serialized } from \"@langchain/core/load/serializable\";\nimport {\n  ChatGeneration,\n  ChatGenerationChunk,\n  LLMResult,\n} from \"@langchain/core/outputs\";\nimport { ChainValues } from \"@langchain/core/utils/types\";\n\nimport { TAG_HIDDEN, TAG_NOSTREAM } from \"../constants.js\";\nimport { StreamChunk } from \"./stream.js\";\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\ntype Meta = [string[], Record<string, any>];\n\nfunction isChatGenerationChunk(x: unknown): x is ChatGenerationChunk {\n  return isBaseMessage((x as ChatGenerationChunk)?.message);\n}\n\n/**\n * A callback handler that implements stream_mode=messages.\n * Collects messages from (1) chat model stream events and (2) node outputs.\n */\n// TODO: Make this import and explicitly implement the\n// CallbackHandlerPrefersStreaming interface once we drop support for core 0.2\nexport class StreamMessagesHandler extends BaseCallbackHandler {\n  name = \"StreamMessagesHandler\";\n\n  streamFn: (streamChunk: StreamChunk) => void;\n\n  metadatas: Record<string, Meta | undefined> = {};\n\n  seen: Record<string, BaseMessage> = {};\n\n  emittedChatModelRunIds: Record<string, boolean> = {};\n\n  stableMessageIdMap: Record<string, string> = {};\n\n  lc_prefer_streaming = true;\n\n  constructor(streamFn: (streamChunk: StreamChunk) => void) {\n    super();\n    this.streamFn = streamFn;\n  }\n\n  _emit(\n    meta: Meta,\n    message: BaseMessage,\n    runId: string | undefined,\n    dedupe = false\n  ) {\n    if (\n      dedupe &&\n      message.id !== undefined &&\n      this.seen[message.id] !== undefined\n    ) {\n      return;\n    }\n\n    let messageId = message.id;\n\n    if (runId != null) {\n      if (isToolMessage(message)) {\n        // Distinguish tool messages by tool call ID.\n        messageId ??= `run-${runId}-tool-${message.tool_call_id}`;\n      } else {\n        // For instance in ChatAnthropic, the first chunk has an message ID\n        // but the subsequent chunks do not. To avoid clients seeing two messages\n        // we rename the message ID if it's being auto-set to `run-${runId}`\n        // (see https://github.com/langchain-ai/langchainjs/pull/6646).\n        if (messageId == null || messageId === `run-${runId}`) {\n          messageId =\n            this.stableMessageIdMap[runId] ?? messageId ?? `run-${runId}`;\n        }\n\n        this.stableMessageIdMap[runId] ??= messageId;\n      }\n    }\n\n    if (messageId !== message.id) {\n      // eslint-disable-next-line no-param-reassign\n      message.id = messageId;\n\n      // eslint-disable-next-line no-param-reassign\n      message.lc_kwargs.id = messageId;\n    }\n\n    if (message.id != null) this.seen[message.id] = message;\n    this.streamFn([meta[0], \"messages\", [message, meta[1]]]);\n  }\n\n  handleChatModelStart(\n    _llm: Serialized,\n    _messages: BaseMessage[][],\n    runId: string,\n    _parentRunId?: string,\n    _extraParams?: Record<string, unknown>,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    name?: string\n  ) {\n    if (\n      metadata &&\n      // Include legacy LangGraph SDK tag\n      (!tags || (!tags.includes(TAG_NOSTREAM) && !tags.includes(\"nostream\")))\n    ) {\n      this.metadatas[runId] = [\n        (metadata.langgraph_checkpoint_ns as string).split(\"|\"),\n        { tags, name, ...metadata },\n      ];\n    }\n  }\n\n  handleLLMNewToken(\n    token: string,\n    _idx: NewTokenIndices,\n    runId: string,\n    _parentRunId?: string,\n    _tags?: string[],\n    fields?: HandleLLMNewTokenCallbackFields\n  ) {\n    const chunk = fields?.chunk;\n    this.emittedChatModelRunIds[runId] = true;\n    if (this.metadatas[runId] !== undefined) {\n      if (isChatGenerationChunk(chunk)) {\n        this._emit(this.metadatas[runId], chunk.message, runId);\n      } else {\n        this._emit(\n          this.metadatas[runId],\n          new AIMessageChunk({ content: token }),\n          runId\n        );\n      }\n    }\n  }\n\n  handleLLMEnd(output: LLMResult, runId: string) {\n    // Filter out runs that we do not have metadata for\n    if (this.metadatas[runId] === undefined) return;\n\n    // In JS, non-streaming runs do not call handleLLMNewToken at the model level\n    if (!this.emittedChatModelRunIds[runId]) {\n      const chatGeneration = output.generations?.[0]?.[0] as ChatGeneration;\n      if (isBaseMessage(chatGeneration?.message)) {\n        this._emit(this.metadatas[runId], chatGeneration?.message, runId, true);\n      }\n      delete this.emittedChatModelRunIds[runId];\n    }\n    delete this.metadatas[runId];\n    delete this.stableMessageIdMap[runId];\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  handleLLMError(_err: any, runId: string) {\n    delete this.metadatas[runId];\n  }\n\n  handleChainStart(\n    _chain: Serialized,\n    inputs: ChainValues,\n    runId: string,\n    _parentRunId?: string,\n    tags?: string[],\n    metadata?: Record<string, unknown>,\n    _runType?: string,\n    name?: string\n  ) {\n    if (\n      metadata !== undefined &&\n      name === metadata.langgraph_node &&\n      (tags === undefined || !tags.includes(TAG_HIDDEN))\n    ) {\n      this.metadatas[runId] = [\n        (metadata.langgraph_checkpoint_ns as string).split(\"|\"),\n        { tags, name, ...metadata },\n      ];\n\n      if (typeof inputs === \"object\") {\n        for (const value of Object.values(inputs)) {\n          if (\n            (isBaseMessage(value) || isBaseMessageChunk(value)) &&\n            value.id !== undefined\n          ) {\n            this.seen[value.id] = value;\n          } else if (Array.isArray(value)) {\n            for (const item of value) {\n              if (\n                (isBaseMessage(item) || isBaseMessageChunk(item)) &&\n                item.id !== undefined\n              ) {\n                this.seen[item.id] = item;\n              }\n            }\n          }\n        }\n      }\n    }\n  }\n\n  handleChainEnd(outputs: ChainValues, runId: string) {\n    const metadata = this.metadatas[runId];\n    delete this.metadatas[runId];\n    if (metadata !== undefined) {\n      if (isBaseMessage(outputs)) {\n        this._emit(metadata, outputs, runId, true);\n      } else if (Array.isArray(outputs)) {\n        for (const value of outputs) {\n          if (isBaseMessage(value)) {\n            this._emit(metadata, value, runId, true);\n          }\n        }\n      } else if (outputs != null && typeof outputs === \"object\") {\n        for (const value of Object.values(outputs)) {\n          if (isBaseMessage(value)) {\n            this._emit(metadata, value, runId, true);\n          } else if (Array.isArray(value)) {\n            for (const item of value) {\n              if (isBaseMessage(item)) {\n                this._emit(metadata, item, runId, true);\n              }\n            }\n          }\n        }\n      }\n    }\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-explicit-any\n  handleChainError(_err: any, runId: string) {\n    delete this.metadatas[runId];\n  }\n}\n"],"mappings":";;;;;AA0BA,SAAS,sBAAsB,GAAsC;AACnE,QAAO,cAAe,GAA2B;;;;;;AASnD,IAAa,wBAAb,cAA2C,oBAAoB;CAC7D,OAAO;CAEP;CAEA,YAA8C;CAE9C,OAAoC;CAEpC,yBAAkD;CAElD,qBAA6C;CAE7C,sBAAsB;CAEtB,YAAY,UAA8C;AACxD;AACA,OAAK,WAAW;;CAGlB,MACE,MACA,SACA,OACA,SAAS,OACT;AACA,MACE,UACA,QAAQ,OAAO,UACf,KAAK,KAAK,QAAQ,QAAQ,OAE1B;EAGF,IAAI,YAAY,QAAQ;AAExB,MAAI,SAAS,KACX,KAAI,cAAc,SAEhB,eAAc,OAAO,MAAM,QAAQ,QAAQ;OACtC;AAKL,OAAI,aAAa,QAAQ,cAAc,OAAO,QAC5C,aACE,KAAK,mBAAmB,UAAU,aAAa,OAAO;AAG1D,QAAK,mBAAmB,WAAW;;AAIvC,MAAI,cAAc,QAAQ,IAAI;AAE5B,WAAQ,KAAK;AAGb,WAAQ,UAAU,KAAK;;AAGzB,MAAI,QAAQ,MAAM,KAAM,MAAK,KAAK,QAAQ,MAAM;AAChD,OAAK,SAAS;GAAC,KAAK;GAAI;GAAY,CAAC,SAAS,KAAK;;;CAGrD,qBACE,MACA,WACA,OACA,cACA,cACA,MACA,UACA,MACA;AACA,MACE,aAEC,CAAC,QAAS,CAAC,KAAK,SAAS,iBAAiB,CAAC,KAAK,SAAS,aAE1D,MAAK,UAAU,SAAS,CACrB,SAAS,wBAAmC,MAAM,MACnD;GAAE;GAAM;GAAM,GAAG;;;CAKvB,kBACE,OACA,MACA,OACA,cACA,OACA,QACA;EACA,MAAM,QAAQ,QAAQ;AACtB,OAAK,uBAAuB,SAAS;AACrC,MAAI,KAAK,UAAU,WAAW,OAC5B,KAAI,sBAAsB,OACxB,MAAK,MAAM,KAAK,UAAU,QAAQ,MAAM,SAAS;MAEjD,MAAK,MACH,KAAK,UAAU,QACf,IAAI,eAAe,EAAE,SAAS,UAC9B;;CAMR,aAAa,QAAmB,OAAe;AAE7C,MAAI,KAAK,UAAU,WAAW,OAAW;AAGzC,MAAI,CAAC,KAAK,uBAAuB,QAAQ;GACvC,MAAM,iBAAiB,OAAO,cAAc,KAAK;AACjD,OAAI,cAAc,gBAAgB,SAChC,MAAK,MAAM,KAAK,UAAU,QAAQ,gBAAgB,SAAS,OAAO;AAEpE,UAAO,KAAK,uBAAuB;;AAErC,SAAO,KAAK,UAAU;AACtB,SAAO,KAAK,mBAAmB;;CAIjC,eAAe,MAAW,OAAe;AACvC,SAAO,KAAK,UAAU;;CAGxB,iBACE,QACA,QACA,OACA,cACA,MACA,UACA,UACA,MACA;AACA,MACE,aAAa,UACb,SAAS,SAAS,mBACjB,SAAS,UAAa,CAAC,KAAK,SAAS,cACtC;AACA,QAAK,UAAU,SAAS,CACrB,SAAS,wBAAmC,MAAM,MACnD;IAAE;IAAM;IAAM,GAAG;;AAGnB,OAAI,OAAO,WAAW,UACpB;SAAK,MAAM,SAAS,OAAO,OAAO,QAChC,MACG,cAAc,UAAU,mBAAmB,WAC5C,MAAM,OAAO,OAEb,MAAK,KAAK,MAAM,MAAM;aACb,MAAM,QAAQ,QACvB;UAAK,MAAM,QAAQ,MACjB,MACG,cAAc,SAAS,mBAAmB,UAC3C,KAAK,OAAO,OAEZ,MAAK,KAAK,KAAK,MAAM;;;;;CASnC,eAAe,SAAsB,OAAe;EAClD,MAAM,WAAW,KAAK,UAAU;AAChC,SAAO,KAAK,UAAU;AACtB,MAAI,aAAa,QACf;OAAI,cAAc,SAChB,MAAK,MAAM,UAAU,SAAS,OAAO;YAC5B,MAAM,QAAQ,UACvB;SAAK,MAAM,SAAS,QAClB,KAAI,cAAc,OAChB,MAAK,MAAM,UAAU,OAAO,OAAO;cAG9B,WAAW,QAAQ,OAAO,YAAY,UAC/C;SAAK,MAAM,SAAS,OAAO,OAAO,SAChC,KAAI,cAAc,OAChB,MAAK,MAAM,UAAU,OAAO,OAAO;aAC1B,MAAM,QAAQ,QACvB;UAAK,MAAM,QAAQ,MACjB,KAAI,cAAc,MAChB,MAAK,MAAM,UAAU,MAAM,OAAO;;;;;CAUhD,iBAAiB,MAAW,OAAe;AACzC,SAAO,KAAK,UAAU"}