illuma-agents 1.0.2

package/dist/esm/messages/prune.mjs

@@ -0,0 +1,393 @@
+import { AIMessage } from '@langchain/core/messages';
+import { ContentTypes, Providers } from '../common/enum.mjs';
+
+function isIndexInContext(arrayA, arrayB, targetIndex) {
+    const startingIndexInA = arrayA.length - arrayB.length;
+    return targetIndex >= startingIndexInA;
+}
+function addThinkingBlock(message, thinkingBlock) {
+    const content = Array.isArray(message.content)
+        ? message.content
+        : [
+            {
+                type: ContentTypes.TEXT,
+                text: message.content,
+            },
+        ];
+    /** Edge case, the message already has the thinking block */
+    if (content[0].type === thinkingBlock.type) {
+        return message;
+    }
+    content.unshift(thinkingBlock);
+    return new AIMessage({
+        ...message,
+        content,
+    });
+}
+/**
+ * Calculates the total tokens from a single usage object
+ *
+ * @param usage The usage metadata object containing token information
+ * @returns An object containing the total input and output tokens
+ */
+function calculateTotalTokens(usage) {
+    const baseInputTokens = Number(usage.input_tokens) || 0;
+    const cacheCreation = Number(usage.input_token_details?.cache_creation) || 0;
+    const cacheRead = Number(usage.input_token_details?.cache_read) || 0;
+    const totalInputTokens = baseInputTokens + cacheCreation + cacheRead;
+    const totalOutputTokens = Number(usage.output_tokens) || 0;
+    return {
+        input_tokens: totalInputTokens,
+        output_tokens: totalOutputTokens,
+        total_tokens: totalInputTokens + totalOutputTokens,
+    };
+}
+/**
+ * Processes an array of messages and returns a context of messages that fit within a specified token limit.
+ * It iterates over the messages from newest to oldest, adding them to the context until the token limit is reached.
+ *
+ * @param options Configuration options for processing messages
+ * @returns Object containing the message context, remaining tokens, messages not included, and summary index
+ */
+function getMessagesWithinTokenLimit({ messages: _messages, maxContextTokens, indexTokenCountMap, startType: _startType, thinkingEnabled, tokenCounter, thinkingStartIndex: _thinkingStartIndex = -1, reasoningType = ContentTypes.THINKING, }) {
+    // Every reply is primed with <|start|>assistant<|message|>, so we
+    // start with 3 tokens for the label after all messages have been counted.
+    let currentTokenCount = 3;
+    const instructions = _messages[0]?.getType() === 'system' ? _messages[0] : undefined;
+    const instructionsTokenCount = instructions != null ? (indexTokenCountMap[0] ?? 0) : 0;
+    const initialContextTokens = maxContextTokens - instructionsTokenCount;
+    let remainingContextTokens = initialContextTokens;
+    let startType = _startType;
+    const originalLength = _messages.length;
+    const messages = [..._messages];
+    /**
+     * IMPORTANT: this context array gets reversed at the end, since the latest messages get pushed first.
+     *
+     * This may be confusing to read, but it is done to ensure the context is in the correct order for the model.
+     * */
+    let context = [];
+    let thinkingStartIndex = _thinkingStartIndex;
+    let thinkingEndIndex = -1;
+    let thinkingBlock;
+    const endIndex = instructions != null ? 1 : 0;
+    const prunedMemory = [];
+    if (_thinkingStartIndex > -1) {
+        const thinkingMessageContent = messages[_thinkingStartIndex]?.content;
+        if (Array.isArray(thinkingMessageContent)) {
+            thinkingBlock = thinkingMessageContent.find((content) => content.type === reasoningType);
+        }
+    }
+    if (currentTokenCount < remainingContextTokens) {
+        let currentIndex = messages.length;
+        while (messages.length > 0 &&
+            currentTokenCount < remainingContextTokens &&
+            currentIndex > endIndex) {
+            currentIndex--;
+            if (messages.length === 1 && instructions) {
+                break;
+            }
+            const poppedMessage = messages.pop();
+            if (!poppedMessage)
+                continue;
+            const messageType = poppedMessage.getType();
+            if (thinkingEnabled === true &&
+                thinkingEndIndex === -1 &&
+                currentIndex === originalLength - 1 &&
+                (messageType === 'ai' || messageType === 'tool')) {
+                thinkingEndIndex = currentIndex;
+            }
+            if (thinkingEndIndex > -1 &&
+                !thinkingBlock &&
+                thinkingStartIndex < 0 &&
+                messageType === 'ai' &&
+                Array.isArray(poppedMessage.content)) {
+                thinkingBlock = poppedMessage.content.find((content) => content.type === reasoningType);
+                thinkingStartIndex = thinkingBlock != null ? currentIndex : -1;
+            }
+            /** False start, the latest message was not part of a multi-assistant/tool sequence of messages */
+            if (thinkingEndIndex > -1 &&
+                currentIndex === thinkingEndIndex - 1 &&
+                messageType !== 'ai' &&
+                messageType !== 'tool') {
+                thinkingEndIndex = -1;
+            }
+            const tokenCount = indexTokenCountMap[currentIndex] ?? 0;
+            if (prunedMemory.length === 0 &&
+                currentTokenCount + tokenCount <= remainingContextTokens) {
+                context.push(poppedMessage);
+                currentTokenCount += tokenCount;
+            }
+            else {
+                prunedMemory.push(poppedMessage);
+                if (thinkingEndIndex > -1 && thinkingStartIndex < 0) {
+                    continue;
+                }
+                break;
+            }
+        }
+        if (context[context.length - 1]?.getType() === 'tool') {
+            startType = ['ai', 'human'];
+        }
+        if (startType != null && startType.length > 0 && context.length > 0) {
+            let requiredTypeIndex = -1;
+            let totalTokens = 0;
+            for (let i = context.length - 1; i >= 0; i--) {
+                const currentType = context[i]?.getType() ?? '';
+                if (Array.isArray(startType)
+                    ? startType.includes(currentType)
+                    : currentType === startType) {
+                    requiredTypeIndex = i + 1;
+                    break;
+                }
+                const originalIndex = originalLength - 1 - i;
+                totalTokens += indexTokenCountMap[originalIndex] ?? 0;
+            }
+            if (requiredTypeIndex > 0) {
+                currentTokenCount -= totalTokens;
+                context = context.slice(0, requiredTypeIndex);
+            }
+        }
+    }
+    if (instructions && originalLength > 0) {
+        context.push(_messages[0]);
+        messages.shift();
+    }
+    remainingContextTokens -= currentTokenCount;
+    const result = {
+        remainingContextTokens,
+        context: [],
+        messagesToRefine: prunedMemory,
+    };
+    if (thinkingStartIndex > -1) {
+        result.thinkingStartIndex = thinkingStartIndex;
+    }
+    if (prunedMemory.length === 0 ||
+        thinkingEndIndex < 0 ||
+        (thinkingStartIndex > -1 &&
+            isIndexInContext(_messages, context, thinkingStartIndex))) {
+        // we reverse at this step to ensure the context is in the correct order for the model, and we need to work backwards
+        result.context = context.reverse();
+        return result;
+    }
+    if (thinkingEndIndex > -1 && thinkingStartIndex < 0) {
+        throw new Error('The payload is malformed. There is a thinking sequence but no "AI" messages with thinking blocks.');
+    }
+    if (!thinkingBlock) {
+        throw new Error('The payload is malformed. There is a thinking sequence but no thinking block found.');
+    }
+    // Since we have a thinking sequence, we need to find the last assistant message
+    // in the latest AI/tool sequence to add the thinking block that falls outside of the current context
+    // Latest messages are ordered first.
+    let assistantIndex = -1;
+    for (let i = 0; i < context.length; i++) {
+        const currentMessage = context[i];
+        const type = currentMessage?.getType();
+        if (type === 'ai') {
+            assistantIndex = i;
+        }
+        if (assistantIndex > -1 && (type === 'human' || type === 'system')) {
+            break;
+        }
+    }
+    if (assistantIndex === -1) {
+        throw new Error('The payload is malformed. There is a thinking sequence but no "AI" messages to append thinking blocks to.');
+    }
+    thinkingStartIndex = originalLength - 1 - assistantIndex;
+    const thinkingTokenCount = tokenCounter(new AIMessage({ content: [thinkingBlock] }));
+    const newRemainingCount = remainingContextTokens - thinkingTokenCount;
+    const newMessage = addThinkingBlock(context[assistantIndex], thinkingBlock);
+    context[assistantIndex] = newMessage;
+    if (newRemainingCount > 0) {
+        result.context = context.reverse();
+        return result;
+    }
+    const thinkingMessage = context[assistantIndex];
+    // now we need to an additional round of pruning but making the thinking block fit
+    const newThinkingMessageTokenCount = (indexTokenCountMap[thinkingStartIndex] ?? 0) + thinkingTokenCount;
+    remainingContextTokens = initialContextTokens - newThinkingMessageTokenCount;
+    currentTokenCount = 3;
+    let newContext = [];
+    const secondRoundMessages = [..._messages];
+    let currentIndex = secondRoundMessages.length;
+    while (secondRoundMessages.length > 0 &&
+        currentTokenCount < remainingContextTokens &&
+        currentIndex > thinkingStartIndex) {
+        currentIndex--;
+        const poppedMessage = secondRoundMessages.pop();
+        if (!poppedMessage)
+            continue;
+        const tokenCount = indexTokenCountMap[currentIndex] ?? 0;
+        if (currentTokenCount + tokenCount <= remainingContextTokens) {
+            newContext.push(poppedMessage);
+            currentTokenCount += tokenCount;
+        }
+        else {
+            messages.push(poppedMessage);
+            break;
+        }
+    }
+    const firstMessage = newContext[newContext.length - 1];
+    const firstMessageType = newContext[newContext.length - 1].getType();
+    if (firstMessageType === 'tool') {
+        startType = ['ai', 'human'];
+    }
+    if (startType != null && startType.length > 0 && newContext.length > 0) {
+        let requiredTypeIndex = -1;
+        let totalTokens = 0;
+        for (let i = newContext.length - 1; i >= 0; i--) {
+            const currentType = newContext[i]?.getType() ?? '';
+            if (Array.isArray(startType)
+                ? startType.includes(currentType)
+                : currentType === startType) {
+                requiredTypeIndex = i + 1;
+                break;
+            }
+            const originalIndex = originalLength - 1 - i;
+            totalTokens += indexTokenCountMap[originalIndex] ?? 0;
+        }
+        if (requiredTypeIndex > 0) {
+            currentTokenCount -= totalTokens;
+            newContext = newContext.slice(0, requiredTypeIndex);
+        }
+    }
+    if (firstMessageType === 'ai') {
+        const newMessage = addThinkingBlock(firstMessage, thinkingBlock);
+        newContext[newContext.length - 1] = newMessage;
+    }
+    else {
+        newContext.push(thinkingMessage);
+    }
+    if (instructions && originalLength > 0) {
+        newContext.push(_messages[0]);
+        secondRoundMessages.shift();
+    }
+    result.context = newContext.reverse();
+    return result;
+}
+function checkValidNumber(value) {
+    return typeof value === 'number' && !isNaN(value) && value > 0;
+}
+function createPruneMessages(factoryParams) {
+    const indexTokenCountMap = { ...factoryParams.indexTokenCountMap };
+    let lastTurnStartIndex = factoryParams.startIndex;
+    let lastCutOffIndex = 0;
+    let totalTokens = Object.values(indexTokenCountMap).reduce((a = 0, b = 0) => a + b, 0);
+    let runThinkingStartIndex = -1;
+    return function pruneMessages(params) {
+        if (factoryParams.provider === Providers.OPENAI &&
+            factoryParams.thinkingEnabled === true) {
+            for (let i = lastTurnStartIndex; i < params.messages.length; i++) {
+                const m = params.messages[i];
+                if (m.getType() === 'ai' &&
+                    typeof m.additional_kwargs.reasoning_content === 'string' &&
+                    Array.isArray(m.additional_kwargs.provider_specific_fields?.thinking_blocks) &&
+                    m.tool_calls &&
+                    (m.tool_calls?.length ?? 0) > 0) {
+                    const message = m;
+                    const thinkingBlocks = message.additional_kwargs.provider_specific_fields.thinking_blocks;
+                    const signature = thinkingBlocks?.[thinkingBlocks.length - 1].signature;
+                    const thinkingBlock = {
+                        signature,
+                        type: ContentTypes.THINKING,
+                        thinking: message.additional_kwargs.reasoning_content,
+                    };
+                    params.messages[i] = new AIMessage({
+                        ...message,
+                        content: [thinkingBlock],
+                        additional_kwargs: {
+                            ...message.additional_kwargs,
+                            reasoning_content: undefined,
+                        },
+                    });
+                }
+            }
+        }
+        let currentUsage;
+        if (params.usageMetadata &&
+            (checkValidNumber(params.usageMetadata.input_tokens) ||
+                (checkValidNumber(params.usageMetadata.input_token_details) &&
+                    (checkValidNumber(params.usageMetadata.input_token_details.cache_creation) ||
+                        checkValidNumber(params.usageMetadata.input_token_details.cache_read)))) &&
+            checkValidNumber(params.usageMetadata.output_tokens)) {
+            currentUsage = calculateTotalTokens(params.usageMetadata);
+            totalTokens = currentUsage.total_tokens;
+        }
+        const newOutputs = new Set();
+        for (let i = lastTurnStartIndex; i < params.messages.length; i++) {
+            const message = params.messages[i];
+            if (i === lastTurnStartIndex &&
+                indexTokenCountMap[i] === undefined &&
+                currentUsage) {
+                indexTokenCountMap[i] = currentUsage.output_tokens;
+            }
+            else if (indexTokenCountMap[i] === undefined) {
+                indexTokenCountMap[i] = factoryParams.tokenCounter(message);
+                if (currentUsage) {
+                    newOutputs.add(i);
+                }
+                totalTokens += indexTokenCountMap[i] ?? 0;
+            }
+        }
+        // If `currentUsage` is defined, we need to distribute the current total tokens to our `indexTokenCountMap`,
+        // We must distribute it in a weighted manner, so that the total token count is equal to `currentUsage.total_tokens`,
+        // relative the manually counted tokens in `indexTokenCountMap`.
+        // EDGE CASE: when the resulting context gets pruned, we should not distribute the usage for messages that are not in the context.
+        if (currentUsage) {
+            let totalIndexTokens = 0;
+            if (params.messages[0].getType() === 'system') {
+                totalIndexTokens += indexTokenCountMap[0] ?? 0;
+            }
+            for (let i = lastCutOffIndex; i < params.messages.length; i++) {
+                if (i === 0 && params.messages[0].getType() === 'system') {
+                    continue;
+                }
+                if (newOutputs.has(i)) {
+                    continue;
+                }
+                totalIndexTokens += indexTokenCountMap[i] ?? 0;
+            }
+            // Calculate ratio based only on messages that remain in the context
+            const ratio = currentUsage.total_tokens / totalIndexTokens;
+            const isRatioSafe = ratio >= 1 / 3 && ratio <= 2.5;
+            // Apply the ratio adjustment only to messages at or after lastCutOffIndex, and only if the ratio is safe
+            if (isRatioSafe) {
+                if (params.messages[0].getType() === 'system' &&
+                    lastCutOffIndex !== 0) {
+                    indexTokenCountMap[0] = Math.round((indexTokenCountMap[0] ?? 0) * ratio);
+                }
+                for (let i = lastCutOffIndex; i < params.messages.length; i++) {
+                    if (newOutputs.has(i)) {
+                        continue;
+                    }
+                    indexTokenCountMap[i] = Math.round((indexTokenCountMap[i] ?? 0) * ratio);
+                }
+            }
+        }
+        lastTurnStartIndex = params.messages.length;
+        if (lastCutOffIndex === 0 && totalTokens <= factoryParams.maxTokens) {
+            return { context: params.messages, indexTokenCountMap };
+        }
+        const { context, thinkingStartIndex } = getMessagesWithinTokenLimit({
+            maxContextTokens: factoryParams.maxTokens,
+            messages: params.messages,
+            indexTokenCountMap,
+            startType: params.startType,
+            thinkingEnabled: factoryParams.thinkingEnabled,
+            tokenCounter: factoryParams.tokenCounter,
+            reasoningType: factoryParams.provider === Providers.BEDROCK
+                ? ContentTypes.REASONING_CONTENT
+                : ContentTypes.THINKING,
+            thinkingStartIndex: factoryParams.thinkingEnabled === true
+                ? runThinkingStartIndex
+                : undefined,
+        });
+        runThinkingStartIndex = thinkingStartIndex ?? -1;
+        /** The index is the first value of `context`, index relative to `params.messages` */
+        lastCutOffIndex = Math.max(params.messages.length -
+            (context.length - (context[0]?.getType() === 'system' ? 1 : 0)), 0);
+        return { context, indexTokenCountMap };
+    };
+}
+
+export { calculateTotalTokens, checkValidNumber, createPruneMessages, getMessagesWithinTokenLimit };
+//# sourceMappingURL=prune.mjs.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"prune.mjs","sources":["../../../src/messages/prune.ts"],"sourcesContent":["import {\r\n  AIMessage,\r\n  BaseMessage,\r\n  UsageMetadata,\r\n} from '@langchain/core/messages';\r\nimport type {\r\n  ThinkingContentText,\r\n  MessageContentComplex,\r\n  ReasoningContentText,\r\n} from '@/types/stream';\r\nimport type { TokenCounter } from '@/types/run';\r\nimport { ContentTypes, Providers } from '@/common';\r\n\r\nexport type PruneMessagesFactoryParams = {\r\n  provider?: Providers;\r\n  maxTokens: number;\r\n  startIndex: number;\r\n  tokenCounter: TokenCounter;\r\n  indexTokenCountMap: Record<string, number | undefined>;\r\n  thinkingEnabled?: boolean;\r\n};\r\nexport type PruneMessagesParams = {\r\n  messages: BaseMessage[];\r\n  usageMetadata?: Partial<UsageMetadata>;\r\n  startType?: ReturnType<BaseMessage['getType']>;\r\n};\r\n\r\nfunction isIndexInContext(\r\n  arrayA: unknown[],\r\n  arrayB: unknown[],\r\n  targetIndex: number\r\n): boolean {\r\n  const startingIndexInA = arrayA.length - arrayB.length;\r\n  return targetIndex >= startingIndexInA;\r\n}\r\n\r\nfunction addThinkingBlock(\r\n  message: AIMessage,\r\n  thinkingBlock: ThinkingContentText | ReasoningContentText\r\n): AIMessage {\r\n  const content: MessageContentComplex[] = Array.isArray(message.content)\r\n    ? (message.content as MessageContentComplex[])\r\n    : [\r\n      {\r\n        type: ContentTypes.TEXT,\r\n        text: message.content,\r\n      },\r\n    ];\r\n  /** Edge case, the message already has the thinking block */\r\n  if (content[0].type === thinkingBlock.type) {\r\n    return message;\r\n  }\r\n  content.unshift(thinkingBlock);\r\n  return new AIMessage({\r\n    ...message,\r\n    content,\r\n  });\r\n}\r\n\r\n/**\r\n * Calculates the total tokens from a single usage object\r\n *\r\n * @param usage The usage metadata object containing token information\r\n * @returns An object containing the total input and output tokens\r\n */\r\nexport function calculateTotalTokens(\r\n  usage: Partial<UsageMetadata>\r\n): UsageMetadata {\r\n  const baseInputTokens = Number(usage.input_tokens) || 0;\r\n  const cacheCreation = Number(usage.input_token_details?.cache_creation) || 0;\r\n  const cacheRead = Number(usage.input_token_details?.cache_read) || 0;\r\n\r\n  const totalInputTokens = baseInputTokens + cacheCreation + cacheRead;\r\n  const totalOutputTokens = Number(usage.output_tokens) || 0;\r\n\r\n  return {\r\n    input_tokens: totalInputTokens,\r\n    output_tokens: totalOutputTokens,\r\n    total_tokens: totalInputTokens + totalOutputTokens,\r\n  };\r\n}\r\n\r\nexport type PruningResult = {\r\n  context: BaseMessage[];\r\n  remainingContextTokens: number;\r\n  messagesToRefine: BaseMessage[];\r\n  thinkingStartIndex?: number;\r\n};\r\n\r\n/**\r\n * Processes an array of messages and returns a context of messages that fit within a specified token limit.\r\n * It iterates over the messages from newest to oldest, adding them to the context until the token limit is reached.\r\n *\r\n * @param options Configuration options for processing messages\r\n * @returns Object containing the message context, remaining tokens, messages not included, and summary index\r\n */\r\nexport function getMessagesWithinTokenLimit({\r\n  messages: _messages,\r\n  maxContextTokens,\r\n  indexTokenCountMap,\r\n  startType: _startType,\r\n  thinkingEnabled,\r\n  tokenCounter,\r\n  thinkingStartIndex: _thinkingStartIndex = -1,\r\n  reasoningType = ContentTypes.THINKING,\r\n}: {\r\n  messages: BaseMessage[];\r\n  maxContextTokens: number;\r\n  indexTokenCountMap: Record<string, number | undefined>;\r\n  startType?: string | string[];\r\n  thinkingEnabled?: boolean;\r\n  tokenCounter: TokenCounter;\r\n  thinkingStartIndex?: number;\r\n  reasoningType?: ContentTypes.THINKING | ContentTypes.REASONING_CONTENT;\r\n}): PruningResult {\r\n  // Every reply is primed with <|start|>assistant<|message|>, so we\r\n  // start with 3 tokens for the label after all messages have been counted.\r\n  let currentTokenCount = 3;\r\n  const instructions =\r\n    _messages[0]?.getType() === 'system' ? _messages[0] : undefined;\r\n  const instructionsTokenCount =\r\n    instructions != null ? (indexTokenCountMap[0] ?? 0) : 0;\r\n  const initialContextTokens = maxContextTokens - instructionsTokenCount;\r\n  let remainingContextTokens = initialContextTokens;\r\n  let startType = _startType;\r\n  const originalLength = _messages.length;\r\n  const messages = [..._messages];\r\n  /**\r\n   * IMPORTANT: this context array gets reversed at the end, since the latest messages get pushed first.\r\n   *\r\n   * This may be confusing to read, but it is done to ensure the context is in the correct order for the model.\r\n   * */\r\n  let context: Array<BaseMessage | undefined> = [];\r\n\r\n  let thinkingStartIndex = _thinkingStartIndex;\r\n  let thinkingEndIndex = -1;\r\n  let thinkingBlock: ThinkingContentText | ReasoningContentText | undefined;\r\n  const endIndex = instructions != null ? 1 : 0;\r\n  const prunedMemory: BaseMessage[] = [];\r\n\r\n  if (_thinkingStartIndex > -1) {\r\n    const thinkingMessageContent = messages[_thinkingStartIndex]?.content;\r\n    if (Array.isArray(thinkingMessageContent)) {\r\n      thinkingBlock = thinkingMessageContent.find(\r\n        (content) => content.type === reasoningType\r\n      ) as ThinkingContentText | undefined;\r\n    }\r\n  }\r\n\r\n  if (currentTokenCount < remainingContextTokens) {\r\n    let currentIndex = messages.length;\r\n    while (\r\n      messages.length > 0 &&\r\n      currentTokenCount < remainingContextTokens &&\r\n      currentIndex > endIndex\r\n    ) {\r\n      currentIndex--;\r\n      if (messages.length === 1 && instructions) {\r\n        break;\r\n      }\r\n      const poppedMessage = messages.pop();\r\n      if (!poppedMessage) continue;\r\n      const messageType = poppedMessage.getType();\r\n      if (\r\n        thinkingEnabled === true &&\r\n        thinkingEndIndex === -1 &&\r\n        currentIndex === originalLength - 1 &&\r\n        (messageType === 'ai' || messageType === 'tool')\r\n      ) {\r\n        thinkingEndIndex = currentIndex;\r\n      }\r\n      if (\r\n        thinkingEndIndex > -1 &&\r\n        !thinkingBlock &&\r\n        thinkingStartIndex < 0 &&\r\n        messageType === 'ai' &&\r\n        Array.isArray(poppedMessage.content)\r\n      ) {\r\n        thinkingBlock = poppedMessage.content.find(\r\n          (content) => content.type === reasoningType\r\n        ) as ThinkingContentText | undefined;\r\n        thinkingStartIndex = thinkingBlock != null ? currentIndex : -1;\r\n      }\r\n      /** False start, the latest message was not part of a multi-assistant/tool sequence of messages */\r\n      if (\r\n        thinkingEndIndex > -1 &&\r\n        currentIndex === thinkingEndIndex - 1 &&\r\n        messageType !== 'ai' &&\r\n        messageType !== 'tool'\r\n      ) {\r\n        thinkingEndIndex = -1;\r\n      }\r\n\r\n      const tokenCount = indexTokenCountMap[currentIndex] ?? 0;\r\n\r\n      if (\r\n        prunedMemory.length === 0 &&\r\n        currentTokenCount + tokenCount <= remainingContextTokens\r\n      ) {\r\n        context.push(poppedMessage);\r\n        currentTokenCount += tokenCount;\r\n      } else {\r\n        prunedMemory.push(poppedMessage);\r\n        if (thinkingEndIndex > -1 && thinkingStartIndex < 0) {\r\n          continue;\r\n        }\r\n        break;\r\n      }\r\n    }\r\n\r\n    if (context[context.length - 1]?.getType() === 'tool') {\r\n      startType = ['ai', 'human'];\r\n    }\r\n\r\n    if (startType != null && startType.length > 0 && context.length > 0) {\r\n      let requiredTypeIndex = -1;\r\n\r\n      let totalTokens = 0;\r\n      for (let i = context.length - 1; i >= 0; i--) {\r\n        const currentType = context[i]?.getType() ?? '';\r\n        if (\r\n          Array.isArray(startType)\r\n            ? startType.includes(currentType)\r\n            : currentType === startType\r\n        ) {\r\n          requiredTypeIndex = i + 1;\r\n          break;\r\n        }\r\n        const originalIndex = originalLength - 1 - i;\r\n        totalTokens += indexTokenCountMap[originalIndex] ?? 0;\r\n      }\r\n\r\n      if (requiredTypeIndex > 0) {\r\n        currentTokenCount -= totalTokens;\r\n        context = context.slice(0, requiredTypeIndex);\r\n      }\r\n    }\r\n  }\r\n\r\n  if (instructions && originalLength > 0) {\r\n    context.push(_messages[0] as BaseMessage);\r\n    messages.shift();\r\n  }\r\n\r\n  remainingContextTokens -= currentTokenCount;\r\n  const result: PruningResult = {\r\n    remainingContextTokens,\r\n    context: [] as BaseMessage[],\r\n    messagesToRefine: prunedMemory,\r\n  };\r\n\r\n  if (thinkingStartIndex > -1) {\r\n    result.thinkingStartIndex = thinkingStartIndex;\r\n  }\r\n\r\n  if (\r\n    prunedMemory.length === 0 ||\r\n    thinkingEndIndex < 0 ||\r\n    (thinkingStartIndex > -1 &&\r\n      isIndexInContext(_messages, context, thinkingStartIndex))\r\n  ) {\r\n    // we reverse at this step to ensure the context is in the correct order for the model, and we need to work backwards\r\n    result.context = context.reverse() as BaseMessage[];\r\n    return result;\r\n  }\r\n\r\n  if (thinkingEndIndex > -1 && thinkingStartIndex < 0) {\r\n    throw new Error(\r\n      'The payload is malformed. There is a thinking sequence but no \"AI\" messages with thinking blocks.'\r\n    );\r\n  }\r\n\r\n  if (!thinkingBlock) {\r\n    throw new Error(\r\n      'The payload is malformed. There is a thinking sequence but no thinking block found.'\r\n    );\r\n  }\r\n\r\n  // Since we have a thinking sequence, we need to find the last assistant message\r\n  // in the latest AI/tool sequence to add the thinking block that falls outside of the current context\r\n  // Latest messages are ordered first.\r\n  let assistantIndex = -1;\r\n  for (let i = 0; i < context.length; i++) {\r\n    const currentMessage = context[i];\r\n    const type = currentMessage?.getType();\r\n    if (type === 'ai') {\r\n      assistantIndex = i;\r\n    }\r\n    if (assistantIndex > -1 && (type === 'human' || type === 'system')) {\r\n      break;\r\n    }\r\n  }\r\n\r\n  if (assistantIndex === -1) {\r\n    throw new Error(\r\n      'The payload is malformed. There is a thinking sequence but no \"AI\" messages to append thinking blocks to.'\r\n    );\r\n  }\r\n\r\n  thinkingStartIndex = originalLength - 1 - assistantIndex;\r\n  const thinkingTokenCount = tokenCounter(\r\n    new AIMessage({ content: [thinkingBlock] })\r\n  );\r\n  const newRemainingCount = remainingContextTokens - thinkingTokenCount;\r\n  const newMessage = addThinkingBlock(\r\n    context[assistantIndex] as AIMessage,\r\n    thinkingBlock\r\n  );\r\n  context[assistantIndex] = newMessage;\r\n  if (newRemainingCount > 0) {\r\n    result.context = context.reverse() as BaseMessage[];\r\n    return result;\r\n  }\r\n\r\n  const thinkingMessage: AIMessage = context[assistantIndex] as AIMessage;\r\n  // now we need to an additional round of pruning but making the thinking block fit\r\n  const newThinkingMessageTokenCount =\r\n    (indexTokenCountMap[thinkingStartIndex] ?? 0) + thinkingTokenCount;\r\n  remainingContextTokens = initialContextTokens - newThinkingMessageTokenCount;\r\n  currentTokenCount = 3;\r\n  let newContext: BaseMessage[] = [];\r\n  const secondRoundMessages = [..._messages];\r\n  let currentIndex = secondRoundMessages.length;\r\n  while (\r\n    secondRoundMessages.length > 0 &&\r\n    currentTokenCount < remainingContextTokens &&\r\n    currentIndex > thinkingStartIndex\r\n  ) {\r\n    currentIndex--;\r\n    const poppedMessage = secondRoundMessages.pop();\r\n    if (!poppedMessage) continue;\r\n    const tokenCount = indexTokenCountMap[currentIndex] ?? 0;\r\n    if (currentTokenCount + tokenCount <= remainingContextTokens) {\r\n      newContext.push(poppedMessage);\r\n      currentTokenCount += tokenCount;\r\n    } else {\r\n      messages.push(poppedMessage);\r\n      break;\r\n    }\r\n  }\r\n\r\n  const firstMessage: AIMessage = newContext[newContext.length - 1];\r\n  const firstMessageType = newContext[newContext.length - 1].getType();\r\n  if (firstMessageType === 'tool') {\r\n    startType = ['ai', 'human'];\r\n  }\r\n\r\n  if (startType != null && startType.length > 0 && newContext.length > 0) {\r\n    let requiredTypeIndex = -1;\r\n\r\n    let totalTokens = 0;\r\n    for (let i = newContext.length - 1; i >= 0; i--) {\r\n      const currentType = newContext[i]?.getType() ?? '';\r\n      if (\r\n        Array.isArray(startType)\r\n          ? startType.includes(currentType)\r\n          : currentType === startType\r\n      ) {\r\n        requiredTypeIndex = i + 1;\r\n        break;\r\n      }\r\n      const originalIndex = originalLength - 1 - i;\r\n      totalTokens += indexTokenCountMap[originalIndex] ?? 0;\r\n    }\r\n\r\n    if (requiredTypeIndex > 0) {\r\n      currentTokenCount -= totalTokens;\r\n      newContext = newContext.slice(0, requiredTypeIndex);\r\n    }\r\n  }\r\n\r\n  if (firstMessageType === 'ai') {\r\n    const newMessage = addThinkingBlock(firstMessage, thinkingBlock);\r\n    newContext[newContext.length - 1] = newMessage;\r\n  } else {\r\n    newContext.push(thinkingMessage);\r\n  }\r\n\r\n  if (instructions && originalLength > 0) {\r\n    newContext.push(_messages[0] as BaseMessage);\r\n    secondRoundMessages.shift();\r\n  }\r\n\r\n  result.context = newContext.reverse();\r\n  return result;\r\n}\r\n\r\nexport function checkValidNumber(value: unknown): value is number {\r\n  return typeof value === 'number' && !isNaN(value) && value > 0;\r\n}\r\n\r\ntype ThinkingBlocks = {\r\n  thinking_blocks?: Array<{\r\n    type: 'thinking';\r\n    thinking: string;\r\n    signature: string;\r\n  }>;\r\n};\r\n\r\nexport function createPruneMessages(factoryParams: PruneMessagesFactoryParams) {\r\n  const indexTokenCountMap = { ...factoryParams.indexTokenCountMap };\r\n  let lastTurnStartIndex = factoryParams.startIndex;\r\n  let lastCutOffIndex = 0;\r\n  let totalTokens = Object.values(indexTokenCountMap).reduce(\r\n    (a = 0, b = 0) => a + b,\r\n    0\r\n  ) as number;\r\n  let runThinkingStartIndex = -1;\r\n  return function pruneMessages(params: PruneMessagesParams): {\r\n    context: BaseMessage[];\r\n    indexTokenCountMap: Record<string, number | undefined>;\r\n  } {\r\n    if (\r\n      factoryParams.provider === Providers.OPENAI &&\r\n      factoryParams.thinkingEnabled === true\r\n    ) {\r\n      for (let i = lastTurnStartIndex; i < params.messages.length; i++) {\r\n        const m = params.messages[i];\r\n        if (\r\n          m.getType() === 'ai' &&\r\n          typeof m.additional_kwargs.reasoning_content === 'string' &&\r\n          Array.isArray(\r\n            (\r\n              m.additional_kwargs.provider_specific_fields as\r\n                | ThinkingBlocks\r\n                | undefined\r\n            )?.thinking_blocks\r\n          ) &&\r\n          (m as AIMessage).tool_calls &&\r\n          ((m as AIMessage).tool_calls?.length ?? 0) > 0\r\n        ) {\r\n          const message = m as AIMessage;\r\n          const thinkingBlocks = (\r\n            message.additional_kwargs.provider_specific_fields as ThinkingBlocks\r\n          ).thinking_blocks;\r\n          const signature =\r\n            thinkingBlocks?.[thinkingBlocks.length - 1].signature;\r\n          const thinkingBlock: ThinkingContentText = {\r\n            signature,\r\n            type: ContentTypes.THINKING,\r\n            thinking: message.additional_kwargs.reasoning_content as string,\r\n          };\r\n\r\n          params.messages[i] = new AIMessage({\r\n            ...message,\r\n            content: [thinkingBlock],\r\n            additional_kwargs: {\r\n              ...message.additional_kwargs,\r\n              reasoning_content: undefined,\r\n            },\r\n          });\r\n        }\r\n      }\r\n    }\r\n\r\n    let currentUsage: UsageMetadata | undefined;\r\n    if (\r\n      params.usageMetadata &&\r\n      (checkValidNumber(params.usageMetadata.input_tokens) ||\r\n        (checkValidNumber(params.usageMetadata.input_token_details) &&\r\n          (checkValidNumber(\r\n            params.usageMetadata.input_token_details.cache_creation\r\n          ) ||\r\n            checkValidNumber(\r\n              params.usageMetadata.input_token_details.cache_read\r\n            )))) &&\r\n      checkValidNumber(params.usageMetadata.output_tokens)\r\n    ) {\r\n      currentUsage = calculateTotalTokens(params.usageMetadata);\r\n      totalTokens = currentUsage.total_tokens;\r\n    }\r\n\r\n    const newOutputs = new Set<number>();\r\n    for (let i = lastTurnStartIndex; i < params.messages.length; i++) {\r\n      const message = params.messages[i];\r\n      if (\r\n        i === lastTurnStartIndex &&\r\n        indexTokenCountMap[i] === undefined &&\r\n        currentUsage\r\n      ) {\r\n        indexTokenCountMap[i] = currentUsage.output_tokens;\r\n      } else if (indexTokenCountMap[i] === undefined) {\r\n        indexTokenCountMap[i] = factoryParams.tokenCounter(message);\r\n        if (currentUsage) {\r\n          newOutputs.add(i);\r\n        }\r\n        totalTokens += indexTokenCountMap[i] ?? 0;\r\n      }\r\n    }\r\n\r\n    // If `currentUsage` is defined, we need to distribute the current total tokens to our `indexTokenCountMap`,\r\n    // We must distribute it in a weighted manner, so that the total token count is equal to `currentUsage.total_tokens`,\r\n    // relative the manually counted tokens in `indexTokenCountMap`.\r\n    // EDGE CASE: when the resulting context gets pruned, we should not distribute the usage for messages that are not in the context.\r\n    if (currentUsage) {\r\n      let totalIndexTokens = 0;\r\n      if (params.messages[0].getType() === 'system') {\r\n        totalIndexTokens += indexTokenCountMap[0] ?? 0;\r\n      }\r\n      for (let i = lastCutOffIndex; i < params.messages.length; i++) {\r\n        if (i === 0 && params.messages[0].getType() === 'system') {\r\n          continue;\r\n        }\r\n        if (newOutputs.has(i)) {\r\n          continue;\r\n        }\r\n        totalIndexTokens += indexTokenCountMap[i] ?? 0;\r\n      }\r\n\r\n      // Calculate ratio based only on messages that remain in the context\r\n      const ratio = currentUsage.total_tokens / totalIndexTokens;\r\n      const isRatioSafe = ratio >= 1 / 3 && ratio <= 2.5;\r\n\r\n      // Apply the ratio adjustment only to messages at or after lastCutOffIndex, and only if the ratio is safe\r\n      if (isRatioSafe) {\r\n        if (\r\n          params.messages[0].getType() === 'system' &&\r\n          lastCutOffIndex !== 0\r\n        ) {\r\n          indexTokenCountMap[0] = Math.round(\r\n            (indexTokenCountMap[0] ?? 0) * ratio\r\n          );\r\n        }\r\n\r\n        for (let i = lastCutOffIndex; i < params.messages.length; i++) {\r\n          if (newOutputs.has(i)) {\r\n            continue;\r\n          }\r\n          indexTokenCountMap[i] = Math.round(\r\n            (indexTokenCountMap[i] ?? 0) * ratio\r\n          );\r\n        }\r\n      }\r\n    }\r\n\r\n    lastTurnStartIndex = params.messages.length;\r\n    if (lastCutOffIndex === 0 && totalTokens <= factoryParams.maxTokens) {\r\n      return { context: params.messages, indexTokenCountMap };\r\n    }\r\n\r\n    const { context, thinkingStartIndex } = getMessagesWithinTokenLimit({\r\n      maxContextTokens: factoryParams.maxTokens,\r\n      messages: params.messages,\r\n      indexTokenCountMap,\r\n      startType: params.startType,\r\n      thinkingEnabled: factoryParams.thinkingEnabled,\r\n      tokenCounter: factoryParams.tokenCounter,\r\n      reasoningType:\r\n        factoryParams.provider === Providers.BEDROCK\r\n          ? ContentTypes.REASONING_CONTENT\r\n          : ContentTypes.THINKING,\r\n      thinkingStartIndex:\r\n        factoryParams.thinkingEnabled === true\r\n          ? runThinkingStartIndex\r\n          : undefined,\r\n    });\r\n    runThinkingStartIndex = thinkingStartIndex ?? -1;\r\n    /** The index is the first value of `context`, index relative to `params.messages` */\r\n    lastCutOffIndex = Math.max(\r\n      params.messages.length -\r\n        (context.length - (context[0]?.getType() === 'system' ? 1 : 0)),\r\n      0\r\n    );\r\n\r\n    return { context, indexTokenCountMap };\r\n  };\r\n}\r\n"],"names":[],"mappings":";;;AA2BA,SAAS,gBAAgB,CACvB,MAAiB,EACjB,MAAiB,EACjB,WAAmB,EAAA;IAEnB,MAAM,gBAAgB,GAAG,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM;IACtD,OAAO,WAAW,IAAI,gBAAgB;AACxC;AAEA,SAAS,gBAAgB,CACvB,OAAkB,EAClB,aAAyD,EAAA;IAEzD,MAAM,OAAO,GAA4B,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO;UACjE,OAAO,CAAC;AACX,UAAE;AACA,YAAA;gBACE,IAAI,EAAE,YAAY,CAAC,IAAI;gBACvB,IAAI,EAAE,OAAO,CAAC,OAAO;AACtB,aAAA;SACF;;IAEH,IAAI,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,IAAI,EAAE;AAC1C,QAAA,OAAO,OAAO;;AAEhB,IAAA,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC;IAC9B,OAAO,IAAI,SAAS,CAAC;AACnB,QAAA,GAAG,OAAO;QACV,OAAO;AACR,KAAA,CAAC;AACJ;AAEA;;;;;AAKG;AACG,SAAU,oBAAoB,CAClC,KAA6B,EAAA;IAE7B,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,CAAC;AACvD,IAAA,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,cAAc,CAAC,IAAI,CAAC;AAC5E,IAAA,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,UAAU,CAAC,IAAI,CAAC;AAEpE,IAAA,MAAM,gBAAgB,GAAG,eAAe,GAAG,aAAa,GAAG,SAAS;IACpE,MAAM,iBAAiB,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC;IAE1D,OAAO;AACL,QAAA,YAAY,EAAE,gBAAgB;AAC9B,QAAA,aAAa,EAAE,iBAAiB;QAChC,YAAY,EAAE,gBAAgB,GAAG,iBAAiB;KACnD;AACH;AASA;;;;;;AAMG;AACa,SAAA,2BAA2B,CAAC,EAC1C,QAAQ,EAAE,SAAS,EACnB,gBAAgB,EAChB,kBAAkB,EAClB,SAAS,EAAE,UAAU,EACrB,eAAe,EACf,YAAY,EACZ,kBAAkB,EAAE,mBAAmB,GAAG,EAAE,EAC5C,aAAa,GAAG,YAAY,CAAC,QAAQ,GAUtC,EAAA;;;IAGC,IAAI,iBAAiB,GAAG,CAAC;IACzB,MAAM,YAAY,GAChB,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,QAAQ,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS;IACjE,MAAM,sBAAsB,GAC1B,YAAY,IAAI,IAAI,IAAI,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;AACzD,IAAA,MAAM,oBAAoB,GAAG,gBAAgB,GAAG,sBAAsB;IACtE,IAAI,sBAAsB,GAAG,oBAAoB;IACjD,IAAI,SAAS,GAAG,UAAU;AAC1B,IAAA,MAAM,cAAc,GAAG,SAAS,CAAC,MAAM;AACvC,IAAA,MAAM,QAAQ,GAAG,CAAC,GAAG,SAAS,CAAC;AAC/B;;;;AAIK;IACL,IAAI,OAAO,GAAmC,EAAE;IAEhD,IAAI,kBAAkB,GAAG,mBAAmB;AAC5C,IAAA,IAAI,gBAAgB,GAAG,EAAE;AACzB,IAAA,IAAI,aAAqE;AACzE,IAAA,MAAM,QAAQ,GAAG,YAAY,IAAI,IAAI,GAAG,CAAC,GAAG,CAAC;IAC7C,MAAM,YAAY,GAAkB,EAAE;AAEtC,IAAA,IAAI,mBAAmB,GAAG,EAAE,EAAE;QAC5B,MAAM,sBAAsB,GAAG,QAAQ,CAAC,mBAAmB,CAAC,EAAE,OAAO;AACrE,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,sBAAsB,CAAC,EAAE;AACzC,YAAA,aAAa,GAAG,sBAAsB,CAAC,IAAI,CACzC,CAAC,OAAO,KAAK,OAAO,CAAC,IAAI,KAAK,aAAa,CACT;;;AAIxC,IAAA,IAAI,iBAAiB,GAAG,sBAAsB,EAAE;AAC9C,QAAA,IAAI,YAAY,GAAG,QAAQ,CAAC,MAAM;AAClC,QAAA,OACE,QAAQ,CAAC,MAAM,GAAG,CAAC;AACnB,YAAA,iBAAiB,GAAG,sBAAsB;YAC1C,YAAY,GAAG,QAAQ,EACvB;AACA,YAAA,YAAY,EAAE;YACd,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,IAAI,YAAY,EAAE;gBACzC;;AAEF,YAAA,MAAM,aAAa,GAAG,QAAQ,CAAC,GAAG,EAAE;AACpC,YAAA,IAAI,CAAC,aAAa;gBAAE;AACpB,YAAA,MAAM,WAAW,GAAG,aAAa,CAAC,OAAO,EAAE;YAC3C,IACE,eAAe,KAAK,IAAI;gBACxB,gBAAgB,KAAK,EAAE;gBACvB,YAAY,KAAK,cAAc,GAAG,CAAC;iBAClC,WAAW,KAAK,IAAI,IAAI,WAAW,KAAK,MAAM,CAAC,EAChD;gBACA,gBAAgB,GAAG,YAAY;;YAEjC,IACE,gBAAgB,GAAG,EAAE;AACrB,gBAAA,CAAC,aAAa;AACd,gBAAA,kBAAkB,GAAG,CAAC;AACtB,gBAAA,WAAW,KAAK,IAAI;gBACpB,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,EACpC;AACA,gBAAA,aAAa,GAAG,aAAa,CAAC,OAAO,CAAC,IAAI,CACxC,CAAC,OAAO,KAAK,OAAO,CAAC,IAAI,KAAK,aAAa,CACT;AACpC,gBAAA,kBAAkB,GAAG,aAAa,IAAI,IAAI,GAAG,YAAY,GAAG,EAAE;;;YAGhE,IACE,gBAAgB,GAAG,EAAE;gBACrB,YAAY,KAAK,gBAAgB,GAAG,CAAC;AACrC,gBAAA,WAAW,KAAK,IAAI;gBACpB,WAAW,KAAK,MAAM,EACtB;gBACA,gBAAgB,GAAG,EAAE;;YAGvB,MAAM,UAAU,GAAG,kBAAkB,CAAC,YAAY,CAAC,IAAI,CAAC;AAExD,YAAA,IACE,YAAY,CAAC,MAAM,KAAK,CAAC;AACzB,gBAAA,iBAAiB,GAAG,UAAU,IAAI,sBAAsB,EACxD;AACA,gBAAA,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC;gBAC3B,iBAAiB,IAAI,UAAU;;iBAC1B;AACL,gBAAA,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC;gBAChC,IAAI,gBAAgB,GAAG,EAAE,IAAI,kBAAkB,GAAG,CAAC,EAAE;oBACnD;;gBAEF;;;AAIJ,QAAA,IAAI,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,MAAM,EAAE;AACrD,YAAA,SAAS,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC;;AAG7B,QAAA,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE;AACnE,YAAA,IAAI,iBAAiB,GAAG,EAAE;YAE1B,IAAI,WAAW,GAAG,CAAC;AACnB,YAAA,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;gBAC5C,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;AAC/C,gBAAA,IACE,KAAK,CAAC,OAAO,CAAC,SAAS;AACrB,sBAAE,SAAS,CAAC,QAAQ,CAAC,WAAW;AAChC,sBAAE,WAAW,KAAK,SAAS,EAC7B;AACA,oBAAA,iBAAiB,GAAG,CAAC,GAAG,CAAC;oBACzB;;AAEF,gBAAA,MAAM,aAAa,GAAG,cAAc,GAAG,CAAC,GAAG,CAAC;AAC5C,gBAAA,WAAW,IAAI,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC;;AAGvD,YAAA,IAAI,iBAAiB,GAAG,CAAC,EAAE;gBACzB,iBAAiB,IAAI,WAAW;gBAChC,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC;;;;AAKnD,IAAA,IAAI,YAAY,IAAI,cAAc,GAAG,CAAC,EAAE;QACtC,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAgB,CAAC;QACzC,QAAQ,CAAC,KAAK,EAAE;;IAGlB,sBAAsB,IAAI,iBAAiB;AAC3C,IAAA,MAAM,MAAM,GAAkB;QAC5B,sBAAsB;AACtB,QAAA,OAAO,EAAE,EAAmB;AAC5B,QAAA,gBAAgB,EAAE,YAAY;KAC/B;AAED,IAAA,IAAI,kBAAkB,GAAG,EAAE,EAAE;AAC3B,QAAA,MAAM,CAAC,kBAAkB,GAAG,kBAAkB;;AAGhD,IAAA,IACE,YAAY,CAAC,MAAM,KAAK,CAAC;AACzB,QAAA,gBAAgB,GAAG,CAAC;SACnB,kBAAkB,GAAG,EAAE;YACtB,gBAAgB,CAAC,SAAS,EAAE,OAAO,EAAE,kBAAkB,CAAC,CAAC,EAC3D;;AAEA,QAAA,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,EAAmB;AACnD,QAAA,OAAO,MAAM;;IAGf,IAAI,gBAAgB,GAAG,EAAE,IAAI,kBAAkB,GAAG,CAAC,EAAE;AACnD,QAAA,MAAM,IAAI,KAAK,CACb,mGAAmG,CACpG;;IAGH,IAAI,CAAC,aAAa,EAAE;AAClB,QAAA,MAAM,IAAI,KAAK,CACb,qFAAqF,CACtF;;;;;AAMH,IAAA,IAAI,cAAc,GAAG,EAAE;AACvB,IAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AACvC,QAAA,MAAM,cAAc,GAAG,OAAO,CAAC,CAAC,CAAC;AACjC,QAAA,MAAM,IAAI,GAAG,cAAc,EAAE,OAAO,EAAE;AACtC,QAAA,IAAI,IAAI,KAAK,IAAI,EAAE;YACjB,cAAc,GAAG,CAAC;;AAEpB,QAAA,IAAI,cAAc,GAAG,EAAE,KAAK,IAAI,KAAK,OAAO,IAAI,IAAI,KAAK,QAAQ,CAAC,EAAE;YAClE;;;AAIJ,IAAA,IAAI,cAAc,KAAK,EAAE,EAAE;AACzB,QAAA,MAAM,IAAI,KAAK,CACb,2GAA2G,CAC5G;;AAGH,IAAA,kBAAkB,GAAG,cAAc,GAAG,CAAC,GAAG,cAAc;AACxD,IAAA,MAAM,kBAAkB,GAAG,YAAY,CACrC,IAAI,SAAS,CAAC,EAAE,OAAO,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAC5C;AACD,IAAA,MAAM,iBAAiB,GAAG,sBAAsB,GAAG,kBAAkB;IACrE,MAAM,UAAU,GAAG,gBAAgB,CACjC,OAAO,CAAC,cAAc,CAAc,EACpC,aAAa,CACd;AACD,IAAA,OAAO,CAAC,cAAc,CAAC,GAAG,UAAU;AACpC,IAAA,IAAI,iBAAiB,GAAG,CAAC,EAAE;AACzB,QAAA,MAAM,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,EAAmB;AACnD,QAAA,OAAO,MAAM;;AAGf,IAAA,MAAM,eAAe,GAAc,OAAO,CAAC,cAAc,CAAc;;AAEvE,IAAA,MAAM,4BAA4B,GAChC,CAAC,kBAAkB,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,kBAAkB;AACpE,IAAA,sBAAsB,GAAG,oBAAoB,GAAG,4BAA4B;IAC5E,iBAAiB,GAAG,CAAC;IACrB,IAAI,UAAU,GAAkB,EAAE;AAClC,IAAA,MAAM,mBAAmB,GAAG,CAAC,GAAG,SAAS,CAAC;AAC1C,IAAA,IAAI,YAAY,GAAG,mBAAmB,CAAC,MAAM;AAC7C,IAAA,OACE,mBAAmB,CAAC,MAAM,GAAG,CAAC;AAC9B,QAAA,iBAAiB,GAAG,sBAAsB;QAC1C,YAAY,GAAG,kBAAkB,EACjC;AACA,QAAA,YAAY,EAAE;AACd,QAAA,MAAM,aAAa,GAAG,mBAAmB,CAAC,GAAG,EAAE;AAC/C,QAAA,IAAI,CAAC,aAAa;YAAE;QACpB,MAAM,UAAU,GAAG,kBAAkB,CAAC,YAAY,CAAC,IAAI,CAAC;AACxD,QAAA,IAAI,iBAAiB,GAAG,UAAU,IAAI,sBAAsB,EAAE;AAC5D,YAAA,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC;YAC9B,iBAAiB,IAAI,UAAU;;aAC1B;AACL,YAAA,QAAQ,CAAC,IAAI,CAAC,aAAa,CAAC;YAC5B;;;IAIJ,MAAM,YAAY,GAAc,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;AACjE,IAAA,MAAM,gBAAgB,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,EAAE;AACpE,IAAA,IAAI,gBAAgB,KAAK,MAAM,EAAE;AAC/B,QAAA,SAAS,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC;;AAG7B,IAAA,IAAI,SAAS,IAAI,IAAI,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE;AACtE,QAAA,IAAI,iBAAiB,GAAG,EAAE;QAE1B,IAAI,WAAW,GAAG,CAAC;AACnB,QAAA,KAAK,IAAI,CAAC,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;YAC/C,MAAM,WAAW,GAAG,UAAU,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE;AAClD,YAAA,IACE,KAAK,CAAC,OAAO,CAAC,SAAS;AACrB,kBAAE,SAAS,CAAC,QAAQ,CAAC,WAAW;AAChC,kBAAE,WAAW,KAAK,SAAS,EAC7B;AACA,gBAAA,iBAAiB,GAAG,CAAC,GAAG,CAAC;gBACzB;;AAEF,YAAA,MAAM,aAAa,GAAG,cAAc,GAAG,CAAC,GAAG,CAAC;AAC5C,YAAA,WAAW,IAAI,kBAAkB,CAAC,aAAa,CAAC,IAAI,CAAC;;AAGvD,QAAA,IAAI,iBAAiB,GAAG,CAAC,EAAE;YACzB,iBAAiB,IAAI,WAAW;YAChC,UAAU,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC;;;AAIvD,IAAA,IAAI,gBAAgB,KAAK,IAAI,EAAE;QAC7B,MAAM,UAAU,GAAG,gBAAgB,CAAC,YAAY,EAAE,aAAa,CAAC;QAChE,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,UAAU;;SACzC;AACL,QAAA,UAAU,CAAC,IAAI,CAAC,eAAe,CAAC;;AAGlC,IAAA,IAAI,YAAY,IAAI,cAAc,GAAG,CAAC,EAAE;QACtC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAgB,CAAC;QAC5C,mBAAmB,CAAC,KAAK,EAAE;;AAG7B,IAAA,MAAM,CAAC,OAAO,GAAG,UAAU,CAAC,OAAO,EAAE;AACrC,IAAA,OAAO,MAAM;AACf;AAEM,SAAU,gBAAgB,CAAC,KAAc,EAAA;AAC7C,IAAA,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC;AAChE;AAUM,SAAU,mBAAmB,CAAC,aAAyC,EAAA;IAC3E,MAAM,kBAAkB,GAAG,EAAE,GAAG,aAAa,CAAC,kBAAkB,EAAE;AAClE,IAAA,IAAI,kBAAkB,GAAG,aAAa,CAAC,UAAU;IACjD,IAAI,eAAe,GAAG,CAAC;AACvB,IAAA,IAAI,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,MAAM,CACxD,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EACvB,CAAC,CACQ;AACX,IAAA,IAAI,qBAAqB,GAAG,EAAE;IAC9B,OAAO,SAAS,aAAa,CAAC,MAA2B,EAAA;AAIvD,QAAA,IACE,aAAa,CAAC,QAAQ,KAAK,SAAS,CAAC,MAAM;AAC3C,YAAA,aAAa,CAAC,eAAe,KAAK,IAAI,EACtC;AACA,YAAA,KAAK,IAAI,CAAC,GAAG,kBAAkB,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAChE,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC5B,gBAAA,IACE,CAAC,CAAC,OAAO,EAAE,KAAK,IAAI;AACpB,oBAAA,OAAO,CAAC,CAAC,iBAAiB,CAAC,iBAAiB,KAAK,QAAQ;oBACzD,KAAK,CAAC,OAAO,CAET,CAAC,CAAC,iBAAiB,CAAC,wBAGrB,EAAE,eAAe,CACnB;AACA,oBAAA,CAAe,CAAC,UAAU;oBAC3B,CAAE,CAAe,CAAC,UAAU,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,EAC9C;oBACA,MAAM,OAAO,GAAG,CAAc;oBAC9B,MAAM,cAAc,GAClB,OAAO,CAAC,iBAAiB,CAAC,wBAC3B,CAAC,eAAe;AACjB,oBAAA,MAAM,SAAS,GACb,cAAc,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,SAAS;AACvD,oBAAA,MAAM,aAAa,GAAwB;wBACzC,SAAS;wBACT,IAAI,EAAE,YAAY,CAAC,QAAQ;AAC3B,wBAAA,QAAQ,EAAE,OAAO,CAAC,iBAAiB,CAAC,iBAA2B;qBAChE;oBAED,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,IAAI,SAAS,CAAC;AACjC,wBAAA,GAAG,OAAO;wBACV,OAAO,EAAE,CAAC,aAAa,CAAC;AACxB,wBAAA,iBAAiB,EAAE;4BACjB,GAAG,OAAO,CAAC,iBAAiB;AAC5B,4BAAA,iBAAiB,EAAE,SAAS;AAC7B,yBAAA;AACF,qBAAA,CAAC;;;;AAKR,QAAA,IAAI,YAAuC;QAC3C,IACE,MAAM,CAAC,aAAa;AACpB,aAAC,gBAAgB,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,CAAC;AAClD,iBAAC,gBAAgB,CAAC,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC;qBACxD,gBAAgB,CACf,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,cAAc,CACxD;wBACC,gBAAgB,CACd,MAAM,CAAC,aAAa,CAAC,mBAAmB,CAAC,UAAU,CACpD,CAAC,CAAC,CAAC;YACV,gBAAgB,CAAC,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,EACpD;AACA,YAAA,YAAY,GAAG,oBAAoB,CAAC,MAAM,CAAC,aAAa,CAAC;AACzD,YAAA,WAAW,GAAG,YAAY,CAAC,YAAY;;AAGzC,QAAA,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU;AACpC,QAAA,KAAK,IAAI,CAAC,GAAG,kBAAkB,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;YAChE,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;YAClC,IACE,CAAC,KAAK,kBAAkB;AACxB,gBAAA,kBAAkB,CAAC,CAAC,CAAC,KAAK,SAAS;AACnC,gBAAA,YAAY,EACZ;AACA,gBAAA,kBAAkB,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,aAAa;;AAC7C,iBAAA,IAAI,kBAAkB,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE;gBAC9C,kBAAkB,CAAC,CAAC,CAAC,GAAG,aAAa,CAAC,YAAY,CAAC,OAAO,CAAC;gBAC3D,IAAI,YAAY,EAAE;AAChB,oBAAA,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;;AAEnB,gBAAA,WAAW,IAAI,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC;;;;;;;QAQ7C,IAAI,YAAY,EAAE;YAChB,IAAI,gBAAgB,GAAG,CAAC;AACxB,YAAA,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE;AAC7C,gBAAA,gBAAgB,IAAI,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC;;AAEhD,YAAA,KAAK,IAAI,CAAC,GAAG,eAAe,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC7D,gBAAA,IAAI,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ,EAAE;oBACxD;;AAEF,gBAAA,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;oBACrB;;AAEF,gBAAA,gBAAgB,IAAI,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC;;;AAIhD,YAAA,MAAM,KAAK,GAAG,YAAY,CAAC,YAAY,GAAG,gBAAgB;YAC1D,MAAM,WAAW,GAAG,KAAK,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,IAAI,GAAG;;YAGlD,IAAI,WAAW,EAAE;gBACf,IACE,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,KAAK,QAAQ;oBACzC,eAAe,KAAK,CAAC,EACrB;AACA,oBAAA,kBAAkB,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAChC,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,CACrC;;AAGH,gBAAA,KAAK,IAAI,CAAC,GAAG,eAAe,EAAE,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC7D,oBAAA,IAAI,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;wBACrB;;AAEF,oBAAA,kBAAkB,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAChC,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,CACrC;;;;AAKP,QAAA,kBAAkB,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM;QAC3C,IAAI,eAAe,KAAK,CAAC,IAAI,WAAW,IAAI,aAAa,CAAC,SAAS,EAAE;YACnE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,QAAQ,EAAE,kBAAkB,EAAE;;AAGzD,QAAA,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,GAAG,2BAA2B,CAAC;YAClE,gBAAgB,EAAE,aAAa,CAAC,SAAS;YACzC,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,kBAAkB;YAClB,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,eAAe,EAAE,aAAa,CAAC,eAAe;YAC9C,YAAY,EAAE,aAAa,CAAC,YAAY;AACxC,YAAA,aAAa,EACX,aAAa,CAAC,QAAQ,KAAK,SAAS,CAAC;kBACjC,YAAY,CAAC;kBACb,YAAY,CAAC,QAAQ;AAC3B,YAAA,kBAAkB,EAChB,aAAa,CAAC,eAAe,KAAK;AAChC,kBAAE;AACF,kBAAE,SAAS;AAChB,SAAA,CAAC;AACF,QAAA,qBAAqB,GAAG,kBAAkB,IAAI,EAAE;;QAEhD,eAAe,GAAG,IAAI,CAAC,GAAG,CACxB,MAAM,CAAC,QAAQ,CAAC,MAAM;aACnB,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EACjE,CAAC,CACF;AAED,QAAA,OAAO,EAAE,OAAO,EAAE,kBAAkB,EAAE;AACxC,KAAC;AACH;;;;"}