illuma-agents 1.0.2 → 1.0.3

package/dist/cjs/messages/content.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"content.cjs","sources":["../../../src/messages/content.ts"],"sourcesContent":["import { ContentTypes } from '@/common';\nimport type { BaseMessage } from '@langchain/core/messages';\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 = (\n  payload: Array<BaseMessage>\n): Array<BaseMessage> => {\n  // Create a new array to store the processed messages\n  const result: Array<BaseMessage> = [];\n\n  for (const message of payload) {\n    const messageType = message.getType();\n    const isValidMessage =\n      messageType === 'human' ||\n      messageType === 'ai' ||\n      messageType === 'system';\n\n    if (!isValidMessage) {\n      result.push(message);\n      continue;\n    }\n\n    // If content is already a string, add as-is\n    if (typeof message.content === 'string') {\n      result.push(message);\n      continue;\n    }\n\n    // If content is not an array, add as-is\n    if (!Array.isArray(message.content)) {\n      result.push(message);\n      continue;\n    }\n\n    // Check if all content blocks are text type\n    const allTextBlocks = message.content.every(\n      (block) => block.type === ContentTypes.TEXT\n    );\n\n    // Only convert to string if all blocks are text type\n    if (!allTextBlocks) {\n      result.push(message);\n      continue;\n    }\n\n    // Reduce text types to a single string\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    result.push(message);\n  }\n\n  return result;\n};\n"],"names":["ContentTypes"],"mappings":";;;;AAGA;;;;AAIG;AACU,MAAA,oBAAoB,GAAG,CAClC,OAA2B,KACL;;IAEtB,MAAM,MAAM,GAAuB,EAAE;AAErC,IAAA,KAAK,MAAM,OAAO,IAAI,OAAO,EAAE;AAC7B,QAAA,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,EAAE;AACrC,QAAA,MAAM,cAAc,GAClB,WAAW,KAAK,OAAO;AACvB,YAAA,WAAW,KAAK,IAAI;YACpB,WAAW,KAAK,QAAQ;QAE1B,IAAI,CAAC,cAAc,EAAE;AACnB,YAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;YACpB;;;AAIF,QAAA,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE;AACvC,YAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;YACpB;;;QAIF,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AACnC,YAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;YACpB;;;QAIF,MAAM,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,CACzC,CAAC,KAAK,KAAK,KAAK,CAAC,IAAI,KAAKA,kBAAY,CAAC,IAAI,CAC5C;;QAGD,IAAI,CAAC,aAAa,EAAE;AAClB,YAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;YACpB;;;AAIF,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,IAAI,KAAI;YACnD,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,CAAC;AAEN,QAAA,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE;AAChC,QAAA,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;;AAGtB,IAAA,OAAO,MAAM;AACf;;;;"}

package/dist/cjs/messages/format.cjs

