UNPKG

@langchain/langgraph-sdk

Version:

Client library for interacting with the LangGraph API

70 lines (68 loc) • 2.47 kB
const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs'); let _langchain_core_messages = require("@langchain/core/messages"); //#region src/ui/messages.ts function tryConvertToChunk(message) { try { if ((0, _langchain_core_messages.isBaseMessageChunk)(message)) return message; return (0, _langchain_core_messages.convertToChunk)(message); } catch { return null; } } function tryCoerceMessageLikeToMessage(message) { if (message.type === "human" || message.type === "user") return new _langchain_core_messages.HumanMessageChunk(message); if (message.type === "ai" || message.type === "assistant") return new _langchain_core_messages.AIMessageChunk(message); if (message.type === "system") return new _langchain_core_messages.SystemMessageChunk(message); if (message.type === "tool" && "tool_call_id" in message) return new _langchain_core_messages.ToolMessageChunk({ ...message, tool_call_id: message.tool_call_id }); if (message.type === "remove" && message.id != null) return new _langchain_core_messages.RemoveMessage({ ...message, id: message.id }); return (0, _langchain_core_messages.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 = ((0, _langchain_core_messages.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 exports.MessageTupleManager = MessageTupleManager; exports.toMessageDict = toMessageDict; //# sourceMappingURL=messages.cjs.map