illuma-agents 1.0.4 → 1.0.6

package/dist/esm/messages/cache.mjs

@@ -2,6 +2,9 @@ import { ContentTypes } from '../common/enum.mjs';
 
 /**
  * Anthropic API: Adds cache control to the appropriate user messages in the payload.
+ * Strips ALL existing cache control (both Anthropic and Bedrock formats) from all messages,
+ * then adds fresh cache control to the last 2 user messages in a single backward pass.
+ * This ensures we don't accumulate stale cache points across multiple turns.
  * @param messages - The array of message objects.
  * @returns - The updated array of message objects with cache control added.
  */
@@ -11,12 +14,20 @@ function addCacheControl(messages) {
     }
     const updatedMessages = [...messages];
     let userMessagesModified = 0;
-    for (let i = updatedMessages.length - 1; i >= 0 && userMessagesModified < 2; i--) {
+    for (let i = updatedMessages.length - 1; i >= 0; i--) {
         const message = updatedMessages[i];
-        if ('getType' in message && message.getType() !== 'human') {
-            continue;
+        const isUserMessage = ('getType' in message && message.getType() === 'human') ||
+            ('role' in message && message.role === 'user');
+        if (Array.isArray(message.content)) {
+            message.content = message.content.filter((block) => !isCachePoint(block));
+            for (let j = 0; j < message.content.length; j++) {
+                const block = message.content[j];
+                if ('cache_control' in block) {
+                    delete block.cache_control;
+                }
+            }
         }
-        else if ('role' in message && message.role !== 'user') {
+        if (userMessagesModified >= 2 || !isUserMessage) {
             continue;
         }
         if (typeof message.content === 'string') {
@@ -44,10 +55,60 @@ function addCacheControl(messages) {
     }
     return updatedMessages;
 }
+/**
+ * Checks if a content block is a cache point
+ */
+function isCachePoint(block) {
+    return 'cachePoint' in block && !('type' in block);
+}
+/**
+ * Removes all Anthropic cache_control fields from messages
+ * Used when switching from Anthropic to Bedrock provider
+ */
+function stripAnthropicCacheControl(messages) {
+    if (!Array.isArray(messages)) {
+        return messages;
+    }
+    const updatedMessages = [...messages];
+    for (let i = 0; i < updatedMessages.length; i++) {
+        const message = updatedMessages[i];
+        const content = message.content;
+        if (Array.isArray(content)) {
+            for (let j = 0; j < content.length; j++) {
+                const block = content[j];
+                if ('cache_control' in block) {
+                    delete block.cache_control;
+                }
+            }
+        }
+    }
+    return updatedMessages;
+}
+/**
+ * Removes all Bedrock cachePoint blocks from messages
+ * Used when switching from Bedrock to Anthropic provider
+ */
+function stripBedrockCacheControl(messages) {
+    if (!Array.isArray(messages)) {
+        return messages;
+    }
+    const updatedMessages = [...messages];
+    for (let i = 0; i < updatedMessages.length; i++) {
+        const message = updatedMessages[i];
+        const content = message.content;
+        if (Array.isArray(content)) {
+            message.content = content.filter((block) => !isCachePoint(block));
+        }
+    }
+    return updatedMessages;
+}
 /**
  * Adds Bedrock Converse API cache points to the last two messages.
  * Inserts `{ cachePoint: { type: 'default' } }` as a separate content block
  * immediately after the last text block in each targeted message.
+ * Strips ALL existing cache control (both Bedrock and Anthropic formats) from all messages,
+ * then adds fresh cache points to the last 2 messages in a single backward pass.
+ * This ensures we don't accumulate stale cache points across multiple turns.
  * @param messages - The array of message objects.
  * @returns - The updated array of message objects with cache points added.
  */
@@ -57,14 +118,24 @@ function addBedrockCacheControl(messages) {
     }
     const updatedMessages = messages.slice();
     let messagesModified = 0;
-    for (let i = updatedMessages.length - 1; i >= 0 && messagesModified < 2; i--) {
+    for (let i = updatedMessages.length - 1; i >= 0; i--) {
         const message = updatedMessages[i];
-        if ('getType' in message &&
+        const isToolMessage = 'getType' in message &&
             typeof message.getType === 'function' &&
-            message.getType() === 'tool') {
+            message.getType() === 'tool';
+        const content = message.content;
+        if (Array.isArray(content)) {
+            message.content = content.filter((block) => !isCachePoint(block));
+            for (let j = 0; j < message.content.length; j++) {
+                const block = message.content[j];
+                if ('cache_control' in block) {
+                    delete block.cache_control;
+                }
+            }
+        }
+        if (messagesModified >= 2 || isToolMessage) {
             continue;
         }
-        const content = message.content;
         if (typeof content === 'string' && content === '') {
             continue;
         }
@@ -76,9 +147,9 @@ function addBedrockCacheControl(messages) {
             messagesModified++;
             continue;
         }
-        if (Array.isArray(content)) {
+        if (Array.isArray(message.content)) {
             let hasCacheableContent = false;
-            for (const block of content) {
+            for (const block of message.content) {
                 if (block.type === ContentTypes.TEXT) {
                     if (typeof block.text === 'string' && block.text !== '') {
                         hasCacheableContent = true;
@@ -90,15 +161,15 @@ function addBedrockCacheControl(messages) {
                 continue;
             }
             let inserted = false;
-            for (let j = content.length - 1; j >= 0; j--) {
-                const block = content[j];
+            for (let j = message.content.length - 1; j >= 0; j--) {
+                const block = message.content[j];
                 const type = block.type;
                 if (type === ContentTypes.TEXT || type === 'text') {
                     const text = block.text;
                     if (text === '' || text === undefined) {
                         continue;
                     }
-                    content.splice(j + 1, 0, {
+                    message.content.splice(j + 1, 0, {
                         cachePoint: { type: 'default' },
                     });
                     inserted = true;
@@ -106,7 +177,7 @@ function addBedrockCacheControl(messages) {
                 }
             }
             if (!inserted) {
-                content.push({
+                message.content.push({
                     cachePoint: { type: 'default' },
                 });
             }
@@ -116,5 +187,5 @@ function addBedrockCacheControl(messages) {
     return updatedMessages;
 }
 
-export { addBedrockCacheControl, addCacheControl };
+export { addBedrockCacheControl, addCacheControl, stripAnthropicCacheControl, stripBedrockCacheControl };
 //# sourceMappingURL=cache.mjs.map

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

@@ -1 +1 @@
-{"version":3,"file":"cache.mjs","sources":["../../../src/messages/cache.ts"],"sourcesContent":["import { BaseMessage, MessageContentComplex } from '@langchain/core/messages';\nimport type { AnthropicMessage } from '@/types/messages';\nimport type Anthropic from '@anthropic-ai/sdk';\nimport { ContentTypes } from '@/common/enum';\n\ntype MessageWithContent = {\n  content?: string | MessageContentComplex[];\n};\n\n/**\n * Anthropic API: Adds cache control to the appropriate user messages in the payload.\n * @param messages - The array of message objects.\n * @returns - The updated array of message objects with cache control added.\n */\nexport function addCacheControl<T extends AnthropicMessage | BaseMessage>(\n  messages: T[]\n): T[] {\n  if (!Array.isArray(messages) || messages.length < 2) {\n    return messages;\n  }\n\n  const updatedMessages = [...messages];\n  let userMessagesModified = 0;\n\n  for (\n    let i = updatedMessages.length - 1;\n    i >= 0 && userMessagesModified < 2;\n    i--\n  ) {\n    const message = updatedMessages[i];\n    if ('getType' in message && message.getType() !== 'human') {\n      continue;\n    } else if ('role' in message && message.role !== 'user') {\n      continue;\n    }\n\n    if (typeof message.content === 'string') {\n      message.content = [\n        {\n          type: 'text',\n          text: message.content,\n          cache_control: { type: 'ephemeral' },\n        },\n      ];\n      userMessagesModified++;\n    } else if (Array.isArray(message.content)) {\n      for (let j = message.content.length - 1; j >= 0; j--) {\n        const contentPart = message.content[j];\n        if ('type' in contentPart && contentPart.type === 'text') {\n          (contentPart as Anthropic.TextBlockParam).cache_control = {\n            type: 'ephemeral',\n          };\n          userMessagesModified++;\n          break;\n        }\n      }\n    }\n  }\n\n  return updatedMessages;\n}\n\n/**\n * Adds Bedrock Converse API cache points to the last two messages.\n * Inserts `{ cachePoint: { type: 'default' } }` as a separate content block\n * immediately after the last text block in each targeted message.\n * @param messages - The array of message objects.\n * @returns - The updated array of message objects with cache points added.\n */\nexport function addBedrockCacheControl<\n  T extends Partial<BaseMessage> & MessageWithContent,\n>(messages: T[]): T[] {\n  if (!Array.isArray(messages) || messages.length < 2) {\n    return messages;\n  }\n\n  const updatedMessages: T[] = messages.slice();\n  let messagesModified = 0;\n\n  for (\n    let i = updatedMessages.length - 1;\n    i >= 0 && messagesModified < 2;\n    i--\n  ) {\n    const message = updatedMessages[i];\n\n    if (\n      'getType' in message &&\n      typeof message.getType === 'function' &&\n      message.getType() === 'tool'\n    ) {\n      continue;\n    }\n\n    const content = message.content;\n\n    if (typeof content === 'string' && content === '') {\n      continue;\n    }\n\n    if (typeof content === 'string') {\n      message.content = [\n        { type: ContentTypes.TEXT, text: content },\n        { cachePoint: { type: 'default' } },\n      ] as MessageContentComplex[];\n      messagesModified++;\n      continue;\n    }\n\n    if (Array.isArray(content)) {\n      let hasCacheableContent = false;\n      for (const block of content) {\n        if (block.type === ContentTypes.TEXT) {\n          if (typeof block.text === 'string' && block.text !== '') {\n            hasCacheableContent = true;\n            break;\n          }\n        }\n      }\n\n      if (!hasCacheableContent) {\n        continue;\n      }\n\n      let inserted = false;\n      for (let j = content.length - 1; j >= 0; j--) {\n        const block = content[j] as MessageContentComplex;\n        const type = (block as { type?: string }).type;\n        if (type === ContentTypes.TEXT || type === 'text') {\n          const text = (block as { text?: string }).text;\n          if (text === '' || text === undefined) {\n            continue;\n          }\n          content.splice(j + 1, 0, {\n            cachePoint: { type: 'default' },\n          } as MessageContentComplex);\n          inserted = true;\n          break;\n        }\n      }\n      if (!inserted) {\n        content.push({\n          cachePoint: { type: 'default' },\n        } as MessageContentComplex);\n      }\n      messagesModified++;\n    }\n  }\n\n  return updatedMessages;\n}\n"],"names":[],"mappings":";;AASA;;;;AAIG;AACG,SAAU,eAAe,CAC7B,QAAa,EAAA;AAEb,IAAA,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;AACnD,QAAA,OAAO,QAAQ;;AAGjB,IAAA,MAAM,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAC;IACrC,IAAI,oBAAoB,GAAG,CAAC;IAE5B,KACE,IAAI,CAAC,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,EAClC,CAAC,IAAI,CAAC,IAAI,oBAAoB,GAAG,CAAC,EAClC,CAAC,EAAE,EACH;AACA,QAAA,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC;QAClC,IAAI,SAAS,IAAI,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,KAAK,OAAO,EAAE;YACzD;;aACK,IAAI,MAAM,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE;YACvD;;AAGF,QAAA,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE;YACvC,OAAO,CAAC,OAAO,GAAG;AAChB,gBAAA;AACE,oBAAA,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO,CAAC,OAAO;AACrB,oBAAA,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;AACrC,iBAAA;aACF;AACD,YAAA,oBAAoB,EAAE;;aACjB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AACzC,YAAA,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;gBACpD,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,IAAI,MAAM,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE;oBACvD,WAAwC,CAAC,aAAa,GAAG;AACxD,wBAAA,IAAI,EAAE,WAAW;qBAClB;AACD,oBAAA,oBAAoB,EAAE;oBACtB;;;;;AAMR,IAAA,OAAO,eAAe;AACxB;AAEA;;;;;;AAMG;AACG,SAAU,sBAAsB,CAEpC,QAAa,EAAA;AACb,IAAA,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;AACnD,QAAA,OAAO,QAAQ;;AAGjB,IAAA,MAAM,eAAe,GAAQ,QAAQ,CAAC,KAAK,EAAE;IAC7C,IAAI,gBAAgB,GAAG,CAAC;IAExB,KACE,IAAI,CAAC,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,EAClC,CAAC,IAAI,CAAC,IAAI,gBAAgB,GAAG,CAAC,EAC9B,CAAC,EAAE,EACH;AACA,QAAA,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC;QAElC,IACE,SAAS,IAAI,OAAO;AACpB,YAAA,OAAO,OAAO,CAAC,OAAO,KAAK,UAAU;AACrC,YAAA,OAAO,CAAC,OAAO,EAAE,KAAK,MAAM,EAC5B;YACA;;AAGF,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO;QAE/B,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,EAAE,EAAE;YACjD;;AAGF,QAAA,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;YAC/B,OAAO,CAAC,OAAO,GAAG;gBAChB,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE;AAC1C,gBAAA,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;aACT;AAC5B,YAAA,gBAAgB,EAAE;YAClB;;AAGF,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAC1B,IAAI,mBAAmB,GAAG,KAAK;AAC/B,YAAA,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE;gBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,EAAE;AACpC,oBAAA,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,KAAK,EAAE,EAAE;wBACvD,mBAAmB,GAAG,IAAI;wBAC1B;;;;YAKN,IAAI,CAAC,mBAAmB,EAAE;gBACxB;;YAGF,IAAI,QAAQ,GAAG,KAAK;AACpB,YAAA,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;AAC5C,gBAAA,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAA0B;AACjD,gBAAA,MAAM,IAAI,GAAI,KAA2B,CAAC,IAAI;gBAC9C,IAAI,IAAI,KAAK,YAAY,CAAC,IAAI,IAAI,IAAI,KAAK,MAAM,EAAE;AACjD,oBAAA,MAAM,IAAI,GAAI,KAA2B,CAAC,IAAI;oBAC9C,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,SAAS,EAAE;wBACrC;;oBAEF,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;AACvB,wBAAA,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;AACP,qBAAA,CAAC;oBAC3B,QAAQ,GAAG,IAAI;oBACf;;;YAGJ,IAAI,CAAC,QAAQ,EAAE;gBACb,OAAO,CAAC,IAAI,CAAC;AACX,oBAAA,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;AACP,iBAAA,CAAC;;AAE7B,YAAA,gBAAgB,EAAE;;;AAItB,IAAA,OAAO,eAAe;AACxB;;;;"}
+{"version":3,"file":"cache.mjs","sources":["../../../src/messages/cache.ts"],"sourcesContent":["import { BaseMessage, MessageContentComplex } from '@langchain/core/messages';\nimport type { AnthropicMessage } from '@/types/messages';\nimport type Anthropic from '@anthropic-ai/sdk';\nimport { ContentTypes } from '@/common/enum';\n\ntype MessageWithContent = {\n  content?: string | MessageContentComplex[];\n};\n\n/**\n * Anthropic API: Adds cache control to the appropriate user messages in the payload.\n * Strips ALL existing cache control (both Anthropic and Bedrock formats) from all messages,\n * then adds fresh cache control to the last 2 user messages in a single backward pass.\n * This ensures we don't accumulate stale cache points across multiple turns.\n * @param messages - The array of message objects.\n * @returns - The updated array of message objects with cache control added.\n */\nexport function addCacheControl<T extends AnthropicMessage | BaseMessage>(\n  messages: T[]\n): T[] {\n  if (!Array.isArray(messages) || messages.length < 2) {\n    return messages;\n  }\n\n  const updatedMessages = [...messages];\n  let userMessagesModified = 0;\n\n  for (let i = updatedMessages.length - 1; i >= 0; i--) {\n    const message = updatedMessages[i];\n    const isUserMessage =\n      ('getType' in message && message.getType() === 'human') ||\n      ('role' in message && message.role === 'user');\n\n    if (Array.isArray(message.content)) {\n      message.content = message.content.filter(\n        (block) => !isCachePoint(block as MessageContentComplex)\n      ) as typeof message.content;\n\n      for (let j = 0; j < message.content.length; j++) {\n        const block = message.content[j] as Record<string, unknown>;\n        if ('cache_control' in block) {\n          delete block.cache_control;\n        }\n      }\n    }\n\n    if (userMessagesModified >= 2 || !isUserMessage) {\n      continue;\n    }\n\n    if (typeof message.content === 'string') {\n      message.content = [\n        {\n          type: 'text',\n          text: message.content,\n          cache_control: { type: 'ephemeral' },\n        },\n      ];\n      userMessagesModified++;\n    } else if (Array.isArray(message.content)) {\n      for (let j = message.content.length - 1; j >= 0; j--) {\n        const contentPart = message.content[j];\n        if ('type' in contentPart && contentPart.type === 'text') {\n          (contentPart as Anthropic.TextBlockParam).cache_control = {\n            type: 'ephemeral',\n          };\n          userMessagesModified++;\n          break;\n        }\n      }\n    }\n  }\n\n  return updatedMessages;\n}\n\n/**\n * Checks if a content block is a cache point\n */\nfunction isCachePoint(block: MessageContentComplex): boolean {\n  return 'cachePoint' in block && !('type' in block);\n}\n\n/**\n * Removes all Anthropic cache_control fields from messages\n * Used when switching from Anthropic to Bedrock provider\n */\nexport function stripAnthropicCacheControl<T extends MessageWithContent>(\n  messages: T[]\n): T[] {\n  if (!Array.isArray(messages)) {\n    return messages;\n  }\n\n  const updatedMessages = [...messages];\n\n  for (let i = 0; i < updatedMessages.length; i++) {\n    const message = updatedMessages[i];\n    const content = message.content;\n\n    if (Array.isArray(content)) {\n      for (let j = 0; j < content.length; j++) {\n        const block = content[j] as Record<string, unknown>;\n        if ('cache_control' in block) {\n          delete block.cache_control;\n        }\n      }\n    }\n  }\n\n  return updatedMessages;\n}\n\n/**\n * Removes all Bedrock cachePoint blocks from messages\n * Used when switching from Bedrock to Anthropic provider\n */\nexport function stripBedrockCacheControl<T extends MessageWithContent>(\n  messages: T[]\n): T[] {\n  if (!Array.isArray(messages)) {\n    return messages;\n  }\n\n  const updatedMessages = [...messages];\n\n  for (let i = 0; i < updatedMessages.length; i++) {\n    const message = updatedMessages[i];\n    const content = message.content;\n\n    if (Array.isArray(content)) {\n      message.content = content.filter(\n        (block) => !isCachePoint(block as MessageContentComplex)\n      ) as typeof content;\n    }\n  }\n\n  return updatedMessages;\n}\n\n/**\n * Adds Bedrock Converse API cache points to the last two messages.\n * Inserts `{ cachePoint: { type: 'default' } }` as a separate content block\n * immediately after the last text block in each targeted message.\n * Strips ALL existing cache control (both Bedrock and Anthropic formats) from all messages,\n * then adds fresh cache points to the last 2 messages in a single backward pass.\n * This ensures we don't accumulate stale cache points across multiple turns.\n * @param messages - The array of message objects.\n * @returns - The updated array of message objects with cache points added.\n */\nexport function addBedrockCacheControl<\n  T extends Partial<BaseMessage> & MessageWithContent,\n>(messages: T[]): T[] {\n  if (!Array.isArray(messages) || messages.length < 2) {\n    return messages;\n  }\n\n  const updatedMessages: T[] = messages.slice();\n  let messagesModified = 0;\n\n  for (let i = updatedMessages.length - 1; i >= 0; i--) {\n    const message = updatedMessages[i];\n    const isToolMessage =\n      'getType' in message &&\n      typeof message.getType === 'function' &&\n      message.getType() === 'tool';\n\n    const content = message.content;\n\n    if (Array.isArray(content)) {\n      message.content = content.filter(\n        (block) => !isCachePoint(block)\n      ) as typeof content;\n\n      for (let j = 0; j < message.content.length; j++) {\n        const block = message.content[j] as Record<string, unknown>;\n        if ('cache_control' in block) {\n          delete block.cache_control;\n        }\n      }\n    }\n\n    if (messagesModified >= 2 || isToolMessage) {\n      continue;\n    }\n\n    if (typeof content === 'string' && content === '') {\n      continue;\n    }\n\n    if (typeof content === 'string') {\n      message.content = [\n        { type: ContentTypes.TEXT, text: content },\n        { cachePoint: { type: 'default' } },\n      ] as MessageContentComplex[];\n      messagesModified++;\n      continue;\n    }\n\n    if (Array.isArray(message.content)) {\n      let hasCacheableContent = false;\n      for (const block of message.content) {\n        if (block.type === ContentTypes.TEXT) {\n          if (typeof block.text === 'string' && block.text !== '') {\n            hasCacheableContent = true;\n            break;\n          }\n        }\n      }\n\n      if (!hasCacheableContent) {\n        continue;\n      }\n\n      let inserted = false;\n      for (let j = message.content.length - 1; j >= 0; j--) {\n        const block = message.content[j] as MessageContentComplex;\n        const type = (block as { type?: string }).type;\n        if (type === ContentTypes.TEXT || type === 'text') {\n          const text = (block as { text?: string }).text;\n          if (text === '' || text === undefined) {\n            continue;\n          }\n          message.content.splice(j + 1, 0, {\n            cachePoint: { type: 'default' },\n          } as MessageContentComplex);\n          inserted = true;\n          break;\n        }\n      }\n      if (!inserted) {\n        message.content.push({\n          cachePoint: { type: 'default' },\n        } as MessageContentComplex);\n      }\n      messagesModified++;\n    }\n  }\n\n  return updatedMessages;\n}\n"],"names":[],"mappings":";;AASA;;;;;;;AAOG;AACG,SAAU,eAAe,CAC7B,QAAa,EAAA;AAEb,IAAA,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;AACnD,QAAA,OAAO,QAAQ;;AAGjB,IAAA,MAAM,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAC;IACrC,IAAI,oBAAoB,GAAG,CAAC;AAE5B,IAAA,KAAK,IAAI,CAAC,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;AACpD,QAAA,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC;AAClC,QAAA,MAAM,aAAa,GACjB,CAAC,SAAS,IAAI,OAAO,IAAI,OAAO,CAAC,OAAO,EAAE,KAAK,OAAO;aACrD,MAAM,IAAI,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC;QAEhD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAClC,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CACtC,CAAC,KAAK,KAAK,CAAC,YAAY,CAAC,KAA8B,CAAC,CAC/B;AAE3B,YAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAA4B;AAC3D,gBAAA,IAAI,eAAe,IAAI,KAAK,EAAE;oBAC5B,OAAO,KAAK,CAAC,aAAa;;;;AAKhC,QAAA,IAAI,oBAAoB,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE;YAC/C;;AAGF,QAAA,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE;YACvC,OAAO,CAAC,OAAO,GAAG;AAChB,gBAAA;AACE,oBAAA,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO,CAAC,OAAO;AACrB,oBAAA,aAAa,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;AACrC,iBAAA;aACF;AACD,YAAA,oBAAoB,EAAE;;aACjB,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AACzC,YAAA,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;gBACpD,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;gBACtC,IAAI,MAAM,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,MAAM,EAAE;oBACvD,WAAwC,CAAC,aAAa,GAAG;AACxD,wBAAA,IAAI,EAAE,WAAW;qBAClB;AACD,oBAAA,oBAAoB,EAAE;oBACtB;;;;;AAMR,IAAA,OAAO,eAAe;AACxB;AAEA;;AAEG;AACH,SAAS,YAAY,CAAC,KAA4B,EAAA;IAChD,OAAO,YAAY,IAAI,KAAK,IAAI,EAAE,MAAM,IAAI,KAAK,CAAC;AACpD;AAEA;;;AAGG;AACG,SAAU,0BAA0B,CACxC,QAAa,EAAA;IAEb,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;AAC5B,QAAA,OAAO,QAAQ;;AAGjB,IAAA,MAAM,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAC;AAErC,IAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC/C,QAAA,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC;AAClC,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO;AAE/B,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AAC1B,YAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AACvC,gBAAA,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAA4B;AACnD,gBAAA,IAAI,eAAe,IAAI,KAAK,EAAE;oBAC5B,OAAO,KAAK,CAAC,aAAa;;;;;AAMlC,IAAA,OAAO,eAAe;AACxB;AAEA;;;AAGG;AACG,SAAU,wBAAwB,CACtC,QAAa,EAAA;IAEb,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;AAC5B,QAAA,OAAO,QAAQ;;AAGjB,IAAA,MAAM,eAAe,GAAG,CAAC,GAAG,QAAQ,CAAC;AAErC,IAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;AAC/C,QAAA,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC;AAClC,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO;AAE/B,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AAC1B,YAAA,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAC9B,CAAC,KAAK,KAAK,CAAC,YAAY,CAAC,KAA8B,CAAC,CACvC;;;AAIvB,IAAA,OAAO,eAAe;AACxB;AAEA;;;;;;;;;AASG;AACG,SAAU,sBAAsB,CAEpC,QAAa,EAAA;AACb,IAAA,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE;AACnD,QAAA,OAAO,QAAQ;;AAGjB,IAAA,MAAM,eAAe,GAAQ,QAAQ,CAAC,KAAK,EAAE;IAC7C,IAAI,gBAAgB,GAAG,CAAC;AAExB,IAAA,KAAK,IAAI,CAAC,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;AACpD,QAAA,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC;AAClC,QAAA,MAAM,aAAa,GACjB,SAAS,IAAI,OAAO;AACpB,YAAA,OAAO,OAAO,CAAC,OAAO,KAAK,UAAU;AACrC,YAAA,OAAO,CAAC,OAAO,EAAE,KAAK,MAAM;AAE9B,QAAA,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO;AAE/B,QAAA,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;AAC1B,YAAA,OAAO,CAAC,OAAO,GAAG,OAAO,CAAC,MAAM,CAC9B,CAAC,KAAK,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,CACd;AAEnB,YAAA,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE;gBAC/C,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAA4B;AAC3D,gBAAA,IAAI,eAAe,IAAI,KAAK,EAAE;oBAC5B,OAAO,KAAK,CAAC,aAAa;;;;AAKhC,QAAA,IAAI,gBAAgB,IAAI,CAAC,IAAI,aAAa,EAAE;YAC1C;;QAGF,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,KAAK,EAAE,EAAE;YACjD;;AAGF,QAAA,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;YAC/B,OAAO,CAAC,OAAO,GAAG;gBAChB,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE;AAC1C,gBAAA,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE;aACT;AAC5B,YAAA,gBAAgB,EAAE;YAClB;;QAGF,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;YAClC,IAAI,mBAAmB,GAAG,KAAK;AAC/B,YAAA,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,EAAE;gBACnC,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,IAAI,EAAE;AACpC,oBAAA,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,KAAK,EAAE,EAAE;wBACvD,mBAAmB,GAAG,IAAI;wBAC1B;;;;YAKN,IAAI,CAAC,mBAAmB,EAAE;gBACxB;;YAGF,IAAI,QAAQ,GAAG,KAAK;AACpB,YAAA,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE;gBACpD,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAA0B;AACzD,gBAAA,MAAM,IAAI,GAAI,KAA2B,CAAC,IAAI;gBAC9C,IAAI,IAAI,KAAK,YAAY,CAAC,IAAI,IAAI,IAAI,KAAK,MAAM,EAAE;AACjD,oBAAA,MAAM,IAAI,GAAI,KAA2B,CAAC,IAAI;oBAC9C,IAAI,IAAI,KAAK,EAAE,IAAI,IAAI,KAAK,SAAS,EAAE;wBACrC;;oBAEF,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;AAC/B,wBAAA,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;AACP,qBAAA,CAAC;oBAC3B,QAAQ,GAAG,IAAI;oBACf;;;YAGJ,IAAI,CAAC,QAAQ,EAAE;AACb,gBAAA,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC;AACnB,oBAAA,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;AACP,iBAAA,CAAC;;AAE7B,YAAA,gBAAgB,EAAE;;;AAItB,IAAA,OAAO,eAAe;AACxB;;;;"}

package/dist/esm/messages/format.mjs

@@ -1,4 +1,4 @@
-import { HumanMessage, AIMessage, SystemMessage, getBufferString, ToolMessage } from '@langchain/core/messages';
+import { HumanMessage, AIMessage, SystemMessage, getBufferString, ToolMessage, AIMessageChunk } from '@langchain/core/messages';
 import { Providers, ContentTypes } from '../common/enum.mjs';
 
 /* eslint-disable @typescript-eslint/no-explicit-any */
@@ -180,19 +180,20 @@ function formatAssistantMessage(message) {
                 }
                 // Create a new AIMessage with this text and prepare for tool calls
                 lastAIMessage = new AIMessage({
-                    content: part.text || '',
+                    content: part.text != null ? part.text : '',
                 });
                 formattedMessages.push(lastAIMessage);
             }
             else if (part.type === ContentTypes.TOOL_CALL) {
                 // Skip malformed tool call entries without tool_call property
-                if (!part.tool_call) {
+                if (part.tool_call == null) {
                     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 === '')) {
+                if (_tool_call.name == null ||
+                    (_tool_call.name === '' && (output == null || output === ''))) {
                     continue;
                 }
                 if (!lastAIMessage) {
@@ -221,7 +222,7 @@ function formatAssistantMessage(message) {
                 formattedMessages.push(new ToolMessage({
                     tool_call_id: tool_call.id ?? '',
                     name: tool_call.name,
-                    content: output || '',
+                    content: output != null ? output : '',
                 }));
             }
             else if (part.type === ContentTypes.THINK) {
@@ -255,6 +256,177 @@ function formatAssistantMessage(message) {
     }
     return formattedMessages;
 }
+/**
+ * Labels all agent content for parallel patterns (fan-out/fan-in)
+ * Groups consecutive content by agent and wraps with clear labels
+ */
+function labelAllAgentContent(contentParts, agentIdMap, agentNames) {
+    const result = [];
+    let currentAgentId;
+    let agentContentBuffer = [];
+    const flushAgentBuffer = () => {
+        if (agentContentBuffer.length === 0) {
+            return;
+        }
+        if (currentAgentId != null && currentAgentId !== '') {
+            const agentName = (agentNames?.[currentAgentId] ?? '') || currentAgentId;
+            const formattedParts = [];
+            formattedParts.push(`--- ${agentName} ---`);
+            for (const part of agentContentBuffer) {
+                if (part.type === ContentTypes.THINK) {
+                    const thinkContent = part.think || '';
+                    if (thinkContent) {
+                        formattedParts.push(`${agentName}: ${JSON.stringify({
+                            type: 'think',
+                            think: thinkContent,
+                        })}`);
+                    }
+                }
+                else if (part.type === ContentTypes.TEXT) {
+                    const textContent = part.text ?? '';
+                    if (textContent) {
+                        formattedParts.push(`${agentName}: ${textContent}`);
+                    }
+                }
+                else if (part.type === ContentTypes.TOOL_CALL) {
+                    formattedParts.push(`${agentName}: ${JSON.stringify({
+                        type: 'tool_call',
+                        tool_call: part.tool_call,
+                    })}`);
+                }
+            }
+            formattedParts.push(`--- End of ${agentName} ---`);
+            // Create a single text content part with all agent content
+            result.push({
+                type: ContentTypes.TEXT,
+                text: formattedParts.join('\n\n'),
+            });
+        }
+        else {
+            // No agent ID, pass through as-is
+            result.push(...agentContentBuffer);
+        }
+        agentContentBuffer = [];
+    };
+    for (let i = 0; i < contentParts.length; i++) {
+        const part = contentParts[i];
+        const agentId = agentIdMap[i];
+        // If agent changed, flush previous buffer
+        if (agentId !== currentAgentId && currentAgentId !== undefined) {
+            flushAgentBuffer();
+        }
+        currentAgentId = agentId;
+        agentContentBuffer.push(part);
+    }
+    // Flush any remaining content
+    flushAgentBuffer();
+    return result;
+}
+/**
+ * Groups content parts by agent and formats them with agent labels
+ * This preprocesses multi-agent content to prevent identity confusion
+ *
+ * @param contentParts - The content parts from a run
+ * @param agentIdMap - Map of content part index to agent ID
+ * @param agentNames - Optional map of agent ID to display name
+ * @param options - Configuration options
+ * @param options.labelNonTransferContent - If true, labels all agent transitions (for parallel patterns)
+ * @returns Modified content parts with agent labels where appropriate
+ */
+const labelContentByAgent = (contentParts, agentIdMap, agentNames, options) => {
+    if (!agentIdMap || Object.keys(agentIdMap).length === 0) {
+        return contentParts;
+    }
+    // If labelNonTransferContent is true, use a different strategy for parallel patterns
+    if (options?.labelNonTransferContent === true) {
+        return labelAllAgentContent(contentParts, agentIdMap, agentNames);
+    }
+    const result = [];
+    let currentAgentId;
+    let agentContentBuffer = [];
+    let transferToolCallIndex;
+    let transferToolCallId;
+    const flushAgentBuffer = () => {
+        if (agentContentBuffer.length === 0) {
+            return;
+        }
+        // If this is content from a transferred agent, format it specially
+        if (currentAgentId != null &&
+            currentAgentId !== '' &&
+            transferToolCallIndex !== undefined) {
+            const agentName = (agentNames?.[currentAgentId] ?? '') || currentAgentId;
+            const formattedParts = [];
+            formattedParts.push(`--- Transfer to ${agentName} ---`);
+            for (const part of agentContentBuffer) {
+                if (part.type === ContentTypes.THINK) {
+                    formattedParts.push(`${agentName}: ${JSON.stringify({
+                        type: 'think',
+                        think: part.think,
+                    })}`);
+                }
+                else if ('text' in part && part.type === ContentTypes.TEXT) {
+                    const textContent = part.text ?? '';
+                    if (textContent) {
+                        formattedParts.push(`${agentName}: ${JSON.stringify({
+                            type: 'text',
+                            text: textContent,
+                        })}`);
+                    }
+                }
+                else if (part.type === ContentTypes.TOOL_CALL) {
+                    formattedParts.push(`${agentName}: ${JSON.stringify({
+                        type: 'tool_call',
+                        tool_call: part.tool_call,
+                    })}`);
+                }
+            }
+            formattedParts.push(`--- End of ${agentName} response ---`);
+            // Find the tool call that triggered this transfer and update its output
+            if (transferToolCallIndex < result.length) {
+                const transferToolCall = result[transferToolCallIndex];
+                if (transferToolCall.type === ContentTypes.TOOL_CALL &&
+                    transferToolCall.tool_call?.id === transferToolCallId) {
+                    transferToolCall.tool_call.output = formattedParts.join('\n\n');
+                }
+            }
+        }
+        else {
+            // Not from a transfer, add as-is
+            result.push(...agentContentBuffer);
+        }
+        agentContentBuffer = [];
+        transferToolCallIndex = undefined;
+        transferToolCallId = undefined;
+    };
+    for (let i = 0; i < contentParts.length; i++) {
+        const part = contentParts[i];
+        const agentId = agentIdMap[i];
+        // Check if this is a transfer tool call
+        const isTransferTool = (part.type === ContentTypes.TOOL_CALL &&
+            part.tool_call?.name?.startsWith('lc_transfer_to_')) ??
+            false;
+        // If agent changed, flush previous buffer
+        if (agentId !== currentAgentId && currentAgentId !== undefined) {
+            flushAgentBuffer();
+        }
+        currentAgentId = agentId;
+        if (isTransferTool) {
+            // Flush any existing buffer first
+            flushAgentBuffer();
+            // Add the transfer tool call to result
+            result.push(part);
+            // Mark that the next agent's content should be captured
+            transferToolCallIndex = result.length - 1;
+            transferToolCallId = part.tool_call?.id;
+            currentAgentId = undefined; // Reset to capture the next agent
+        }
+        else {
+            agentContentBuffer.push(part);
+        }
+    }
+    flushAgentBuffer();
+    return result;
+};
 /**
  * Formats an array of messages for LangChain, handling tool calls and creating ToolMessage instances.
  *
@@ -305,7 +477,7 @@ const formatAgentMessages = (payload, indexTokenCountMap, tools) => {
                             break;
                         }
                         // Protect against malformed tool call entries
-                        if (!part.tool_call ||
+                        if (part.tool_call == null ||
                             part.tool_call.name == null ||
                             part.tool_call.name === '') {
                             hasInvalidTool = true;
@@ -332,7 +504,7 @@ const formatAgentMessages = (payload, indexTokenCountMap, tools) => {
                     // Check if this is a continuation of the tool sequence
                     let isToolResponse = false;
                     const content = payload[j].content;
-                    if (content && Array.isArray(content)) {
+                    if (content != null && Array.isArray(content)) {
                         for (const part of content) {
                             if (part.type === ContentTypes.TOOL_CALL) {
                                 isToolResponse = true;
@@ -431,6 +603,68 @@ function shiftIndexTokenCountMap(indexTokenCountMap, instructionsTokenCount) {
     }
     return shiftedMap;
 }
+/**
+ * Ensures compatibility when switching from a non-thinking agent to a thinking-enabled agent.
+ * Converts AI messages with tool calls (that lack thinking blocks) into buffer strings,
+ * avoiding the thinking block signature requirement.
+ *
+ * @param messages - Array of messages to process
+ * @param provider - The provider being used (unused but kept for future compatibility)
+ * @returns The messages array with tool sequences converted to buffer strings if necessary
+ */
+function ensureThinkingBlockInMessages(messages, _provider) {
+    const result = [];
+    let i = 0;
+    while (i < messages.length) {
+        const msg = messages[i];
+        const isAI = msg instanceof AIMessage || msg instanceof AIMessageChunk;
+        if (!isAI) {
+            result.push(msg);
+            i++;
+            continue;
+        }
+        const aiMsg = msg;
+        const hasToolCalls = aiMsg.tool_calls && aiMsg.tool_calls.length > 0;
+        const contentIsArray = Array.isArray(aiMsg.content);
+        // Check if the message has tool calls or tool_use content
+        let hasToolUse = hasToolCalls ?? false;
+        let firstContentType;
+        if (contentIsArray && aiMsg.content.length > 0) {
+            const content = aiMsg.content;
+            firstContentType = content[0]?.type;
+            hasToolUse =
+                hasToolUse ||
+                    content.some((c) => typeof c === 'object' && c.type === 'tool_use');
+        }
+        // If message has tool use but no thinking block, convert to buffer string
+        if (hasToolUse &&
+            firstContentType !== ContentTypes.THINKING &&
+            firstContentType !== 'redacted_thinking') {
+            // Collect the AI message and any following tool messages
+            const toolSequence = [msg];
+            let j = i + 1;
+            // Look ahead for tool messages that belong to this AI message
+            while (j < messages.length && messages[j] instanceof ToolMessage) {
+                toolSequence.push(messages[j]);
+                j++;
+            }
+            // Convert the sequence to a buffer string and wrap in a HumanMessage
+            // This avoids the thinking block requirement which only applies to AI messages
+            const bufferString = getBufferString(toolSequence);
+            result.push(new HumanMessage({
+                content: `[Previous agent context]\n${bufferString}`,
+            }));
+            // Skip the messages we've processed
+            i = j;
+        }
+        else {
+            // Keep the message as is
+            result.push(msg);
+            i++;
+        }
+    }
+    return result;
+}
 
-export { formatAgentMessages, formatFromLangChain, formatLangChainMessages, formatMediaMessage, formatMessage, shiftIndexTokenCountMap };
+export { ensureThinkingBlockInMessages, formatAgentMessages, formatFromLangChain, formatLangChainMessages, formatMediaMessage, formatMessage, labelContentByAgent, shiftIndexTokenCountMap };
 //# sourceMappingURL=format.mjs.map