@librechat/agents 2.2.1 → 2.2.2

package/dist/esm/messages/format.mjs

@@ -0,0 +1,306 @@
+import { HumanMessage, AIMessage, SystemMessage, ToolMessage } from '@langchain/core/messages';
+import { Providers, ContentTypes } from '../common/enum.mjs';
+
+/**
+ * Formats a message to OpenAI Vision API payload format.
+ *
+ * @param {VisionMessageParams} params - The parameters for formatting.
+ * @returns {Object} - The formatted message.
+ */
+const formatVisionMessage = ({ message, image_urls, endpoint }) => {
+    // Create a new object to avoid mutating the input
+    const result = {
+        ...message,
+        content: []
+    };
+    if (endpoint === Providers.ANTHROPIC) {
+        result.content = [
+            ...image_urls,
+            { type: ContentTypes.TEXT, text: message.content }
+        ];
+        return result;
+    }
+    result.content = [
+        { type: ContentTypes.TEXT, text: message.content },
+        ...image_urls
+    ];
+    return result;
+};
+/**
+ * Formats a message to OpenAI payload format based on the provided options.
+ *
+ * @param {FormatMessageParams} params - The parameters for formatting.
+ * @returns {FormattedMessage | HumanMessage | AIMessage | SystemMessage} - The formatted message.
+ */
+const formatMessage = ({ message, userName, assistantName, endpoint, langChain = false }) => {
+    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 && sender?.toLowerCase() === 'user' ? 'user' : 'assistant');
+    const content = _content ?? text ?? '';
+    const formattedMessage = {
+        role,
+        content,
+    };
+    const { image_urls } = message;
+    if (Array.isArray(image_urls) && image_urls.length > 0 && role === 'user') {
+        return formatVisionMessage({
+            message: {
+                ...formattedMessage,
+                content: typeof formattedMessage.content === 'string' ? formattedMessage.content : ''
+            },
+            image_urls,
+            endpoint,
+        });
+    }
+    if (_name) {
+        formattedMessage.name = _name;
+    }
+    if (userName && formattedMessage.role === 'user') {
+        formattedMessage.name = userName;
+    }
+    if (assistantName && formattedMessage.role === 'assistant') {
+        formattedMessage.name = assistantName;
+    }
+    if (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);
+        }
+    }
+    if (!langChain) {
+        return formattedMessage;
+    }
+    if (role === 'user') {
+        return new HumanMessage(formattedMessage);
+    }
+    else if (role === 'assistant') {
+        return new AIMessage(formattedMessage);
+    }
+    else {
+        return new SystemMessage(formattedMessage);
+    }
+};
+/**
+ * Formats an array of messages for LangChain.
+ *
+ * @param {Array<MessageInput>} messages - The array of messages to format.
+ * @param {Omit<FormatMessageParams, 'message' | 'langChain'>} formatOptions - The options for formatting each message.
+ * @returns {Array<HumanMessage | AIMessage | SystemMessage>} - 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 {LangChainMessage} message - The message object to format.
+ * @returns {Record<string, any>} 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,
+    };
+};
+/**
+ * Formats an array of messages for LangChain, handling tool calls and creating ToolMessage instances.
+ *
+ * @param {Array<Partial<TMessage>>} payload - The array of messages to format.
+ * @param {Record<number, number>} [indexTokenCountMap] - Optional map of message indices to token counts.
+ * @returns {Object} - Object containing formatted messages and updated indexTokenCountMap if provided.
+ */
+const formatAgentMessages = (payload, indexTokenCountMap) => {
+    const messages = [];
+    // 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 = {};
+    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: ContentTypes.TEXT, [ContentTypes.TEXT]: message.content }];
+        }
+        if (message.role !== 'assistant') {
+            messages.push(formatMessage({
+                message: message,
+                langChain: true
+            }));
+            // Update the index mapping for this message
+            indexMapping[i] = [messages.length - 1];
+            continue;
+        }
+        // For assistant messages, track the starting index before processing
+        const startMessageIndex = messages.length;
+        let currentContent = [];
+        let lastAIMessage = null;
+        let hasReasoning = false;
+        if (Array.isArray(message.content)) {
+            for (const part of message.content) {
+                if (part.type === 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 === ContentTypes.TEXT) {
+                                return `${acc}${curr[ContentTypes.TEXT] || ''}\n`;
+                            }
+                            return acc;
+                        }, '');
+                        content = `${content}\n${part[ContentTypes.TEXT] ?? part.text ?? ''}`.trim();
+                        lastAIMessage = new AIMessage({ content });
+                        messages.push(lastAIMessage);
+                        currentContent = [];
+                        continue;
+                    }
+                    // Create a new AIMessage with this text and prepare for tool calls
+                    lastAIMessage = new AIMessage({
+                        content: part.text || '',
+                    });
+                    messages.push(lastAIMessage);
+                }
+                else if (part.type === ContentTypes.TOOL_CALL) {
+                    if (!lastAIMessage) {
+                        throw new Error('Invalid tool call structure: No preceding AIMessage with tool_call_ids');
+                    }
+                    // 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;
+                    // 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 (e) {
+                        if (typeof _args === 'string') {
+                            args = { input: _args };
+                        }
+                    }
+                    tool_call.args = args;
+                    if (!lastAIMessage.tool_calls) {
+                        lastAIMessage.tool_calls = [];
+                    }
+                    lastAIMessage.tool_calls.push(tool_call);
+                    // Add the corresponding ToolMessage
+                    messages.push(new ToolMessage({
+                        tool_call_id: tool_call.id,
+                        name: tool_call.name,
+                        content: output || '',
+                    }));
+                }
+                else if (part.type === ContentTypes.THINK) {
+                    hasReasoning = true;
+                    continue;
+                }
+                else if (part.type === ContentTypes.ERROR || part.type === ContentTypes.AGENT_UPDATE) {
+                    continue;
+                }
+                else {
+                    currentContent.push(part);
+                }
+            }
+        }
+        if (hasReasoning && currentContent.length > 0) {
+            const content = currentContent
+                .reduce((acc, curr) => {
+                if (curr.type === ContentTypes.TEXT) {
+                    return `${acc}${curr[ContentTypes.TEXT] || ''}\n`;
+                }
+                return acc;
+            }, '')
+                .trim();
+            if (content) {
+                messages.push(new AIMessage({ content }));
+            }
+        }
+        else if (currentContent.length > 0) {
+            messages.push(new AIMessage({ content: currentContent }));
+        }
+        // Update the index mapping for this assistant message
+        // Store all indices that were created from this original message
+        const endMessageIndex = messages.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,
+        indexTokenCountMap: indexTokenCountMap ? updatedIndexTokenCountMap : undefined
+    };
+};
+/**
+ * Formats an array of messages for LangChain, making sure all content fields are strings
+ * @param {Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>} payload - The array of messages to format.
+ * @returns {Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>} - 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 === ContentTypes.TEXT) {
+                return `${acc}${curr[ContentTypes.TEXT] || ''}\n`;
+            }
+            return acc;
+        }, '');
+        message.content = content.trim();
+    }
+    return result;
+};
+
+export { formatAgentMessages, formatContentStrings, formatFromLangChain, formatLangChainMessages, formatMessage, formatVisionMessage };
+//# sourceMappingURL=format.mjs.map

package/dist/esm/messages/format.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"format.mjs","sources":["../../../src/messages/format.ts"],"sourcesContent":["import { ToolMessage, BaseMessage } from '@langchain/core/messages';\nimport { HumanMessage, AIMessage, SystemMessage } from '@langchain/core/messages';\nimport { MessageContentImageUrl } from '@langchain/core/messages';\nimport type { ToolCall } from '@langchain/core/messages/tool';\nimport type { MessageContentComplex } from '@/types';\nimport { Providers, ContentTypes } from '@/common';\n\ninterface VisionMessageParams {\n  message: {\n    role: string;\n    content: string;\n    name?: string;\n    [key: string]: any;\n  };\n  image_urls: MessageContentImageUrl[];\n  endpoint?: Providers;\n}\n\n/**\n * Formats a message to OpenAI Vision API payload format.\n *\n * @param {VisionMessageParams} params - The parameters for formatting.\n * @returns {Object} - The formatted message.\n */\nexport const formatVisionMessage = ({ message, image_urls, endpoint }: VisionMessageParams): {\n  role: string;\n  content: MessageContentComplex[];\n  name?: string;\n  [key: string]: any;\n} => {\n  // Create a new object to avoid mutating the input\n  const result: {\n    role: string;\n    content: MessageContentComplex[];\n    name?: string;\n    [key: string]: any;\n  } = {\n    ...message,\n    content: [] as MessageContentComplex[]\n  };\n  \n  if (endpoint === Providers.ANTHROPIC) {\n    result.content = [\n      ...image_urls, \n      { type: ContentTypes.TEXT, text: message.content }\n    ] as MessageContentComplex[];\n    return result;\n  }\n\n  result.content = [\n    { type: ContentTypes.TEXT, text: message.content }, \n    ...image_urls\n  ] as MessageContentComplex[];\n\n  return result;\n};\n\ninterface MessageInput {\n  role?: string;\n  _name?: string;\n  sender?: string;\n  text?: string;\n  content?: string | MessageContentComplex[];\n  image_urls?: MessageContentImageUrl[];\n  lc_id?: string[];\n  [key: string]: any;\n}\n\ninterface FormatMessageParams {\n  message: MessageInput;\n  userName?: string;\n  assistantName?: string;\n  endpoint?: Providers;\n  langChain?: boolean;\n}\n\ninterface FormattedMessage {\n  role: string;\n  content: string | MessageContentComplex[];\n  name?: string;\n  [key: string]: any;\n}\n\n/**\n * Formats a message to OpenAI payload format based on the provided options.\n *\n * @param {FormatMessageParams} params - The parameters for formatting.\n * @returns {FormattedMessage | HumanMessage | AIMessage | SystemMessage} - The formatted message.\n */\nexport const formatMessage = ({ \n  message, \n  userName, \n  assistantName, \n  endpoint, \n  langChain = false \n}: FormatMessageParams): FormattedMessage | HumanMessage | AIMessage | SystemMessage => {\n  let { role: _role, _name, sender, text, content: _content, lc_id } = message;\n  if (lc_id && lc_id[2] && !langChain) {\n    const roleMapping: Record<string, string> = {\n      SystemMessage: 'system',\n      HumanMessage: 'user',\n      AIMessage: 'assistant',\n    };\n    _role = roleMapping[lc_id[2]] || _role;\n  }\n  const role = _role ?? (sender && sender?.toLowerCase() === 'user' ? 'user' : 'assistant');\n  const content = _content ?? text ?? '';\n  const formattedMessage: FormattedMessage = {\n    role,\n    content,\n  };\n\n  const { image_urls } = message;\n  if (Array.isArray(image_urls) && image_urls.length > 0 && role === 'user') {\n    return formatVisionMessage({\n      message: {\n        ...formattedMessage,\n        content: typeof formattedMessage.content === 'string' ? formattedMessage.content : ''\n      },\n      image_urls,\n      endpoint,\n    });\n  }\n\n  if (_name) {\n    formattedMessage.name = _name;\n  }\n\n  if (userName && formattedMessage.role === 'user') {\n    formattedMessage.name = userName;\n  }\n\n  if (assistantName && formattedMessage.role === 'assistant') {\n    formattedMessage.name = assistantName;\n  }\n\n  if (formattedMessage.name) {\n    // Conform to API regex: ^[a-zA-Z0-9_-]{1,64}$\n    // https://community.openai.com/t/the-format-of-the-name-field-in-the-documentation-is-incorrect/175684/2\n    formattedMessage.name = formattedMessage.name.replace(/[^a-zA-Z0-9_-]/g, '_');\n\n    if (formattedMessage.name.length > 64) {\n      formattedMessage.name = formattedMessage.name.substring(0, 64);\n    }\n  }\n\n  if (!langChain) {\n    return formattedMessage;\n  }\n\n  if (role === 'user') {\n    return new HumanMessage(formattedMessage);\n  } else if (role === 'assistant') {\n    return new AIMessage(formattedMessage);\n  } else {\n    return new SystemMessage(formattedMessage);\n  }\n};\n\n/**\n * Formats an array of messages for LangChain.\n *\n * @param {Array<MessageInput>} messages - The array of messages to format.\n * @param {Omit<FormatMessageParams, 'message' | 'langChain'>} formatOptions - The options for formatting each message.\n * @returns {Array<HumanMessage | AIMessage | SystemMessage>} - The array of formatted LangChain messages.\n */\nexport const formatLangChainMessages = (\n  messages: Array<MessageInput>, \n  formatOptions: Omit<FormatMessageParams, 'message' | 'langChain'>\n): Array<HumanMessage | AIMessage | SystemMessage> => {\n  return messages.map((msg) => {\n    const formatted = formatMessage({ ...formatOptions, message: msg, langChain: true });\n    return formatted as HumanMessage | AIMessage | SystemMessage;\n  });\n};\n\ninterface LangChainMessage {\n  lc_kwargs?: {\n    additional_kwargs?: Record<string, any>;\n    [key: string]: any;\n  };\n  kwargs?: {\n    additional_kwargs?: Record<string, any>;\n    [key: string]: any;\n  };\n  [key: string]: any;\n}\n\n/**\n * Formats a LangChain message object by merging properties from `lc_kwargs` or `kwargs` and `additional_kwargs`.\n *\n * @param {LangChainMessage} message - The message object to format.\n * @returns {Record<string, any>} The formatted LangChain message.\n */\nexport const formatFromLangChain = (message: LangChainMessage): Record<string, any> => {\n  const kwargs = message.lc_kwargs ?? message.kwargs ?? {};\n  const { additional_kwargs = {}, ...message_kwargs } = kwargs;\n  return {\n    ...message_kwargs,\n    ...additional_kwargs,\n  };\n};\n\ninterface TMessage {\n  role?: string;\n  content?: string | Array<{\n    type: ContentTypes;\n    [ContentTypes.TEXT]?: string;\n    text?: string;\n    tool_call_ids?: string[];\n    [key: string]: any;\n  }>;\n  [key: string]: any;\n}\n\ninterface ToolCallPart {\n  type: ContentTypes.TOOL_CALL;\n  tool_call: {\n    id: string;\n    name: string;\n    args: string | Record<string, unknown>;\n    output?: string;\n    [key: string]: any;\n  };\n}\n\n/**\n * Formats an array of messages for LangChain, handling tool calls and creating ToolMessage instances.\n *\n * @param {Array<Partial<TMessage>>} payload - The array of messages to format.\n * @param {Record<number, number>} [indexTokenCountMap] - Optional map of message indices to token counts.\n * @returns {Object} - Object containing formatted messages and updated indexTokenCountMap if provided.\n */\nexport const formatAgentMessages = (\n  payload: Array<Partial<TMessage>>, \n  indexTokenCountMap?: Record<number, number>\n): {\n  messages: Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>;\n  indexTokenCountMap?: Record<number, number>;\n} => {\n  const messages: Array<HumanMessage | AIMessage | SystemMessage | ToolMessage> = [];\n  // If indexTokenCountMap is provided, create a new map to track the updated indices\n  const updatedIndexTokenCountMap: Record<number, number> = {};\n  // Keep track of the mapping from original payload indices to result indices\n  const indexMapping: Record<number, number[]> = {};\n\n  for (let i = 0; i < payload.length; i++) {\n    const message = payload[i];\n    // Q: Store the current length of messages to track where this payload message starts in the result?\n    // const startIndex = messages.length;\n    if (typeof message.content === 'string') {\n      message.content = [{ type: ContentTypes.TEXT, [ContentTypes.TEXT]: message.content }];\n    }\n    if (message.role !== 'assistant') {\n      messages.push(formatMessage({ \n        message: message as MessageInput, \n        langChain: true \n      }) as HumanMessage | AIMessage | SystemMessage);\n      \n      // Update the index mapping for this message\n      indexMapping[i] = [messages.length - 1];\n      continue;\n    }\n\n    // For assistant messages, track the starting index before processing\n    const startMessageIndex = messages.length;\n\n    let currentContent: any[] = [];\n    let lastAIMessage: AIMessage | null = null;\n\n    let hasReasoning = false;\n    if (Array.isArray(message.content)) {\n      for (const part of message.content) {\n        if (part.type === ContentTypes.TEXT && part.tool_call_ids) {\n          /*\n          If there's pending content, it needs to be aggregated as a single string to prepare for tool calls.\n          For Anthropic models, the \"tool_calls\" field on a message is only respected if content is a string.\n          */\n          if (currentContent.length > 0) {\n            let content = currentContent.reduce((acc, curr) => {\n              if (curr.type === ContentTypes.TEXT) {\n                return `${acc}${curr[ContentTypes.TEXT] || ''}\\n`;\n              }\n              return acc;\n            }, '');\n            content = `${content}\\n${part[ContentTypes.TEXT] ?? part.text ?? ''}`.trim();\n            lastAIMessage = new AIMessage({ content });\n            messages.push(lastAIMessage);\n            currentContent = [];\n            continue;\n          }\n\n          // Create a new AIMessage with this text and prepare for tool calls\n          lastAIMessage = new AIMessage({\n            content: part.text || '',\n          });\n\n          messages.push(lastAIMessage);\n        } else if (part.type === ContentTypes.TOOL_CALL) {\n          if (!lastAIMessage) {\n            throw new Error('Invalid tool call structure: No preceding AIMessage with tool_call_ids');\n          }\n\n          // Note: `tool_calls` list is defined when constructed by `AIMessage` class, and outputs should be excluded from it\n          const { output, args: _args, ...tool_call } = (part.tool_call as any);\n          // TODO: investigate; args as dictionary may need to be providers-or-tool-specific\n          let args: any = _args;\n          try {\n            if (typeof _args === 'string') {\n              args = JSON.parse(_args);\n            }\n          } catch (e) {\n            if (typeof _args === 'string') {\n              args = { input: _args };\n            }\n          }\n\n          tool_call.args = args;\n          if (!lastAIMessage.tool_calls) {\n            lastAIMessage.tool_calls = [];\n          }\n          lastAIMessage.tool_calls.push(tool_call as ToolCall);\n\n          // Add the corresponding ToolMessage\n          messages.push(\n            new ToolMessage({\n              tool_call_id: tool_call.id,\n              name: tool_call.name,\n              content: output || '',\n            }),\n          );\n        } else if (part.type === ContentTypes.THINK) {\n          hasReasoning = true;\n          continue;\n        } else if (part.type === ContentTypes.ERROR || part.type === ContentTypes.AGENT_UPDATE) {\n          continue;\n        } else {\n          currentContent.push(part);\n        }\n      }\n    }\n\n    if (hasReasoning && currentContent.length > 0) {\n      const content = currentContent\n        .reduce((acc, curr) => {\n          if (curr.type === ContentTypes.TEXT) {\n            return `${acc}${curr[ContentTypes.TEXT] || ''}\\n`;\n          }\n          return acc;\n        }, '')\n        .trim();\n      \n      if (content) {\n        messages.push(new AIMessage({ content }));\n      }\n    } else if (currentContent.length > 0) {\n      messages.push(new AIMessage({ content: currentContent }));\n    }\n    \n    // Update the index mapping for this assistant message\n    // Store all indices that were created from this original message\n    const endMessageIndex = messages.length;\n    const resultIndices = [];\n    for (let j = startMessageIndex; j < endMessageIndex; j++) {\n      resultIndices.push(j);\n    }\n    indexMapping[i] = resultIndices;\n  }\n\n  // Update the token count map if it was provided\n  if (indexTokenCountMap) {\n    for (let originalIndex = 0; originalIndex < payload.length; originalIndex++) {\n      const resultIndices = indexMapping[originalIndex] || [];\n      const tokenCount = indexTokenCountMap[originalIndex];\n      \n      if (tokenCount !== undefined) {\n        if (resultIndices.length === 1) {\n          // Simple 1:1 mapping\n          updatedIndexTokenCountMap[resultIndices[0]] = tokenCount;\n        } else if (resultIndices.length > 1) {\n          // If one message was split into multiple, distribute the token count\n          // This is a simplification - in reality, you might want a more sophisticated distribution\n          const countPerMessage = Math.floor(tokenCount / resultIndices.length);\n          resultIndices.forEach((resultIndex, idx) => {\n            if (idx === resultIndices.length - 1) {\n              // Give any remainder to the last message\n              updatedIndexTokenCountMap[resultIndex] = tokenCount - (countPerMessage * (resultIndices.length - 1));\n            } else {\n              updatedIndexTokenCountMap[resultIndex] = countPerMessage;\n            }\n          });\n        }\n      }\n    }\n  }\n\n  return {\n    messages,\n    indexTokenCountMap: indexTokenCountMap ? updatedIndexTokenCountMap : undefined\n  };\n};\n\n/**\n * Formats an array of messages for LangChain, making sure all content fields are strings\n * @param {Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>} payload - The array of messages to format.\n * @returns {Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>} - The array of formatted LangChain messages, including ToolMessages for tool calls.\n */\nexport const formatContentStrings = (payload: Array<BaseMessage>): Array<BaseMessage> => {\n  // Create a copy of the payload to avoid modifying the original\n  const result = [...payload];\n\n  for (const message of result) {\n    if (typeof message.content === 'string') {\n      continue;\n    }\n\n    if (!Array.isArray(message.content)) {\n      continue;\n    }\n\n    // Reduce text types to a single string, ignore all other types\n    const content = message.content.reduce((acc, curr) => {\n      if (curr.type === ContentTypes.TEXT) {\n        return `${acc}${curr[ContentTypes.TEXT] || ''}\\n`;\n      }\n      return acc;\n    }, '');\n\n    message.content = content.trim();\n  }\n\n  return result;\n};\n"],"names":[],"mappings":";;;AAkBA;;;;;AAKG;AACI,MAAM,mBAAmB,GAAG,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAuB,KAKtF;;AAEF,IAAA,MAAM,MAAM,GAKR;AACF,QAAA,GAAG,OAAO;AACV,QAAA,OAAO,EAAE;KACV;AAED,IAAA,IAAI,QAAQ,KAAK,SAAS,CAAC,SAAS,EAAE;QACpC,MAAM,CAAC,OAAO,GAAG;AACf,YAAA,GAAG,UAAU;YACb,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO;SACtB;AAC5B,QAAA,OAAO,MAAM;;IAGf,MAAM,CAAC,OAAO,GAAG;QACf,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE;AAClD,QAAA,GAAG;KACuB;AAE5B,IAAA,OAAO,MAAM;AACf;AA4BA;;;;;AAKG;AACU,MAAA,aAAa,GAAG,CAAC,EAC5B,OAAO,EACP,QAAQ,EACR,aAAa,EACb,QAAQ,EACR,SAAS,GAAG,KAAK,EACG,KAAiE;AACrF,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,GAAG,KAAK,KAAK,MAAM,IAAI,MAAM,EAAE,WAAW,EAAE,KAAK,MAAM,GAAG,MAAM,GAAG,WAAW,CAAC;AACzF,IAAA,MAAM,OAAO,GAAG,QAAQ,IAAI,IAAI,IAAI,EAAE;AACtC,IAAA,MAAM,gBAAgB,GAAqB;QACzC,IAAI;QACJ,OAAO;KACR;AAED,IAAA,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO;AAC9B,IAAA,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,KAAK,MAAM,EAAE;AACzE,QAAA,OAAO,mBAAmB,CAAC;AACzB,YAAA,OAAO,EAAE;AACP,gBAAA,GAAG,gBAAgB;AACnB,gBAAA,OAAO,EAAE,OAAO,gBAAgB,CAAC,OAAO,KAAK,QAAQ,GAAG,gBAAgB,CAAC,OAAO,GAAG;AACpF,aAAA;YACD,UAAU;YACV,QAAQ;AACT,SAAA,CAAC;;IAGJ,IAAI,KAAK,EAAE;AACT,QAAA,gBAAgB,CAAC,IAAI,GAAG,KAAK;;IAG/B,IAAI,QAAQ,IAAI,gBAAgB,CAAC,IAAI,KAAK,MAAM,EAAE;AAChD,QAAA,gBAAgB,CAAC,IAAI,GAAG,QAAQ;;IAGlC,IAAI,aAAa,IAAI,gBAAgB,CAAC,IAAI,KAAK,WAAW,EAAE;AAC1D,QAAA,gBAAgB,CAAC,IAAI,GAAG,aAAa;;AAGvC,IAAA,IAAI,gBAAgB,CAAC,IAAI,EAAE;;;AAGzB,QAAA,gBAAgB,CAAC,IAAI,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC;QAE7E,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,IAAI,CAAC,SAAS,EAAE;AACd,QAAA,OAAO,gBAAgB;;AAGzB,IAAA,IAAI,IAAI,KAAK,MAAM,EAAE;AACnB,QAAA,OAAO,IAAI,YAAY,CAAC,gBAAgB,CAAC;;AACpC,SAAA,IAAI,IAAI,KAAK,WAAW,EAAE;AAC/B,QAAA,OAAO,IAAI,SAAS,CAAC,gBAAgB,CAAC;;SACjC;AACL,QAAA,OAAO,IAAI,aAAa,CAAC,gBAAgB,CAAC;;AAE9C;AAEA;;;;;;AAMG;MACU,uBAAuB,GAAG,CACrC,QAA6B,EAC7B,aAAiE,KACd;AACnD,IAAA,OAAO,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,KAAI;AAC1B,QAAA,MAAM,SAAS,GAAG,aAAa,CAAC,EAAE,GAAG,aAAa,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;AACpF,QAAA,OAAO,SAAqD;AAC9D,KAAC,CAAC;AACJ;AAcA;;;;;AAKG;AACU,MAAA,mBAAmB,GAAG,CAAC,OAAyB,KAAyB;IACpF,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;AAyBA;;;;;;AAMG;MACU,mBAAmB,GAAG,CACjC,OAAiC,EACjC,kBAA2C,KAIzC;IACF,MAAM,QAAQ,GAAkE,EAAE;;IAElF,MAAM,yBAAyB,GAA2B,EAAE;;IAE5D,MAAM,YAAY,GAA6B,EAAE;AAEjD,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,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,CAAC,YAAY,CAAC,IAAI,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;;AAEvF,QAAA,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE;AAChC,YAAA,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC;AAC1B,gBAAA,OAAO,EAAE,OAAuB;AAChC,gBAAA,SAAS,EAAE;AACZ,aAAA,CAA6C,CAAC;;YAG/C,YAAY,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC;YACvC;;;AAIF,QAAA,MAAM,iBAAiB,GAAG,QAAQ,CAAC,MAAM;QAEzC,IAAI,cAAc,GAAU,EAAE;QAC9B,IAAI,aAAa,GAAqB,IAAI;QAE1C,IAAI,YAAY,GAAG,KAAK;QACxB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AAClC,YAAA,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,OAAO,EAAE;AAClC,gBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,IAAI,IAAI,CAAC,aAAa,EAAE;AACzD;;;AAGE;AACF,oBAAA,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;wBAC7B,IAAI,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,KAAI;4BAChD,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,EAAE;AACnC,gCAAA,OAAO,CAAG,EAAA,GAAG,CAAG,EAAA,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA,EAAA,CAAI;;AAEnD,4BAAA,OAAO,GAAG;yBACX,EAAE,EAAE,CAAC;wBACN,OAAO,GAAG,GAAG,OAAO,CAAA,EAAA,EAAK,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE;wBAC5E,aAAa,GAAG,IAAI,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC;AAC1C,wBAAA,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC;wBAC5B,cAAc,GAAG,EAAE;wBACnB;;;oBAIF,aAAa,GAAG,IAAI,SAAS,CAAC;AAC5B,wBAAA,OAAO,EAAE,IAAI,CAAC,IAAI,IAAI,EAAE;AACzB,qBAAA,CAAC;AAEF,oBAAA,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC;;qBACvB,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,SAAS,EAAE;oBAC/C,IAAI,CAAC,aAAa,EAAE;AAClB,wBAAA,MAAM,IAAI,KAAK,CAAC,wEAAwE,CAAC;;;AAI3F,oBAAA,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,SAAS,EAAE,GAAI,IAAI,CAAC,SAAiB;;oBAErE,IAAI,IAAI,GAAQ,KAAK;AACrB,oBAAA,IAAI;AACF,wBAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;AAC7B,4BAAA,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;;;oBAE1B,OAAO,CAAC,EAAE;AACV,wBAAA,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;AAC7B,4BAAA,IAAI,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE;;;AAI3B,oBAAA,SAAS,CAAC,IAAI,GAAG,IAAI;AACrB,oBAAA,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE;AAC7B,wBAAA,aAAa,CAAC,UAAU,GAAG,EAAE;;AAE/B,oBAAA,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,SAAqB,CAAC;;AAGpD,oBAAA,QAAQ,CAAC,IAAI,CACX,IAAI,WAAW,CAAC;wBACd,YAAY,EAAE,SAAS,CAAC,EAAE;wBAC1B,IAAI,EAAE,SAAS,CAAC,IAAI;wBACpB,OAAO,EAAE,MAAM,IAAI,EAAE;AACtB,qBAAA,CAAC,CACH;;qBACI,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,KAAK,EAAE;oBAC3C,YAAY,GAAG,IAAI;oBACnB;;AACK,qBAAA,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,KAAK,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,YAAY,EAAE;oBACtF;;qBACK;AACL,oBAAA,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;;;;QAK/B,IAAI,YAAY,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;YAC7C,MAAM,OAAO,GAAG;AACb,iBAAA,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,KAAI;gBACpB,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,EAAE;AACnC,oBAAA,OAAO,CAAG,EAAA,GAAG,CAAG,EAAA,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA,EAAA,CAAI;;AAEnD,gBAAA,OAAO,GAAG;aACX,EAAE,EAAE;AACJ,iBAAA,IAAI,EAAE;YAET,IAAI,OAAO,EAAE;gBACX,QAAQ,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC;;;AAEtC,aAAA,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;AACpC,YAAA,QAAQ,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;;;;AAK3D,QAAA,MAAM,eAAe,GAAG,QAAQ,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,KAAK,IAAI,aAAa,GAAG,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE;YAC3E,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;;AAEpC,4BAAA,yBAAyB,CAAC,WAAW,CAAC,GAAG,UAAU,IAAI,eAAe,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;;6BAC/F;AACL,4BAAA,yBAAyB,CAAC,WAAW,CAAC,GAAG,eAAe;;AAE5D,qBAAC,CAAC;;;;;IAMV,OAAO;QACL,QAAQ;QACR,kBAAkB,EAAE,kBAAkB,GAAG,yBAAyB,GAAG;KACtE;AACH;AAEA;;;;AAIG;AACU,MAAA,oBAAoB,GAAG,CAAC,OAA2B,KAAwB;;AAEtF,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,KAAK,YAAY,CAAC,IAAI,EAAE;AACnC,gBAAA,OAAO,CAAG,EAAA,GAAG,CAAG,EAAA,IAAI,CAAC,YAAY,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;;;;"}

package/dist/esm/messages/transformers.mjs

@@ -0,0 +1,316 @@
+import { RunnableLambda } from '@langchain/core/runnables';
+import { isBaseMessageChunk, ChatMessageChunk, ChatMessage, FunctionMessageChunk, FunctionMessage, ToolMessageChunk, ToolMessage, SystemMessageChunk, SystemMessage, AIMessageChunk, AIMessage, HumanMessageChunk, HumanMessage, defaultTextSplitter } from '@langchain/core/messages';
+
+const _isMessageType = (msg, types) => {
+    const typesAsStrings = [
+        ...new Set(types?.map((t) => {
+            if (typeof t === "string") {
+                return t;
+            }
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const instantiatedMsgClass = new t({});
+            if (!("getType" in instantiatedMsgClass) ||
+                typeof instantiatedMsgClass.getType !== "function") {
+                throw new Error("Invalid type provided.");
+            }
+            return instantiatedMsgClass.getType();
+        })),
+    ];
+    const msgType = msg.getType();
+    return typesAsStrings.some((t) => t === msgType);
+};
+function trimMessages(messagesOrOptions, options) {
+    if (Array.isArray(messagesOrOptions)) {
+        const messages = messagesOrOptions;
+        if (!options) {
+            throw new Error("Options parameter is required when providing messages.");
+        }
+        return _trimMessagesHelper(messages, options);
+    }
+    else {
+        const trimmerOptions = messagesOrOptions;
+        return RunnableLambda.from((input) => _trimMessagesHelper(input, trimmerOptions)).withConfig({
+            runName: "trim_messages",
+        });
+    }
+}
+async function _trimMessagesHelper(messages, options) {
+    const { maxTokens, tokenCounter, strategy = "last", allowPartial = false, endOn, startOn, includeSystem = false, textSplitter, } = options;
+    if (startOn && strategy === "first") {
+        throw new Error("`startOn` should only be specified if `strategy` is 'last'.");
+    }
+    if (includeSystem && strategy === "first") {
+        throw new Error("`includeSystem` should only be specified if `strategy` is 'last'.");
+    }
+    let listTokenCounter;
+    if ("getNumTokens" in tokenCounter) {
+        listTokenCounter = async (msgs) => {
+            const tokenCounts = await Promise.all(msgs.map((msg) => tokenCounter.getNumTokens(msg.content)));
+            return tokenCounts.reduce((sum, count) => sum + count, 0);
+        };
+    }
+    else {
+        listTokenCounter = async (msgs) => tokenCounter(msgs);
+    }
+    let textSplitterFunc = defaultTextSplitter;
+    if (textSplitter) {
+        if ("splitText" in textSplitter) {
+            textSplitterFunc = textSplitter.splitText;
+        }
+        else {
+            textSplitterFunc = async (text) => textSplitter(text);
+        }
+    }
+    if (strategy === "first") {
+        return _firstMaxTokens(messages, {
+            maxTokens,
+            tokenCounter: listTokenCounter,
+            textSplitter: textSplitterFunc,
+            partialStrategy: allowPartial ? "first" : undefined,
+            endOn,
+        });
+    }
+    else if (strategy === "last") {
+        return _lastMaxTokens(messages, {
+            maxTokens,
+            tokenCounter: listTokenCounter,
+            textSplitter: textSplitterFunc,
+            allowPartial,
+            includeSystem,
+            startOn,
+            endOn,
+        });
+    }
+    else {
+        throw new Error(`Unrecognized strategy: '${strategy}'. Must be one of 'first' or 'last'.`);
+    }
+}
+async function _firstMaxTokens(messages, options) {
+    const { maxTokens, tokenCounter, textSplitter, partialStrategy, endOn } = options;
+    let messagesCopy = [...messages];
+    let idx = 0;
+    for (let i = 0; i < messagesCopy.length; i += 1) {
+        const remainingMessages = i > 0 ? messagesCopy.slice(0, -i) : messagesCopy;
+        if ((await tokenCounter(remainingMessages)) <= maxTokens) {
+            idx = messagesCopy.length - i;
+            break;
+        }
+    }
+    if (idx < messagesCopy.length - 1 && partialStrategy) {
+        let includedPartial = false;
+        if (Array.isArray(messagesCopy[idx].content)) {
+            const excluded = messagesCopy[idx];
+            if (typeof excluded.content === "string") {
+                throw new Error("Expected content to be an array.");
+            }
+            const numBlock = excluded.content.length;
+            const reversedContent = partialStrategy === "last"
+                ? [...excluded.content].reverse()
+                : excluded.content;
+            for (let i = 1; i <= numBlock; i += 1) {
+                const partialContent = partialStrategy === "first"
+                    ? reversedContent.slice(0, i)
+                    : reversedContent.slice(-i);
+                const fields = Object.fromEntries(Object.entries(excluded).filter(([k]) => k !== "type" && !k.startsWith("lc_")));
+                const updatedMessage = _switchTypeToMessage(excluded.getType(), {
+                    ...fields,
+                    content: partialContent,
+                });
+                const slicedMessages = [...messagesCopy.slice(0, idx), updatedMessage];
+                if ((await tokenCounter(slicedMessages)) <= maxTokens) {
+                    messagesCopy = slicedMessages;
+                    idx += 1;
+                    includedPartial = true;
+                }
+                else {
+                    break;
+                }
+            }
+            if (includedPartial && partialStrategy === "last") {
+                excluded.content = [...reversedContent].reverse();
+            }
+        }
+        if (!includedPartial) {
+            const excluded = messagesCopy[idx];
+            let text;
+            if (Array.isArray(excluded.content) &&
+                excluded.content.some((block) => typeof block === "string" || block.type === "text")) {
+                const textBlock = excluded.content.find((block) => block.type === "text" && block.text);
+                text = textBlock?.text;
+            }
+            else if (typeof excluded.content === "string") {
+                text = excluded.content;
+            }
+            if (text) {
+                const splitTexts = await textSplitter(text);
+                const numSplits = splitTexts.length;
+                if (partialStrategy === "last") {
+                    splitTexts.reverse();
+                }
+                for (let _ = 0; _ < numSplits - 1; _ += 1) {
+                    splitTexts.pop();
+                    excluded.content = splitTexts.join("");
+                    if ((await tokenCounter([...messagesCopy.slice(0, idx), excluded])) <=
+                        maxTokens) {
+                        if (partialStrategy === "last") {
+                            excluded.content = [...splitTexts].reverse().join("");
+                        }
+                        messagesCopy = [...messagesCopy.slice(0, idx), excluded];
+                        idx += 1;
+                        break;
+                    }
+                }
+            }
+        }
+    }
+    if (endOn) {
+        const endOnArr = Array.isArray(endOn) ? endOn : [endOn];
+        while (idx > 0 && !_isMessageType(messagesCopy[idx - 1], endOnArr)) {
+            idx -= 1;
+        }
+    }
+    return messagesCopy.slice(0, idx);
+}
+async function _lastMaxTokens(messages, options) {
+    const { allowPartial = false, includeSystem = false, endOn, startOn, ...rest } = options;
+    // Create a copy of messages to avoid mutation
+    let messagesCopy = messages.map((message) => {
+        const fields = Object.fromEntries(Object.entries(message).filter(([k]) => k !== "type" && !k.startsWith("lc_")));
+        return _switchTypeToMessage(message.getType(), fields, isBaseMessageChunk(message));
+    });
+    if (endOn) {
+        const endOnArr = Array.isArray(endOn) ? endOn : [endOn];
+        while (messagesCopy.length > 0 &&
+            !_isMessageType(messagesCopy[messagesCopy.length - 1], endOnArr)) {
+            messagesCopy = messagesCopy.slice(0, -1);
+        }
+    }
+    const swappedSystem = includeSystem && messagesCopy[0]?.getType() === "system";
+    let reversed_ = swappedSystem
+        ? messagesCopy.slice(0, 1).concat(messagesCopy.slice(1).reverse())
+        : messagesCopy.reverse();
+    reversed_ = await _firstMaxTokens(reversed_, {
+        ...rest,
+        partialStrategy: allowPartial ? "last" : undefined,
+        endOn: startOn,
+    });
+    if (swappedSystem) {
+        return [reversed_[0], ...reversed_.slice(1).reverse()];
+    }
+    else {
+        return reversed_.reverse();
+    }
+}
+function _switchTypeToMessage(messageType, fields, returnChunk) {
+    let chunk;
+    let msg;
+    switch (messageType) {
+        case "human":
+            if (returnChunk) {
+                chunk = new HumanMessageChunk(fields);
+            }
+            else {
+                msg = new HumanMessage(fields);
+            }
+            break;
+        case "ai":
+            if (returnChunk) {
+                let aiChunkFields = {
+                    ...fields,
+                };
+                if ("tool_calls" in aiChunkFields) {
+                    aiChunkFields = {
+                        ...aiChunkFields,
+                        tool_call_chunks: aiChunkFields.tool_calls?.map((tc) => ({
+                            ...tc,
+                            type: "tool_call_chunk",
+                            index: undefined,
+                            args: JSON.stringify(tc.args),
+                        })),
+                    };
+                }
+                chunk = new AIMessageChunk(aiChunkFields);
+            }
+            else {
+                msg = new AIMessage(fields);
+            }
+            break;
+        case "system":
+            if (returnChunk) {
+                chunk = new SystemMessageChunk(fields);
+            }
+            else {
+                msg = new SystemMessage(fields);
+            }
+            break;
+        case "developer":
+            if (returnChunk) {
+                chunk = new SystemMessageChunk({
+                    ...fields,
+                    additional_kwargs: {
+                        ...fields.additional_kwargs,
+                        __openai_role__: "developer",
+                    },
+                });
+            }
+            else {
+                msg = new SystemMessage({
+                    ...fields,
+                    additional_kwargs: {
+                        ...fields.additional_kwargs,
+                        __openai_role__: "developer",
+                    },
+                });
+            }
+            break;
+        case "tool":
+            if ("tool_call_id" in fields) {
+                if (returnChunk) {
+                    chunk = new ToolMessageChunk(fields);
+                }
+                else {
+                    msg = new ToolMessage(fields);
+                }
+            }
+            else {
+                throw new Error("Can not convert ToolMessage to ToolMessageChunk if 'tool_call_id' field is not defined.");
+            }
+            break;
+        case "function":
+            if (returnChunk) {
+                chunk = new FunctionMessageChunk(fields);
+            }
+            else {
+                if (!fields.name) {
+                    throw new Error("FunctionMessage must have a 'name' field");
+                }
+                msg = new FunctionMessage(fields);
+            }
+            break;
+        case "generic":
+            if ("role" in fields) {
+                if (returnChunk) {
+                    chunk = new ChatMessageChunk(fields);
+                }
+                else {
+                    msg = new ChatMessage(fields);
+                }
+            }
+            else {
+                throw new Error("Can not convert ChatMessage to ChatMessageChunk if 'role' field is not defined.");
+            }
+            break;
+        default:
+            throw new Error(`Unrecognized message type ${messageType}`);
+    }
+    if (returnChunk && chunk) {
+        return chunk;
+    }
+    if (msg) {
+        return msg;
+    }
+    throw new Error(`Unrecognized message type ${messageType}`);
+}
+
+export { trimMessages };
+//# sourceMappingURL=transformers.mjs.map