@langchain/langgraph-sdk/dist/ui/messages.js

Client library for interacting with the LangGraph API

import { AIMessageChunk, HumanMessageChunk, RemoveMessage, SystemMessageChunk, ToolMessageChunk, coerceMessageLikeToMessage, convertToChunk, isBaseMessageChunk } from "@langchain/core/messages";

//#region src/ui/messages.ts
function tryConvertToChunk(message) {
	try {
		if (isBaseMessageChunk(message)) return message;
		return convertToChunk(message);
	} catch {
		return null;
	}
}
function tryCoerceMessageLikeToMessage(message) {
	if (message.type === "human" || message.type === "user") return new HumanMessageChunk(message);
	if (message.type === "ai" || message.type === "assistant") return new AIMessageChunk(message);
	if (message.type === "system") return new SystemMessageChunk(message);
	if (message.type === "tool" && "tool_call_id" in message) return new ToolMessageChunk({
		...message,
		tool_call_id: message.tool_call_id
	});
	if (message.type === "remove" && message.id != null) return new RemoveMessage({
		...message,
		id: message.id
	});
	return coerceMessageLikeToMessage(message);
}
var MessageTupleManager = class {
	chunks = {};
	constructor() {
		this.chunks = {};
	}
	add(serialized, metadata) {
		if (serialized.type.endsWith("MessageChunk")) serialized.type = serialized.type.slice(0, -12).toLowerCase();
		const message = tryCoerceMessageLikeToMessage(serialized);
		const chunk = tryConvertToChunk(message);
		const { id } = chunk ?? message;
		if (!id) {
			console.warn("No message ID found for chunk, ignoring in state", serialized);
			return null;
		}
		this.chunks[id] ??= {};
		this.chunks[id].metadata = metadata ?? this.chunks[id].metadata;
		if (chunk) {
			const prev = this.chunks[id].chunk;
			this.chunks[id].chunk = (isBaseMessageChunk(prev) ? prev : null)?.concat(chunk) ?? chunk;
		} else this.chunks[id].chunk = message;
		return id;
	}
	clear() {
		this.chunks = {};
	}
	get(id, defaultIndex) {
		if (id == null) return null;
		if (this.chunks[id] == null) return null;
		if (defaultIndex != null) this.chunks[id].index ??= defaultIndex;
		return this.chunks[id];
	}
};
const toMessageDict = (chunk) => {
	const { type, data } = chunk.toDict();
	return {
		...data,
		type
	};
};

//#endregion
export { MessageTupleManager, toMessageDict };
//# sourceMappingURL=messages.js.map