illuma-agents 1.0.2

package/dist/cjs/messages/format.cjs

@@ -0,0 +1,455 @@
+'use strict';
+
+var messages = require('@langchain/core/messages');
+var _enum = require('../common/enum.cjs');
+
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/**
+ * Formats a message with media content (images, documents, videos, audios) to API payload format.
+ *
+ * @param params - The parameters for formatting.
+ * @returns - The formatted message.
+ */
+const formatMediaMessage = ({ message, endpoint, mediaParts, }) => {
+    // Create a new object to avoid mutating the input
+    const result = {
+        ...message,
+        content: [],
+    };
+    if (endpoint === _enum.Providers.ANTHROPIC) {
+        result.content = [
+            ...mediaParts,
+            { type: _enum.ContentTypes.TEXT, text: message.content },
+        ];
+        return result;
+    }
+    result.content = [
+        { type: _enum.ContentTypes.TEXT, text: message.content },
+        ...mediaParts,
+    ];
+    return result;
+};
+/**
+ * Formats a message to OpenAI payload format based on the provided options.
+ *
+ * @param params - The parameters for formatting.
+ * @returns - The formatted message.
+ */
+const formatMessage = ({ message, userName, endpoint, assistantName, langChain = false, }) => {
+    // eslint-disable-next-line prefer-const
+    let { role: _role, _name, sender, text, content: _content, lc_id } = message;
+    if (lc_id && lc_id[2] && !langChain) {
+        const roleMapping = {
+            SystemMessage: 'system',
+            HumanMessage: 'user',
+            AIMessage: 'assistant',
+        };
+        _role = roleMapping[lc_id[2]] || _role;
+    }
+    const role = _role ??
+        (sender != null && sender && sender.toLowerCase() === 'user'
+            ? 'user'
+            : 'assistant');
+    const content = _content ?? text ?? '';
+    const formattedMessage = {
+        role,
+        content,
+    };
+    // Set name fields first
+    if (_name != null && _name) {
+        formattedMessage.name = _name;
+    }
+    if (userName != null && userName && formattedMessage.role === 'user') {
+        formattedMessage.name = userName;
+    }
+    if (assistantName != null &&
+        assistantName &&
+        formattedMessage.role === 'assistant') {
+        formattedMessage.name = assistantName;
+    }
+    if (formattedMessage.name != null && formattedMessage.name) {
+        // Conform to API regex: ^[a-zA-Z0-9_-]{1,64}$
+        // https://community.openai.com/t/the-format-of-the-name-field-in-the-documentation-is-incorrect/175684/2
+        formattedMessage.name = formattedMessage.name.replace(/[^a-zA-Z0-9_-]/g, '_');
+        if (formattedMessage.name.length > 64) {
+            formattedMessage.name = formattedMessage.name.substring(0, 64);
+        }
+    }
+    const { image_urls, documents, videos, audios } = message;
+    const mediaParts = [];
+    if (Array.isArray(documents) && documents.length > 0) {
+        mediaParts.push(...documents);
+    }
+    if (Array.isArray(videos) && videos.length > 0) {
+        mediaParts.push(...videos);
+    }
+    if (Array.isArray(audios) && audios.length > 0) {
+        mediaParts.push(...audios);
+    }
+    if (Array.isArray(image_urls) && image_urls.length > 0) {
+        mediaParts.push(...image_urls);
+    }
+    if (mediaParts.length > 0 && role === 'user') {
+        const mediaMessage = formatMediaMessage({
+            message: {
+                ...formattedMessage,
+                content: typeof formattedMessage.content === 'string'
+                    ? formattedMessage.content
+                    : '',
+            },
+            mediaParts,
+            endpoint,
+        });
+        if (!langChain) {
+            return mediaMessage;
+        }
+        return new messages.HumanMessage(mediaMessage);
+    }
+    if (!langChain) {
+        return formattedMessage;
+    }
+    if (role === 'user') {
+        return new messages.HumanMessage(formattedMessage);
+    }
+    else if (role === 'assistant') {
+        return new messages.AIMessage(formattedMessage);
+    }
+    else {
+        return new messages.SystemMessage(formattedMessage);
+    }
+};
+/**
+ * Formats an array of messages for LangChain.
+ *
+ * @param messages - The array of messages to format.
+ * @param formatOptions - The options for formatting each message.
+ * @returns - The array of formatted LangChain messages.
+ */
+const formatLangChainMessages = (messages, formatOptions) => {
+    return messages.map((msg) => {
+        const formatted = formatMessage({
+            ...formatOptions,
+            message: msg,
+            langChain: true,
+        });
+        return formatted;
+    });
+};
+/**
+ * Formats a LangChain message object by merging properties from `lc_kwargs` or `kwargs` and `additional_kwargs`.
+ *
+ * @param message - The message object to format.
+ * @returns - The formatted LangChain message.
+ */
+const formatFromLangChain = (message) => {
+    const kwargs = message.lc_kwargs ?? message.kwargs ?? {};
+    const { additional_kwargs = {}, ...message_kwargs } = kwargs;
+    return {
+        ...message_kwargs,
+        ...additional_kwargs,
+    };
+};
+/**
+ * Helper function to format an assistant message
+ * @param message The message to format
+ * @returns Array of formatted messages
+ */
+function formatAssistantMessage(message) {
+    const formattedMessages = [];
+    let currentContent = [];
+    let lastAIMessage = null;
+    let hasReasoning = false;
+    if (Array.isArray(message.content)) {
+        for (const part of message.content) {
+            if (part.type === _enum.ContentTypes.TEXT && part.tool_call_ids) {
+                /*
+                If there's pending content, it needs to be aggregated as a single string to prepare for tool calls.
+                For Anthropic models, the "tool_calls" field on a message is only respected if content is a string.
+                */
+                if (currentContent.length > 0) {
+                    let content = currentContent.reduce((acc, curr) => {
+                        if (curr.type === _enum.ContentTypes.TEXT) {
+                            return `${acc}${curr[_enum.ContentTypes.TEXT] || ''}\n`;
+                        }
+                        return acc;
+                    }, '');
+                    content =
+                        `${content}\n${part[_enum.ContentTypes.TEXT] ?? part.text ?? ''}`.trim();
+                    lastAIMessage = new messages.AIMessage({ content });
+                    formattedMessages.push(lastAIMessage);
+                    currentContent = [];
+                    continue;
+                }
+                // Create a new AIMessage with this text and prepare for tool calls
+                lastAIMessage = new messages.AIMessage({
+                    content: part.text || '',
+                });
+                formattedMessages.push(lastAIMessage);
+            }
+            else if (part.type === _enum.ContentTypes.TOOL_CALL) {
+                if (!lastAIMessage) {
+                    // "Heal" the payload by creating an AIMessage to precede the tool call
+                    lastAIMessage = new messages.AIMessage({ content: '' });
+                    formattedMessages.push(lastAIMessage);
+                }
+                // Note: `tool_calls` list is defined when constructed by `AIMessage` class, and outputs should be excluded from it
+                const { output, args: _args, ..._tool_call } = part.tool_call;
+                const tool_call = _tool_call;
+                // TODO: investigate; args as dictionary may need to be providers-or-tool-specific
+                let args = _args;
+                try {
+                    if (typeof _args === 'string') {
+                        args = JSON.parse(_args);
+                    }
+                }
+                catch {
+                    if (typeof _args === 'string') {
+                        args = { input: _args };
+                    }
+                }
+                tool_call.args = args;
+                if (!lastAIMessage.tool_calls) {
+                    lastAIMessage.tool_calls = [];
+                }
+                lastAIMessage.tool_calls.push(tool_call);
+                formattedMessages.push(new messages.ToolMessage({
+                    tool_call_id: tool_call.id ?? '',
+                    name: tool_call.name,
+                    content: output || '',
+                }));
+            }
+            else if (part.type === _enum.ContentTypes.THINK) {
+                hasReasoning = true;
+                continue;
+            }
+            else if (part.type === _enum.ContentTypes.ERROR ||
+                part.type === _enum.ContentTypes.AGENT_UPDATE) {
+                continue;
+            }
+            else {
+                currentContent.push(part);
+            }
+        }
+    }
+    if (hasReasoning && currentContent.length > 0) {
+        const content = currentContent
+            .reduce((acc, curr) => {
+            if (curr.type === _enum.ContentTypes.TEXT) {
+                return `${acc}${curr[_enum.ContentTypes.TEXT] || ''}\n`;
+            }
+            return acc;
+        }, '')
+            .trim();
+        if (content) {
+            formattedMessages.push(new messages.AIMessage({ content }));
+        }
+    }
+    else if (currentContent.length > 0) {
+        formattedMessages.push(new messages.AIMessage({ content: currentContent }));
+    }
+    return formattedMessages;
+}
+/**
+ * Formats an array of messages for LangChain, handling tool calls and creating ToolMessage instances.
+ *
+ * @param payload - The array of messages to format.
+ * @param indexTokenCountMap - Optional map of message indices to token counts.
+ * @param tools - Optional set of tool names that are allowed in the request.
+ * @returns - Object containing formatted messages and updated indexTokenCountMap if provided.
+ */
+const formatAgentMessages = (payload, indexTokenCountMap, tools) => {
+    const messages$1 = [];
+    // If indexTokenCountMap is provided, create a new map to track the updated indices
+    const updatedIndexTokenCountMap = {};
+    // Keep track of the mapping from original payload indices to result indices
+    const indexMapping = {};
+    // Process messages with tool conversion if tools set is provided
+    for (let i = 0; i < payload.length; i++) {
+        const message = payload[i];
+        // Q: Store the current length of messages to track where this payload message starts in the result?
+        // const startIndex = messages.length;
+        if (typeof message.content === 'string') {
+            message.content = [
+                { type: _enum.ContentTypes.TEXT, [_enum.ContentTypes.TEXT]: message.content },
+            ];
+        }
+        if (message.role !== 'assistant') {
+            messages$1.push(formatMessage({
+                message: message,
+                langChain: true,
+            }));
+            // Update the index mapping for this message
+            indexMapping[i] = [messages$1.length - 1];
+            continue;
+        }
+        // For assistant messages, track the starting index before processing
+        const startMessageIndex = messages$1.length;
+        // If tools set is provided, we need to check if we need to convert tool messages to a string
+        if (tools) {
+            // First, check if this message contains tool calls
+            let hasToolCalls = false;
+            let hasInvalidTool = false;
+            const content = message.content;
+            if (content && Array.isArray(content)) {
+                for (const part of content) {
+                    if (part.type === _enum.ContentTypes.TOOL_CALL) {
+                        hasToolCalls = true;
+                        if (tools.size === 0) {
+                            hasInvalidTool = true;
+                            break;
+                        }
+                        const toolName = part.tool_call.name;
+                        if (!tools.has(toolName)) {
+                            hasInvalidTool = true;
+                        }
+                    }
+                }
+            }
+            // If this message has tool calls and at least one is invalid, we need to convert it
+            if (hasToolCalls && hasInvalidTool) {
+                // We need to collect all related messages (this message and any subsequent tool messages)
+                const toolSequence = [];
+                let sequenceEndIndex = i;
+                // Process the current assistant message to get the AIMessage with tool calls
+                const formattedMessages = formatAssistantMessage(message);
+                toolSequence.push(...formattedMessages);
+                // Look ahead for any subsequent assistant messages that might be part of this tool sequence
+                let j = i + 1;
+                while (j < payload.length && payload[j].role === 'assistant') {
+                    // Check if this is a continuation of the tool sequence
+                    let isToolResponse = false;
+                    const content = payload[j].content;
+                    if (content && Array.isArray(content)) {
+                        for (const part of content) {
+                            if (part.type === _enum.ContentTypes.TOOL_CALL) {
+                                isToolResponse = true;
+                                break;
+                            }
+                        }
+                    }
+                    if (isToolResponse) {
+                        // This is part of the tool sequence, add it
+                        const nextMessages = formatAssistantMessage(payload[j]);
+                        toolSequence.push(...nextMessages);
+                        sequenceEndIndex = j;
+                        j++;
+                    }
+                    else {
+                        // This is not part of the tool sequence, stop looking
+                        break;
+                    }
+                }
+                // Convert the sequence to a string
+                const bufferString = messages.getBufferString(toolSequence);
+                messages$1.push(new messages.AIMessage({ content: bufferString }));
+                // Skip the messages we've already processed
+                i = sequenceEndIndex;
+                // Update the index mapping for this sequence
+                const resultIndices = [messages$1.length - 1];
+                for (let k = i; k >= i && k <= sequenceEndIndex; k++) {
+                    indexMapping[k] = resultIndices;
+                }
+                continue;
+            }
+        }
+        // Process the assistant message using the helper function
+        const formattedMessages = formatAssistantMessage(message);
+        messages$1.push(...formattedMessages);
+        // Update the index mapping for this assistant message
+        // Store all indices that were created from this original message
+        const endMessageIndex = messages$1.length;
+        const resultIndices = [];
+        for (let j = startMessageIndex; j < endMessageIndex; j++) {
+            resultIndices.push(j);
+        }
+        indexMapping[i] = resultIndices;
+    }
+    // Update the token count map if it was provided
+    if (indexTokenCountMap) {
+        for (let originalIndex = 0; originalIndex < payload.length; originalIndex++) {
+            const resultIndices = indexMapping[originalIndex] || [];
+            const tokenCount = indexTokenCountMap[originalIndex];
+            if (tokenCount !== undefined) {
+                if (resultIndices.length === 1) {
+                    // Simple 1:1 mapping
+                    updatedIndexTokenCountMap[resultIndices[0]] = tokenCount;
+                }
+                else if (resultIndices.length > 1) {
+                    // If one message was split into multiple, distribute the token count
+                    // This is a simplification - in reality, you might want a more sophisticated distribution
+                    const countPerMessage = Math.floor(tokenCount / resultIndices.length);
+                    resultIndices.forEach((resultIndex, idx) => {
+                        if (idx === resultIndices.length - 1) {
+                            // Give any remainder to the last message
+                            updatedIndexTokenCountMap[resultIndex] =
+                                tokenCount - countPerMessage * (resultIndices.length - 1);
+                        }
+                        else {
+                            updatedIndexTokenCountMap[resultIndex] = countPerMessage;
+                        }
+                    });
+                }
+            }
+        }
+    }
+    return {
+        messages: messages$1,
+        indexTokenCountMap: indexTokenCountMap
+            ? updatedIndexTokenCountMap
+            : undefined,
+    };
+};
+/**
+ * Formats an array of messages for LangChain, making sure all content fields are strings
+ * @param payload - The array of messages to format.
+ * @returns - The array of formatted LangChain messages, including ToolMessages for tool calls.
+ */
+const formatContentStrings = (payload) => {
+    // Create a copy of the payload to avoid modifying the original
+    const result = [...payload];
+    for (const message of result) {
+        if (typeof message.content === 'string') {
+            continue;
+        }
+        if (!Array.isArray(message.content)) {
+            continue;
+        }
+        // Reduce text types to a single string, ignore all other types
+        const content = message.content.reduce((acc, curr) => {
+            if (curr.type === _enum.ContentTypes.TEXT) {
+                return `${acc}${curr[_enum.ContentTypes.TEXT] || ''}\n`;
+            }
+            return acc;
+        }, '');
+        message.content = content.trim();
+    }
+    return result;
+};
+/**
+ * Adds a value at key 0 for system messages and shifts all key indices by one in an indexTokenCountMap.
+ * This is useful when adding a system message at the beginning of a conversation.
+ *
+ * @param indexTokenCountMap - The original map of message indices to token counts
+ * @param instructionsTokenCount - The token count for the system message to add at index 0
+ * @returns A new map with the system message at index 0 and all other indices shifted by 1
+ */
+function shiftIndexTokenCountMap(indexTokenCountMap, instructionsTokenCount) {
+    // Create a new map to avoid modifying the original
+    const shiftedMap = {};
+    shiftedMap[0] = instructionsTokenCount;
+    // Shift all existing indices by 1
+    for (const [indexStr, tokenCount] of Object.entries(indexTokenCountMap)) {
+        const index = Number(indexStr);
+        shiftedMap[index + 1] = tokenCount;
+    }
+    return shiftedMap;
+}
+
+exports.formatAgentMessages = formatAgentMessages;
+exports.formatContentStrings = formatContentStrings;
+exports.formatFromLangChain = formatFromLangChain;
+exports.formatLangChainMessages = formatLangChainMessages;
+exports.formatMediaMessage = formatMediaMessage;
+exports.formatMessage = formatMessage;
+exports.shiftIndexTokenCountMap = shiftIndexTokenCountMap;
+//# sourceMappingURL=format.cjs.map