@@ -187,13 +187,21 @@ function formatAssistantMessage(message) {
                 formattedMessages.push(lastAIMessage);
             }
             else if (part.type === _enum.ContentTypes.TOOL_CALL) {
+                // Skip malformed tool call entries without tool_call property
+                if (!part.tool_call) {
+                    continue;
+                }
+                // 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;
+                // Skip invalid tool calls that have no name AND no output
+                if (!_tool_call.name && (output == null || output === '')) {
+                    continue;
+                }
                 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;
@@ -298,6 +306,13 @@ const formatAgentMessages = (payload, indexTokenCountMap, tools) => {
                             hasInvalidTool = true;
                             break;
                         }
+                        // Protect against malformed tool call entries
+                        if (!part.tool_call ||
+                            part.tool_call.name == null ||
+                            part.tool_call.name === '') {
+                            hasInvalidTool = true;
+                            continue;
+                        }
                         const toolName = part.tool_call.name;
                         if (!tools.has(toolName)) {
                             hasInvalidTool = true;
@@ -399,32 +414,6 @@ const formatAgentMessages = (payload, indexTokenCountMap, tools) => {
             : 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.
@@ -446,7 +435,6 @@ function shiftIndexTokenCountMap(indexTokenCountMap, instructionsTokenCount) {
 }
 
 exports.formatAgentMessages = formatAgentMessages;
-exports.formatContentStrings = formatContentStrings;
 exports.formatFromLangChain = formatFromLangChain;
 exports.formatLangChainMessages = formatLangChainMessages;
 exports.formatMediaMessage = formatMediaMessage;

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

@@ -1 +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;;;;;;;;;;"}
+{"version":3,"file":"format.cjs","sources":["../../../src/messages/format.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\nimport {\n  AIMessage,\n  ToolMessage,\n  BaseMessage,\n  HumanMessage,\n  SystemMessage,\n  getBufferString,\n} from '@langchain/core/messages';\nimport type { MessageContentImageUrl } from '@langchain/core/messages';\nimport type { ToolCall } from '@langchain/core/messages/tool';\nimport type {\n  MessageContentComplex,\n  ToolCallPart,\n  TPayload,\n  TMessage,\n} from '@/types';\nimport { Providers, ContentTypes } from '@/common';\n\ninterface MediaMessageParams {\n  message: {\n    role: string;\n    content: string;\n    name?: string;\n    [key: string]: any;\n  };\n  mediaParts: MessageContentComplex[];\n  endpoint?: Providers;\n}\n\n/**\n * Formats a message with media content (images, documents, videos, audios) to API payload format.\n *\n * @param params - The parameters for formatting.\n * @returns - The formatted message.\n */\nexport const formatMediaMessage = ({\n  message,\n  endpoint,\n  mediaParts,\n}: MediaMessageParams): {\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      ...mediaParts,\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    ...mediaParts,\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  documents?: MessageContentComplex[];\n  videos?: MessageContentComplex[];\n  audios?: MessageContentComplex[];\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 params - The parameters for formatting.\n * @returns - The formatted message.\n */\nexport const formatMessage = ({\n  message,\n  userName,\n  endpoint,\n  assistantName,\n  langChain = false,\n}: FormatMessageParams):\n  | FormattedMessage\n  | HumanMessage\n  | AIMessage\n  | SystemMessage => {\n  // eslint-disable-next-line prefer-const\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 =\n    _role ??\n    (sender != null && sender && sender.toLowerCase() === 'user'\n      ? 'user'\n      : 'assistant');\n  const content = _content ?? text ?? '';\n  const formattedMessage: FormattedMessage = {\n    role,\n    content,\n  };\n\n  // Set name fields first\n  if (_name != null && _name) {\n    formattedMessage.name = _name;\n  }\n\n  if (userName != null && userName && formattedMessage.role === 'user') {\n    formattedMessage.name = userName;\n  }\n\n  if (\n    assistantName != null &&\n    assistantName &&\n    formattedMessage.role === 'assistant'\n  ) {\n    formattedMessage.name = assistantName;\n  }\n\n  if (formattedMessage.name != null && 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(\n      /[^a-zA-Z0-9_-]/g,\n      '_'\n    );\n\n    if (formattedMessage.name.length > 64) {\n      formattedMessage.name = formattedMessage.name.substring(0, 64);\n    }\n  }\n\n  const { image_urls, documents, videos, audios } = message;\n  const mediaParts: MessageContentComplex[] = [];\n\n  if (Array.isArray(documents) && documents.length > 0) {\n    mediaParts.push(...documents);\n  }\n\n  if (Array.isArray(videos) && videos.length > 0) {\n    mediaParts.push(...videos);\n  }\n\n  if (Array.isArray(audios) && audios.length > 0) {\n    mediaParts.push(...audios);\n  }\n\n  if (Array.isArray(image_urls) && image_urls.length > 0) {\n    mediaParts.push(...image_urls);\n  }\n\n  if (mediaParts.length > 0 && role === 'user') {\n    const mediaMessage = formatMediaMessage({\n      message: {\n        ...formattedMessage,\n        content:\n          typeof formattedMessage.content === 'string'\n            ? formattedMessage.content\n            : '',\n      },\n      mediaParts,\n      endpoint,\n    });\n\n    if (!langChain) {\n      return mediaMessage;\n    }\n\n    return new HumanMessage(mediaMessage);\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 messages - The array of messages to format.\n * @param formatOptions - The options for formatting each message.\n * @returns - 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({\n      ...formatOptions,\n      message: msg,\n      langChain: true,\n    });\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 message - The message object to format.\n * @returns - The formatted LangChain message.\n */\nexport const formatFromLangChain = (\n  message: LangChainMessage\n): 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\n/**\n * Helper function to format an assistant message\n * @param message The message to format\n * @returns Array of formatted messages\n */\nfunction formatAssistantMessage(\n  message: Partial<TMessage>\n): Array<AIMessage | ToolMessage> {\n  const formattedMessages: Array<AIMessage | ToolMessage> = [];\n  let currentContent: MessageContentComplex[] = [];\n  let lastAIMessage: AIMessage | null = null;\n  let hasReasoning = false;\n\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 =\n            `${content}\\n${part[ContentTypes.TEXT] ?? part.text ?? ''}`.trim();\n          lastAIMessage = new AIMessage({ content });\n          formattedMessages.push(lastAIMessage);\n          currentContent = [];\n          continue;\n        }\n        // Create a new AIMessage with this text and prepare for tool calls\n        lastAIMessage = new AIMessage({\n          content: part.text || '',\n        });\n        formattedMessages.push(lastAIMessage);\n      } else if (part.type === ContentTypes.TOOL_CALL) {\n        // Skip malformed tool call entries without tool_call property\n        if (!part.tool_call) {\n          continue;\n        }\n\n        // Note: `tool_calls` list is defined when constructed by `AIMessage` class, and outputs should be excluded from it\n        const {\n          output,\n          args: _args,\n          ..._tool_call\n        } = part.tool_call as ToolCallPart;\n\n        // Skip invalid tool calls that have no name AND no output\n        if (!_tool_call.name && (output == null || output === '')) {\n          continue;\n        }\n\n        if (!lastAIMessage) {\n          // \"Heal\" the payload by creating an AIMessage to precede the tool call\n          lastAIMessage = new AIMessage({ content: '' });\n          formattedMessages.push(lastAIMessage);\n        }\n\n        const tool_call: ToolCallPart = _tool_call;\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 {\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        formattedMessages.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 (\n        part.type === ContentTypes.ERROR ||\n        part.type === ContentTypes.AGENT_UPDATE\n      ) {\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      formattedMessages.push(new AIMessage({ content }));\n    }\n  } else if (currentContent.length > 0) {\n    formattedMessages.push(new AIMessage({ content: currentContent }));\n  }\n\n  return formattedMessages;\n}\n\n/**\n * Formats an array of messages for LangChain, handling tool calls and creating ToolMessage instances.\n *\n * @param payload - The array of messages to format.\n * @param indexTokenCountMap - Optional map of message indices to token counts.\n * @param tools - Optional set of tool names that are allowed in the request.\n * @returns - Object containing formatted messages and updated indexTokenCountMap if provided.\n */\nexport const formatAgentMessages = (\n  payload: TPayload,\n  indexTokenCountMap?: Record<number, number>,\n  tools?: Set<string>\n): {\n  messages: Array<HumanMessage | AIMessage | SystemMessage | ToolMessage>;\n  indexTokenCountMap?: Record<number, number>;\n} => {\n  const messages: Array<\n    HumanMessage | AIMessage | SystemMessage | ToolMessage\n  > = [];\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  // Process messages with tool conversion if tools set is provided\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 = [\n        { type: ContentTypes.TEXT, [ContentTypes.TEXT]: message.content },\n      ];\n    }\n    if (message.role !== 'assistant') {\n      messages.push(\n        formatMessage({\n          message: message as MessageInput,\n          langChain: true,\n        }) as HumanMessage | AIMessage | SystemMessage\n      );\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    // If tools set is provided, we need to check if we need to convert tool messages to a string\n    if (tools) {\n      // First, check if this message contains tool calls\n      let hasToolCalls = false;\n      let hasInvalidTool = false;\n      const toolNames: string[] = [];\n\n      const content = message.content;\n      if (content && Array.isArray(content)) {\n        for (const part of content) {\n          if (part.type === ContentTypes.TOOL_CALL) {\n            hasToolCalls = true;\n            if (tools.size === 0) {\n              hasInvalidTool = true;\n              break;\n            }\n            // Protect against malformed tool call entries\n            if (\n              !part.tool_call ||\n              part.tool_call.name == null ||\n              part.tool_call.name === ''\n            ) {\n              hasInvalidTool = true;\n              continue;\n            }\n            const toolName = part.tool_call.name;\n            toolNames.push(toolName);\n            if (!tools.has(toolName)) {\n              hasInvalidTool = true;\n            }\n          }\n        }\n      }\n\n      // If this message has tool calls and at least one is invalid, we need to convert it\n      if (hasToolCalls && hasInvalidTool) {\n        // We need to collect all related messages (this message and any subsequent tool messages)\n        const toolSequence: BaseMessage[] = [];\n        let sequenceEndIndex = i;\n\n        // Process the current assistant message to get the AIMessage with tool calls\n        const formattedMessages = formatAssistantMessage(message);\n        toolSequence.push(...formattedMessages);\n\n        // Look ahead for any subsequent assistant messages that might be part of this tool sequence\n        let j = i + 1;\n        while (j < payload.length && payload[j].role === 'assistant') {\n          // Check if this is a continuation of the tool sequence\n          let isToolResponse = false;\n          const content = payload[j].content;\n          if (content && Array.isArray(content)) {\n            for (const part of content) {\n              if (part.type === ContentTypes.TOOL_CALL) {\n                isToolResponse = true;\n                break;\n              }\n            }\n          }\n\n          if (isToolResponse) {\n            // This is part of the tool sequence, add it\n            const nextMessages = formatAssistantMessage(payload[j]);\n            toolSequence.push(...nextMessages);\n            sequenceEndIndex = j;\n            j++;\n          } else {\n            // This is not part of the tool sequence, stop looking\n            break;\n          }\n        }\n\n        // Convert the sequence to a string\n        const bufferString = getBufferString(toolSequence);\n        messages.push(new AIMessage({ content: bufferString }));\n\n        // Skip the messages we've already processed\n        i = sequenceEndIndex;\n\n        // Update the index mapping for this sequence\n        const resultIndices = [messages.length - 1];\n        for (let k = i; k >= i && k <= sequenceEndIndex; k++) {\n          indexMapping[k] = resultIndices;\n        }\n\n        continue;\n      }\n    }\n\n    // Process the assistant message using the helper function\n    const formattedMessages = formatAssistantMessage(message);\n    messages.push(...formattedMessages);\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 (\n      let originalIndex = 0;\n      originalIndex < payload.length;\n      originalIndex++\n    ) {\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] =\n                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\n      ? updatedIndexTokenCountMap\n      : undefined,\n  };\n};\n\n/**\n * Adds a value at key 0 for system messages and shifts all key indices by one in an indexTokenCountMap.\n * This is useful when adding a system message at the beginning of a conversation.\n *\n * @param indexTokenCountMap - The original map of message indices to token counts\n * @param instructionsTokenCount - The token count for the system message to add at index 0\n * @returns A new map with the system message at index 0 and all other indices shifted by 1\n */\nexport function shiftIndexTokenCountMap(\n  indexTokenCountMap: Record<number, number>,\n  instructionsTokenCount: number\n): Record<number, number> {\n  // Create a new map to avoid modifying the original\n  const shiftedMap: Record<number, number> = {};\n  shiftedMap[0] = instructionsTokenCount;\n\n  // Shift all existing indices by 1\n  for (const [indexStr, tokenCount] of Object.entries(indexTokenCountMap)) {\n    const index = Number(indexStr);\n    shiftedMap[index + 1] = tokenCount;\n  }\n\n  return shiftedMap;\n}\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;;AAE/C,gBAAA,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE;oBACnB;;;AAIF,gBAAA,MAAM,EACJ,MAAM,EACN,IAAI,EAAE,KAAK,EACX,GAAG,UAAU,EACd,GAAG,IAAI,CAAC,SAAyB;;AAGlC,gBAAA,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,IAAI,MAAM,KAAK,EAAE,CAAC,EAAE;oBACzD;;gBAGF,IAAI,CAAC,aAAa,EAAE;;oBAElB,aAAa,GAAG,IAAIE,kBAAS,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;AAC9C,oBAAA,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC;;gBAGvC,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;;;wBAGF,IACE,CAAC,IAAI,CAAC,SAAS;AACf,4BAAA,IAAI,CAAC,SAAS,CAAC,IAAI,IAAI,IAAI;AAC3B,4BAAA,IAAI,CAAC,SAAS,CAAC,IAAI,KAAK,EAAE,EAC1B;4BACA,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;;;;;;;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/run.cjs

@@ -1,16 +1,13 @@
 'use strict';
 
 require('./instrumentation.cjs');
-var zodToJsonSchema = require('zod-to-json-schema');
 var langchain = require('@langfuse/langchain');
 var prompts = require('@langchain/core/prompts');
-var messages = require('@langchain/core/messages');
 var runnables = require('@langchain/core/runnables');
 var openai = require('@langchain/openai');
-var _enum = require('./common/enum.cjs');
-var providers = require('./llm/providers.cjs');
-var format = require('./messages/format.cjs');
 var title = require('./utils/title.cjs');
+var _enum = require('./common/enum.cjs');
+var MultiAgentGraph = require('./graphs/MultiAgentGraph.cjs');
 var tokens = require('./utils/tokens.cjs');
 var Graph = require('./graphs/Graph.cjs');
 var events = require('./events.cjs');
@@ -31,13 +28,12 @@ const defaultOmitOptions = new Set([
     'additionalModelRequestFields',
 ]);
 class Run {
-    graphRunnable;
-    // private collab!: CollabGraph;
-    // private taskManager!: TaskManager;
-    handlerRegistry;
     id;
+    tokenCounter;
+    handlerRegistry;
+    indexTokenCountMap;
+    graphRunnable;
     Graph;
-    provider;
     returnContent = false;
     constructor(config) {
         const runId = config.runId ?? '';
@@ -45,6 +41,8 @@ class Run {
             throw new Error('Run ID not provided');
         }
         this.id = runId;
+        this.tokenCounter = config.tokenCounter;
+        this.indexTokenCountMap = config.indexTokenCountMap;
         const handlerRegistry = new events.HandlerRegistry();
         if (config.customHandlers) {
             for (const [eventType, handler] of Object.entries(config.customHandlers)) {
@@ -55,29 +53,80 @@ class Run {
         if (!config.graphConfig) {
             throw new Error('Graph config not provided');
         }
-        if (config.graphConfig.type === 'standard' || !config.graphConfig.type) {
-            this.provider = config.graphConfig.llmConfig.provider;
-            this.graphRunnable = this.createStandardGraph(config.graphConfig);
+        /** Handle different graph types */
+        if (config.graphConfig.type === 'multi-agent') {
+            this.graphRunnable = this.createMultiAgentGraph(config.graphConfig);
+            if (this.Graph) {
+                this.Graph.handlerRegistry = handlerRegistry;
+            }
+        }
+        else {
+            /** Default to legacy graph for 'standard' or undefined type */
+            this.graphRunnable = this.createLegacyGraph(config.graphConfig);
             if (this.Graph) {
+                this.Graph.compileOptions =
+                    config.graphConfig.compileOptions ?? this.Graph.compileOptions;
                 this.Graph.handlerRegistry = handlerRegistry;
             }
         }
         this.returnContent = config.returnContent ?? false;
     }
-    createStandardGraph(config) {
-        const { llmConfig, tools = [], ...graphInput } = config;
-        const { provider, ...clientOptions } = llmConfig;
+    createLegacyGraph(config) {
+        let agentConfig;
+        let signal;
+        /** Check if this is a multi-agent style config (has agents array) */
+        if ('agents' in config && Array.isArray(config.agents)) {
+            if (config.agents.length === 0) {
+                throw new Error('At least one agent must be provided');
+            }
+            agentConfig = config.agents[0];
+            signal = config.signal;
+        }
+        else {
+            /** Legacy path: build agent config from llmConfig */
+            const { type: _type, llmConfig, signal: legacySignal, tools = [], ...agentInputs } = config;
+            const { provider, ...clientOptions } = llmConfig;
+            agentConfig = {
+                ...agentInputs,
+                tools,
+                provider,
+                clientOptions,
+                agentId: 'default',
+            };
+            signal = legacySignal;
+        }
         const standardGraph = new Graph.StandardGraph({
-            tools,
-            provider,
-            clientOptions,
-            ...graphInput,
+            signal,
             runId: this.id,
+            agents: [agentConfig],
+            tokenCounter: this.tokenCounter,
+            indexTokenCountMap: this.indexTokenCountMap,
         });
+        /** Propagate compile options from graph config */
+        standardGraph.compileOptions = config.compileOptions;
         this.Graph = standardGraph;
         return standardGraph.createWorkflow();
     }
+    createMultiAgentGraph(config) {
+        const { agents, edges, compileOptions } = config;
+        const multiAgentGraph = new MultiAgentGraph.MultiAgentGraph({
+            runId: this.id,
+            agents,
+            edges,
+            tokenCounter: this.tokenCounter,
+            indexTokenCountMap: this.indexTokenCountMap,
+        });
+        if (compileOptions != null) {
+            multiAgentGraph.compileOptions = compileOptions;
+        }
+        this.Graph = multiAgentGraph;
+        return multiAgentGraph.createWorkflow();
+    }
     static async create(config) {
+        /** Create tokenCounter if indexTokenCountMap is provided but tokenCounter is not */
+        if (config.indexTokenCountMap && !config.tokenCounter) {
+            config.tokenCounter = await tokens.createTokenCounter();
+        }
         return new Run(config);
     }
     getRunMessages() {
@@ -86,21 +135,39 @@ class Run {
         }
         return this.Graph.getRunMessages();
     }
+    /**
+     * Creates a custom event callback handler that intercepts custom events
+     * and processes them through our handler registry instead of EventStreamCallbackHandler
+     */
+    createCustomEventCallback() {
+        return async (eventName, data, runId, tags, metadata) => {
+            if (data['emitted'] === true &&
+                eventName === _enum.GraphEvents.CHAT_MODEL_STREAM) {
+                return;
+            }
+            const handler = this.handlerRegistry?.getHandler(eventName);
+            if (handler && this.Graph) {
+                await handler.handle(eventName, data, metadata, this.Graph);
+            }
+        };
+    }
     async processStream(inputs, config, streamOptions) {
-        if (!this.graphRunnable) {
+        if (this.graphRunnable == null) {
             throw new Error('Run not initialized. Make sure to use Run.create() to instantiate the Run.');
         }
         if (!this.Graph) {
             throw new Error('Graph not initialized. Make sure to use Run.create() to instantiate the Run.');
         }
         this.Graph.resetValues(streamOptions?.keepContent);
-        const provider = this.Graph.provider;
-        const hasTools = this.Graph.tools ? this.Graph.tools.length > 0 : false;
-        if (streamOptions?.callbacks) {
-            /* TODO: conflicts with callback manager */
-            const callbacks = config.callbacks ?? [];
-            config.callbacks = callbacks.concat(this.getCallbacks(streamOptions.callbacks));
-        }
+        /** Custom event callback to intercept and handle custom events */
+        const customEventCallback = this.createCustomEventCallback();
+        const baseCallbacks = config.callbacks ?? [];
+        const streamCallbacks = streamOptions?.callbacks
+            ? this.getCallbacks(streamOptions.callbacks)
+            : [];
+        config.callbacks = baseCallbacks.concat(streamCallbacks).concat({
+            [_enum.Callback.CUSTOM_EVENT]: customEventCallback,
+        });
         if (misc.isPresent(process.env.LANGFUSE_SECRET_KEY) &&
             misc.isPresent(process.env.LANGFUSE_PUBLIC_KEY) &&
             misc.isPresent(process.env.LANGFUSE_BASE_URL)) {
@@ -120,60 +187,23 @@ class Run {
         if (!this.id) {
             throw new Error('Run ID not provided');
         }
-        const tokenCounter = streamOptions?.tokenCounter ??
-            (streamOptions?.indexTokenCountMap
-                ? await tokens.createTokenCounter()
-                : undefined);
-        const tools = this.Graph.tools;
-        const toolTokens = tokenCounter
-            ? (tools?.reduce((acc, tool) => {
-                if (!tool.schema) {
-                    return acc;
-                }
-                const jsonSchema = zodToJsonSchema.zodToJsonSchema((tool?.schema).describe(tool?.description ?? ''), tool?.name ?? '');
-                return (acc + tokenCounter(new messages.SystemMessage(JSON.stringify(jsonSchema))));
-            }, 0) ?? 0)
-            : 0;
-        let instructionTokens = toolTokens;
-        if (this.Graph.systemMessage && tokenCounter) {
-            instructionTokens += tokenCounter(this.Graph.systemMessage);
-        }
-        const tokenMap = streamOptions?.indexTokenCountMap ?? {};
-        if (this.Graph.systemMessage && instructionTokens > 0) {
-            this.Graph.indexTokenCountMap = format.shiftIndexTokenCountMap(tokenMap, instructionTokens);
-        }
-        else if (instructionTokens > 0) {
-            tokenMap[0] = tokenMap[0] + instructionTokens;
-            this.Graph.indexTokenCountMap = tokenMap;
-        }
-        else {
-            this.Graph.indexTokenCountMap = tokenMap;
-        }
-        this.Graph.maxContextTokens = streamOptions?.maxContextTokens;
-        this.Graph.tokenCounter = tokenCounter;
         config.run_id = this.id;
         config.configurable = Object.assign(config.configurable ?? {}, {
             run_id: this.id,
-            provider: this.provider,
         });
         const stream = this.graphRunnable.streamEvents(inputs, config, {
             raiseError: true,
         });
         for await (const event of stream) {
-            const { data, name, metadata, ...info } = event;
-            let eventName = info.event;
-            if (hasTools &&
-                providers.manualToolStreamProviders.has(provider) &&
-                eventName === _enum.GraphEvents.CHAT_MODEL_STREAM) {
-                /* Skipping CHAT_MODEL_STREAM event due to double-call edge case */
+            const { data, metadata, ...info } = event;
+            const eventName = info.event;
+            /** Skip custom events as they're handled by our callback */
+            if (eventName === _enum.GraphEvents.ON_CUSTOM_EVENT) {
                 continue;
             }
-            if (eventName && eventName === _enum.GraphEvents.ON_CUSTOM_EVENT) {
-                eventName = name;
-            }
-            const handler = this.handlerRegistry.getHandler(eventName);
+            const handler = this.handlerRegistry?.getHandler(eventName);
             if (handler) {
-                handler.handle(eventName, data, metadata, this.Graph);
+                await handler.handle(eventName, data, metadata, this.Graph);
             }
         }
         if (this.returnContent) {
@@ -195,7 +225,7 @@ class Run {
             [_enum.Callback.TOOL_END]: this.createSystemCallback(clientCallbacks, _enum.Callback.TOOL_END),
         };
     }
-    async generateTitle({ provider, inputText, contentParts, titlePrompt, clientOptions, chainOptions, skipLanguage, omitOptions = defaultOmitOptions, titleMethod = _enum.TitleMethod.COMPLETION, titlePromptTemplate, }) {
+    async generateTitle({ provider, inputText, contentParts, titlePrompt, clientOptions, chainOptions, skipLanguage, titleMethod = _enum.TitleMethod.COMPLETION, titlePromptTemplate, }) {
         if (chainOptions != null &&
             misc.isPresent(process.env.LANGFUSE_SECRET_KEY) &&
             misc.isPresent(process.env.LANGFUSE_PUBLIC_KEY) &&
@@ -222,7 +252,6 @@ class Run {
             .join('\n');
         const model = this.Graph?.getNewModel({
             provider,
-            omitOptions,
             clientOptions,
         });
         if (!model) {
@@ -255,7 +284,23 @@ class Run {
             .pipe(convoToTitleInput)
             .pipe(titleChain)
             .withConfig({ runName: 'TitleChain' });
-        return await fullChain.invoke({ input: inputText, output: response }, chainOptions);
+        const invokeConfig = Object.assign({}, chainOptions, {
+            run_id: this.id,
+            runId: this.id,
+        });
+        try {
+            return await fullChain.invoke({ input: inputText, output: response }, invokeConfig);
+        }
+        catch (_e) {
+            // Fallback: strip callbacks to avoid EventStream tracer errors in certain environments
+            // But preserve langfuse handler if it exists
+            const langfuseHandler = invokeConfig.callbacks?.find((cb) => cb instanceof langchain.CallbackHandler);
+            const { callbacks: _cb, ...rest } = invokeConfig;
+            const safeConfig = Object.assign({}, rest, {
+                callbacks: langfuseHandler ? [langfuseHandler] : [],
+            });
+            return await fullChain.invoke({ input: inputText, output: response }, safeConfig);
+        }
     }
 }