package/dist/cjs/messages/format.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"format.cjs","sources":["../../../src/messages/format.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\r\nimport {\r\n  AIMessage,\r\n  ToolMessage,\r\n  BaseMessage,\r\n  HumanMessage,\r\n  SystemMessage,\r\n  getBufferString,\r\n} from '@langchain/core/messages';\r\nimport type { MessageContentImageUrl } from '@langchain/core/messages';\r\nimport type { ToolCall } from '@langchain/core/messages/tool';\r\nimport type {\r\n  MessageContentComplex,\r\n  ToolCallPart,\r\n  TPayload,\r\n  TMessage,\r\n} from '@/types';\r\nimport { Providers, ContentTypes } from '@/common';\r\n\r\ninterface MediaMessageParams {\r\n  message: {\r\n    role: string;\r\n    content: string;\r\n    name?: string;\r\n    [key: string]: any;\r\n  };\r\n  mediaParts: MessageContentComplex[];\r\n  endpoint?: Providers;\r\n}\r\n\r\n/**\r\n * Formats a message with media content (images, documents, videos, audios) to API payload format.\r\n *\r\n * @param params - The parameters for formatting.\r\n * @returns - The formatted message.\r\n */\r\nexport const formatMediaMessage = ({\r\n  message,\r\n  endpoint,\r\n  mediaParts,\r\n}: MediaMessageParams): {\r\n  role: string;\r\n  content: MessageContentComplex[];\r\n  name?: string;\r\n  [key: string]: any;\r\n} => {\r\n  // Create a new object to avoid mutating the input\r\n  const result: {\r\n    role: string;\r\n    content: MessageContentComplex[];\r\n    name?: string;\r\n    [key: string]: any;\r\n  } = {\r\n    ...message,\r\n    content: [] as MessageContentComplex[],\r\n  };\r\n\r\n  if (endpoint === Providers.ANTHROPIC) {\r\n    result.content = [\r\n      ...mediaParts,\r\n      { type: ContentTypes.TEXT, text: message.content },\r\n    ] as MessageContentComplex[];\r\n    return result;\r\n  }\r\n\r\n  result.content = [\r\n    { type: ContentTypes.TEXT, text: message.content },\r\n    ...mediaParts,\r\n  ] as MessageContentComplex[];\r\n\r\n  return result;\r\n};\r\n\r\ninterface MessageInput {\r\n  role?: string;\r\n  _name?: string;\r\n  sender?: string;\r\n  text?: string;\r\n  content?: string | MessageContentComplex[];\r\n  image_urls?: MessageContentImageUrl[];\r\n  documents?: MessageContentComplex[];\r\n  videos?: MessageContentComplex[];\r\n  audios?: MessageContentComplex[];\r\n  lc_id?: string[];\r\n  [key: string]: any;\r\n}\r\n\r\ninterface FormatMessageParams {\r\n  message: MessageInput;\r\n  userName?: string;\r\n  assistantName?: string;\r\n  endpoint?: Providers;\r\n  langChain?: boolean;\r\n}\r\n\r\ninterface FormattedMessage {\r\n  role: string;\r\n  content: string | MessageContentComplex[];\r\n  name?: string;\r\n  [key: string]: any;\r\n}\r\n\r\n/**\r\n * Formats a message to OpenAI payload format based on the provided options.\r\n *\r\n * @param params - The parameters for formatting.\r\n * @returns - The formatted message.\r\n */\r\nexport const formatMessage = ({\r\n  message,\r\n  userName,\r\n  endpoint,\r\n  assistantName,\r\n  langChain = false,\r\n}: FormatMessageParams):\r\n  | FormattedMessage\r\n  | HumanMessage\r\n  | AIMessage\r\n  | SystemMessage => {\r\n  // eslint-disable-next-line prefer-const\r\n  let { role: _role, _name, sender, text, content: _content, lc_id } = message;\r\n  if (lc_id && lc_id[2] && !langChain) {\r\n    const roleMapping: Record<string, string> = {\r\n      SystemMessage: 'system',\r\n      HumanMessage: 'user',\r\n      AIMessage: 'assistant',\r\n    };\r\n    _role = roleMapping[lc_id[2]] || _role;\r\n  }\r\n  const role =\r\n    _role ??\r\n    (sender != null && sender && sender.toLowerCase() === 'user'\r\n      ? 'user'\r\n      : 'assistant');\r\n  const content = _content ?? text ?? '';\r\n  const formattedMessage: FormattedMessage = {\r\n    role,\r\n    content,\r\n  };\r\n\r\n  // Set name fields first\r\n  if (_name != null && _name) {\r\n    formattedMessage.name = _name;\r\n  }\r\n\r\n  if (userName != null && userName && formattedMessage.role === 'user') {\r\n    formattedMessage.name = userName;\r\n  }\r\n\r\n  if (\r\n    assistantName != null &&\r\n    assistantName &&\r\n    formattedMessage.role === 'assistant'\r\n  ) {\r\n    formattedMessage.name = assistantName;\r\n  }\r\n\r\n  if (formattedMessage.name != null && formattedMessage.name) {\r\n    // Conform to API regex: ^[a-zA-Z0-9_-]{1,64}$\r\n    // https://community.openai.com/t/the-format-of-the-name-field-in-the-documentation-is-incorrect/175684/2\r\n    formattedMessage.name = formattedMessage.name.replace(\r\n      /[^a-zA-Z0-9_-]/g,\r\n      '_'\r\n    );\r\n\r\n    if (formattedMessage.name.length > 64) {\r\n      formattedMessage.name = formattedMessage.name.substring(0, 64);\r\n    }\r\n  }\r\n\r\n  const { image_urls, documents, videos, audios } = message;\r\n  const mediaParts: MessageContentComplex[] = [];\r\n\r\n  if (Array.isArray(documents) && documents.length > 0) {\r\n    mediaParts.push(...documents);\r\n  }\r\n\r\n  if (Array.isArray(videos) && videos.length > 0) {\r\n    mediaParts.push(...videos);\r\n  }\r\n\r\n  if (Array.isArray(audios) && audios.length > 0) {\r\n    mediaParts.push(...audios);\r\n  }\r\n\r\n  if (Array.isArray(image_urls) && image_urls.length > 0) {\r\n    mediaParts.push(...image_urls);\r\n  }\r\n\r\n  if (mediaParts.length > 0 && role === 'user') {\r\n    const mediaMessage = formatMediaMessage({\r\n      message: {\r\n        ...formattedMessage,\r\n        content:\r\n          typeof formattedMessage.content === 'string'\r\n            ? formattedMessage.content\r\n            : '',\r\n      },\r\n      mediaParts,\r\n      endpoint,\r\n    });\r\n\r\n    if (!langChain) {\r\n      return mediaMessage;\r\n    }\r\n\r\n    return new HumanMessage(mediaMessage);\r\n  }\r\n\r\n  if (!langChain) {\r\n    return formattedMessage;\r\n  }\r\n\r\n  if (role === 'user') {\r\n    return new HumanMessage(formattedMessage);\r\n  } else if (role === 'assistant') {\r\n    return new AIMessage(formattedMessage);\r\n  } else {\r\n    return new SystemMessage(formattedMessage);\r\n  }\r\n};\r\n\r\n/**\r\n * Formats an array of messages for LangChain.\r\n *\r\n * @param messages - The array of messages to format.\r\n * @param formatOptions - The options for formatting each message.\r\n * @returns - The array of formatted LangChain messages.\r\n */\r\nexport const formatLangChainMessages = (\r\n  messages: Array<MessageInput>,\r\n  formatOptions: Omit<FormatMessageParams, 'message' | 'langChain'>\r\n): Array<HumanMessage | AIMessage | SystemMessage> => {\r\n  return messages.map((msg) => {\r\n    const formatted = formatMessage({\r\n      ...formatOptions,\r\n      message: msg,\r\n      langChain: true,\r\n    });\r\n    return formatted as HumanMessage | AIMessage | SystemMessage;\r\n  });\r\n};\r\n\r\ninterface LangChainMessage {\r\n  lc_kwargs?: {\r\n    additional_kwargs?: Record<string, any>;\r\n    [key: string]: any;\r\n  };\r\n  kwargs?: {\r\n    additional_kwargs?: Record<string, any>;\r\n    [key: string]: any;\r\n  };\r\n  [key: string]: any;\r\n}\r\n\r\n/**\r\n * Formats a LangChain message object by merging properties from `lc_kwargs` or `kwargs` and `additional_kwargs`.\r\n *\r\n * @param message - The message object to format.\r\n * @returns - The formatted LangChain message.\r\n */\r\nexport const formatFromLangChain = (\r\n  message: LangChainMessage\r\n): Record<string, any> => {\r\n  const kwargs = message.lc_kwargs ?? message.kwargs ?? {};\r\n  const { additional_kwargs = {}, ...message_kwargs } = kwargs;\r\n  return {\r\n    ...message_kwargs,\r\n    ...additional_kwargs,\r\n  };\r\n};\r\n\r\n/**\r\n * Helper function to format an assistant message\r\n * @param message The message to format\r\n * @returns Array of formatted messages\r\n */\r\nfunction formatAssistantMessage(\r\n  message: Partial<TMessage>\r\n): Array<AIMessage | ToolMessage> {\r\n  const formattedMessages: Array<AIMessage | ToolMessage> = [];\r\n  let currentContent: MessageContentComplex[] = [];\r\n  let lastAIMessage: AIMessage | null = null;\r\n  let hasReasoning = false;\r\n\r\n  if (Array.isArray(message.content)) {\r\n    for (const part of message.content) {\r\n      if (part.type === ContentTypes.TEXT && part.tool_call_ids) {\r\n        /*\r\n        If there's pending content, it needs to be aggregated as a single string to prepare for tool calls.\r\n        For Anthropic models, the \"tool_calls\" field on a message is only respected if content is a string.\r\n        */\r\n        if (currentContent.length > 0) {\r\n          let content = currentContent.reduce((acc, curr) => {\r\n            if (curr.type === ContentTypes.TEXT) {\r\n              return `${acc}${curr[ContentTypes.TEXT] || ''}\\n`;\r\n            }\r\n            return acc;\r\n          }, '');\r\n          content =\r\n            `${content}\\n${part[ContentTypes.TEXT] ?? part.text ?? ''}`.trim();\r\n          lastAIMessage = new AIMessage({ content });\r\n          formattedMessages.push(lastAIMessage);\r\n          currentContent = [];\r\n          continue;\r\n        }\r\n        // Create a new AIMessage with this text and prepare for tool calls\r\n        lastAIMessage = new AIMessage({\r\n          content: part.text || '',\r\n        });\r\n        formattedMessages.push(lastAIMessage);\r\n      } else if (part.type === ContentTypes.TOOL_CALL) {\r\n        if (!lastAIMessage) {\r\n          // \"Heal\" the payload by creating an AIMessage to precede the tool call\r\n          lastAIMessage = new AIMessage({ content: '' });\r\n          formattedMessages.push(lastAIMessage);\r\n        }\r\n\r\n        // Note: `tool_calls` list is defined when constructed by `AIMessage` class, and outputs should be excluded from it\r\n        const {\r\n          output,\r\n          args: _args,\r\n          ..._tool_call\r\n        } = part.tool_call as ToolCallPart;\r\n        const tool_call: ToolCallPart = _tool_call;\r\n        // TODO: investigate; args as dictionary may need to be providers-or-tool-specific\r\n        let args: any = _args;\r\n        try {\r\n          if (typeof _args === 'string') {\r\n            args = JSON.parse(_args);\r\n          }\r\n        } catch {\r\n          if (typeof _args === 'string') {\r\n            args = { input: _args };\r\n          }\r\n        }\r\n\r\n        tool_call.args = args;\r\n        if (!lastAIMessage.tool_calls) {\r\n          lastAIMessage.tool_calls = [];\r\n        }\r\n        lastAIMessage.tool_calls.push(tool_call as ToolCall);\r\n\r\n        formattedMessages.push(\r\n          new ToolMessage({\r\n            tool_call_id: tool_call.id ?? '',\r\n            name: tool_call.name,\r\n            content: output || '',\r\n          })\r\n        );\r\n      } else if (part.type === ContentTypes.THINK) {\r\n        hasReasoning = true;\r\n        continue;\r\n      } else if (\r\n        part.type === ContentTypes.ERROR ||\r\n        part.type === ContentTypes.AGENT_UPDATE\r\n      ) {\r\n        continue;\r\n      } else {\r\n        currentContent.push(part);\r\n      }\r\n    }\r\n  }\r\n\r\n  if (hasReasoning && currentContent.length > 0) {\r\n    const content = currentContent\r\n      .reduce((acc, curr) => {\r\n        if (curr.type === ContentTypes.TEXT) {\r\n          return `${acc}${curr[ContentTypes.TEXT] || ''}\\n`;\r\n        }\r\n        return acc;\r\n      }, '')\r\n      .trim();\r\n\r\n    if (content) {\r\n      formattedMessages.push(new AIMessage({ content }));\r\n    }\r\n  } else if (currentContent.length > 0) {\r\n    formattedMessages.push(new AIMessage({ content: currentContent }));\r\n  }\r\n\r\n  return formattedMessages;\r\n}\r\n\r\n/**\r\n * Formats an array of messages for LangChain, handling tool calls and creating ToolMessage instances.\r\n *\r\n * @param payload - The array of messages to format.\r\n * @param indexTokenCountMap - Optional map of message indices to token counts.\r\n * @param tools - Optional set of tool names that are allowed in the request.\r\n * @returns - Object containing formatted messages and updated indexTokenCountMap if provided.\r\n */\r\nexport const formatAgentMessages = (\r\n  payload: TPayload,\r\n  indexTokenCountMap?: Record<number, number>,\r\n  tools?: Set<string>\r\n): {\r\n  messages: Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>;\r\n  indexTokenCountMap?: Record<number, number>;\r\n} => {\r\n  const messages: Array<\r\n    HumanMessage | AIMessage | SystemMessage | ToolMessage\r\n  > = [];\r\n  // If indexTokenCountMap is provided, create a new map to track the updated indices\r\n  const updatedIndexTokenCountMap: Record<number, number> = {};\r\n  // Keep track of the mapping from original payload indices to result indices\r\n  const indexMapping: Record<number, number[]> = {};\r\n\r\n  // Process messages with tool conversion if tools set is provided\r\n  for (let i = 0; i < payload.length; i++) {\r\n    const message = payload[i];\r\n    // Q: Store the current length of messages to track where this payload message starts in the result?\r\n    // const startIndex = messages.length;\r\n    if (typeof message.content === 'string') {\r\n      message.content = [\r\n        { type: ContentTypes.TEXT, [ContentTypes.TEXT]: message.content },\r\n      ];\r\n    }\r\n    if (message.role !== 'assistant') {\r\n      messages.push(\r\n        formatMessage({\r\n          message: message as MessageInput,\r\n          langChain: true,\r\n        }) as HumanMessage | AIMessage | SystemMessage\r\n      );\r\n\r\n      // Update the index mapping for this message\r\n      indexMapping[i] = [messages.length - 1];\r\n      continue;\r\n    }\r\n\r\n    // For assistant messages, track the starting index before processing\r\n    const startMessageIndex = messages.length;\r\n\r\n    // If tools set is provided, we need to check if we need to convert tool messages to a string\r\n    if (tools) {\r\n      // First, check if this message contains tool calls\r\n      let hasToolCalls = false;\r\n      let hasInvalidTool = false;\r\n      const toolNames: string[] = [];\r\n\r\n      const content = message.content;\r\n      if (content && Array.isArray(content)) {\r\n        for (const part of content) {\r\n          if (part.type === ContentTypes.TOOL_CALL) {\r\n            hasToolCalls = true;\r\n            if (tools.size === 0) {\r\n              hasInvalidTool = true;\r\n              break;\r\n            }\r\n            const toolName = part.tool_call.name;\r\n            toolNames.push(toolName);\r\n            if (!tools.has(toolName)) {\r\n              hasInvalidTool = true;\r\n            }\r\n          }\r\n        }\r\n      }\r\n\r\n      // If this message has tool calls and at least one is invalid, we need to convert it\r\n      if (hasToolCalls && hasInvalidTool) {\r\n        // We need to collect all related messages (this message and any subsequent tool messages)\r\n        const toolSequence: BaseMessage[] = [];\r\n        let sequenceEndIndex = i;\r\n\r\n        // Process the current assistant message to get the AIMessage with tool calls\r\n        const formattedMessages = formatAssistantMessage(message);\r\n        toolSequence.push(...formattedMessages);\r\n\r\n        // Look ahead for any subsequent assistant messages that might be part of this tool sequence\r\n        let j = i + 1;\r\n        while (j < payload.length && payload[j].role === 'assistant') {\r\n          // Check if this is a continuation of the tool sequence\r\n          let isToolResponse = false;\r\n          const content = payload[j].content;\r\n          if (content && Array.isArray(content)) {\r\n            for (const part of content) {\r\n              if (part.type === ContentTypes.TOOL_CALL) {\r\n                isToolResponse = true;\r\n                break;\r\n              }\r\n            }\r\n          }\r\n\r\n          if (isToolResponse) {\r\n            // This is part of the tool sequence, add it\r\n            const nextMessages = formatAssistantMessage(payload[j]);\r\n            toolSequence.push(...nextMessages);\r\n            sequenceEndIndex = j;\r\n            j++;\r\n          } else {\r\n            // This is not part of the tool sequence, stop looking\r\n            break;\r\n          }\r\n        }\r\n\r\n        // Convert the sequence to a string\r\n        const bufferString = getBufferString(toolSequence);\r\n        messages.push(new AIMessage({ content: bufferString }));\r\n\r\n        // Skip the messages we've already processed\r\n        i = sequenceEndIndex;\r\n\r\n        // Update the index mapping for this sequence\r\n        const resultIndices = [messages.length - 1];\r\n        for (let k = i; k >= i && k <= sequenceEndIndex; k++) {\r\n          indexMapping[k] = resultIndices;\r\n        }\r\n\r\n        continue;\r\n      }\r\n    }\r\n\r\n    // Process the assistant message using the helper function\r\n    const formattedMessages = formatAssistantMessage(message);\r\n    messages.push(...formattedMessages);\r\n\r\n    // Update the index mapping for this assistant message\r\n    // Store all indices that were created from this original message\r\n    const endMessageIndex = messages.length;\r\n    const resultIndices = [];\r\n    for (let j = startMessageIndex; j < endMessageIndex; j++) {\r\n      resultIndices.push(j);\r\n    }\r\n    indexMapping[i] = resultIndices;\r\n  }\r\n\r\n  // Update the token count map if it was provided\r\n  if (indexTokenCountMap) {\r\n    for (\r\n      let originalIndex = 0;\r\n      originalIndex < payload.length;\r\n      originalIndex++\r\n    ) {\r\n      const resultIndices = indexMapping[originalIndex] || [];\r\n      const tokenCount = indexTokenCountMap[originalIndex];\r\n\r\n      if (tokenCount !== undefined) {\r\n        if (resultIndices.length === 1) {\r\n          // Simple 1:1 mapping\r\n          updatedIndexTokenCountMap[resultIndices[0]] = tokenCount;\r\n        } else if (resultIndices.length > 1) {\r\n          // If one message was split into multiple, distribute the token count\r\n          // This is a simplification - in reality, you might want a more sophisticated distribution\r\n          const countPerMessage = Math.floor(tokenCount / resultIndices.length);\r\n          resultIndices.forEach((resultIndex, idx) => {\r\n            if (idx === resultIndices.length - 1) {\r\n              // Give any remainder to the last message\r\n              updatedIndexTokenCountMap[resultIndex] =\r\n                tokenCount - countPerMessage * (resultIndices.length - 1);\r\n            } else {\r\n              updatedIndexTokenCountMap[resultIndex] = countPerMessage;\r\n            }\r\n          });\r\n        }\r\n      }\r\n    }\r\n  }\r\n\r\n  return {\r\n    messages,\r\n    indexTokenCountMap: indexTokenCountMap\r\n      ? updatedIndexTokenCountMap\r\n      : undefined,\r\n  };\r\n};\r\n\r\n/**\r\n * Formats an array of messages for LangChain, making sure all content fields are strings\r\n * @param payload - The array of messages to format.\r\n * @returns - The array of formatted LangChain messages, including ToolMessages for tool calls.\r\n */\r\nexport const formatContentStrings = (\r\n  payload: Array<BaseMessage>\r\n): Array<BaseMessage> => {\r\n  // Create a copy of the payload to avoid modifying the original\r\n  const result = [...payload];\r\n\r\n  for (const message of result) {\r\n    if (typeof message.content === 'string') {\r\n      continue;\r\n    }\r\n\r\n    if (!Array.isArray(message.content)) {\r\n      continue;\r\n    }\r\n\r\n    // Reduce text types to a single string, ignore all other types\r\n    const content = message.content.reduce((acc, curr) => {\r\n      if (curr.type === ContentTypes.TEXT) {\r\n        return `${acc}${curr[ContentTypes.TEXT] || ''}\\n`;\r\n      }\r\n      return acc;\r\n    }, '');\r\n\r\n    message.content = content.trim();\r\n  }\r\n\r\n  return result;\r\n};\r\n\r\n/**\r\n * Adds a value at key 0 for system messages and shifts all key indices by one in an indexTokenCountMap.\r\n * This is useful when adding a system message at the beginning of a conversation.\r\n *\r\n * @param indexTokenCountMap - The original map of message indices to token counts\r\n * @param instructionsTokenCount - The token count for the system message to add at index 0\r\n * @returns A new map with the system message at index 0 and all other indices shifted by 1\r\n */\r\nexport function shiftIndexTokenCountMap(\r\n  indexTokenCountMap: Record<number, number>,\r\n  instructionsTokenCount: number\r\n): Record<number, number> {\r\n  // Create a new map to avoid modifying the original\r\n  const shiftedMap: Record<number, number> = {};\r\n  shiftedMap[0] = instructionsTokenCount;\r\n\r\n  // Shift all existing indices by 1\r\n  for (const [indexStr, tokenCount] of Object.entries(indexTokenCountMap)) {\r\n    const index = Number(indexStr);\r\n    shiftedMap[index + 1] = tokenCount;\r\n  }\r\n\r\n  return shiftedMap;\r\n}\r\n"],"names":["Providers","ContentTypes","HumanMessage","AIMessage","SystemMessage","ToolMessage","messages","getBufferString"],"mappings":";;;;;AAAA;AA8BA;;;;;AAKG;AACI,MAAM,kBAAkB,GAAG,CAAC,EACjC,OAAO,EACP,QAAQ,EACR,UAAU,GACS,KAKjB;;AAEF,IAAA,MAAM,MAAM,GAKR;AACF,QAAA,GAAG,OAAO;AACV,QAAA,OAAO,EAAE,EAA6B;KACvC;AAED,IAAA,IAAI,QAAQ,KAAKA,eAAS,CAAC,SAAS,EAAE;QACpC,MAAM,CAAC,OAAO,GAAG;AACf,YAAA,GAAG,UAAU;YACb,EAAE,IAAI,EAAEC,kBAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE;SACxB;AAC5B,QAAA,OAAO,MAAM;;IAGf,MAAM,CAAC,OAAO,GAAG;QACf,EAAE,IAAI,EAAEA,kBAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE;AAClD,QAAA,GAAG,UAAU;KACa;AAE5B,IAAA,OAAO,MAAM;AACf;AA+BA;;;;;AAKG;AACU,MAAA,aAAa,GAAG,CAAC,EAC5B,OAAO,EACP,QAAQ,EACR,QAAQ,EACR,aAAa,EACb,SAAS,GAAG,KAAK,GACG,KAIF;;AAElB,IAAA,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,GAAG,OAAO;IAC5E,IAAI,KAAK,IAAI,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,EAAE;AACnC,QAAA,MAAM,WAAW,GAA2B;AAC1C,YAAA,aAAa,EAAE,QAAQ;AACvB,YAAA,YAAY,EAAE,MAAM;AACpB,YAAA,SAAS,EAAE,WAAW;SACvB;QACD,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK;;IAExC,MAAM,IAAI,GACR,KAAK;SACJ,MAAM,IAAI,IAAI,IAAI,MAAM,IAAI,MAAM,CAAC,WAAW,EAAE,KAAK;AACpD,cAAE;cACA,WAAW,CAAC;AAClB,IAAA,MAAM,OAAO,GAAG,QAAQ,IAAI,IAAI,IAAI,EAAE;AACtC,IAAA,MAAM,gBAAgB,GAAqB;QACzC,IAAI;QACJ,OAAO;KACR;;AAGD,IAAA,IAAI,KAAK,IAAI,IAAI,IAAI,KAAK,EAAE;AAC1B,QAAA,gBAAgB,CAAC,IAAI,GAAG,KAAK;;AAG/B,IAAA,IAAI,QAAQ,IAAI,IAAI,IAAI,QAAQ,IAAI,gBAAgB,CAAC,IAAI,KAAK,MAAM,EAAE;AACpE,QAAA,gBAAgB,CAAC,IAAI,GAAG,QAAQ;;IAGlC,IACE,aAAa,IAAI,IAAI;QACrB,aAAa;AACb,QAAA,gBAAgB,CAAC,IAAI,KAAK,WAAW,EACrC;AACA,QAAA,gBAAgB,CAAC,IAAI,GAAG,aAAa;;IAGvC,IAAI,gBAAgB,CAAC,IAAI,IAAI,IAAI,IAAI,gBAAgB,CAAC,IAAI,EAAE;;;AAG1D,QAAA,gBAAgB,CAAC,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CACnD,iBAAiB,EACjB,GAAG,CACJ;QAED,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,GAAG,EAAE,EAAE;AACrC,YAAA,gBAAgB,CAAC,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC;;;IAIlE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO;IACzD,MAAM,UAAU,GAA4B,EAAE;AAE9C,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE;AACpD,QAAA,UAAU,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;;AAG/B,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;AAC9C,QAAA,UAAU,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;;AAG5B,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE;AAC9C,QAAA,UAAU,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;;AAG5B,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;AACtD,QAAA,UAAU,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC;;IAGhC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,KAAK,MAAM,EAAE;QAC5C,MAAM,YAAY,GAAG,kBAAkB,CAAC;AACtC,YAAA,OAAO,EAAE;AACP,gBAAA,GAAG,gBAAgB;AACnB,gBAAA,OAAO,EACL,OAAO,gBAAgB,CAAC,OAAO,KAAK;sBAChC,gBAAgB,CAAC;AACnB,sBAAE,EAAE;AACT,aAAA;YACD,UAAU;YACV,QAAQ;AACT,SAAA,CAAC;QAEF,IAAI,CAAC,SAAS,EAAE;AACd,YAAA,OAAO,YAAY;;AAGrB,QAAA,OAAO,IAAIC,qBAAY,CAAC,YAAY,CAAC;;IAGvC,IAAI,CAAC,SAAS,EAAE;AACd,QAAA,OAAO,gBAAgB;;AAGzB,IAAA,IAAI,IAAI,KAAK,MAAM,EAAE;AACnB,QAAA,OAAO,IAAIA,qBAAY,CAAC,gBAAgB,CAAC;;AACpC,SAAA,IAAI,IAAI,KAAK,WAAW,EAAE;AAC/B,QAAA,OAAO,IAAIC,kBAAS,CAAC,gBAAgB,CAAC;;SACjC;AACL,QAAA,OAAO,IAAIC,sBAAa,CAAC,gBAAgB,CAAC;;AAE9C;AAEA;;;;;;AAMG;MACU,uBAAuB,GAAG,CACrC,QAA6B,EAC7B,aAAiE,KACd;AACnD,IAAA,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,KAAI;QAC1B,MAAM,SAAS,GAAG,aAAa,CAAC;AAC9B,YAAA,GAAG,aAAa;AAChB,YAAA,OAAO,EAAE,GAAG;AACZ,YAAA,SAAS,EAAE,IAAI;AAChB,SAAA,CAAC;AACF,QAAA,OAAO,SAAqD;AAC9D,KAAC,CAAC;AACJ;AAcA;;;;;AAKG;AACU,MAAA,mBAAmB,GAAG,CACjC,OAAyB,KACF;IACvB,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,MAAM,IAAI,EAAE;IACxD,MAAM,EAAE,iBAAiB,GAAG,EAAE,EAAE,GAAG,cAAc,EAAE,GAAG,MAAM;IAC5D,OAAO;AACL,QAAA,GAAG,cAAc;AACjB,QAAA,GAAG,iBAAiB;KACrB;AACH;AAEA;;;;AAIG;AACH,SAAS,sBAAsB,CAC7B,OAA0B,EAAA;IAE1B,MAAM,iBAAiB,GAAmC,EAAE;IAC5D,IAAI,cAAc,GAA4B,EAAE;IAChD,IAAI,aAAa,GAAqB,IAAI;IAC1C,IAAI,YAAY,GAAG,KAAK;IAExB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AAClC,QAAA,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE;AAClC,YAAA,IAAI,IAAI,CAAC,IAAI,KAAKH,kBAAY,CAAC,IAAI,IAAI,IAAI,CAAC,aAAa,EAAE;AACzD;;;AAGE;AACF,gBAAA,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;oBAC7B,IAAI,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,KAAI;wBAChD,IAAI,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,IAAI,EAAE;AACnC,4BAAA,OAAO,CAAG,EAAA,GAAG,CAAG,EAAA,IAAI,CAACA,kBAAY,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA,EAAA,CAAI;;AAEnD,wBAAA,OAAO,GAAG;qBACX,EAAE,EAAE,CAAC;oBACN,OAAO;AACL,wBAAA,CAAA,EAAG,OAAO,CAAK,EAAA,EAAA,IAAI,CAACA,kBAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE;oBACpE,aAAa,GAAG,IAAIE,kBAAS,CAAC,EAAE,OAAO,EAAE,CAAC;AAC1C,oBAAA,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC;oBACrC,cAAc,GAAG,EAAE;oBACnB;;;gBAGF,aAAa,GAAG,IAAIA,kBAAS,CAAC;AAC5B,oBAAA,OAAO,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;AACzB,iBAAA,CAAC;AACF,gBAAA,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC;;iBAChC,IAAI,IAAI,CAAC,IAAI,KAAKF,kBAAY,CAAC,SAAS,EAAE;gBAC/C,IAAI,CAAC,aAAa,EAAE;;oBAElB,aAAa,GAAG,IAAIE,kBAAS,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AAC9C,oBAAA,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC;;;AAIvC,gBAAA,MAAM,EACJ,MAAM,EACN,IAAI,EAAE,KAAK,EACX,GAAG,UAAU,EACd,GAAG,IAAI,CAAC,SAAyB;gBAClC,MAAM,SAAS,GAAiB,UAAU;;gBAE1C,IAAI,IAAI,GAAQ,KAAK;AACrB,gBAAA,IAAI;AACF,oBAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;AAC7B,wBAAA,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;;;AAE1B,gBAAA,MAAM;AACN,oBAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;AAC7B,wBAAA,IAAI,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE;;;AAI3B,gBAAA,SAAS,CAAC,IAAI,GAAG,IAAI;AACrB,gBAAA,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE;AAC7B,oBAAA,aAAa,CAAC,UAAU,GAAG,EAAE;;AAE/B,gBAAA,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,SAAqB,CAAC;AAEpD,gBAAA,iBAAiB,CAAC,IAAI,CACpB,IAAIE,oBAAW,CAAC;AACd,oBAAA,YAAY,EAAE,SAAS,CAAC,EAAE,IAAI,EAAE;oBAChC,IAAI,EAAE,SAAS,CAAC,IAAI;oBACpB,OAAO,EAAE,MAAM,IAAI,EAAE;AACtB,iBAAA,CAAC,CACH;;iBACI,IAAI,IAAI,CAAC,IAAI,KAAKJ,kBAAY,CAAC,KAAK,EAAE;gBAC3C,YAAY,GAAG,IAAI;gBACnB;;AACK,iBAAA,IACL,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,KAAK;AAChC,gBAAA,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,YAAY,EACvC;gBACA;;iBACK;AACL,gBAAA,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;;;;IAK/B,IAAI,YAAY,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;QAC7C,MAAM,OAAO,GAAG;AACb,aAAA,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,KAAI;YACpB,IAAI,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,IAAI,EAAE;AACnC,gBAAA,OAAO,CAAG,EAAA,GAAG,CAAG,EAAA,IAAI,CAACA,kBAAY,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA,EAAA,CAAI;;AAEnD,YAAA,OAAO,GAAG;SACX,EAAE,EAAE;AACJ,aAAA,IAAI,EAAE;QAET,IAAI,OAAO,EAAE;YACX,iBAAiB,CAAC,IAAI,CAAC,IAAIE,kBAAS,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;;;AAE/C,SAAA,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;AACpC,QAAA,iBAAiB,CAAC,IAAI,CAAC,IAAIA,kBAAS,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;;AAGpE,IAAA,OAAO,iBAAiB;AAC1B;AAEA;;;;;;;AAOG;AACU,MAAA,mBAAmB,GAAG,CACjC,OAAiB,EACjB,kBAA2C,EAC3C,KAAmB,KAIjB;IACF,MAAMG,UAAQ,GAEV,EAAE;;IAEN,MAAM,yBAAyB,GAA2B,EAAE;;IAE5D,MAAM,YAAY,GAA6B,EAAE;;AAGjD,IAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AACvC,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC;;;AAG1B,QAAA,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE;YACvC,OAAO,CAAC,OAAO,GAAG;AAChB,gBAAA,EAAE,IAAI,EAAEL,kBAAY,CAAC,IAAI,EAAE,CAACA,kBAAY,CAAC,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE;aAClE;;AAEH,QAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE;AAChC,YAAAK,UAAQ,CAAC,IAAI,CACX,aAAa,CAAC;AACZ,gBAAA,OAAO,EAAE,OAAuB;AAChC,gBAAA,SAAS,EAAE,IAAI;AAChB,aAAA,CAA6C,CAC/C;;YAGD,YAAY,CAAC,CAAC,CAAC,GAAG,CAACA,UAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YACvC;;;AAIF,QAAA,MAAM,iBAAiB,GAAGA,UAAQ,CAAC,MAAM;;QAGzC,IAAI,KAAK,EAAE;;YAET,IAAI,YAAY,GAAG,KAAK;YACxB,IAAI,cAAc,GAAG,KAAK;AAG1B,YAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO;YAC/B,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AACrC,gBAAA,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE;oBAC1B,IAAI,IAAI,CAAC,IAAI,KAAKL,kBAAY,CAAC,SAAS,EAAE;wBACxC,YAAY,GAAG,IAAI;AACnB,wBAAA,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,EAAE;4BACpB,cAAc,GAAG,IAAI;4BACrB;;AAEF,wBAAA,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI;wBAEpC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE;4BACxB,cAAc,GAAG,IAAI;;;;;;AAO7B,YAAA,IAAI,YAAY,IAAI,cAAc,EAAE;;gBAElC,MAAM,YAAY,GAAkB,EAAE;gBACtC,IAAI,gBAAgB,GAAG,CAAC;;AAGxB,gBAAA,MAAM,iBAAiB,GAAG,sBAAsB,CAAC,OAAO,CAAC;AACzD,gBAAA,YAAY,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC;;AAGvC,gBAAA,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AACb,gBAAA,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE;;oBAE5D,IAAI,cAAc,GAAG,KAAK;oBAC1B,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO;oBAClC,IAAI,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AACrC,wBAAA,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE;4BAC1B,IAAI,IAAI,CAAC,IAAI,KAAKA,kBAAY,CAAC,SAAS,EAAE;gCACxC,cAAc,GAAG,IAAI;gCACrB;;;;oBAKN,IAAI,cAAc,EAAE;;wBAElB,MAAM,YAAY,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;AACvD,wBAAA,YAAY,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC;wBAClC,gBAAgB,GAAG,CAAC;AACpB,wBAAA,CAAC,EAAE;;yBACE;;wBAEL;;;;AAKJ,gBAAA,MAAM,YAAY,GAAGM,wBAAe,CAAC,YAAY,CAAC;AAClD,gBAAAD,UAAQ,CAAC,IAAI,CAAC,IAAIH,kBAAS,CAAC,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;;gBAGvD,CAAC,GAAG,gBAAgB;;gBAGpB,MAAM,aAAa,GAAG,CAACG,UAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;AAC3C,gBAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,gBAAgB,EAAE,CAAC,EAAE,EAAE;AACpD,oBAAA,YAAY,CAAC,CAAC,CAAC,GAAG,aAAa;;gBAGjC;;;;AAKJ,QAAA,MAAM,iBAAiB,GAAG,sBAAsB,CAAC,OAAO,CAAC;AACzD,QAAAA,UAAQ,CAAC,IAAI,CAAC,GAAG,iBAAiB,CAAC;;;AAInC,QAAA,MAAM,eAAe,GAAGA,UAAQ,CAAC,MAAM;QACvC,MAAM,aAAa,GAAG,EAAE;AACxB,QAAA,KAAK,IAAI,CAAC,GAAG,iBAAiB,EAAE,CAAC,GAAG,eAAe,EAAE,CAAC,EAAE,EAAE;AACxD,YAAA,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;;AAEvB,QAAA,YAAY,CAAC,CAAC,CAAC,GAAG,aAAa;;;IAIjC,IAAI,kBAAkB,EAAE;AACtB,QAAA,KACE,IAAI,aAAa,GAAG,CAAC,EACrB,aAAa,GAAG,OAAO,CAAC,MAAM,EAC9B,aAAa,EAAE,EACf;YACA,MAAM,aAAa,GAAG,YAAY,CAAC,aAAa,CAAC,IAAI,EAAE;AACvD,YAAA,MAAM,UAAU,GAAG,kBAAkB,CAAC,aAAa,CAAC;AAEpD,YAAA,IAAI,UAAU,KAAK,SAAS,EAAE;AAC5B,gBAAA,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE;;oBAE9B,yBAAyB,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU;;AACnD,qBAAA,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;;;AAGnC,oBAAA,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC;oBACrE,aAAa,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,GAAG,KAAI;wBACzC,IAAI,GAAG,KAAK,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE;;4BAEpC,yBAAyB,CAAC,WAAW,CAAC;gCACpC,UAAU,GAAG,eAAe,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;;6BACtD;AACL,4BAAA,yBAAyB,CAAC,WAAW,CAAC,GAAG,eAAe;;AAE5D,qBAAC,CAAC;;;;;IAMV,OAAO;kBACLA,UAAQ;AACR,QAAA,kBAAkB,EAAE;AAClB,cAAE;AACF,cAAE,SAAS;KACd;AACH;AAEA;;;;AAIG;AACU,MAAA,oBAAoB,GAAG,CAClC,OAA2B,KACL;;AAEtB,IAAA,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC;AAE3B,IAAA,KAAK,MAAM,OAAO,IAAI,MAAM,EAAE;AAC5B,QAAA,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE;YACvC;;QAGF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YACnC;;;AAIF,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,KAAI;YACnD,IAAI,IAAI,CAAC,IAAI,KAAKL,kBAAY,CAAC,IAAI,EAAE;AACnC,gBAAA,OAAO,CAAG,EAAA,GAAG,CAAG,EAAA,IAAI,CAACA,kBAAY,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA,EAAA,CAAI;;AAEnD,YAAA,OAAO,GAAG;SACX,EAAE,EAAE,CAAC;AAEN,QAAA,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE;;AAGlC,IAAA,OAAO,MAAM;AACf;AAEA;;;;;;;AAOG;AACa,SAAA,uBAAuB,CACrC,kBAA0C,EAC1C,sBAA8B,EAAA;;IAG9B,MAAM,UAAU,GAA2B,EAAE;AAC7C,IAAA,UAAU,CAAC,CAAC,CAAC,GAAG,sBAAsB;;AAGtC,IAAA,KAAK,MAAM,CAAC,QAAQ,EAAE,UAAU,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE;AACvE,QAAA,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC;AAC9B,QAAA,UAAU,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,UAAU;;AAGpC,IAAA,OAAO,UAAU;AACnB;;;;;;;;;;"}

package/dist/cjs/messages/ids.cjs

@@ -0,0 +1,23 @@
+'use strict';
+
+var nanoid = require('nanoid');
+
+// src/stream.ts
+const getMessageId = (stepKey, graph, returnExistingId = false) => {
+    const messageId = graph.messageIdsByStepKey.get(stepKey);
+    if (messageId != null && messageId) {
+        return returnExistingId ? messageId : undefined;
+    }
+    const prelimMessageId = graph.prelimMessageIdsByStepKey.get(stepKey);
+    if (prelimMessageId != null && prelimMessageId) {
+        graph.prelimMessageIdsByStepKey.delete(stepKey);
+        graph.messageIdsByStepKey.set(stepKey, prelimMessageId);
+        return prelimMessageId;
+    }
+    const message_id = `msg_${nanoid.nanoid()}`;
+    graph.messageIdsByStepKey.set(stepKey, message_id);
+    return message_id;
+};
+
+exports.getMessageId = getMessageId;
+//# sourceMappingURL=ids.cjs.map

package/dist/cjs/messages/ids.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"ids.cjs","sources":["../../../src/messages/ids.ts"],"sourcesContent":["// src/stream.ts\r\nimport { nanoid } from 'nanoid';\r\nimport type { Graph } from '@/graphs';\r\nimport type * as t from '@/types';\r\n\r\nexport const getMessageId = (\r\n  stepKey: string,\r\n  graph: Graph<t.BaseGraphState>,\r\n  returnExistingId = false\r\n): string | undefined => {\r\n  const messageId = graph.messageIdsByStepKey.get(stepKey);\r\n  if (messageId != null && messageId) {\r\n    return returnExistingId ? messageId : undefined;\r\n  }\r\n\r\n  const prelimMessageId = graph.prelimMessageIdsByStepKey.get(stepKey);\r\n  if (prelimMessageId != null && prelimMessageId) {\r\n    graph.prelimMessageIdsByStepKey.delete(stepKey);\r\n    graph.messageIdsByStepKey.set(stepKey, prelimMessageId);\r\n    return prelimMessageId;\r\n  }\r\n\r\n  const message_id = `msg_${nanoid()}`;\r\n  graph.messageIdsByStepKey.set(stepKey, message_id);\r\n  return message_id;\r\n};\r\n"],"names":["nanoid"],"mappings":";;;;AAAA;AAKO,MAAM,YAAY,GAAG,CAC1B,OAAe,EACf,KAA8B,EAC9B,gBAAgB,GAAG,KAAK,KACF;IACtB,MAAM,SAAS,GAAG,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC;AACxD,IAAA,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,EAAE;QAClC,OAAO,gBAAgB,GAAG,SAAS,GAAG,SAAS;;IAGjD,MAAM,eAAe,GAAG,KAAK,CAAC,yBAAyB,CAAC,GAAG,CAAC,OAAO,CAAC;AACpE,IAAA,IAAI,eAAe,IAAI,IAAI,IAAI,eAAe,EAAE;AAC9C,QAAA,KAAK,CAAC,yBAAyB,CAAC,MAAM,CAAC,OAAO,CAAC;QAC/C,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,EAAE,eAAe,CAAC;AACvD,QAAA,OAAO,eAAe;;AAGxB,IAAA,MAAM,UAAU,GAAG,CAAA,IAAA,EAAOA,aAAM,EAAE,EAAE;IACpC,KAAK,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU,CAAC;AAClD,IAAA,OAAO,UAAU;AACnB;;;;"}